shanshanzhong
6c370485d1
Some checks failed
Build docker and publish / build (20.15.1) (push) Failing after 8m1s
103 lines
2.8 KiB
Go
103 lines
2.8 KiB
Go
package user
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"net"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/perfect-panel/server/internal/config"
|
|
"github.com/perfect-panel/server/internal/svc"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
type handlerResponse struct {
|
|
Code uint32 `json:"code"`
|
|
Msg string `json:"msg"`
|
|
Data json.RawMessage `json:"data"`
|
|
}
|
|
|
|
func newDeleteAccountTestRouter(serverCtx *svc.ServiceContext) *gin.Engine {
|
|
gin.SetMode(gin.TestMode)
|
|
router := gin.New()
|
|
router.POST("/v1/public/user/delete_account", DeleteAccountHandler(serverCtx))
|
|
return router
|
|
}
|
|
|
|
func TestDeleteAccountHandlerInvalidParamsUsesUnifiedResponse(t *testing.T) {
|
|
router := newDeleteAccountTestRouter(&svc.ServiceContext{})
|
|
|
|
reqBody := bytes.NewBufferString(`{"email":"invalid-email"}`)
|
|
req := httptest.NewRequest(http.MethodPost, "/v1/public/user/delete_account", reqBody)
|
|
req.Header.Set("Content-Type", "application/json")
|
|
recorder := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(recorder, req)
|
|
|
|
if recorder.Code != http.StatusOK {
|
|
t.Fatalf("expected HTTP 200, got %d", recorder.Code)
|
|
}
|
|
|
|
var resp handlerResponse
|
|
if err := json.Unmarshal(recorder.Body.Bytes(), &resp); err != nil {
|
|
t.Fatalf("failed to decode response: %v", err)
|
|
}
|
|
|
|
if resp.Code != 400 {
|
|
t.Fatalf("expected business code 400, got %d, body=%s", resp.Code, recorder.Body.String())
|
|
}
|
|
|
|
var raw map[string]interface{}
|
|
if err := json.Unmarshal(recorder.Body.Bytes(), &raw); err != nil {
|
|
t.Fatalf("failed to decode raw response: %v", err)
|
|
}
|
|
if _, exists := raw["error"]; exists {
|
|
t.Fatalf("unexpected raw error field in response: %s", recorder.Body.String())
|
|
}
|
|
}
|
|
|
|
func TestDeleteAccountHandlerVerifyCodeErrorUsesUnifiedResponse(t *testing.T) {
|
|
redisClient := redis.NewClient(&redis.Options{
|
|
Addr: "invalid:6379",
|
|
Dialer: func(_ context.Context, _, _ string) (net.Conn, error) {
|
|
return nil, errors.New("dial disabled in test")
|
|
},
|
|
})
|
|
defer redisClient.Close()
|
|
|
|
serverCtx := &svc.ServiceContext{
|
|
Redis: redisClient,
|
|
Config: config.Config{
|
|
VerifyCode: config.VerifyCode{
|
|
VerifyCodeExpireTime: 900,
|
|
},
|
|
},
|
|
}
|
|
router := newDeleteAccountTestRouter(serverCtx)
|
|
|
|
reqBody := bytes.NewBufferString(`{"email":"user@example.com","code":"123456"}`)
|
|
req := httptest.NewRequest(http.MethodPost, "/v1/public/user/delete_account", reqBody)
|
|
req.Header.Set("Content-Type", "application/json")
|
|
recorder := httptest.NewRecorder()
|
|
|
|
router.ServeHTTP(recorder, req)
|
|
|
|
if recorder.Code != http.StatusOK {
|
|
t.Fatalf("expected HTTP 200, got %d", recorder.Code)
|
|
}
|
|
|
|
var resp handlerResponse
|
|
if err := json.Unmarshal(recorder.Body.Bytes(), &resp); err != nil {
|
|
t.Fatalf("failed to decode response: %v", err)
|
|
}
|
|
|
|
if resp.Code != 70001 {
|
|
t.Fatalf("expected business code 70001, got %d, body=%s", resp.Code, recorder.Body.String())
|
|
}
|
|
}
|