store_command_test.go RawHistoryFolder

1package auth
2
3import (
4	"os"
5	"path/filepath"
6	"testing"
7)
8
9func TestStoreRecognizesAndClearsAPIKeyCommand(t *testing.T) {
10	path := filepath.Join(t.TempDir(), "auth.json")
11	if err := os.WriteFile(path, []byte(`{
12  "anthropic": {
13    "api_key_command": {
14      "program": "op",
15      "args": ["read", "op://example/credential"],
16      "timeout_ms": 120000
17    }
18  }
19}`), 0o600); err != nil {
20		t.Fatal(err)
21	}
22	store := NewStore(path)
23	creds, err := store.Load()
24	if err != nil {
25		t.Fatal(err)
26	}
27	if !creds.Has("anthropic") || creds.Method("anthropic") != "apikey" {
28		t.Fatalf("command credential not recognized: %+v", creds.Anthropic)
29	}
30
31	if err := store.ClearAPIKey("anthropic"); err != nil {
32		t.Fatal(err)
33	}
34	creds, err = store.Load()
35	if err != nil {
36		t.Fatal(err)
37	}
38	if creds.Anthropic.APIKeyCommand != nil || creds.Has("anthropic") {
39		t.Fatalf("command credential not cleared: %+v", creds.Anthropic)
40	}
41}
42
43func TestSetAPIKeyReplacesAPIKeyCommand(t *testing.T) {
44	path := filepath.Join(t.TempDir(), "auth.json")
45	if err := os.WriteFile(path, []byte(`{"additional_api_key_creds":{"custom":{"api_key_command":{"program":"op"}}}}`), 0o600); err != nil {
46		t.Fatal(err)
47	}
48	store := NewStore(path)
49	if err := store.SetAPIKey("custom", "literal-key"); err != nil {
50		t.Fatal(err)
51	}
52	creds, err := store.Load()
53	if err != nil {
54		t.Fatal(err)
55	}
56	got := creds.AdditionalAPIKeyCreds["custom"]
57	if got.APIKey != "literal-key" || got.APIKeyCommand != nil {
58		t.Fatalf("stored credential = %+v", got)
59	}
60}