opencode/picoclaw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix(auth): improve error handling in credential storage tests

Browse source
This commit is contained in:
lc6464 2026-04-20 21:52:38 +08:00
parent c738497820
commit 8257f25fdc
No known key found for this signature in database
GPG key ID: 53C61B42FEC71D6D
1 changed files with 21 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
pkg/auth/store_test.go
View file

@ -206,10 +206,12 @@ func TestGetCredentialCanonicalizesLegacyAntigravityProvider(t *testing.T) {
t.Fatalf("json.Marshal() error: %v", err) t.Fatalf("json.Marshal() error: %v", err)
} }
path := filepath.Join(tmpDir, ".picoclaw", "auth.json") path := filepath.Join(tmpDir, ".picoclaw", "auth.json")
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil { err = os.MkdirAll(filepath.Dir(path), 0o755)
if err != nil {
t.Fatalf("MkdirAll() error: %v", err) t.Fatalf("MkdirAll() error: %v", err)
} }
if err := os.WriteFile(path, data, 0o600); err != nil { err = os.WriteFile(path, data, 0o600)
if err != nil {
t.Fatalf("WriteFile() error: %v", err) t.Fatalf("WriteFile() error: %v", err)
} }
@ -257,10 +259,12 @@ func TestLoadStoreMergesAntigravityAliasesPreferringNewerExpiry(t *testing.T) {
t.Fatalf("json.Marshal() error: %v", err) t.Fatalf("json.Marshal() error: %v", err)
} }
path := filepath.Join(tmpDir, ".picoclaw", "auth.json") path := filepath.Join(tmpDir, ".picoclaw", "auth.json")
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil { err = os.MkdirAll(filepath.Dir(path), 0o755)
if err != nil {
t.Fatalf("MkdirAll() error: %v", err) t.Fatalf("MkdirAll() error: %v", err)
} }
if err := os.WriteFile(path, data, 0o600); err != nil { err = os.WriteFile(path, data, 0o600)
if err != nil {
t.Fatalf("WriteFile() error: %v", err) t.Fatalf("WriteFile() error: %v", err)
} }
@ -311,20 +315,23 @@ func TestSetCredentialReplacesLegacyAntigravityEntry(t *testing.T) {
t.Fatalf("json.Marshal() error: %v", err) t.Fatalf("json.Marshal() error: %v", err)
} }
path := filepath.Join(tmpDir, ".picoclaw", "auth.json") path := filepath.Join(tmpDir, ".picoclaw", "auth.json")
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil { err = os.MkdirAll(filepath.Dir(path), 0o755)
if err != nil {
t.Fatalf("MkdirAll() error: %v", err) t.Fatalf("MkdirAll() error: %v", err)
} }
if err := os.WriteFile(path, data, 0o600); err != nil { err = os.WriteFile(path, data, 0o600)
if err != nil {
t.Fatalf("WriteFile() error: %v", err) t.Fatalf("WriteFile() error: %v", err)
} }
refreshedExpiry := time.Date(2026, 4, 16, 12, 30, 0, 0, time.UTC) refreshedExpiry := time.Date(2026, 4, 16, 12, 30, 0, 0, time.UTC)
if err := SetCredential("google-antigravity", &AuthCredential{ err = SetCredential("google-antigravity", &AuthCredential{
AccessToken: "fresh-token", AccessToken: "fresh-token",
ExpiresAt: refreshedExpiry, ExpiresAt: refreshedExpiry,
Provider: "google-antigravity", Provider: "google-antigravity",
AuthMethod: "oauth", AuthMethod: "oauth",
}); err != nil { })
if err != nil {
t.Fatalf("SetCredential() error: %v", err) t.Fatalf("SetCredential() error: %v", err)
} }
@ -365,14 +372,17 @@ func TestDeleteCredentialRemovesLegacyAntigravityAlias(t *testing.T) {
t.Fatalf("json.Marshal() error: %v", err) t.Fatalf("json.Marshal() error: %v", err)
} }
path := filepath.Join(tmpDir, ".picoclaw", "auth.json") path := filepath.Join(tmpDir, ".picoclaw", "auth.json")
if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil { err = os.MkdirAll(filepath.Dir(path), 0o755)
if err != nil {
t.Fatalf("MkdirAll() error: %v", err) t.Fatalf("MkdirAll() error: %v", err)
} }
if err := os.WriteFile(path, data, 0o600); err != nil { err = os.WriteFile(path, data, 0o600)
if err != nil {
t.Fatalf("WriteFile() error: %v", err) t.Fatalf("WriteFile() error: %v", err)
} }
if err := DeleteCredential(" google-antigravity "); err != nil { err = DeleteCredential(" google-antigravity ")
if err != nil {
t.Fatalf("DeleteCredential() error: %v", err) t.Fatalf("DeleteCredential() error: %v", err)
} }