package passkey import ( "testing" "github.com/glebarez/sqlite" "github.com/openteam/server/internal/store" "gorm.io/gorm" ) func newTestService(t *testing.T) *Service { t.Helper() db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{}) if err != nil { t.Fatal(err) } if err := db.AutoMigrate(&store.User{}, &store.Passkey{}); err != nil { t.Fatal(err) } s, err := New(db, Config{RPID: "localhost", Origin: "http://localhost:8080", Name: "openteam test"}) if err != nil { t.Fatal(err) } return s } // TestChallengeSessionRoundTrip 验证 begin 产生的 challenge 能被 complete 取回。 // go-webauthn 默认 Enforce=false,SessionData.Expires 为零值;takeSession 的 // 过期判断必须跳过零值时间,否则每个 challenge 都被当成已过期(线上必现 // "challenge 已过期或不存在")。 func TestChallengeSessionRoundTrip(t *testing.T) { s := newTestService(t) u := &store.User{Username: "alice", Email: "alice@example.com", Role: "user", Status: "active"} if err := s.db.Create(u).Error; err != nil { t.Fatal(err) } creation, err := s.BeginRegistration(u) if err != nil { t.Fatal(err) } challenge := creation.Response.Challenge.String() // 与前端回传一致的 base64url sess, ok := s.takeSession(challenge) if !ok { t.Fatal("takeSession 返回 false:challenge 被误判为已过期(Expires 零值 bug)") } if sess.Challenge != challenge { t.Fatalf("session challenge 不匹配: %q != %q", sess.Challenge, challenge) } // challenge 一次性消费 if _, ok := s.takeSession(challenge); ok { t.Fatal("takeSession 应一次性消费 challenge,二次取应失败") } }