Function testRouterConfig

internal/router/rule_router_test.go:12–43

()

)

func testRouterConfig() *config.Config {
	return &config.Config{
		Backends: []config.BackendConfig{
			{
				Name:         "small-model",
				Cost:         config.CostConfig{Unit: 1},
				Capabilities: []string{"chat", "summarization"},
			},
			{
				Name:         "large-model",
				Cost:         config.CostConfig{Unit: 5},
				Capabilities: []string{"chat", "reasoning", "summarization"},
			},
		},
		Tenants: []config.TenantConfig{
			{ID: "team-a", Class: "premium", BudgetPerRequest: 10},
		},
		Routing: config.RoutingConfig{
			DefaultBackend: "small-model",
			Rules: []config.RouteRule{
				{
					Name: "premium-simple-expensive-route",
					When: config.RouteWhen{
						TenantClass: "premium",
						TaskType:    "simple",
					},
					UseBackend: "large-model",
				},
			},
		},
	}
}

func TestRuleRouter_UsesRuleMatch(t *testing.T) {
	r := NewRuleRouter(testRouterConfig(), nil)