second commit

internal/handler/api/v1/agent_bulk_handler_test.go

@@ -0,0 +1,129 @@
+package v1
+
+import (
+	"bytes"
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/gin-gonic/gin"
+	"github.com/stretchr/testify/assert"
+
+	"github.com/gochat/gochat/internal/service"
+)
+
+func setupAgentBulkRouter(handler *AgentBulkHandler) *gin.Engine {
+	gin.SetMode(gin.TestMode)
+	r := gin.New()
+	r.Use(gin.Recovery())
+	r.POST("/api/v1/accounts/:account_id/agents/bulk_assign", handler.BulkAssign)
+	r.POST("/api/v1/accounts/:account_id/agents/bulk_unassign", handler.BulkUnassign)
+	return r
+}
+
+func TestAgentBulkHandler_BulkAssign_BadJSON(t *testing.T) {
+	handler := NewAgentBulkHandler(&service.ConversationService{})
+	router := setupAgentBulkRouter(handler)
+
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("POST", "/api/v1/accounts/1/agents/bulk_assign", bytes.NewReader([]byte("invalid json")))
+	req.Header.Set("Content-Type", "application/json")
+	router.ServeHTTP(w, req)
+
+	assert.Equal(t, http.StatusBadRequest, w.Code)
+}
+
+func TestAgentBulkHandler_BulkAssign_EmptyConversationIDs(t *testing.T) {
+	handler := NewAgentBulkHandler(&service.ConversationService{})
+	router := setupAgentBulkRouter(handler)
+
+	body := map[string]interface{}{
+		"conversation_ids": []uint{},
+		"agent_id":         10,
+	}
+	b, _ := json.Marshal(body)
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("POST", "/api/v1/accounts/1/agents/bulk_assign", bytes.NewReader(b))
+	req.Header.Set("Content-Type", "application/json")
+	router.ServeHTTP(w, req)
+
+	assert.Equal(t, http.StatusBadRequest, w.Code)
+}
+
+func TestAgentBulkHandler_BulkAssign_MissingAgentID(t *testing.T) {
+	handler := NewAgentBulkHandler(&service.ConversationService{})
+	router := setupAgentBulkRouter(handler)
+
+	body := map[string]interface{}{
+		"conversation_ids": []uint{1, 2, 3},
+	}
+	b, _ := json.Marshal(body)
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("POST", "/api/v1/accounts/1/agents/bulk_assign", bytes.NewReader(b))
+	req.Header.Set("Content-Type", "application/json")
+	router.ServeHTTP(w, req)
+
+	assert.Equal(t, http.StatusBadRequest, w.Code)
+}
+
+func TestAgentBulkHandler_BulkAssign_InvalidAccountID(t *testing.T) {
+	handler := NewAgentBulkHandler(&service.ConversationService{})
+	router := setupAgentBulkRouter(handler)
+
+	body := map[string]interface{}{
+		"conversation_ids": []uint{1, 2},
+		"agent_id":         10,
+	}
+	b, _ := json.Marshal(body)
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("POST", "/api/v1/accounts/abc/agents/bulk_assign", bytes.NewReader(b))
+	req.Header.Set("Content-Type", "application/json")
+	router.ServeHTTP(w, req)
+
+	assert.Equal(t, http.StatusBadRequest, w.Code)
+}
+
+func TestAgentBulkHandler_BulkUnassign_BadJSON(t *testing.T) {
+	handler := NewAgentBulkHandler(&service.ConversationService{})
+	router := setupAgentBulkRouter(handler)
+
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("POST", "/api/v1/accounts/1/agents/bulk_unassign", bytes.NewReader([]byte("invalid json")))
+	req.Header.Set("Content-Type", "application/json")
+	router.ServeHTTP(w, req)
+
+	assert.Equal(t, http.StatusBadRequest, w.Code)
+}
+
+func TestAgentBulkHandler_BulkUnassign_EmptyConversationIDs(t *testing.T) {
+	handler := NewAgentBulkHandler(&service.ConversationService{})
+	router := setupAgentBulkRouter(handler)
+
+	body := map[string]interface{}{
+		"conversation_ids": []uint{},
+	}
+	b, _ := json.Marshal(body)
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("POST", "/api/v1/accounts/1/agents/bulk_unassign", bytes.NewReader(b))
+	req.Header.Set("Content-Type", "application/json")
+	router.ServeHTTP(w, req)
+
+	assert.Equal(t, http.StatusBadRequest, w.Code)
+}
+
+func TestAgentBulkHandler_BulkUnassign_InvalidAccountID(t *testing.T) {
+	handler := NewAgentBulkHandler(&service.ConversationService{})
+	router := setupAgentBulkRouter(handler)
+
+	body := map[string]interface{}{
+		"conversation_ids": []uint{1, 2},
+	}
+	b, _ := json.Marshal(body)
+	w := httptest.NewRecorder()
+	req, _ := http.NewRequest("POST", "/api/v1/accounts/abc/agents/bulk_unassign", bytes.NewReader(b))
+	req.Header.Set("Content-Type", "application/json")
+	router.ServeHTTP(w, req)
+
+	assert.Equal(t, http.StatusBadRequest, w.Code)
+}