molecule-ai/molecule-core
35
0
Fork 2
Code Issues 16 Pull Requests 2 Actions 49 Packages Projects Releases Wiki Activity

fix(orgtoken): update Validate test mock to include org_id column

Browse Source 
Validate now SELECTs id/prefix/org_id; the test mock row only had two
columns, so the actual query against sqlmock errored with 'invalid or
revoked org api token' at runtime (the row couldn't Scan). Add org_id
to the mocked row and assert it propagates to the 4th return value.

This is a test-only change — the production code path already had the
third column selected; CI was the canary.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
Hongming Wang 2026-04-21 04:20:47 -07:00
parent 343bffdf26
commit 8065d7ef03
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
workspace-server/internal/orgtoken/tokens_test.go
View File

@ -72,14 +72,14 @@ func TestValidate_HappyPath(t *testing.T) {
plaintext := "known-plaintext-for-test"
hash := sha256.Sum256([]byte(plaintext))
mock.ExpectQuery(`SELECT id, prefix FROM org_api_tokens`).
mock.ExpectQuery(`SELECT id, prefix, org_id FROM org_api_tokens`).
WithArgs(hash[:]).
WillReturnRows(sqlmock.NewRows([]string{"id", "prefix"}).AddRow("tok-live", "abcd1234"))
WillReturnRows(sqlmock.NewRows([]string{"id", "prefix", "org_id"}).AddRow("tok-live", "abcd1234", "org-happy"))
mock.ExpectExec(`UPDATE org_api_tokens SET last_used_at`).
WithArgs("tok-live").
WillReturnResult(sqlmock.NewResult(0, 1))
id, prefix, _, err := Validate(context.Background(), db, plaintext)
id, prefix, orgID, err := Validate(context.Background(), db, plaintext)
if err != nil {
t.Fatalf("Validate: %v", err)
}
@ -89,6 +89,9 @@ func TestValidate_HappyPath(t *testing.T) {
if prefix != "abcd1234" {
t.Errorf("prefix = %q, want abcd1234", prefix)
}
if orgID != "org-happy" {
t.Errorf("orgID = %q, want org-happy", orgID)
}
}
func TestValidate_EmptyPlaintextRejected(t *testing.T) {