1 files changed, 62 insertions, 0 deletions

diff --git a/api/cmd/main_test.go b/api/cmd/main_test.go
new file mode 100644
index 0000000..a4db57a
--- /dev/null
+++ b/api/cmd/main_test.go
@@ -0,0 +1,62 @@
+package main
+
+import (
+        "encoding/json"
+        "log"
+        "net/http"
+        "net/http/httptest"
+        "testing"
+        "water/api/internal/router"
+
+        "github.com/stretchr/testify/assert"
+        "water/api/internal/config"
+)
+
+func getTestUserCredentials() (string, string) {
+        viper, err := config.Load()
+        if err != nil {
+                log.Fatalf("Error while reading config file %s", err)
+        }
+        
+        testUser := viper.GetString("TEST_USER")
+        testPass := viper.GetString("TEST_PASS")
+        return testUser, testPass
+}
+
+func TestAuthRoute(t *testing.T) {
+        r := router.SetupRouter()
+        
+        username, password := getTestUserCredentials()
+
+        w := httptest.NewRecorder()
+        req, err := http.NewRequest("POST", "/api/v1/auth", nil)
+        if err != nil {
+                t.Fatalf("Failed to create request: %v", err)
+        }
+        req.SetBasicAuth(username, password)
+        r.ServeHTTP(w, req)
+
+        assert.Equal(t, http.StatusOK, w.Code, "response should return a 200 code")
+
+        var response map[string]interface{}
+        err = json.Unmarshal(w.Body.Bytes(), &response)
+        if err != nil {
+                t.Fatalf("Failed to unmarshal response: %v", err)
+        }
+        _, exists := response["token"]
+        assert.True(t, exists, "response should return a token")
+        if !exists {
+                t.Fatalf("response did not contain token")
+        }
+}
+
+func TestAuthRouteFailure(t *testing.T) {
+        r := router.SetupRouter()
+
+        w := httptest.NewRecorder()
+        req, _ := http.NewRequest("POST", "/api/v1/auth", nil)
+        req.SetBasicAuth("asdf", "asdf")
+        r.ServeHTTP(w, req)
+
+        assert.Equal(t, http.StatusUnauthorized, w.Code, "should return a 401 code")
+}