shoal/vibe-proxy
Watch
4
0
Fork
You've already forked vibe-proxy
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add projects

Browse source
This commit is contained in:
Alois 2026-08-24 00:10:41 +02:00
commit 8b607dd700
Signed by: alois
SSH key fingerprint: SHA256:GBzT2DXvAuGV9XIV5W3WrzVpjU54FThmxHXdbz95J24
1802 changed files with 503346 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

116
backend/internal/api/handlers/management/config_claude_key_test.go Normal file
View file

@ -0,0 +1,116 @@
package management
import (
"net/http"
"net/http/httptest"
"strings"
"testing"
"github.com/gin-gonic/gin"
"github.com/router-for-me/CLIProxyAPI/v7/internal/config"
)
func TestPatchClaudeKeyFingerprintProfile(t *testing.T) {
cfg := &config.Config{
ClaudeKey: []config.ClaudeKey{
{APIKey: "test-claude-key"},
},
}
h := &Handler{cfg: cfg, configFilePath: writeTestConfigFile(t)}
// Patch fingerprint-profile to claude-code-cli
rec := httptest.NewRecorder()
ctx, _ := gin.CreateTestContext(rec)
ctx.Request = httptest.NewRequest(http.MethodPatch, "/v0/management/claude-api-key",
strings.NewReader(`{"index":0,"value":{"fingerprint-profile":"claude-code-cli"}}`))
ctx.Request.Header.Set("Content-Type", "application/json")
h.PatchClaudeKey(ctx)
if rec.Code != http.StatusOK {
t.Fatalf("status = %d, want 200; body=%s", rec.Code, rec.Body.String())
}
if got := cfg.ClaudeKey[0].FingerprintProfile; got != "claude-code-cli" {
t.Fatalf("FingerprintProfile = %q, want %q", got, "claude-code-cli")
}
// Patch fingerprint-profile back to empty
rec = httptest.NewRecorder()
ctx, _ = gin.CreateTestContext(rec)
ctx.Request = httptest.NewRequest(http.MethodPatch, "/v0/management/claude-api-key",
strings.NewReader(`{"index":0,"value":{"fingerprint-profile":""}}`))
ctx.Request.Header.Set("Content-Type", "application/json")
h.PatchClaudeKey(ctx)
if rec.Code != http.StatusOK {
t.Fatalf("status = %d, want 200; body=%s", rec.Code, rec.Body.String())
}
if got := cfg.ClaudeKey[0].FingerprintProfile; got != "" {
t.Fatalf("FingerprintProfile = %q, want empty", got)
}
// A legacy alias is stored in canonical form so the config file and the request
// path agree on one spelling.
rec = httptest.NewRecorder()
ctx, _ = gin.CreateTestContext(rec)
ctx.Request = httptest.NewRequest(http.MethodPatch, "/v0/management/claude-api-key",
strings.NewReader(`{"index":0,"value":{"fingerprint-profile":" OAuth-CLI "}}`))
ctx.Request.Header.Set("Content-Type", "application/json")
h.PatchClaudeKey(ctx)
if rec.Code != http.StatusOK {
t.Fatalf("status = %d, want 200; body=%s", rec.Code, rec.Body.String())
}
if got := cfg.ClaudeKey[0].FingerprintProfile; got != "claude-code-cli" {
t.Fatalf("FingerprintProfile = %q, want canonical %q", got, "claude-code-cli")
}
}
// A typo must fail the write instead of reaching the request path, where it can
// only be reported as a warning behind every later request.
func TestPatchClaudeKeyRejectsUnknownFingerprintProfile(t *testing.T) {
cfg := &config.Config{
ClaudeKey: []config.ClaudeKey{
{APIKey: "test-claude-key", FingerprintProfile: "claude-code-cli"},
},
}
h := &Handler{cfg: cfg, configFilePath: writeTestConfigFile(t)}
rec := httptest.NewRecorder()
ctx, _ := gin.CreateTestContext(rec)
ctx.Request = httptest.NewRequest(http.MethodPatch, "/v0/management/claude-api-key",
strings.NewReader(`{"index":0,"value":{"fingerprint-profile":"claude-code"}}`))
ctx.Request.Header.Set("Content-Type", "application/json")
h.PatchClaudeKey(ctx)
if rec.Code != http.StatusBadRequest {
t.Fatalf("status = %d, want 400; body=%s", rec.Code, rec.Body.String())
}
if !strings.Contains(rec.Body.String(), "fingerprint-profile") {
t.Fatalf("error body = %s, want it to name the field", rec.Body.String())
}
if got := cfg.ClaudeKey[0].FingerprintProfile; got != "claude-code-cli" {
t.Fatalf("FingerprintProfile = %q, want the rejected patch to leave it unchanged", got)
}
}
func TestPutClaudeKeysRejectsUnknownFingerprintProfile(t *testing.T) {
cfg := &config.Config{}
h := &Handler{cfg: cfg, configFilePath: writeTestConfigFile(t)}
rec := httptest.NewRecorder()
ctx, _ := gin.CreateTestContext(rec)
ctx.Request = httptest.NewRequest(http.MethodPut, "/v0/management/claude-api-key",
strings.NewReader(`[{"api-key":"k1"},{"api-key":"k2","fingerprint-profile":"claude-cli"}]`))
ctx.Request.Header.Set("Content-Type", "application/json")
h.PutClaudeKeys(ctx)
if rec.Code != http.StatusBadRequest {
t.Fatalf("status = %d, want 400; body=%s", rec.Code, rec.Body.String())
}
if !strings.Contains(rec.Body.String(), "claude-api-key[1].fingerprint-profile") {
t.Fatalf("error body = %s, want the offending index", rec.Body.String())
}
if len(cfg.ClaudeKey) != 0 {
t.Fatalf("ClaudeKey = %+v, want the rejected write to change nothing", cfg.ClaudeKey)
}
}