auth_service_test.go 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540
  1. // AuthService 单测(T4-3 验收):两步登录 + me + refresh + logout + 惰性升级。
  2. //
  3. // 用 mock 仓储 + 真 Issuer + miniredis 票据(不连真实 DB/Redis),覆盖 Service 编排逻辑。
  4. //
  5. // 断言纪律(响应契约 = code-based):
  6. // - Service 返回"自定义文案"的业务错误时用 errors.New(code, customMsg)(如"账号或密码错误"),
  7. // 与 errors 包的 sentinel(如 ErrUnauth="未登录")**指针不同但 code 相同**;
  8. // 故对这类断言 **code**(response.Fail 只读 Code/Msg,不靠指针);
  9. // - Service 返回"真 sentinel"(ErrNotFound/ErrBadParam/ErrTokenExpired 等)时断言 errors.Is。
  10. package service
  11. import (
  12. "context"
  13. "crypto/md5"
  14. "encoding/hex"
  15. stderrors "errors"
  16. "testing"
  17. "yuansiwei/unified-backend/internal/identity/credential"
  18. "yuansiwei/unified-backend/internal/identity/model"
  19. "yuansiwei/unified-backend/internal/identity/token"
  20. apperrors "yuansiwei/unified-backend/internal/infra/errors"
  21. orgmodel "yuansiwei/unified-backend/internal/org/model"
  22. "yuansiwei/unified-backend/internal/org/rbac"
  23. )
  24. // codeOf 取业务错误码(*AppError.Code);非 *AppError 或 nil 时 Fatal(测试 bug)。
  25. // 用于断言"自定义文案错误"的 code(与 sentinel 指针不同但 code 相同)。
  26. func codeOf(t *testing.T, err error) int {
  27. t.Helper()
  28. if err == nil {
  29. t.Fatal("err 为 nil(期望有业务错误)")
  30. return 0
  31. }
  32. var ae *apperrors.AppError
  33. if !stderrors.As(err, &ae) {
  34. t.Fatalf("err 非 *AppError:类型 %T,值 %v", err, err)
  35. return 0
  36. }
  37. return ae.Code
  38. }
  39. // ---- Login(两步第一步)----
  40. // TestLogin_Arion2 命中 argon2 密码:LoginStep1 含 loginTicket + user + memberships。
  41. func TestLogin_Arion2(t *testing.T) {
  42. fx := newFixture(t)
  43. ctx := context.Background()
  44. argon2Hash, err := credential.HashArgon2("correct-password")
  45. if err != nil {
  46. t.Fatalf("HashArgon2: %v", err)
  47. }
  48. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  49. fx.addUser(uid, model.HashVersionArgon2, argon2Hash, "", model.IdentityTypeUsername, "testuser")
  50. fx.memberships.add(uid, orgmodel.Membership{
  51. MembershipID: 1, CanonicalUserID: uid,
  52. InstitutionID: 11, CampusID: 3, Role: "teacher",
  53. Status: "active",
  54. })
  55. res, err := fx.svc.Login(ctx, model.IdentityTypeUsername, "testuser", "correct-password")
  56. if err != nil {
  57. t.Fatalf("Login(argon2 命中): %v", err)
  58. }
  59. if res.LoginTicket == "" {
  60. t.Error("LoginTicket 应非空")
  61. }
  62. if res.User == nil || res.User.ID != uid {
  63. t.Errorf("User.ID = %v,期望 %q", res.User, uid)
  64. }
  65. if len(res.Memberships) != 1 {
  66. t.Fatalf("Memberships 应有 1 条,实际 %d", len(res.Memberships))
  67. }
  68. if res.Memberships[0].InstitutionID != 11 {
  69. t.Errorf("Membership[0].InstitutionID = %d,期望 11", res.Memberships[0].InstitutionID)
  70. }
  71. }
  72. // TestLogin_WrongPassword 密码错 → errAuthFailed(1001)。
  73. func TestLogin_WrongPassword(t *testing.T) {
  74. fx := newFixture(t)
  75. ctx := context.Background()
  76. argon2Hash, _ := credential.HashArgon2("correct-password")
  77. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  78. fx.addUser(uid, model.HashVersionArgon2, argon2Hash, "", model.IdentityTypeUsername, "testuser")
  79. _, err := fx.svc.Login(ctx, model.IdentityTypeUsername, "testuser", "wrong-password")
  80. if codeOf(t, err) != apperrors.CodeUnauth {
  81. t.Fatalf("密码错应 ErrUnauth(1001),实际 %v", err)
  82. }
  83. }
  84. // TestLogin_UnknownIdentity 账号不存在 → errAuthFailed(1001),不泄漏"账号不存在"(防枚举)。
  85. func TestLogin_UnknownIdentity(t *testing.T) {
  86. fx := newFixture(t)
  87. ctx := context.Background()
  88. _, err := fx.svc.Login(ctx, model.IdentityTypeUsername, "ghost-user", "any-password")
  89. if codeOf(t, err) != apperrors.CodeUnauth {
  90. t.Fatalf("账号不存在应 ErrUnauth(1001)(防枚举统一文案),实际 %v", err)
  91. }
  92. }
  93. // TestLogin_DisabledUser 账号 disabled → errAuthFailed(1001)。
  94. func TestLogin_DisabledUser(t *testing.T) {
  95. fx := newFixture(t)
  96. ctx := context.Background()
  97. argon2Hash, _ := credential.HashArgon2("pw")
  98. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  99. fx.addUser(uid, model.HashVersionArgon2, argon2Hash, "", model.IdentityTypeUsername, "disabled")
  100. // 改 status 为 disabled(addUser 默认 active,这里手动改)
  101. fx.users.byUID[uid].Status = model.UserStatusDisabled
  102. _, err := fx.svc.Login(ctx, model.IdentityTypeUsername, "disabled", "pw")
  103. if codeOf(t, err) != apperrors.CodeUnauth {
  104. t.Fatalf("disabled 用户应 ErrUnauth(1001),实际 %v", err)
  105. }
  106. }
  107. // TestLogin_NoHash 无密码哈希 → errAuthFailed(1001)。
  108. func TestLogin_NoHash(t *testing.T) {
  109. fx := newFixture(t)
  110. ctx := context.Background()
  111. // hash="" 表示无哈希(addUser hashVersion="" → HashVersion=nil)
  112. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  113. fx.addUser(uid, "", "", "", model.IdentityTypeUsername, "nohash")
  114. _, err := fx.svc.Login(ctx, model.IdentityTypeUsername, "nohash", "pw")
  115. if codeOf(t, err) != apperrors.CodeUnauth {
  116. t.Fatalf("无密码哈希应 ErrUnauth(1001),实际 %v", err)
  117. }
  118. }
  119. // TestLogin_Md5Salt_LazyUpgrade md5_salt 用户命中 → 惰性升级到 argon2 + 发 ticket。
  120. func TestLogin_Md5Salt_LazyUpgrade(t *testing.T) {
  121. fx := newFixture(t)
  122. ctx := context.Background()
  123. // 造 md5("password"+"salt") 的 hex 串
  124. raw := md5.Sum([]byte("password" + "salt"))
  125. md5Hash := hex.EncodeToString(raw[:])
  126. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  127. fx.addUser(uid, model.HashVersionMd5Salt, md5Hash, "salt", model.IdentityTypeUsername, "legacy")
  128. res, err := fx.svc.Login(ctx, model.IdentityTypeUsername, "legacy", "password")
  129. if err != nil {
  130. t.Fatalf("Login(md5_salt 命中): %v", err)
  131. }
  132. if res.LoginTicket == "" {
  133. t.Error("LazyUpgrade 后应发 loginTicket")
  134. }
  135. // 断言惰性升级已写回:hashVersion 应为 argon2,salt 应为 nil
  136. u := fx.users.byUID[uid]
  137. if u.HashVersion == nil || *u.HashVersion != model.HashVersionArgon2 {
  138. t.Errorf("升级后 hashVersion = %v,期望 argon2", u.HashVersion)
  139. }
  140. if u.PasswordSalt != nil {
  141. t.Errorf("升级后 salt 应置 nil,实际 %q", *u.PasswordSalt)
  142. }
  143. if u.PasswordHash == nil || *u.PasswordHash == md5Hash {
  144. t.Error("升级后 password_hash 应已替换为 argon2 PHC 串(非原 md5 值)")
  145. }
  146. }
  147. // TestLogin_Phone_BadFormat phone 身份 identifier 非 11 位 → ErrBadParam(2001)。
  148. func TestLogin_Phone_BadFormat(t *testing.T) {
  149. fx := newFixture(t)
  150. ctx := context.Background()
  151. _, err := fx.svc.Login(ctx, model.IdentityTypePhone, "12345", "pw")
  152. if !stderrors.Is(err, apperrors.ErrBadParam) {
  153. t.Fatalf("phone 非 11 位应 ErrBadParam(2001),实际 %v", err)
  154. }
  155. }
  156. // ---- Issue(两步第二步)----
  157. // TestIssue_Happy 消费有效 ticket + 合法作用域 → TokenPair(真 access 可验签)。
  158. func TestIssue_Happy(t *testing.T) {
  159. fx := newFixture(t)
  160. ctx := context.Background()
  161. argon2Hash, _ := credential.HashArgon2("pw")
  162. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  163. fx.addUser(uid, model.HashVersionArgon2, argon2Hash, "", model.IdentityTypeUsername, "testuser")
  164. fx.memberships.add(uid, orgmodel.Membership{
  165. MembershipID: 1, CanonicalUserID: uid,
  166. InstitutionID: 11, CampusID: 3, Role: "teacher", Status: "active",
  167. })
  168. // 第一步:Login 拿 ticket
  169. step1, err := fx.svc.Login(ctx, model.IdentityTypeUsername, "testuser", "pw")
  170. if err != nil {
  171. t.Fatalf("Login: %v", err)
  172. }
  173. // 第二步:Issue 签 token
  174. pair, err := fx.svc.Issue(ctx, step1.LoginTicket, 11, 3)
  175. if err != nil {
  176. t.Fatalf("Issue: %v", err)
  177. }
  178. if pair.AccessToken == "" {
  179. t.Error("AccessToken 应非空")
  180. }
  181. if pair.RefreshToken == "" {
  182. t.Error("RefreshToken 应非空")
  183. }
  184. if pair.InstitutionID != 11 {
  185. t.Errorf("InstitutionID = %d,期望 11", pair.InstitutionID)
  186. }
  187. // 验证 access token 真可验签(RS256)
  188. claims, verr := token.Verify(ctx, fx.pubKeys, nil, "https://unified.test", pair.AccessToken)
  189. if verr != nil {
  190. t.Fatalf("token.Verify(新签 access): %v", verr)
  191. }
  192. if claims.Sub != uid {
  193. t.Errorf("claims.Sub = %q,期望 %q", claims.Sub, uid)
  194. }
  195. if claims.InstitutionID != 11 {
  196. t.Errorf("claims.InstitutionID = %d,期望 11", claims.InstitutionID)
  197. }
  198. }
  199. // TestIssue_InvalidTicket 无效/不存在的 ticket → ErrUnauth(1001)。
  200. func TestIssue_InvalidTicket(t *testing.T) {
  201. fx := newFixture(t)
  202. ctx := context.Background()
  203. _, err := fx.svc.Issue(ctx, "invalid-ticket-hex", 11, 3)
  204. if codeOf(t, err) != apperrors.CodeUnauth {
  205. t.Fatalf("无效 ticket 应 ErrUnauth(1001),实际 %v", err)
  206. }
  207. }
  208. // TestIssue_ReplayTicket ticket 已消费(重放)→ ErrUnauth(1001)。
  209. func TestIssue_ReplayTicket(t *testing.T) {
  210. fx := newFixture(t)
  211. ctx := context.Background()
  212. argon2Hash, _ := credential.HashArgon2("pw")
  213. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  214. fx.addUser(uid, model.HashVersionArgon2, argon2Hash, "", model.IdentityTypeUsername, "replay")
  215. fx.memberships.add(uid, orgmodel.Membership{
  216. MembershipID: 1, CanonicalUserID: uid,
  217. InstitutionID: 22, CampusID: 5, Role: "academic", Status: "active",
  218. })
  219. step1, _ := fx.svc.Login(ctx, model.IdentityTypeUsername, "replay", "pw")
  220. // 第一次 Issue(成功)
  221. if _, err := fx.svc.Issue(ctx, step1.LoginTicket, 22, 5); err != nil {
  222. t.Fatalf("第一次 Issue: %v", err)
  223. }
  224. // 第二次 Issue(重放 → 票据已 GETDEL 删除 → 拒绝)
  225. _, err := fx.svc.Issue(ctx, step1.LoginTicket, 22, 5)
  226. if codeOf(t, err) != apperrors.CodeUnauth {
  227. t.Fatalf("重放 ticket 应 ErrUnauth(1001),实际 %v", err)
  228. }
  229. }
  230. // TestIssue_ScopeNotBelong 作用域不属于该用户 → ErrForbidden(1003)。
  231. func TestIssue_ScopeNotBelong(t *testing.T) {
  232. fx := newFixture(t)
  233. ctx := context.Background()
  234. argon2Hash, _ := credential.HashArgon2("pw")
  235. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  236. fx.addUser(uid, model.HashVersionArgon2, argon2Hash, "", model.IdentityTypeUsername, "scopetest")
  237. // 用户只有 instA/campus1 的 membership
  238. fx.memberships.add(uid, orgmodel.Membership{
  239. MembershipID: 1, CanonicalUserID: uid,
  240. InstitutionID: 11, CampusID: 1, Role: "teacher", Status: "active",
  241. })
  242. step1, _ := fx.svc.Login(ctx, model.IdentityTypeUsername, "scopetest", "pw")
  243. // 试图签 instB/campus99(不属于该用户)
  244. _, err := fx.svc.Issue(ctx, step1.LoginTicket, 99, 99)
  245. if codeOf(t, err) != apperrors.CodeForbidden {
  246. t.Fatalf("作用域越权应 ErrForbidden(1003),实际 %v", err)
  247. }
  248. }
  249. // ---- Me ----
  250. // TestMe_Happy 登录态自查:凭 claims 回读 user + memberships。
  251. func TestMe_Happy(t *testing.T) {
  252. fx := newFixture(t)
  253. ctx := context.Background()
  254. argon2Hash, _ := credential.HashArgon2("pw")
  255. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  256. fx.addUser(uid, model.HashVersionArgon2, argon2Hash, "", model.IdentityTypeUsername, "meuser")
  257. fx.memberships.add(uid,
  258. orgmodel.Membership{MembershipID: 1, CanonicalUserID: uid, InstitutionID: 11, CampusID: 3, Role: "teacher", Status: "active"},
  259. orgmodel.Membership{MembershipID: 2, CanonicalUserID: uid, InstitutionID: 11, CampusID: 5, Role: "finance", Status: "active"},
  260. )
  261. claims := token.Claims{Sub: uid, Roles: []string{"teacher"}, InstitutionID: 11, CampusID: 3}
  262. me, err := fx.svc.Me(ctx, claims)
  263. if err != nil {
  264. t.Fatalf("Me: %v", err)
  265. }
  266. if me.UserView == nil || me.UserView.ID != uid {
  267. t.Errorf("Me.UserView.ID = %v,期望 %q", me.UserView, uid)
  268. }
  269. if len(me.Memberships) != 2 {
  270. t.Errorf("Me.Memberships 应有 2 条,实际 %d", len(me.Memberships))
  271. }
  272. }
  273. // TestMe_UnknownSub claims.Sub 对应用户不存在 → ErrNotFound(3001)。
  274. func TestMe_UnknownSub(t *testing.T) {
  275. fx := newFixture(t)
  276. ctx := context.Background()
  277. claims := token.Claims{Sub: "00000000000000000000000000"}
  278. _, err := fx.svc.Me(ctx, claims)
  279. if !stderrors.Is(err, apperrors.ErrNotFound) {
  280. t.Fatalf("Me(未知 sub) 应 ErrNotFound(3001),实际 %v", err)
  281. }
  282. }
  283. // ---- T6.5-d:MenuProvider 注入后 Issue/Me 下发 menuTree + permissions ----
  284. // fakeMenuProvider MenuProvider 测试假实现(按角色返回固定菜单树/权限码集;不连 DB)。
  285. type fakeMenuProvider struct {
  286. codes map[string][]string
  287. tree map[string][]rbac.MenuNode
  288. }
  289. func (f *fakeMenuProvider) Permissions(_ context.Context, role string) ([]string, error) {
  290. if c, ok := f.codes[role]; ok {
  291. return c, nil
  292. }
  293. return []string{}, nil
  294. }
  295. func (f *fakeMenuProvider) MenuTree(_ context.Context, role string) ([]rbac.MenuNode, error) {
  296. if tr, ok := f.tree[role]; ok {
  297. return tr, nil
  298. }
  299. return []rbac.MenuNode{}, nil
  300. }
  301. // TestIssue_WithMenuProvider 注入 MenuProvider 后 Issue 出参含 menuTree + permissions(按角色)。
  302. func TestIssue_WithMenuProvider(t *testing.T) {
  303. fx := newFixture(t)
  304. ctx := context.Background()
  305. argon2Hash, _ := credential.HashArgon2("pw")
  306. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  307. fx.addUser(uid, model.HashVersionArgon2, argon2Hash, "", model.IdentityTypeUsername, "menuser")
  308. fx.memberships.add(uid, orgmodel.Membership{
  309. MembershipID: 1, CanonicalUserID: uid,
  310. InstitutionID: 11, CampusID: 3, Role: "academic", Status: "active",
  311. })
  312. // 注入假解析器:academic 角色 2 权限码 + 1 菜单节点
  313. fx.svc.menuProvider = &fakeMenuProvider{
  314. codes: map[string][]string{"academic": {"org.institution.list", "org.campus.list"}},
  315. tree: map[string][]rbac.MenuNode{
  316. "academic": {{Code: "org.institution.list", Label: "机构管理", Path: strPtr("/org/institutions")}},
  317. },
  318. }
  319. step1, err := fx.svc.Login(ctx, model.IdentityTypeUsername, "menuser", "pw")
  320. if err != nil {
  321. t.Fatalf("Login: %v", err)
  322. }
  323. pair, err := fx.svc.Issue(ctx, step1.LoginTicket, 11, 3)
  324. if err != nil {
  325. t.Fatalf("Issue: %v", err)
  326. }
  327. if len(pair.Permissions) != 2 || pair.Permissions[0] != "org.institution.list" {
  328. t.Fatalf("Issue.Permissions=%v(期望 2 个 org 权限码)", pair.Permissions)
  329. }
  330. if len(pair.MenuTree) != 1 || pair.MenuTree[0].Code != "org.institution.list" {
  331. t.Fatalf("Issue.MenuTree=%+v(期望 1 个机构管理菜单)", pair.MenuTree)
  332. }
  333. }
  334. // TestMe_WithMenuProvider 注入 MenuProvider 后 Me 出参含 menuTree + permissions(菜单刷新恢复)。
  335. func TestMe_WithMenuProvider(t *testing.T) {
  336. fx := newFixture(t)
  337. ctx := context.Background()
  338. argon2Hash, _ := credential.HashArgon2("pw")
  339. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  340. fx.addUser(uid, model.HashVersionArgon2, argon2Hash, "", model.IdentityTypeUsername, "memuuser")
  341. fx.svc.menuProvider = &fakeMenuProvider{
  342. codes: map[string][]string{"super_admin": {"org.institution.list", "org.campus.list", "sys.menu"}},
  343. tree: map[string][]rbac.MenuNode{
  344. "super_admin": {
  345. {Code: "org.institution.list", Label: "机构管理"},
  346. {Code: "org.campus.list", Label: "校区管理"},
  347. },
  348. },
  349. }
  350. claims := token.Claims{Sub: uid, Roles: []string{"super_admin"}, InstitutionID: 0, CampusID: 0}
  351. me, err := fx.svc.Me(ctx, claims)
  352. if err != nil {
  353. t.Fatalf("Me: %v", err)
  354. }
  355. if len(me.Permissions) != 3 {
  356. t.Fatalf("Me.Permissions=%v(超管期望 3 个权限码)", me.Permissions)
  357. }
  358. if len(me.MenuTree) != 2 {
  359. t.Fatalf("Me.MenuTree=%+v(超管期望 2 个顶级菜单)", me.MenuTree)
  360. }
  361. }
  362. // TestIssue_NilMenuProvider MenuProvider 未注入(默认 fixture):menuTree/permissions 非 nil 空集(不阻断)。
  363. func TestIssue_NilMenuProvider(t *testing.T) {
  364. fx := newFixture(t) // 不注入 menuProvider
  365. ctx := context.Background()
  366. argon2Hash, _ := credential.HashArgon2("pw")
  367. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  368. fx.addUser(uid, model.HashVersionArgon2, argon2Hash, "", model.IdentityTypeUsername, "nilmenu")
  369. fx.memberships.add(uid, orgmodel.Membership{
  370. MembershipID: 1, CanonicalUserID: uid,
  371. InstitutionID: 11, CampusID: 3, Role: "teacher", Status: "active",
  372. })
  373. step1, _ := fx.svc.Login(ctx, model.IdentityTypeUsername, "nilmenu", "pw")
  374. pair, err := fx.svc.Issue(ctx, step1.LoginTicket, 11, 3)
  375. if err != nil {
  376. t.Fatalf("Issue: %v", err)
  377. }
  378. if pair.MenuTree == nil || pair.Permissions == nil {
  379. t.Fatalf("MenuTree/Permissions 必须非 nil(实际 %v / %v),空集应是 [] 而非 null", pair.MenuTree, pair.Permissions)
  380. }
  381. if len(pair.MenuTree) != 0 || len(pair.Permissions) != 0 {
  382. t.Fatalf("nil MenuProvider 时 menuTree/permissions 应空集(实际 %d / %d)", len(pair.MenuTree), len(pair.Permissions))
  383. }
  384. }
  385. // strPtr 字符串指针辅助(测试用)。
  386. func strPtr(s string) *string { return &s }
  387. // ---- Refresh ----
  388. // TestRefresh_Happy 用有效 refresh 轮换 → 新 TokenPair。
  389. func TestRefresh_Happy(t *testing.T) {
  390. fx := newFixture(t)
  391. ctx := context.Background()
  392. argon2Hash, _ := credential.HashArgon2("pw")
  393. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  394. fx.addUser(uid, model.HashVersionArgon2, argon2Hash, "", model.IdentityTypeUsername, "refreshuser")
  395. fx.memberships.add(uid, orgmodel.Membership{
  396. MembershipID: 1, CanonicalUserID: uid,
  397. InstitutionID: 11, CampusID: 3, Role: "teacher", Status: "active",
  398. })
  399. step1, _ := fx.svc.Login(ctx, model.IdentityTypeUsername, "refreshuser", "pw")
  400. pair1, err := fx.svc.Issue(ctx, step1.LoginTicket, 11, 3)
  401. if err != nil {
  402. t.Fatalf("Issue: %v", err)
  403. }
  404. // 用 pair1.RefreshToken 轮换
  405. pair2, err := fx.svc.Refresh(ctx, pair1.RefreshToken)
  406. if err != nil {
  407. t.Fatalf("Refresh: %v", err)
  408. }
  409. if pair2.AccessToken == "" {
  410. t.Error("Refresh 后 AccessToken 应非空")
  411. }
  412. if pair2.AccessToken == pair1.AccessToken {
  413. t.Error("Refresh 后 access 应换新(jti 不同)")
  414. }
  415. if pair2.User == nil || pair2.User.ID != uid {
  416. t.Errorf("Refresh 后 User.ID = %v,期望 %q", pair2.User, uid)
  417. }
  418. // 旧 refresh 应已失效(轮换后旧 jti 已删)
  419. _, err = fx.svc.Refresh(ctx, pair1.RefreshToken)
  420. if !stderrors.Is(err, apperrors.ErrTokenExpired) {
  421. t.Fatalf("旧 refresh 重放应 ErrTokenExpired(1002),实际 %v", err)
  422. }
  423. }
  424. // TestRefresh_InvalidRefresh 无效 refresh → ErrTokenExpired(1002)。
  425. func TestRefresh_InvalidRefresh(t *testing.T) {
  426. fx := newFixture(t)
  427. ctx := context.Background()
  428. _, err := fx.svc.Refresh(ctx, "not-a-valid-refresh-token")
  429. if !stderrors.Is(err, apperrors.ErrTokenExpired) {
  430. t.Fatalf("无效 refresh 应 ErrTokenExpired(1002),实际 %v", err)
  431. }
  432. }
  433. // ---- Logout ----
  434. // TestLogout_Happy 登出:删 refresh + 吊销 access(幂等,返回 nil)。
  435. func TestLogout_Happy(t *testing.T) {
  436. fx := newFixture(t)
  437. ctx := context.Background()
  438. argon2Hash, _ := credential.HashArgon2("pw")
  439. uid := "01GAQF0Z6C1V9Y3E5H7K2M8B4T"
  440. fx.addUser(uid, model.HashVersionArgon2, argon2Hash, "", model.IdentityTypeUsername, "logoutuser")
  441. fx.memberships.add(uid, orgmodel.Membership{
  442. MembershipID: 1, CanonicalUserID: uid,
  443. InstitutionID: 11, CampusID: 3, Role: "teacher", Status: "active",
  444. })
  445. step1, _ := fx.svc.Login(ctx, model.IdentityTypeUsername, "logoutuser", "pw")
  446. pair, _ := fx.svc.Issue(ctx, step1.LoginTicket, 11, 3)
  447. // 登出(传 refresh + 一个 jti + 剩余 TTL)
  448. err := fx.svc.Logout(ctx, pair.RefreshToken, "some-jti", 30*60*1e9) // 30min = 1.8e9 ns
  449. if err != nil {
  450. t.Fatalf("Logout: %v", err)
  451. }
  452. // 登出后旧 refresh 应失效(已删白名单)
  453. _, err = fx.svc.Refresh(ctx, pair.RefreshToken)
  454. if !stderrors.Is(err, apperrors.ErrTokenExpired) {
  455. t.Errorf("Logout 后旧 refresh 应 ErrTokenExpired(1002),实际 %v", err)
  456. }
  457. }
  458. // TestLogout_EmptyArgs 空参登出(幂等,不报错)。
  459. func TestLogout_EmptyArgs(t *testing.T) {
  460. fx := newFixture(t)
  461. ctx := context.Background()
  462. err := fx.svc.Logout(ctx, "", "", 0)
  463. if err != nil {
  464. t.Fatalf("Logout(空参) 应 nil(幂等),实际 %v", err)
  465. }
  466. }