test(quality): cover upsert validation and redis open branches

This commit is contained in:
phamnazage-jpg
2026-05-30 18:48:20 +08:00
parent d9216d5712
commit 9e32a24f8c
4 changed files with 142 additions and 0 deletions

View File

@@ -457,6 +457,44 @@ func TestRedisStickyStoreOpenRejectsAuthError(t *testing.T) {
} }
} }
func TestRedisStickyStoreOpenRejectsAuthUnexpectedResponse(t *testing.T) {
t.Parallel()
server := newScriptedRedisServer(t, func(conn net.Conn, reader *bufio.Reader) {
command, err := readRESPArray(reader)
if err != nil {
return
}
if len(command) > 0 && strings.ToUpper(command[0]) == "AUTH" {
_, _ = io.WriteString(conn, ":1\r\n")
}
})
defer server.Close()
store := &RedisStickyStore{cfg: RedisConfig{Addr: server.Addr(), Password: "secret"}}
_, _, err := store.open(context.Background())
if err == nil || !strings.Contains(err.Error(), "redis AUTH unexpected response") {
t.Fatalf("open() error = %v, want auth unexpected response", err)
}
}
func TestRedisStickyStoreOpenRejectsAuthReadFailure(t *testing.T) {
t.Parallel()
server := newScriptedRedisServer(t, func(conn net.Conn, reader *bufio.Reader) {
if _, err := readRESPArray(reader); err != nil {
return
}
})
defer server.Close()
store := &RedisStickyStore{cfg: RedisConfig{Addr: server.Addr(), Password: "secret"}}
_, _, err := store.open(context.Background())
if err == nil || !strings.Contains(err.Error(), "redis AUTH read") {
t.Fatalf("open() error = %v, want auth read failure", err)
}
}
func TestRedisStickyStoreOpenRejectsSelectUnexpectedResponse(t *testing.T) { func TestRedisStickyStoreOpenRejectsSelectUnexpectedResponse(t *testing.T) {
t.Parallel() t.Parallel()
@@ -485,6 +523,30 @@ func TestRedisStickyStoreOpenRejectsSelectUnexpectedResponse(t *testing.T) {
} }
} }
func TestRedisStickyStoreOpenRejectsSelectReadFailure(t *testing.T) {
t.Parallel()
server := newScriptedRedisServer(t, func(conn net.Conn, reader *bufio.Reader) {
command, err := readRESPArray(reader)
if err != nil {
return
}
if len(command) > 0 && strings.ToUpper(command[0]) == "AUTH" {
_, _ = io.WriteString(conn, "+OK\r\n")
}
if _, err := readRESPArray(reader); err != nil {
return
}
})
defer server.Close()
store := &RedisStickyStore{cfg: RedisConfig{Addr: server.Addr(), Password: "secret", DB: 2}}
_, _, err := store.open(context.Background())
if err == nil || !strings.Contains(err.Error(), "redis SELECT read") {
t.Fatalf("open() error = %v, want select read failure", err)
}
}
func TestRedisStickyStoreGetJSONRejectsUnexpectedResponse(t *testing.T) { func TestRedisStickyStoreGetJSONRejectsUnexpectedResponse(t *testing.T) {
t.Parallel() t.Parallel()

View File

@@ -363,6 +363,41 @@ func TestImportRunItemsRepoUpsertDefaultsOptionalJSONAndNullableFields(t *testin
} }
} }
func TestImportRunItemsRepoUpsertValidationErrors(t *testing.T) {
t.Parallel()
ctx := context.Background()
store := openTestDB(t)
run := ImportRun{RunID: "run-upsert-validation", HostID: "host-upsert-validation", Mode: "strict", AccessMode: "subscription", State: "running"}
if err := store.ImportRuns().Create(ctx, run); err != nil {
t.Fatalf("ImportRuns().Create() error = %v", err)
}
tests := []struct {
name string
item ImportRunItem
}{
{name: "missing item_id", item: ImportRunItem{RunID: "run-upsert-validation", BaseURL: "https://api.example.com/v1", ProviderID: "provider", APIKeyFingerprint: "fp", CurrentStage: "confirm", ConfirmationStatus: "pending", AccessStatus: "unknown", MatchedAccountState: "active", AccountResolution: "created"}},
{name: "missing run_id", item: ImportRunItem{ItemID: "item", BaseURL: "https://api.example.com/v1", ProviderID: "provider", APIKeyFingerprint: "fp", CurrentStage: "confirm", ConfirmationStatus: "pending", AccessStatus: "unknown", MatchedAccountState: "active", AccountResolution: "created"}},
{name: "missing base_url", item: ImportRunItem{ItemID: "item", RunID: "run-upsert-validation", ProviderID: "provider", APIKeyFingerprint: "fp", CurrentStage: "confirm", ConfirmationStatus: "pending", AccessStatus: "unknown", MatchedAccountState: "active", AccountResolution: "created"}},
{name: "missing provider_id", item: ImportRunItem{ItemID: "item", RunID: "run-upsert-validation", BaseURL: "https://api.example.com/v1", APIKeyFingerprint: "fp", CurrentStage: "confirm", ConfirmationStatus: "pending", AccessStatus: "unknown", MatchedAccountState: "active", AccountResolution: "created"}},
{name: "missing fingerprint", item: ImportRunItem{ItemID: "item", RunID: "run-upsert-validation", BaseURL: "https://api.example.com/v1", ProviderID: "provider", CurrentStage: "confirm", ConfirmationStatus: "pending", AccessStatus: "unknown", MatchedAccountState: "active", AccountResolution: "created"}},
{name: "missing current_stage", item: ImportRunItem{ItemID: "item", RunID: "run-upsert-validation", BaseURL: "https://api.example.com/v1", ProviderID: "provider", APIKeyFingerprint: "fp", ConfirmationStatus: "pending", AccessStatus: "unknown", MatchedAccountState: "active", AccountResolution: "created"}},
{name: "missing confirmation_status", item: ImportRunItem{ItemID: "item", RunID: "run-upsert-validation", BaseURL: "https://api.example.com/v1", ProviderID: "provider", APIKeyFingerprint: "fp", CurrentStage: "confirm", AccessStatus: "unknown", MatchedAccountState: "active", AccountResolution: "created"}},
{name: "missing access_status", item: ImportRunItem{ItemID: "item", RunID: "run-upsert-validation", BaseURL: "https://api.example.com/v1", ProviderID: "provider", APIKeyFingerprint: "fp", CurrentStage: "confirm", ConfirmationStatus: "pending", MatchedAccountState: "active", AccountResolution: "created"}},
{name: "missing matched_account_state", item: ImportRunItem{ItemID: "item", RunID: "run-upsert-validation", BaseURL: "https://api.example.com/v1", ProviderID: "provider", APIKeyFingerprint: "fp", CurrentStage: "confirm", ConfirmationStatus: "pending", AccessStatus: "unknown", AccountResolution: "created"}},
{name: "missing account_resolution", item: ImportRunItem{ItemID: "item", RunID: "run-upsert-validation", BaseURL: "https://api.example.com/v1", ProviderID: "provider", APIKeyFingerprint: "fp", CurrentStage: "confirm", ConfirmationStatus: "pending", AccessStatus: "unknown", MatchedAccountState: "active"}},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if err := store.ImportRunItems().Upsert(ctx, tt.item); err == nil {
t.Fatal("Upsert() error = nil, want validation error")
}
})
}
}
func TestImportRunEventsRepoCreateAndHelpers(t *testing.T) { func TestImportRunEventsRepoCreateAndHelpers(t *testing.T) {
t.Parallel() t.Parallel()

View File

@@ -180,6 +180,27 @@ func TestPacksRepoValidationErrors(t *testing.T) {
} }
} }
func TestPacksRepoUpsertValidationErrors(t *testing.T) {
store := openTestDB(t)
tests := []struct {
name string
pack Pack
}{
{"empty pack_id", Pack{Version: "v", Checksum: "c"}},
{"empty version", Pack{PackID: "p", Checksum: "c"}},
{"empty checksum", Pack{PackID: "p", Version: "v"}},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
_, err := store.Packs().Upsert(context.Background(), tt.pack)
if err == nil {
t.Fatal("Upsert() error = nil, want validation error")
}
})
}
}
func TestPacksRepoGetByIDNotFound(t *testing.T) { func TestPacksRepoGetByIDNotFound(t *testing.T) {
store := openTestDB(t) store := openTestDB(t)
_, err := store.Packs().GetByID(context.Background(), 999) _, err := store.Packs().GetByID(context.Background(), 999)

View File

@@ -253,6 +253,30 @@ func TestProvidersRepoValidationErrors(t *testing.T) {
} }
} }
func TestProvidersRepoUpsertValidationErrors(t *testing.T) {
store := openTestDB(t)
packID := createTestPack(t, store)
tests := []struct {
name string
provider Provider
}{
{"pack_id zero", Provider{ProviderID: "p", DisplayName: "d", BaseURL: "b", Platform: "openai"}},
{"empty provider_id", Provider{PackID: packID, DisplayName: "d", BaseURL: "b", Platform: "openai"}},
{"empty display_name", Provider{PackID: packID, ProviderID: "p", BaseURL: "b", Platform: "openai"}},
{"empty base_url", Provider{PackID: packID, ProviderID: "p", DisplayName: "d", Platform: "openai"}},
{"empty platform", Provider{PackID: packID, ProviderID: "p", DisplayName: "d", BaseURL: "b"}},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
_, err := store.Providers().Upsert(context.Background(), tt.provider)
if err == nil {
t.Fatal("Upsert() error = nil, want validation error")
}
})
}
}
func TestProvidersRepoGetByPackIDAndProviderIDNotFound(t *testing.T) { func TestProvidersRepoGetByPackIDAndProviderIDNotFound(t *testing.T) {
store := openTestDB(t) store := openTestDB(t)
_, err := store.Providers().GetByPackIDAndProviderID(context.Background(), 999, "p") _, err := store.Providers().GetByPackIDAndProviderID(context.Background(), 999, "p")