mirror of
https://github.com/CJackHwang/ds2api.git
synced 2026-05-13 04:38:00 +08:00
refactor backend API structure
This commit is contained in:
143
internal/httpapi/admin/handler_test.go
Normal file
143
internal/httpapi/admin/handler_test.go
Normal file
@@ -0,0 +1,143 @@
|
||||
package admin
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"sync/atomic"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"ds2api/internal/config"
|
||||
)
|
||||
|
||||
func TestToAccountMissingFieldsRemainEmpty(t *testing.T) {
|
||||
acc := toAccount(map[string]any{
|
||||
"email": "user@example.com",
|
||||
"password": "secret",
|
||||
})
|
||||
if acc.Email != "user@example.com" {
|
||||
t.Fatalf("unexpected email: %q", acc.Email)
|
||||
}
|
||||
if acc.Mobile != "" {
|
||||
t.Fatalf("expected empty mobile, got %q", acc.Mobile)
|
||||
}
|
||||
if acc.Token != "" {
|
||||
t.Fatalf("expected empty token, got %q", acc.Token)
|
||||
}
|
||||
}
|
||||
|
||||
func TestFieldStringNilToEmpty(t *testing.T) {
|
||||
if got := fieldString(map[string]any{"token": nil}, "token"); got != "" {
|
||||
t.Fatalf("expected empty string for nil field, got %q", got)
|
||||
}
|
||||
if got := fieldString(map[string]any{}, "token"); got != "" {
|
||||
t.Fatalf("expected empty string for missing field, got %q", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMaskSecretPreviewKeepsOnlyFirstAndLastTwoChars(t *testing.T) {
|
||||
cases := map[string]string{
|
||||
"": "",
|
||||
"a": "*",
|
||||
"ab": "**",
|
||||
"abcd": "****",
|
||||
"abcdef": "ab****ef",
|
||||
"abc12345": "ab****45",
|
||||
}
|
||||
|
||||
for input, want := range cases {
|
||||
if got := maskSecretPreview(input); got != want {
|
||||
t.Fatalf("maskSecretPreview(%q)=%q want %q", input, got, want)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetConfigMasksAccountTokenPreview(t *testing.T) {
|
||||
h := newAdminTestHandler(t, `{
|
||||
"accounts":[{"email":"u@example.com","password":"pwd"}]
|
||||
}`)
|
||||
if err := h.Store.UpdateAccountToken("u@example.com", "abcdefgh"); err != nil {
|
||||
t.Fatalf("seed runtime token: %v", err)
|
||||
}
|
||||
|
||||
req := httptest.NewRequest(http.MethodGet, "/admin/config", nil)
|
||||
rec := httptest.NewRecorder()
|
||||
h.getConfig(rec, req)
|
||||
|
||||
if rec.Code != http.StatusOK {
|
||||
t.Fatalf("unexpected status: %d body=%s", rec.Code, rec.Body.String())
|
||||
}
|
||||
|
||||
var payload map[string]any
|
||||
if err := json.Unmarshal(rec.Body.Bytes(), &payload); err != nil {
|
||||
t.Fatalf("decode response failed: %v", err)
|
||||
}
|
||||
accounts, _ := payload["accounts"].([]any)
|
||||
if len(accounts) != 1 {
|
||||
t.Fatalf("expected 1 account, got %d", len(accounts))
|
||||
}
|
||||
first, _ := accounts[0].(map[string]any)
|
||||
if got, _ := first["token_preview"].(string); got != "ab****gh" {
|
||||
t.Fatalf("expected masked token preview, got %q", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRunAccountTestsConcurrentlyKeepsInputOrder(t *testing.T) {
|
||||
accounts := []config.Account{
|
||||
{Email: "a@example.com"},
|
||||
{Email: "b@example.com"},
|
||||
{Email: "c@example.com"},
|
||||
}
|
||||
results := runAccountTestsConcurrently(accounts, 2, func(idx int, acc config.Account) map[string]any {
|
||||
return map[string]any{
|
||||
"idx": idx,
|
||||
"account": acc.Identifier(),
|
||||
}
|
||||
})
|
||||
if len(results) != len(accounts) {
|
||||
t.Fatalf("unexpected result length: got %d want %d", len(results), len(accounts))
|
||||
}
|
||||
for i := range accounts {
|
||||
gotIdx, _ := results[i]["idx"].(int)
|
||||
if gotIdx != i {
|
||||
t.Fatalf("result index mismatch at %d: got %d", i, gotIdx)
|
||||
}
|
||||
gotID, _ := results[i]["account"].(string)
|
||||
if gotID != accounts[i].Identifier() {
|
||||
t.Fatalf("result order mismatch at %d: got %q want %q", i, gotID, accounts[i].Identifier())
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestRunAccountTestsConcurrentlyRespectsLimit(t *testing.T) {
|
||||
const limit = 3
|
||||
accounts := []config.Account{
|
||||
{Email: "1@example.com"},
|
||||
{Email: "2@example.com"},
|
||||
{Email: "3@example.com"},
|
||||
{Email: "4@example.com"},
|
||||
{Email: "5@example.com"},
|
||||
{Email: "6@example.com"},
|
||||
}
|
||||
var current int32
|
||||
var maxSeen int32
|
||||
_ = runAccountTestsConcurrently(accounts, limit, func(_ int, _ config.Account) map[string]any {
|
||||
c := atomic.AddInt32(¤t, 1)
|
||||
for {
|
||||
m := atomic.LoadInt32(&maxSeen)
|
||||
if c <= m || atomic.CompareAndSwapInt32(&maxSeen, m, c) {
|
||||
break
|
||||
}
|
||||
}
|
||||
time.Sleep(20 * time.Millisecond)
|
||||
atomic.AddInt32(¤t, -1)
|
||||
return map[string]any{"success": true}
|
||||
})
|
||||
if maxSeen > limit {
|
||||
t.Fatalf("concurrency exceeded limit: got %d > %d", maxSeen, limit)
|
||||
}
|
||||
if maxSeen < 2 {
|
||||
t.Fatalf("expected concurrent execution, max seen %d", maxSeen)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user