CLIProxyAPIPlus/sdk/cliproxy/service_gitlab_models_test.go

package cliproxy

import (
	"testing"

	"github.com/router-for-me/CLIProxyAPI/v6/internal/registry"
	coreauth "github.com/router-for-me/CLIProxyAPI/v6/sdk/cliproxy/auth"
	"github.com/router-for-me/CLIProxyAPI/v6/sdk/config"
)

func TestRegisterModelsForAuth_GitLabUsesDiscoveredModels(t *testing.T) {
	service := &Service{cfg: &config.Config{}}
	auth := &coreauth.Auth{
		ID:       "gitlab-auth.json",
		Provider: "gitlab",
		Status:   coreauth.StatusActive,
		Metadata: map[string]any{
			"model_details": map[string]any{
				"model_provider": "anthropic",
				"model_name":     "claude-sonnet-4-5",
			},
		},
	}

	reg := registry.GetGlobalRegistry()
	reg.UnregisterClient(auth.ID)
	t.Cleanup(func() { reg.UnregisterClient(auth.ID) })

	service.registerModelsForAuth(auth)
	models := reg.GetModelsForClient(auth.ID)
	if len(models) < 2 {
		t.Fatalf("expected stable alias and discovered model, got %d entries", len(models))
	}

	seenAlias := false
	seenDiscovered := false
	for _, model := range models {
		switch model.ID {
		case "gitlab-duo":
			seenAlias = true
		case "claude-sonnet-4-5":
			seenDiscovered = true
		}
	}
	if !seenAlias || !seenDiscovered {
		t.Fatalf("expected gitlab-duo and discovered model, got %+v", models)
	}
}

func TestRegisterModelsForAuth_GitLabIncludesAgenticCatalog(t *testing.T) {
	service := &Service{cfg: &config.Config{}}
	auth := &coreauth.Auth{
		ID:       "gitlab-agentic-auth.json",
		Provider: "gitlab",
		Status:   coreauth.StatusActive,
	}

	reg := registry.GetGlobalRegistry()
	reg.UnregisterClient(auth.ID)
	t.Cleanup(func() { reg.UnregisterClient(auth.ID) })

	service.registerModelsForAuth(auth)
	models := reg.GetModelsForClient(auth.ID)
	if len(models) < 5 {
		t.Fatalf("expected stable alias plus built-in agentic catalog, got %d entries", len(models))
	}

	required := map[string]bool{
		"gitlab-duo":           false,
		"duo-chat-opus-4-6":    false,
		"duo-chat-haiku-4-5":   false,
		"duo-chat-sonnet-4-5":  false,
		"duo-chat-opus-4-5":    false,
		"duo-chat-gpt-5-codex": false,
	}
	for _, model := range models {
		if _, ok := required[model.ID]; ok {
			required[model.ID] = true
		}
	}
	for id, seen := range required {
		if !seen {
			t.Fatalf("expected built-in GitLab Duo model %q, got %+v", id, models)
		}
	}
}