see if i get token correctly to config

This reverts commit 99ed529600.

.github/workflows/nightly.yml

@@ -0,0 +1,70 @@
+name: Run nightly tests
+
+on:
+  workflow_dispatch:
+  schedule:
+    - cron: '0 20 * * *' # Every day at 20:00 UTC
+  push:
+jobs:
+  build:
+    runs-on: [ubuntu-latest, cloud]
+    steps:
+    - uses: actions/checkout@v4
+
+    # - name: Set up Go
+    #   uses: actions/setup-go@v4
+    #   with:
+    #     go-version: '1.23'
+    # - name: build
+    #   working-directory: ./src
+    #   run: |
+    #     go build
+    # - name: Upload artifacts
+    #   uses: actions/upload-artifact@v3
+    #   with:
+    #     path: ${{ github.workspace }}/src/home-automation-backend
+    #     name: home-automation-backend_build_${{ github.run_number }}
+    #     retention-days: 1
+
+  nightly-tests-play:
+    runs-on: [ubuntu-latest, cloud]
+    needs: build
+    steps:
+      - name: Put token into config file
+        run: |
+          cat <<EOF > ${GITHUB_WORKSPACE}/config.json
+          {
+            "token": "${{ secrets.TEST_TOKEN }}"
+          }
+          EOF
+      - name: print config file
+        run: cat ${GITHUB_WORKSPACE}/config.json
+      # - name: Download artifacts
+      #   uses: actions/download-artifact@v3
+      #   with:
+      #     name: home-automation-backend_build_${{ github.run_number }}
+      # - name: Print helping message
+      #   run: |
+      #     chmod +x ${GITHUB_WORKSPACE}/home-automation-backend
+      #     ${GITHUB_WORKSPACE}/home-automation-backend help
+
+  # nightly-tests:
+  #   runs-on: [ubuntu-latest, cloud]
+  #   needs: build
+  #   steps:
+  #   - uses: actions/checkout@v4
+  #   - name: Download artifacts
+  #     uses: actions/download-artifact@v3
+  #     with:
+  #       name: home-automation-backend_build_${{ github.run_number }}
+  #   - name: Check files
+  #     run: ls -lR
+    # - name: Set up Go
+    #   uses: actions/setup-go@v4
+    #   with:
+    #     go-version: '1.23'
+    # - name: build
+    #   run: go build
+    # - name: Run nightly tests
+    #   working-directory: ./tests
+    #   run: go test -v --short ./...

.github/workflows/short-tests.yml

@@ -0,0 +1,20 @@
+name: Run short tests
+
+on:
+  push:
+  pull_request:
+
+jobs:
+  run-tests:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4
+
+    - name: Set up Go
+      uses: actions/setup-go@v4
+      with:
+        go-version: '1.23'
+
+    - name: Test
+      working-directory: ./src
+      run: go test -v --short ./...

.gitignore

@@ -33,3 +33,5 @@ __pycache__/
 config.yaml
 bin/
 *.db
+
+cover.html

src/cmd/serve.go

@@ -21,11 +21,15 @@ import (
 	"github.com/t-liu93/home-automation-backend/components/locationRecorder"
 	"github.com/t-liu93/home-automation-backend/components/pooRecorder"
 	"github.com/t-liu93/home-automation-backend/util/notion"
-	"github.com/t-liu93/home-automation-backend/util/ticktick"
+	"github.com/t-liu93/home-automation-backend/util/ticktickutil"
 )
 
-var port string
+var (
-var scheduler gocron.Scheduler
+	port      string
+	scheduler gocron.Scheduler
+	ticktick  ticktickutil.TicktickUtil
+	ha        *homeassistant.HomeAssistant
+)
 
 // serveCmd represents the serve command
 var serveCmd = &cobra.Command{
@@ -43,7 +47,7 @@ func initUtil() {
 		os.Exit(1)
 	}
 	// init ticktick
-	ticktick.Init()
+	ticktick = ticktickutil.Init()
 }
 
 func initComponent() {
@@ -51,6 +55,8 @@ func initComponent() {
 	pooRecorder.Init(&scheduler)
 	// init location recorder
 	locationRecorder.Init()
+	// init homeassistant
+	ha = homeassistant.NewHomeAssistant(ticktick)
 }
 
 func serve(cmd *cobra.Command, args []string) {
@@ -103,8 +109,7 @@ func serve(cmd *cobra.Command, args []string) {
 
 	router.HandleFunc("/poo/latest", pooRecorder.HandleNotifyLatestPoo).Methods("GET")
 	router.HandleFunc("/poo/record", pooRecorder.HandleRecordPoo).Methods("POST")
-
+	router.HandleFunc("/homeassistant/publish", ha.HandleHaMessage).Methods("POST")
-	router.HandleFunc("/homeassistant/publish", homeassistant.HandleHaMessage).Methods("POST")
 
 	router.HandleFunc("/location/record", locationRecorder.HandleRecordLocation).Methods("POST")
 

src/components/homeassistant/homeassistant.go

@@ -9,7 +9,7 @@ import (
 	"time"
 
 	"github.com/spf13/viper"
-	"github.com/t-liu93/home-automation-backend/util/ticktick"
+	"github.com/t-liu93/home-automation-backend/util/ticktickutil"
 )
 
 type haMessage struct {
@@ -18,104 +18,135 @@ type haMessage struct {
 	Content string `json:"content"`
 }
 
+type HomeAssistant struct {
+	ticktickUtil ticktickutil.TicktickUtil
+}
+
 type actionTask struct {
 	Action  string `json:"action"`
 	DueHour int    `json:"due_hour"`
 }
 
-func HandleHaMessage(w http.ResponseWriter, r *http.Request) {
+func NewHomeAssistant(ticktick ticktickutil.TicktickUtil) *HomeAssistant {
+	return &HomeAssistant{
+		ticktickUtil: ticktick,
+	}
+}
+
+func (ha *HomeAssistant) HandleHaMessage(w http.ResponseWriter, r *http.Request) {
 	var message haMessage
 	decoder := json.NewDecoder(r.Body)
 	decoder.DisallowUnknownFields()
 	err := decoder.Decode(&message)
 	if err != nil {
-		slog.Warn(fmt.Sprintln("HandleHaMessage Error decoding request body", err))
+		slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Error decoding request body", err))
-		http.Error(w, err.Error(), http.StatusBadRequest)
+		http.Error(w, "", http.StatusInternalServerError)
 		return
 	}
 
 	switch message.Target {
 	case "poo_recorder":
-		handlePooRecorderMsg(message)
+		res := ha.handlePooRecorderMsg(message)
-	case "location_recorder":
+		if !res {
-		handleLocationRecorderMsg(message)
+			slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Error handling poo recorder message"))
-	case "ticktick":
+			http.Error(w, "", http.StatusInternalServerError)
-		handleTicktickMsg(message)
+		}
+	case "location_recorder":
+		res := ha.handleLocationRecorderMsg(message)
+		if !res {
+			slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Error handling location recorder message"))
+			http.Error(w, "", http.StatusInternalServerError)
+		}
+	case "ticktick":
+		res := ha.handleTicktickMsg(message)
+		if !res {
+			slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Error handling ticktick message"))
+			http.Error(w, "", http.StatusInternalServerError)
+		}
+	default:
+		slog.Warn(fmt.Sprintln("homeassistant.HandleHaMessage: Unknown target", message.Target))
+		http.Error(w, "", http.StatusInternalServerError)
 	}
-
 }
 
-func handlePooRecorderMsg(message haMessage) {
+func (ha *HomeAssistant) handlePooRecorderMsg(message haMessage) bool {
 	switch message.Action {
 	case "get_latest":
-		handleGetLatestPoo()
+		return ha.handleGetLatestPoo()
+	default:
+		slog.Warn(fmt.Sprintln("homeassistant.handlePooRecorderMsg: Unknown action", message.Action))
+		return false
 	}
-
 }
 
-func handleLocationRecorderMsg(message haMessage) {
+func (ha *HomeAssistant) handleLocationRecorderMsg(message haMessage) bool {
 	if message.Action == "record" {
 		port := viper.GetString("port")
-		req, err := http.NewRequest("POST", "http://localhost:"+port+"/location/record", strings.NewReader(strings.ReplaceAll(message.Content, "'", "\"")))
-		if err != nil {
-			slog.Warn(fmt.Sprintln("handleLocationRecorderMsg Error creating request to location recorder", err))
-			return
-		}
-		req.Header.Set("Content-Type", "application/json")
 		client := &http.Client{
 			Timeout: time.Second * 1,
 		}
-		_, err = client.Do(req)
+		_, err := client.Post("http://localhost:"+port+"/location/record", "application/json", strings.NewReader(strings.ReplaceAll(message.Content, "'", "\"")))
 		if err != nil {
-			slog.Warn(fmt.Sprintln("handleLocationRecorderMsg Error sending request to location recorder", err))
+			slog.Warn(fmt.Sprintln("homeassistant.handleLocationRecorderMsg: Error sending request to location recorder", err))
+			return false
 		}
 	} else {
-		slog.Warn(fmt.Sprintln("handleLocationRecorderMsg Unknown action", message.Action))
+		slog.Warn(fmt.Sprintln("homeassistant.handleLocationRecorderMsg: Unknown action", message.Action))
+		return false
 	}
+	return true
 }
 
-func handleTicktickMsg(message haMessage) {
+func (ha *HomeAssistant) handleTicktickMsg(message haMessage) bool {
 	switch message.Action {
 	case "create_action_task":
-		createActionTask(message)
+		return ha.createActionTask(message)
+	default:
+		slog.Warn(fmt.Sprintln("homeassistant.handleTicktickMsg: Unknown action", message.Action))
+		return false
 	}
 }
 
-func handleGetLatestPoo() {
+func (ha *HomeAssistant) handleGetLatestPoo() bool {
 	client := &http.Client{
 		Timeout: time.Second * 1,
 	}
 	port := viper.GetString("port")
 	_, err := client.Get("http://localhost:" + port + "/poo/latest")
 	if err != nil {
-		slog.Warn(fmt.Sprintln("handleGetLatestPoo Error sending request to poo recorder", err))
+		slog.Warn(fmt.Sprintln("homeassistant.handleGetLatestPoo: Error sending request to poo recorder", err))
+		return false
 	}
+
+	return true
 }
 
-func createActionTask(message haMessage) {
+func (ha *HomeAssistant) createActionTask(message haMessage) bool {
-	if !viper.InConfig("homeassistant.actionTaskProjectId") {
+	if !viper.IsSet("homeassistant.actionTaskProjectId") {
-		slog.Warn("Homeassistant.createActionTask actionTaskProjectId not found in config file")
+		slog.Warn("homeassistant.createActionTask: actionTaskProjectId not found in config file")
-		return
+		return false
 	}
 	projectId := viper.GetString("homeassistant.actionTaskProjectId")
 	detail := strings.ReplaceAll(message.Content, "'", "\"")
 	var task actionTask
 	err := json.Unmarshal([]byte(detail), &task)
 	if err != nil {
-		slog.Warn(fmt.Sprintln("Homeassistant.createActionTask Error unmarshalling", err))
+		slog.Warn(fmt.Sprintln("homeassistant.createActionTask: Error unmarshalling", err))
-		return
+		return false
 	}
 	dueHour := task.DueHour
 	due := time.Now().Add(time.Hour * time.Duration(dueHour))
 	dueNextMidnight := time.Date(due.Year(), due.Month(), due.Day(), 0, 0, 0, 0, time.Local).AddDate(0, 0, 1)
-	dueTicktick := dueNextMidnight.UTC().Format(ticktick.DateTimeLayout)
+	dueTicktick := dueNextMidnight.UTC().Format(ticktickutil.DateTimeLayout)
-	ticktickTask := ticktick.Task{
+	ticktickTask := ticktickutil.Task{
 		ProjectId: projectId,
 		Title:     task.Action,
 		DueDate:   dueTicktick,
 	}
-	err = ticktick.CreateTask(ticktickTask)
+	err = ha.ticktickUtil.CreateTask(ticktickTask)
 	if err != nil {
-		slog.Warn(fmt.Sprintf("Homeassistant.createActionTask Error creating task %s", err))
+		slog.Warn(fmt.Sprintf("homeassistant.createActionTask: Error creating task %s", err))
+		return false
 	}
+	return true
 }

src/components/homeassistant/homeassistant_test.go

@@ -0,0 +1,280 @@
+package homeassistant
+
+import (
+	"bytes"
+	"errors"
+	"log/slog"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/spf13/viper"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
+	"github.com/t-liu93/home-automation-backend/util/ticktickutil"
+)
+
+var (
+	loggerText = new(bytes.Buffer)
+)
+
+type MockTicktickUtil struct {
+	mock.Mock
+}
+
+func (m *MockTicktickUtil) HandleAuthCode(w http.ResponseWriter, r *http.Request) {
+	m.Called(w, r)
+}
+
+func (m *MockTicktickUtil) GetTasks(projectId string) []ticktickutil.Task {
+	args := m.Called(projectId)
+	return args.Get(0).([]ticktickutil.Task)
+}
+
+func (m *MockTicktickUtil) HasDuplicateTask(projectId string, taskTitile string) bool {
+	args := m.Called(projectId, taskTitile)
+	return args.Bool(0)
+}
+
+func (m *MockTicktickUtil) CreateTask(task ticktickutil.Task) error {
+	args := m.Called(task)
+	return args.Error(0)
+}
+
+func SetupTearDown(t *testing.T) (func(), *HomeAssistant) {
+	loggertearDown := loggerSetupTeardown()
+	mockTicktick := &MockTicktickUtil{}
+	ha := NewHomeAssistant(mockTicktick)
+
+	return func() {
+		loggertearDown()
+		viper.Reset()
+	}, ha
+}
+
+func loggerSetupTeardown() func() {
+	logger := slog.New(slog.NewTextHandler(loggerText, nil))
+	defaultLogger := slog.Default()
+	slog.SetDefault(logger)
+
+	return func() {
+		slog.SetDefault(defaultLogger)
+		loggerText.Reset()
+	}
+}
+
+func TestHandleHaMessageJsonDecodeError(t *testing.T) {
+	teardown, ha := SetupTearDown(t)
+	defer teardown()
+
+	invalidRequestBody := ` { "target": "poo_recorder", "action": "get_latest", "content": " }`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(invalidRequestBody))
+	w := httptest.NewRecorder()
+
+	ha.HandleHaMessage(w, req)
+	assert.Equal(t, http.StatusInternalServerError, w.Code)
+	assert.Contains(t, loggerText.String(), "homeassistant.HandleHaMessage: Error decoding request body")
+}
+
+func TestHandlePooRecorderMsgGetLatest(t *testing.T) {
+	teardown, ha := SetupTearDown(t)
+	defer teardown()
+	requestBody := `{"target": "poo_recorder", "action": "get_latest", "content": ""}`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
+	w := httptest.NewRecorder()
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		assert.Equal(t, http.MethodGet, r.Method)
+		assert.Equal(t, "/poo/latest", r.URL.Path)
+	}))
+	defer server.Close()
+	port := strings.Split(server.URL, ":")[2]
+	viper.Set("port", port)
+
+	ha.HandleHaMessage(w, req)
+	assert.Equal(t, http.StatusOK, w.Code)
+	assert.Empty(t, loggerText.String())
+}
+
+func TestHandlePooRecorderMsgUnknownAction(t *testing.T) {
+	teardown, ha := SetupTearDown(t)
+	defer teardown()
+
+	requestBody := `{"target": "poo_recorder", "action": "unknown_action", "content": ""}`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
+	w := httptest.NewRecorder()
+
+	ha.HandleHaMessage(w, req)
+	assert.Equal(t, http.StatusInternalServerError, w.Code)
+	assert.Contains(t, loggerText.String(), "homeassistant.handlePooRecorderMsg: Unknown action")
+}
+
+func TestHandlePooRecorderMsgGetLatestError(t *testing.T) {
+	teardown, ha := SetupTearDown(t)
+	defer teardown()
+
+	requestBody := `{"target": "poo_recorder", "action": "get_latest", "content": ""}`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
+	w := httptest.NewRecorder()
+
+	port := "invalid port"
+	viper.Set("port", port)
+
+	ha.HandleHaMessage(w, req)
+	assert.Equal(t, http.StatusInternalServerError, w.Code)
+	assert.Contains(t, loggerText.String(), "homeassistant.handleGetLatestPoo: Error sending request to poo recorder")
+}
+
+func TestHandleLocationRecorderMsg(t *testing.T) {
+	teardown, ha := SetupTearDown(t)
+	defer teardown()
+
+	requestBody := `{"target": "location_recorder", "action": "record", "content": "{'person': 'test', 'latitude': '1.0', 'longitude': '2.0', 'altitude': '3.0'}"}`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
+	w := httptest.NewRecorder()
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		assert.Equal(t, http.MethodPost, r.Method)
+		assert.Equal(t, "/location/record", r.URL.Path)
+		assert.Equal(t, "application/json", r.Header.Get("Content-Type"))
+	}))
+	defer server.Close()
+
+	port := strings.Split(server.URL, ":")[2]
+	viper.Set("port", port)
+
+	ha.HandleHaMessage(w, req)
+	assert.Equal(t, http.StatusOK, w.Code)
+	assert.Empty(t, loggerText.String())
+}
+
+func TestHandleLocationRecorderMsgUnknownAction(t *testing.T) {
+	teardown, ha := SetupTearDown(t)
+	defer teardown()
+
+	requestBody := `{"target": "location_recorder", "action": "unknown_action", "content": ""}`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
+	w := httptest.NewRecorder()
+
+	ha.HandleHaMessage(w, req)
+	assert.Equal(t, http.StatusInternalServerError, w.Code)
+	assert.Contains(t, loggerText.String(), "homeassistant.handleLocationRecorderMsg: Unknown action")
+}
+
+func TestHandleLocationRecorderMsgRequestErr(t *testing.T) {
+	teardown, ha := SetupTearDown(t)
+	defer teardown()
+
+	requestBody := `{"target": "location_recorder", "action": "record", "content": "{'person': 'test', 'latitude': '1.0', 'longitude': '2.0', 'altitude': '3.0'}"}`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
+	w := httptest.NewRecorder()
+
+	port := "invalid port"
+	viper.Set("port", port)
+
+	ha.HandleHaMessage(w, req)
+	assert.Equal(t, http.StatusInternalServerError, w.Code)
+	assert.Contains(t, loggerText.String(), "homeassistant.handleLocationRecorderMsg: Error sending request to location recorder")
+}
+
+func TestHandleTicktickMsgCreateActionTask(t *testing.T) {
+	teardown, _ := SetupTearDown(t)
+	defer teardown()
+	const expectedProjectId = "test_project_id"
+	const dueHour = 12
+	due := time.Now().Add(time.Hour * time.Duration(dueHour))
+	dueNextMidnight := time.Date(due.Year(), due.Month(), due.Day(), 0, 0, 0, 0, time.Local).AddDate(0, 0, 1)
+	dueTicktick := dueNextMidnight.UTC().Format(ticktickutil.DateTimeLayout)
+
+	requestBody := `{"target": "ticktick", "action": "create_action_task", "content": "{'title': 'test', 'action': 'test_action', 'due_hour': 12}"}`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
+	w := httptest.NewRecorder()
+
+	mockTicktick := &MockTicktickUtil{}
+	mockTicktick.On("CreateTask", mock.Anything).Return(nil)
+	ha := NewHomeAssistant(mockTicktick)
+	viper.Set("homeassistant.actionTaskProjectId", expectedProjectId)
+	ha.HandleHaMessage(w, req)
+	expectedTask := ticktickutil.Task{
+		Title:     "test_action",
+		DueDate:   dueTicktick,
+		ProjectId: expectedProjectId,
+	}
+	mockTicktick.AssertCalled(t, "CreateTask", expectedTask)
+	mockTicktick.AssertNumberOfCalls(t, "CreateTask", 1)
+	assert.Equal(t, http.StatusOK, w.Code)
+	assert.Empty(t, loggerText.String())
+}
+
+func TestHandleTicktickMsgUnknownAction(t *testing.T) {
+	teardown, ha := SetupTearDown(t)
+	defer teardown()
+
+	requestBody := `{"target": "ticktick", "action": "unknown_action", "content": ""}`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
+	w := httptest.NewRecorder()
+
+	ha.HandleHaMessage(w, req)
+	assert.Equal(t, http.StatusInternalServerError, w.Code)
+	assert.Contains(t, loggerText.String(), "homeassistant.handleTicktickMsg: Unknown action")
+}
+
+func TestHandleTicktickMsgProjectIdUnset(t *testing.T) {
+	teardown, ha := SetupTearDown(t)
+	defer teardown()
+
+	requestBody := `{"target": "ticktick", "action": "create_action_task", "content": "{'title': 'test', 'action': 'test_action', 'due_hour': 12}"}`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
+	w := httptest.NewRecorder()
+
+	ha.HandleHaMessage(w, req)
+	assert.Equal(t, http.StatusInternalServerError, w.Code)
+	assert.Contains(t, loggerText.String(), "homeassistant.createActionTask: actionTaskProjectId not found in config file")
+}
+
+func TestHandleTicktickMsgJsonError(t *testing.T) {
+	teardown, ha := SetupTearDown(t)
+	defer teardown()
+
+	invalidRequestBody := ` { "target": "ticktick", "action": "create_action_task", "content": "{'title': 'tes, 'action': 'test_action', 'due_hour': 12}"}`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(invalidRequestBody))
+	w := httptest.NewRecorder()
+	viper.Set("homeassistant.actionTaskProjectId", "some project id")
+	ha.HandleHaMessage(w, req)
+	assert.Equal(t, http.StatusInternalServerError, w.Code)
+	assert.Contains(t, loggerText.String(), "homeassistant.createActionTask: Error unmarshalling")
+}
+
+func TestHandleTicktickMsgTicktickUtilErr(t *testing.T) {
+	teardown, _ := SetupTearDown(t)
+	defer teardown()
+
+	requestBody := `{"target": "ticktick", "action": "create_action_task", "content": "{'title': 'test', 'action': 'test_action', 'due_hour': 12}"}`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
+	w := httptest.NewRecorder()
+	mockedTicktickUtil := &MockTicktickUtil{}
+	viper.Set("homeassistant.actionTaskProjectId", "some project id")
+
+	mockedTicktickUtil.On("CreateTask", mock.Anything).Return(errors.New("some error"))
+
+	ha := NewHomeAssistant(mockedTicktickUtil)
+
+	ha.HandleHaMessage(w, req)
+
+	mockedTicktickUtil.AssertCalled(t, "CreateTask", mock.Anything)
+	assert.Equal(t, http.StatusInternalServerError, w.Code)
+	assert.Contains(t, loggerText.String(), "homeassistant.createActionTask: Error creating task")
+}
+
+func TestHandleHaMessageUnknownTarget(t *testing.T) {
+	teardown, ha := SetupTearDown(t)
+	defer teardown()
+
+	requestBody := `{"target": "unknown_target", "action": "record", "content": ""}`
+	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
+	w := httptest.NewRecorder()
+
+	ha.HandleHaMessage(w, req)
+	assert.Equal(t, http.StatusInternalServerError, w.Code)
+	assert.Contains(t, loggerText.String(), "homeassistant.HandleHaMessage: Unknown target")
+}

src/go.mod

@@ -8,11 +8,13 @@ require (
 	github.com/jomei/notionapi v1.13.2
 	github.com/spf13/cobra v1.8.1
 	github.com/spf13/viper v1.19.0
+	github.com/stretchr/testify v1.10.0
 	golang.org/x/term v0.24.0
 	modernc.org/sqlite v1.33.1
 )
 
 require (
+	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
 	github.com/dustin/go-humanize v1.0.1 // indirect
 	github.com/fsnotify/fsnotify v1.7.0 // indirect
 	github.com/google/uuid v1.6.0 // indirect
@@ -25,6 +27,7 @@ require (
 	github.com/mitchellh/mapstructure v1.5.0 // indirect
 	github.com/ncruces/go-strftime v0.1.9 // indirect
 	github.com/pelletier/go-toml/v2 v2.2.2 // indirect
+	github.com/pmezard/go-difflib v1.0.1-0.20181226105442-5d4384ee4fb2 // indirect
 	github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
 	github.com/robfig/cron/v3 v3.0.1 // indirect
 	github.com/sagikazarmark/locafero v0.4.0 // indirect
@@ -33,6 +36,7 @@ require (
 	github.com/spf13/afero v1.11.0 // indirect
 	github.com/spf13/cast v1.6.0 // indirect
 	github.com/spf13/pflag v1.0.5 // indirect
+	github.com/stretchr/objx v0.5.2 // indirect
 	github.com/subosito/gotenv v1.6.0 // indirect
 	go.uber.org/atomic v1.9.0 // indirect
 	go.uber.org/multierr v1.9.0 // indirect

src/go.sum

@@ -72,13 +72,15 @@ github.com/spf13/viper v1.19.0/go.mod h1:GQUN9bilAbhU/jgc1bKs99f/suXKeUMct8Adx5+
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=
 github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=
+github.com/stretchr/objx v0.5.2 h1:xuMeJ0Sdp5ZMRXx/aWO6RZxdr3beISkG5/G/aIRr3pY=
 github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=
 github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
 github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
 github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
 github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=
-github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
 github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
+github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA=
+github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 github.com/subosito/gotenv v1.6.0 h1:9NlTDc1FTs4qu0DDq7AEtTPNw6SVm7uBMsUCUjABIf8=
 github.com/subosito/gotenv v1.6.0/go.mod h1:Dk4QP5c2W3ibzajGcXpNraDfq2IrhjMIvMSWPKKo0FU=
 go.uber.org/atomic v1.9.0 h1:ECmE8Bn/WFTYwEW/bpKD3M8VtR/zQVbavAoalC1PYyE=

src/util/ticktickutil/ticktickutil.go

@@ -1,4 +1,4 @@
-package ticktick
+package ticktickutil
 
 import (
 	"bytes"
@@ -15,15 +15,25 @@ import (
 	"github.com/spf13/viper"
 )
 
-var (
-	authState string
-)
-
 const (
 	DateTimeLayout = "2006-01-02T15:04:05-0700"
 )
 
-type Project struct {
+type (
+	TicktickUtil interface {
+		HandleAuthCode(w http.ResponseWriter, r *http.Request)
+		GetTasks(projectId string) []Task
+		HasDuplicateTask(projectId string, taskTitile string) bool
+		CreateTask(task Task) error
+	}
+
+	TicktickUtilImpl struct {
+		authState string
+	}
+)
+
+type (
+	Project struct {
 		Id         string `json:"id"`
 		Name       string `json:"name"`
 		Color      string `json:"color,omitempty"`
@@ -33,16 +43,16 @@ type Project struct {
 		ViewMode   string `json:"viewMode,omitempty"`
 		Permission string `json:"permission,omitempty"`
 		Kind       string `json:"kind,omitempty"`
-}
+	}
 
-type Column struct {
+	Column struct {
 		Id        string `json:"id"`
 		Name      string `json:"name"`
 		ProjectId string `json:"projectId"`
 		SortOrder int64  `json:"sortOrder,omitempty"`
-}
+	}
 
-type Task struct {
+	Task struct {
 		Id            string        `json:"id"`
 		ProjectId     string        `json:"projectId"`
 		Title         string        `json:"title"`
@@ -59,15 +69,17 @@ type Task struct {
 		StartDate     string        `json:"startDate,omitempty"`
 		Status        int32         `json:"status,omitempty"`
 		TimeZone      string        `json:"timeZone,omitempty"`
-}
+	}
 
-type ProjectData struct {
+	ProjectData struct {
 		Project Project  `json:"project"`
 		Tasks   []Task   `json:"tasks"`
 		Columns []Column `json:"columns,omitempty"`
-}
+	}
+)
 
-func Init() {
+func Init() TicktickUtil { // TODO: Will modify Init to a proper behavior
+	ticktickUtilImpl := &TicktickUtilImpl{}
 	if !viper.InConfig("ticktick.clientId") {
 		slog.Error("TickTick clientId not found in config file, exiting..")
 		os.Exit(1)
@@ -83,14 +95,15 @@ func Init() {
 			}
 		}
 	} else {
-		beginAuth()
+		ticktickUtilImpl.beginAuth()
 	}
+	return ticktickUtilImpl
 }
 
-func HandleAuthCode(w http.ResponseWriter, r *http.Request) {
+func (t *TicktickUtilImpl) HandleAuthCode(w http.ResponseWriter, r *http.Request) {
 	state := r.URL.Query().Get("state")
 	code := r.URL.Query().Get("code")
-	if state != authState {
+	if state != t.authState {
 		slog.Warn(fmt.Sprintln("HandleAuthCode Invalid state", state))
 		http.Error(w, "Invalid state", http.StatusBadRequest)
 		return
@@ -148,7 +161,7 @@ func HandleAuthCode(w http.ResponseWriter, r *http.Request) {
 	w.Write([]byte("Authorization successful"))
 }
 
-func GetTasks(projectId string) []Task {
+func (t *TicktickUtilImpl) GetTasks(projectId string) []Task {
 	getTaskUrl := fmt.Sprintf("https://api.ticktick.com/open/v1/project/%s/data", projectId)
 	token := viper.GetString("ticktick.token")
 	req, err := http.NewRequest("GET", getTaskUrl, nil)
@@ -184,8 +197,8 @@ func GetTasks(projectId string) []Task {
 	return projectData.Tasks
 }
 
-func HasDuplicateTask(projectId string, taskTitile string) bool {
+func (t *TicktickUtilImpl) HasDuplicateTask(projectId string, taskTitile string) bool {
-	tasks := GetTasks(projectId)
+	tasks := t.GetTasks(projectId)
 	for _, task := range tasks {
 		if task.Title == taskTitile {
 			return true
@@ -194,8 +207,8 @@ func HasDuplicateTask(projectId string, taskTitile string) bool {
 	return false
 }
 
-func CreateTask(task Task) error {
+func (t *TicktickUtilImpl) CreateTask(task Task) error {
-	if HasDuplicateTask(task.ProjectId, task.Title) {
+	if t.HasDuplicateTask(task.ProjectId, task.Title) {
 		return nil
 	}
 	token := viper.GetString("ticktick.token")
@@ -259,7 +272,7 @@ func getProjects() ([]Project, error) {
 	return projects, nil
 }
 
-func beginAuth() {
+func (t *TicktickUtilImpl) beginAuth() {
 	if !viper.InConfig("ticktick.redirectUri") {
 		slog.Error("TickTick redirectUri not found in config file, exiting..")
 		os.Exit(1)
@@ -272,12 +285,12 @@ func beginAuth() {
 		slog.Error(fmt.Sprintln("Error generating auth state", err))
 		os.Exit(1)
 	}
-	authState = hex.EncodeToString(authStateBytes)
+	t.authState = hex.EncodeToString(authStateBytes)
 	params := url.Values{}
 	params.Add("client_id", viper.GetString("ticktick.clientId"))
 	params.Add("response_type", "code")
 	params.Add("redirect_uri", viper.GetString("ticktick.redirectUri"))
-	params.Add("state", authState)
+	params.Add("state", t.authState)
 	params.Add("scope", "tasks:read tasks:write")
 	authUrl.RawQuery = params.Encode()
 	slog.Info(fmt.Sprintln("Please visit the following URL to authorize TickTick:", authUrl.String()))