From a2e59e3217030fb22f3958b784b7303d0b054498 Mon Sep 17 00:00:00 2001
From: ARREY-ETTA BESSONG EKEP OBASI <arreyettaekep@gmail.com>
Date: Thu, 19 Dec 2024 01:25:21 -0500
Subject: [PATCH] tests: add tests for chat update functionality

---
 handlers/chat_test.go | 151 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 151 insertions(+)
 create mode 100644 handlers/chat_test.go

diff --git a/handlers/chat_test.go b/handlers/chat_test.go
new file mode 100644
index 000000000..82e80aef6
--- /dev/null
+++ b/handlers/chat_test.go
@@ -0,0 +1,151 @@
+package handlers
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/go-chi/chi"
+	"github.com/google/uuid"
+	"github.com/stakwork/sphinx-tribes/db"
+	"github.com/stretchr/testify/assert"
+)
+
+type Chat struct {
+	ID          string `json:"id"`
+	WorkspaceID string `json:"workspaceId"`
+	Title       string `json:"title"`
+}
+
+func TestUpdateChat(t *testing.T) {
+	teardownSuite := SetupSuite(t)
+	defer teardownSuite(t)
+
+	chatHandler := NewChatHandler(&http.Client{}, db.TestDB)
+
+	t.Run("should successfully update chat when valid data is provided", func(t *testing.T) {
+		rr := httptest.NewRecorder()
+		handler := http.HandlerFunc(chatHandler.UpdateChat)
+
+		chat := &db.Chat{
+			ID:          uuid.New().String(),
+			WorkspaceID: uuid.New().String(),
+			Title:       "Old Title",
+		}
+		db.TestDB.AddChat(chat)
+
+		requestBody := map[string]string{
+			"workspaceId": chat.WorkspaceID,
+			"title":       "New Title",
+		}
+		bodyBytes, _ := json.Marshal(requestBody)
+
+		rctx := chi.NewRouteContext()
+		rctx.URLParams.Add("chat_id", chat.ID)
+		req, err := http.NewRequestWithContext(
+			context.WithValue(context.Background(), chi.RouteCtxKey, rctx),
+			http.MethodPut,
+			"/hivechat/"+chat.ID,
+			bytes.NewReader(bodyBytes),
+		)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		handler.ServeHTTP(rr, req)
+
+		var response ChatResponse
+		_ = json.Unmarshal(rr.Body.Bytes(), &response)
+		assert.Equal(t, http.StatusOK, rr.Code)
+		assert.True(t, response.Success)
+		assert.Equal(t, "Chat updated successfully", response.Message)
+
+		updatedChat := response.Data.(Chat)
+		assert.Equal(t, chat.ID, updatedChat.ID)
+		assert.Equal(t, "New Title", updatedChat.Title)
+	})
+
+	t.Run("should return bad request when chat_id is missing", func(t *testing.T) {
+		rr := httptest.NewRecorder()
+		handler := http.HandlerFunc(chatHandler.UpdateChat)
+
+		rctx := chi.NewRouteContext()
+		req, err := http.NewRequestWithContext(
+			context.WithValue(context.Background(), chi.RouteCtxKey, rctx),
+			http.MethodPut,
+			"/hivechat/",
+			nil,
+		)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		handler.ServeHTTP(rr, req)
+
+		var response ChatResponse
+		_ = json.Unmarshal(rr.Body.Bytes(), &response)
+		assert.Equal(t, http.StatusBadRequest, rr.Code)
+		assert.False(t, response.Success)
+		assert.Equal(t, "Chat ID is required", response.Message)
+	})
+
+	t.Run("should return bad request when request body is invalid", func(t *testing.T) {
+		rr := httptest.NewRecorder()
+		handler := http.HandlerFunc(chatHandler.UpdateChat)
+
+		invalidJson := []byte(`{"title": "New Title"`)
+		rctx := chi.NewRouteContext()
+		rctx.URLParams.Add("chat_id", uuid.New().String())
+		req, err := http.NewRequestWithContext(
+			context.WithValue(context.Background(), chi.RouteCtxKey, rctx),
+			http.MethodPut,
+			"/hivechat/"+uuid.New().String(),
+			bytes.NewReader(invalidJson),
+		)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		handler.ServeHTTP(rr, req)
+
+		var response ChatResponse
+		_ = json.Unmarshal(rr.Body.Bytes(), &response)
+		assert.Equal(t, http.StatusBadRequest, rr.Code)
+		assert.False(t, response.Success)
+		assert.Equal(t, "Invalid request body", response.Message)
+	})
+
+	t.Run("should return not found when chat doesn't exist", func(t *testing.T) {
+		rr := httptest.NewRecorder()
+		handler := http.HandlerFunc(chatHandler.UpdateChat)
+
+		requestBody := map[string]string{
+			"workspaceId": uuid.New().String(),
+			"title":       "New Title",
+		}
+		bodyBytes, _ := json.Marshal(requestBody)
+
+		rctx := chi.NewRouteContext()
+		rctx.URLParams.Add("chat_id", uuid.New().String())
+		req, err := http.NewRequestWithContext(
+			context.WithValue(context.Background(), chi.RouteCtxKey, rctx),
+			http.MethodPut,
+			"/hivechat/"+uuid.New().String(),
+			bytes.NewReader(bodyBytes),
+		)
+		if err != nil {
+			t.Fatal(err)
+		}
+
+		handler.ServeHTTP(rr, req)
+
+		var response ChatResponse
+		_ = json.Unmarshal(rr.Body.Bytes(), &response)
+		assert.Equal(t, http.StatusNotFound, rr.Code)
+		assert.False(t, response.Success)
+		assert.Equal(t, "Chat not found", response.Message)
+	})
+}