93 lines
4.2 KiB
Go
93 lines
4.2 KiB
Go
package backend_test
|
|
|
|
import (
|
|
"errors"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"gitea.maximumdirect.net/eric/promptkit/internal/backend"
|
|
"gitea.maximumdirect.net/eric/promptkit/internal/domain"
|
|
)
|
|
|
|
func TestEffectiveRegistryOwnsDefinitionsAndCapacityPolicies(t *testing.T) {
|
|
params := map[string]any{"nested": map[string]any{"enabled": true}}
|
|
defaultProfile := "catalog-profile"
|
|
reachabilityEndpoint := "https://probe.example/ready"
|
|
authenticationKind := "openrouter_key"
|
|
authenticationEndpoint := "https://probe.example/key"
|
|
smokeTests := &domain.BackendSmokeTestConfig{
|
|
DefaultProfile: &defaultProfile,
|
|
Reachability: &domain.ReachabilityProbeConfig{Endpoint: &reachabilityEndpoint},
|
|
Authentication: &domain.AuthenticationProbeConfig{
|
|
Kind: &authenticationKind, Endpoint: &authenticationEndpoint,
|
|
},
|
|
}
|
|
registry, err := backend.NewEffectiveRegistry(map[string]domain.Backend{
|
|
" local.v1 ": {
|
|
Endpoint: "https://backend.example/v1", ExtraParams: params,
|
|
ConcurrencyLimit: 2, QueueCapacity: 3, QueueCapacitySet: true,
|
|
SmokeTests: smokeTests,
|
|
},
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
params["nested"].(map[string]any)["enabled"] = false
|
|
defaultProfile = "mutated-profile"
|
|
reachabilityEndpoint = "https://mutated.example/ready"
|
|
authenticationKind = "none"
|
|
authenticationEndpoint = "https://mutated.example/key"
|
|
selected, err := registry.GetBackend(" local.v1 ")
|
|
if err != nil || selected.ID != " local.v1 " || selected.ExtraParams["nested"].(map[string]any)["enabled"] != true {
|
|
t.Fatalf("effective backend was not captured under its literal ID: %+v, %v", selected, err)
|
|
}
|
|
if selected.SmokeTests == nil || selected.SmokeTests.DefaultProfile == nil ||
|
|
*selected.SmokeTests.DefaultProfile != "catalog-profile" || selected.SmokeTests.Reachability == nil ||
|
|
selected.SmokeTests.Reachability.Endpoint == nil || *selected.SmokeTests.Reachability.Endpoint != "https://probe.example/ready" ||
|
|
selected.SmokeTests.Authentication == nil || selected.SmokeTests.Authentication.Kind == nil ||
|
|
*selected.SmokeTests.Authentication.Kind != "openrouter_key" || selected.SmokeTests.Authentication.Endpoint == nil ||
|
|
*selected.SmokeTests.Authentication.Endpoint != "https://probe.example/key" {
|
|
t.Fatalf("registry did not capture nested diagnostic metadata: %+v", selected.SmokeTests)
|
|
}
|
|
selected.ExtraParams["nested"].(map[string]any)["enabled"] = false
|
|
*selected.SmokeTests.DefaultProfile = "mutated-return"
|
|
*selected.SmokeTests.Reachability.Endpoint = "https://mutated-return.example/ready"
|
|
*selected.SmokeTests.Authentication.Kind = "none"
|
|
*selected.SmokeTests.Authentication.Endpoint = "https://mutated-return.example/key"
|
|
again, err := registry.GetBackend(" local.v1 ")
|
|
if err != nil || again.ExtraParams["nested"].(map[string]any)["enabled"] != true {
|
|
t.Fatalf("lookup exposed registry values: %+v, %v", again, err)
|
|
}
|
|
if again.SmokeTests == nil || again.SmokeTests.DefaultProfile == nil ||
|
|
*again.SmokeTests.DefaultProfile != "catalog-profile" || again.SmokeTests.Reachability == nil ||
|
|
again.SmokeTests.Reachability.Endpoint == nil || *again.SmokeTests.Reachability.Endpoint != "https://probe.example/ready" ||
|
|
again.SmokeTests.Authentication == nil || again.SmokeTests.Authentication.Kind == nil ||
|
|
*again.SmokeTests.Authentication.Kind != "openrouter_key" || again.SmokeTests.Authentication.Endpoint == nil ||
|
|
*again.SmokeTests.Authentication.Endpoint != "https://probe.example/key" {
|
|
t.Fatalf("lookup exposed nested diagnostic metadata: %+v", again.SmokeTests)
|
|
}
|
|
policies := registry.CapacityPolicies()
|
|
if !reflect.DeepEqual(policies, map[string]domain.BackendCapacityPolicy{
|
|
" local.v1 ": {ConcurrencyLimit: 2, QueueCapacity: 3},
|
|
}) {
|
|
t.Fatalf("capacity policy = %+v", policies)
|
|
}
|
|
delete(policies, " local.v1 ")
|
|
if len(registry.CapacityPolicies()) != 1 {
|
|
t.Fatal("capacity policy map exposed registry state")
|
|
}
|
|
}
|
|
|
|
func TestEffectiveRegistryMissingBackend(t *testing.T) {
|
|
registry, err := backend.NewEffectiveRegistry(nil)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := registry.GetBackend("missing"); !errors.Is(err, backend.ErrBackendNotFound) {
|
|
t.Fatalf("missing backend error = %v", err)
|
|
}
|
|
if _, err := registry.GetBackend("local.v1"); !errors.Is(err, backend.ErrBackendNotFound) {
|
|
t.Fatalf("trimmed lookup unexpectedly matched: %v", err)
|
|
}
|
|
}
|