internal/adminapi/usage_test.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

package adminapi

import (
	"testing"

	"aigw/internal/controlplane"
)

func TestTenantUsageRedactionRemovesRouteInternals(t *testing.T) {
	page := controlplane.UsagePage{Data: []controlplane.UsageRecord{{
		ProviderID: "provider-uuid", ProviderName: "Internal Provider", UpstreamModel: "vendor/model-v2",
	}}}
	redactTenantUsagePage(&page)
	if page.Data[0].ProviderID != "" || page.Data[0].ProviderName != "" || page.Data[0].UpstreamModel != "" {
		t.Fatalf("tenant usage leaked route internals: %+v", page.Data[0])
	}

	analytics := controlplane.UsageAnalytics{
		Models:    []controlplane.UsageModelAnalytics{{PublicModel: "public/model", ProviderCount: 3}},
		Providers: []controlplane.UsageProviderAnalytics{{ProviderID: "provider-uuid", ProviderName: "Internal Provider"}},
	}
	redactTenantUsageAnalytics(&analytics)
	if len(analytics.Providers) != 0 || analytics.Models[0].ProviderCount != 0 {
		t.Fatalf("tenant analytics leaked provider topology: %+v", analytics)
	}
}