Files
gochat/internal/handler/api/v1/csat_survey_handler_test.go_BAK
T
2026-06-04 15:44:48 +08:00

87 lines
2.6 KiB
Plaintext

package v1
import (
"net/http"
"net/http/httptest"
"strings"
"testing"
"github.com/gin-gonic/gin"
"github.com/stretchr/testify/assert"
"gorm.io/gorm"
"github.com/gochat/gochat/internal/automation"
)
// csatMockDBProvider implements automation.DBProvider for validation-only tests.
// DB() returns nil — service methods are never reached since validation fails first.
type csatMockDBProvider struct{}
func (m *csatMockDBProvider) DB() *gorm.DB { return nil }
func setupCsatSurveyRouter(handler *CsatSurveyHandler) *gin.Engine {
gin.SetMode(gin.TestMode)
r := gin.New()
r.GET("/api/v1/accounts/:account_id/csats", handler.List)
r.GET("/api/v1/accounts/:account_id/csats/metrics", handler.Metrics)
r.POST("/api/v1/accounts/:account_id/csats/:id/update_review_notes", handler.UpdateReviewNotes)
return r
}
func newTestCsatSurveyHandler() *CsatSurveyHandler {
svc := automation.NewCsatSurveyService(&csatMockDBProvider{})
return NewCsatSurveyHandler(svc)
}
// --- List endpoint tests ---
func TestCsatSurveyList_InvalidAccountID(t *testing.T) {
handler := newTestCsatSurveyHandler()
router := setupCsatSurveyRouter(handler)
w := httptest.NewRecorder()
req, _ := http.NewRequest("GET", "/api/v1/accounts/xyz/csats", nil)
router.ServeHTTP(w, req)
assert.Equal(t, http.StatusBadRequest, w.Code)
}
// --- Metrics endpoint tests ---
func TestCsatSurveyMetrics_InvalidAccountID(t *testing.T) {
handler := newTestCsatSurveyHandler()
router := setupCsatSurveyRouter(handler)
w := httptest.NewRecorder()
req, _ := http.NewRequest("GET", "/api/v1/accounts/xyz/csats/metrics", nil)
router.ServeHTTP(w, req)
assert.Equal(t, http.StatusBadRequest, w.Code)
}
// --- UpdateReviewNotes endpoint tests ---
func TestCsatSurveyUpdateReviewNotes_InvalidID(t *testing.T) {
handler := newTestCsatSurveyHandler()
router := setupCsatSurveyRouter(handler)
body := `{"csat_review_notes":"looks good"}`
w := httptest.NewRecorder()
req, _ := http.NewRequest("POST", "/api/v1/accounts/1/csats/xyz/update_review_notes", strings.NewReader(body))
req.Header.Set("Content-Type", "application/json")
router.ServeHTTP(w, req)
assert.Equal(t, http.StatusBadRequest, w.Code)
}
func TestCsatSurveyUpdateReviewNotes_InvalidJSON(t *testing.T) {
handler := newTestCsatSurveyHandler()
router := setupCsatSurveyRouter(handler)
w := httptest.NewRecorder()
req, _ := http.NewRequest("POST", "/api/v1/accounts/1/csats/1/update_review_notes", strings.NewReader("{invalid}"))
req.Header.Set("Content-Type", "application/json")
router.ServeHTTP(w, req)
assert.Equal(t, http.StatusBadRequest, w.Code)
}