jackyu66gitandCursor
c81f57a7d1
聚合 GET /admin/users/:id/entitlements（Membership∪DeepAccess∪问答额度）与 admin-h5 权益 Tab；无 migration / 无真支付。 Co-authored-by: Cursor <cursoragent@cursor.com>
106 lines
3.2 KiB
Go
106 lines
3.2 KiB
Go
package integration_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestUserEntitlements(t *testing.T) {
|
|
r, _ := setupAPIPool(t)
|
|
tok := adminLogin(t, r, "admin", "change-me")
|
|
|
|
_, code := doAdminJSON(t, r, http.MethodGet, "/api/v1/admin/users/"+fakeUUID()+"/entitlements", nil, "")
|
|
if code != http.StatusUnauthorized {
|
|
t.Fatalf("expected 401, got %d", code)
|
|
}
|
|
|
|
testBearer = ""
|
|
phone := fmt.Sprintf("1%010d", time.Now().UnixNano()%10_000_000_000)
|
|
nick := "ent_" + phone[7:]
|
|
env, key := doJSON(t, r, http.MethodPost, "/api/v1/auth/register", map[string]any{
|
|
"phone": phone, "password": "secret12", "nickname": nick,
|
|
}, "")
|
|
sess := decodeData[map[string]any](t, env.Data)
|
|
testBearer = sess["token"].(string)
|
|
t.Cleanup(func() { testBearer = "" })
|
|
|
|
env, key = doJSON(t, r, http.MethodPost, "/api/v1/profiles", map[string]any{
|
|
"relation": "self", "birth_date": "1990-01-01", "display_name": "权",
|
|
}, key)
|
|
profileID := decodeData[map[string]any](t, env.Data)["id"].(string)
|
|
|
|
env, key = doJSON(t, r, http.MethodPost, "/api/v1/reports/portrait", map[string]any{
|
|
"profile_id": profileID,
|
|
}, key)
|
|
reportID := decodeData[map[string]any](t, env.Data)["id"].(string)
|
|
|
|
env, key = doJSON(t, r, http.MethodPost, "/api/v1/orders", map[string]any{
|
|
"kind": "deep_access", "report_id": reportID,
|
|
}, key)
|
|
orderID := decodeData[map[string]any](t, env.Data)["order_id"].(string)
|
|
_, key = doJSON(t, r, http.MethodPost, "/api/v1/orders/"+orderID+"/pay-mock", nil, key)
|
|
|
|
env, code = doAdminJSON(t, r, http.MethodGet, "/api/v1/admin/users?q="+url.QueryEscape(nick), nil, tok)
|
|
if code != 200 {
|
|
t.Fatalf("list users %d", code)
|
|
}
|
|
var list struct {
|
|
Items []struct {
|
|
ID string `json:"id"`
|
|
} `json:"items"`
|
|
}
|
|
_ = json.Unmarshal(env.Data, &list)
|
|
userID := list.Items[0].ID
|
|
|
|
start := time.Now()
|
|
env, code = doAdminJSON(t, r, http.MethodGet, "/api/v1/admin/users/"+userID+"/entitlements", nil, tok)
|
|
if code != 200 || env.Code != 0 {
|
|
t.Fatalf("entitlements http=%d msg=%s", code, env.Message)
|
|
}
|
|
if time.Since(start) > 500*time.Millisecond {
|
|
t.Fatalf("too slow %v", time.Since(start))
|
|
}
|
|
var before struct {
|
|
Flags struct {
|
|
ViaMem bool `json:"report_detail_via_membership"`
|
|
Count int `json:"deep_access_count"`
|
|
} `json:"flags"`
|
|
Deep []any `json:"deep_accesses"`
|
|
}
|
|
_ = json.Unmarshal(env.Data, &before)
|
|
if before.Flags.Count < 1 || len(before.Deep) < 1 {
|
|
t.Fatalf("expected deep_access: %#v", before)
|
|
}
|
|
if before.Flags.ViaMem {
|
|
t.Fatal("expected membership inactive before grant")
|
|
}
|
|
|
|
_, code = doAdminJSON(t, r, http.MethodPost, "/api/v1/admin/users/"+userID+"/membership/grant",
|
|
map[string]string{"plan": "month"}, tok)
|
|
if code != 200 {
|
|
t.Fatalf("grant %d", code)
|
|
}
|
|
|
|
env, code = doAdminJSON(t, r, http.MethodGet, "/api/v1/admin/users/"+userID+"/entitlements", nil, tok)
|
|
if code != 200 {
|
|
t.Fatalf("after grant %d", code)
|
|
}
|
|
var after struct {
|
|
Flags struct {
|
|
ViaMem bool `json:"report_detail_via_membership"`
|
|
} `json:"flags"`
|
|
Membership struct {
|
|
Active bool `json:"active"`
|
|
} `json:"membership"`
|
|
}
|
|
_ = json.Unmarshal(env.Data, &after)
|
|
if !after.Flags.ViaMem || !after.Membership.Active {
|
|
t.Fatalf("expected active membership entitlement: %#v", after)
|
|
}
|
|
_ = key
|
|
}
|