Add test structure

src/components/homeassistant/homeassistant_test.go

@@ -1,65 +1,46 @@
 package homeassistant
 
 import (
-	"fmt"
-	"math/rand/v2"
+	"bytes"
+	"log/slog"
 	"net/http"
 	"net/http/httptest"
 	"strings"
 	"testing"
 
-	"github.com/stretchr/testify/mock"
+	"github.com/spf13/viper"
+	"github.com/stretchr/testify/assert"
 )
 
-type MockedHaHandler struct {
-	mock.Mock
+var (
+	loggerText = new(bytes.Buffer)
+)
+
+func LoggerSetupTeardown(t *testing.T) func() {
+	logger := slog.New(slog.NewTextHandler(loggerText, nil))
+	defaultLogger := slog.Default()
+	slog.SetDefault(logger)
+
+	return func() {
+		slog.SetDefault(defaultLogger)
+		loggerText.Reset()
+	}
 }
 
-func (m *MockedHaHandler) handlePooRecorderMsg(message haMessage) {
-	m.Called(message)
-}
-
-func (m *MockedHaHandler) handleLocationRecorderMsg(message haMessage) {
-	m.Called(message)
-}
-
-func (m *MockedHaHandler) handleTicktickMsg(message haMessage) {
-	m.Called(message)
-}
-
-func TestHandleHaMessagePooRecorder(t *testing.T) {
+func TestHandlePooRecorderMsgGetLatest(t *testing.T) {
+	teardown := LoggerSetupTeardown(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, "/poo/latest", r.URL.Path)
+	}))
+	defer server.Close()
+	port := strings.Split(server.URL, ":")[2]
+	viper.Set("port", port)
 
-	testObj := new(MockedHaHandler)
-	mockedHandler := HaMessageHandler{
-		PooRecorderHandler:      testObj.handlePooRecorderMsg,
-		LocationRecorderHandler: testObj.handleLocationRecorderMsg,
-		TicktickHandler:         testObj.handleTicktickMsg,
-	}
-	testObj.On("handlePooRecorderMsg", haMessage{Target: "poo_recorder", Action: "get_latest"}).Return()
-	mockedHandler.HandleHaMessage(w, req)
-	testObj.AssertCalled(t, "handlePooRecorderMsg", haMessage{Target: "poo_recorder", Action: "get_latest"})
-}
-
-func TestHandleHaMessageLocationRecorder(t *testing.T) {
-	expectedLatitude := rand.Float64() * 359
-	expectedLongitude := rand.Float64() * 359
-	expectedAltitude := rand.Float64() * 10000
-	requestBody := fmt.Sprintf(`{"target": "location_recorder", "action": "record",
-	"content": "{'Person': 'Tom', 'latitude': %f, 'longitude': %f, 'altitude': %f}"}`, expectedLatitude, expectedLongitude, expectedAltitude)
-	req := httptest.NewRequest(http.MethodPost, "/homeassistant/publish", strings.NewReader(requestBody))
-	w := httptest.NewRecorder()
-	testobj := new(MockedHaHandler)
-	mockedHandler := HaMessageHandler{
-		PooRecorderHandler:      testobj.handlePooRecorderMsg,
-		LocationRecorderHandler: testobj.handleLocationRecorderMsg,
-		TicktickHandler:         testobj.handleTicktickMsg,
-	}
-	testobj.On("handleLocationRecorderMsg", haMessage{Target: "location_recorder", Action: "record",
-		Content: fmt.Sprintf("{'Person': 'Tom', 'latitude': %f, 'longitude': %f, 'altitude': %f}", expectedLatitude, expectedLongitude, expectedAltitude)}).Return()
-	mockedHandler.HandleHaMessage(w, req)
-	testobj.AssertCalled(t, "handleLocationRecorderMsg", haMessage{Target: "location_recorder", Action: "record",
-		Content: fmt.Sprintf("{'Person': 'Tom', 'latitude': %f, 'longitude': %f, 'altitude': %f}", expectedLatitude, expectedLongitude, expectedAltitude)})
+	HandleHaMessage(w, req)
+	assert.Equal(t, http.StatusOK, w.Code)
+	assert.Empty(t, loggerText.String())
 }