diff options
Diffstat (limited to 'internal/controlplane/preferences_test.go')
| -rw-r--r-- | internal/controlplane/preferences_test.go | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/internal/controlplane/preferences_test.go b/internal/controlplane/preferences_test.go new file mode 100644 index 0000000..88a3d50 --- /dev/null +++ b/internal/controlplane/preferences_test.go @@ -0,0 +1,31 @@ +package controlplane + +import ( + "context" + "testing" +) + +func TestGetTenantPreferencesWithoutTenantUsesConfiguredDefault(t *testing.T) { + result, err := (&Store{}).GetTenantPreferences(context.Background(), "", 12_500_000) + if err != nil { + t.Fatal(err) + } + if !result.LowBalanceEnabled || result.LowBalanceThresholdMicros != 12_500_000 { + t.Fatalf("unexpected defaults: %+v", result) + } +} + +func TestPreferenceValidationRejectsUnsafeValuesBeforeDatabaseAccess(t *testing.T) { + store := &Store{} + if _, err := store.SetDeveloperPreferences(context.Background(), SetDeveloperPreferencesInput{ + TenantID: "tenant", DefaultModel: "same", FallbackModel: "same", + }); err == nil { + t.Fatal("expected identical default and fallback models to fail") + } + threshold := maxLowBalanceThresholdMicros + 1 + if _, err := store.SetBillingPreferences(context.Background(), SetBillingPreferencesInput{ + TenantID: "tenant", LowBalanceThresholdMicros: &threshold, + }, 5_000_000); err == nil { + t.Fatal("expected excessive low balance threshold to fail") + } +} |
