feat(backend): add kiro account support

2026-04-30 14:02:05 +08:00
parent 9d801595c9
commit 05bc424c9a
60 changed files with 11916 additions and 38 deletions
@@ -0,0 +1,40 @@
+//go:build unit
+
+package service
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestAccountUsageService_GetUsage_KiroAPIKeyUnsupported(t *testing.T) {
+	account := &Account{
+		ID:       9101,
+		Platform: PlatformKiro,
+		Type:     AccountTypeAPIKey,
+	}
+	repo := &mockAccountRepoForGemini{accountsByID: map[int64]*Account{account.ID: account}}
+	svc := NewAccountUsageService(repo, nil, nil, nil, nil, NewUsageCache(), nil, nil)
+
+	usage, err := svc.GetUsage(context.Background(), account.ID)
+	require.Nil(t, usage)
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "does not support usage query")
+}
+
+func TestAccountUsageService_GetPassiveUsage_KiroAPIKeyUnsupported(t *testing.T) {
+	account := &Account{
+		ID:       9102,
+		Platform: PlatformKiro,
+		Type:     AccountTypeAPIKey,
+	}
+	repo := &mockAccountRepoForGemini{accountsByID: map[int64]*Account{account.ID: account}}
+	svc := NewAccountUsageService(repo, nil, nil, nil, nil, NewUsageCache(), nil, nil)
+
+	usage, err := svc.GetPassiveUsage(context.Background(), account.ID)
+	require.Nil(t, usage)
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "Kiro OAuth")
+}