store_additional_test.go RawHistoryFolder

1package auth
2
3import (
4	"path/filepath"
5	"testing"
6)
7
8func TestStoreEndpointCredentialAllowsOptionalAPIKey(t *testing.T) {
9	store := NewStore(filepath.Join(t.TempDir(), "auth.json"))
10	if err := store.SetEndpointCredential("llama.cpp", "http://127.0.0.1:8080", ""); err != nil {
11		t.Fatal(err)
12	}
13	baseURL, key, err := store.EndpointCredential("llama.cpp")
14	if err != nil {
15		t.Fatal(err)
16	}
17	if baseURL != "http://127.0.0.1:8080" || key != "" {
18		t.Fatalf("endpoint = (%q, %q)", baseURL, key)
19	}
20	creds, err := store.Load()
21	if err != nil {
22		t.Fatal(err)
23	}
24	if !creds.Has("llama.cpp") || creds.Method("llama.cpp") != "apikey" {
25		t.Fatalf("stored credentials not recognized: %+v", creds.AdditionalAPIKeyCreds)
26	}
27}
28
29func TestStoreAdditionalAPIKeyClear(t *testing.T) {
30	store := NewStore(filepath.Join(t.TempDir(), "auth.json"))
31	if err := store.SetAPIKey("groq", "gsk_test"); err != nil {
32		t.Fatal(err)
33	}
34	creds, err := store.Load()
35	if err != nil {
36		t.Fatal(err)
37	}
38	if got := creds.Method("groq"); got != "apikey" {
39		t.Fatalf("method before clear=%q", got)
40	}
41	if err := store.Clear("groq"); err != nil {
42		t.Fatal(err)
43	}
44	creds, err = store.Load()
45	if err != nil {
46		t.Fatal(err)
47	}
48	if got := creds.Method("groq"); got != "" {
49		t.Fatalf("method after clear=%q", got)
50	}
51	if len(creds.AdditionalAPIKeyCreds) != 0 {
52		t.Fatalf("additional creds not cleared: %+v", creds.AdditionalAPIKeyCreds)
53	}
54}