wehub-resource-sync
107 lines
3.3 KiB
Go
107 lines
3.3 KiB
Go
package response_test
|
||
|
||
import (
|
||
"encoding/json"
|
||
"net/http/httptest"
|
||
"testing"
|
||
|
||
"github.com/EthanCodeCraft/xlgo-core/response"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// withMode 切换响应模式并在测试结束后恢复为默认 ModeBusiness。
|
||
func withMode(t *testing.T, m response.Mode) {
|
||
t.Helper()
|
||
response.SetMode(m)
|
||
t.Cleanup(func() { response.SetMode(response.ModeBusiness) })
|
||
}
|
||
|
||
func TestSetGetMode(t *testing.T) {
|
||
withMode(t, response.ModeREST)
|
||
if response.GetMode() != response.ModeREST {
|
||
t.Errorf("GetMode = %v, want ModeREST", response.GetMode())
|
||
}
|
||
}
|
||
|
||
func TestModeBusinessReturns200(t *testing.T) {
|
||
withMode(t, response.ModeBusiness)
|
||
r := setupTestRouter()
|
||
r.GET("/u", func(c *gin.Context) { response.Unauthorized(c, "no") })
|
||
|
||
w := httptest.NewRecorder()
|
||
r.ServeHTTP(w, httptest.NewRequest("GET", "/u", nil))
|
||
|
||
if w.Code != 200 {
|
||
t.Errorf("ModeBusiness Unauthorized status = %d, want 200", w.Code)
|
||
}
|
||
}
|
||
|
||
func TestModeRESTMapsStatus(t *testing.T) {
|
||
withMode(t, response.ModeREST)
|
||
cases := []struct {
|
||
path string
|
||
fn func(c *gin.Context)
|
||
wantCode int
|
||
}{
|
||
{"/unauth", func(c *gin.Context) { response.Unauthorized(c, "no") }, 401},
|
||
{"/notfound", func(c *gin.Context) { response.NotFound(c, "no") }, 404},
|
||
{"/server", func(c *gin.Context) { response.ServerError(c, "err") }, 500},
|
||
{"/ratelimit", func(c *gin.Context) { response.RateLimit(c) }, 429},
|
||
{"/fail", func(c *gin.Context) { response.Fail(c, "bad") }, 200}, // CodeFail 不映射
|
||
}
|
||
for _, tc := range cases {
|
||
r := setupTestRouter()
|
||
r.GET(tc.path, tc.fn)
|
||
w := httptest.NewRecorder()
|
||
r.ServeHTTP(w, httptest.NewRequest("GET", tc.path, nil))
|
||
if w.Code != tc.wantCode {
|
||
t.Errorf("%s status = %d, want %d", tc.path, w.Code, tc.wantCode)
|
||
}
|
||
// body 仍带业务码
|
||
var body response.Response
|
||
if err := json.Unmarshal(w.Body.Bytes(), &body); err != nil {
|
||
t.Errorf("%s body unmarshal: %v", tc.path, err)
|
||
}
|
||
if body.Code == 0 && tc.path != "/fail" {
|
||
// 非 Fail 路径 code 应非 0（Fail 的 CodeFail=1 也非 0，跳过）
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestCustomIgnoresMode(t *testing.T) {
|
||
withMode(t, response.ModeBusiness) // 即使 business 模式，Custom 也用指定 status
|
||
r := setupTestRouter()
|
||
r.GET("/c", func(c *gin.Context) { response.Custom(c, 418, 999, "teapot", nil) })
|
||
|
||
w := httptest.NewRecorder()
|
||
r.ServeHTTP(w, httptest.NewRequest("GET", "/c", nil))
|
||
if w.Code != 418 {
|
||
t.Errorf("Custom status = %d, want 418", w.Code)
|
||
}
|
||
}
|
||
|
||
func TestFailWithErrorREST(t *testing.T) {
|
||
withMode(t, response.ModeREST)
|
||
r := setupTestRouter()
|
||
r.GET("/e", func(c *gin.Context) { response.FailWithError(c, response.ErrUnauthorized) })
|
||
|
||
w := httptest.NewRecorder()
|
||
r.ServeHTTP(w, httptest.NewRequest("GET", "/e", nil))
|
||
if w.Code != 401 {
|
||
t.Errorf("FailWithError(ErrUnauthorized) status = %d, want 401", w.Code)
|
||
}
|
||
}
|
||
|
||
// TestDataAlreadyExistsMapsConflict_B7：CodeDataAlreadyExists 应映射 409（与 CodeDataConflict 一致）。
|
||
func TestDataAlreadyExistsMapsConflict_B7(t *testing.T) {
|
||
withMode(t, response.ModeREST)
|
||
r := setupTestRouter()
|
||
r.GET("/dup", func(c *gin.Context) { response.FailWithError(c, response.ErrDataAlreadyExists) })
|
||
|
||
w := httptest.NewRecorder()
|
||
r.ServeHTTP(w, httptest.NewRequest("GET", "/dup", nil))
|
||
if w.Code != 409 {
|
||
t.Errorf("ErrDataAlreadyExists status = %d, want 409", w.Code)
|
||
}
|
||
}
|