Starts adding tests and github workflow

2024-10-18 17:14:10 +02:00
parent 7827ec794d
commit ad6b7cdbc1
6 changed files with 87 additions and 20 deletions
@@ -0,0 +1,65 @@
+package homeassistant
+
+import (
+	"fmt"
+	"math/rand/v2"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/mock"
+)
+
+type MockedHaHandler struct {
+	mock.Mock
+}
+
+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) {
+	requestBody := `{"target": "poo_recorder", "action": "get_latest", "content": ""}`
+	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("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)})
+}