Sakurasan
ef3025dd80
Major rewrite of the Go backend to support:
- Three API format imports: openai, anthropic, compatible
- Three protocol conversions: Chat Completions, Responses, Messages
- Hub-and-spoke architecture with Chat as intermediate format
Deleted:
- opencat.go (old entry)
- store/, team/, pkg/team/, pkg/store/ (old data layer)
- internal/model/, internal/consts/ (old types)
- internal/service/team/, internal/controller/team/ (old handlers)
- llm/ (removed LLM client library, pure proxy mode)
- dist/, assets/ (old build artifacts)
Added:
- internal/store/ — 9 GORM models + multi-DB support
- internal/pkg/ — crypto (AES-GCM), apikey, jwt, ratelimit, resp, tokenizer
- internal/channel/ — channel selection, weighted LB, health checks
- internal/proxy/convert/ — 6 protocol conversion functions + SSE streaming
- internal/proxy/ — gateway with request dispatch and upstream selection
- internal/usage/ — async usage recorder with batch writes
- internal/api/ — management API (auth, users, keys, channels, models)
- Makefile for build/test/deploy
Fixed API to match frontend expectations:
- Login response wraps token in { data: { token } }
- GET /api/profile route added
- Profile response wraps user in { code, data }
- Role returned as number (10=admin, 1=user)
56 lines
1.1 KiB
Go
56 lines
1.1 KiB
Go
package crypto
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestEncryptDecrypt(t *testing.T) {
|
|
plaintext := "sk-test-api-key-12345"
|
|
|
|
encrypted, err := Encrypt(plaintext)
|
|
if err != nil {
|
|
t.Fatalf("Encrypt() error = %v", err)
|
|
}
|
|
|
|
if encrypted == plaintext {
|
|
t.Error("Encrypt() returned plaintext")
|
|
}
|
|
|
|
decrypted, err := Decrypt(encrypted)
|
|
if err != nil {
|
|
t.Fatalf("Decrypt() error = %v", err)
|
|
}
|
|
|
|
if decrypted != plaintext {
|
|
t.Errorf("Decrypt() = %q, want %q", decrypted, plaintext)
|
|
}
|
|
}
|
|
|
|
func TestSha256Hex(t *testing.T) {
|
|
input := "test"
|
|
result := Sha256Hex(input)
|
|
|
|
if len(result) != 64 {
|
|
t.Errorf("Sha256Hex() returned %d chars, want 64", len(result))
|
|
}
|
|
|
|
// Same input should produce same hash
|
|
result2 := Sha256Hex(input)
|
|
if result != result2 {
|
|
t.Error("Sha256Hex() not deterministic")
|
|
}
|
|
|
|
// Different input should produce different hash
|
|
result3 := Sha256Hex("different")
|
|
if result == result3 {
|
|
t.Error("Sha256Hex() same hash for different inputs")
|
|
}
|
|
}
|
|
|
|
func TestEncryptorInvalidKey(t *testing.T) {
|
|
_, err := NewEncryptor([]byte("short"))
|
|
if err == nil {
|
|
t.Error("NewEncryptor() should error with invalid key length")
|
|
}
|
|
}
|