rogee
aeddedf2a3
Restructure the monorepo into clear top-level directories: - backend/: Go module root (cmd, internal, pkg, configs, migrations, docs/swagger, scripts, tests, go.mod, Makefile, .air.toml) - deploy/: Docker (Dockerfile, docker-compose*), quickstart, fluentd - docs/: project documentation + reports/ (moved from repo root) - AGENTS.md: new AI coding-agent guide at repo root Update all references to the new layout: - Dockerfile: COPY backend/go.mod, COPY backend/ (context = repo root) - docker-compose files: context ../.., dockerfile deploy/docker/Dockerfile, env_file ../../.env, volume mounts ../../backend:/app - deploy/quickstart/compose.yaml: dockerfile deploy/docker/Dockerfile - CI: working-directory: backend for go commands, file deploy/docker/Dockerfile, coverage path backend/coverage.out, health_check backend/scripts/ - backend/Makefile: docker target uses -f ../deploy/docker/Dockerfile ../ - README: architecture tree, quickstart, config paths updated Move root stray scripts (rename_models.*, run_m11_tests.sh, verify_build.sh, gorm_bool_main.go) to backend/scripts/legacy/. All moves via git mv to preserve history. Build, vet, SQLite tests, and docker compose config verified.
129 lines
4.0 KiB
Go
129 lines
4.0 KiB
Go
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)
|
|
} |