CLIProxyAPI/internal/watcher/diff/oauth_settings_test.go

package diff

import (
	"strings"
	"testing"

	"github.com/router-for-me/CLIProxyAPI/v8/internal/config"
)

func TestDiffOAuthSettingsChanges(t *testing.T) {
	oldMap := map[string][]config.OAuthModelSetting{
		"codex": {
			{Name: "gpt-6-sol", MaxContextLength: 272000},
		},
		"vertex": {
			{Name: "gemini-2.5-pro", MaxContextLength: 1048576},
		},
	}

	newMap := map[string][]config.OAuthModelSetting{
		"codex": {
			{Name: "gpt-6-sol", MaxContextLength: 524288},
		},
		"claude": {
			{Name: "claude-sonnet-4-5-20250929", MaxContextLength: 200000},
		},
	}

	changes, affected := DiffOAuthSettingsChanges(oldMap, newMap)

	// vertex removed, claude added, codex updated
	expectContains := func(slice []string, expected string) {
		t.Helper()
		found := false
		for _, s := range slice {
			if strings.Contains(s, expected) {
				found = true
				break
			}
		}
		if !found {
			t.Errorf("expected slice to contain %q, but got %#v", expected, slice)
		}
	}

	expectContains(changes, "oauth-settings[vertex]: removed")
	expectContains(changes, "oauth-settings[claude]: added (1 entries)")
	expectContains(changes, "oauth-settings[codex]: updated (1 -> 1 entries)")

	expectContains(affected, "vertex")
	expectContains(affected, "claude")
	expectContains(affected, "codex")
}

func TestDiffOAuthSettingsChanges_Reordering(t *testing.T) {
	oldMap := map[string][]config.OAuthModelSetting{
		"codex": {
			{Name: "gpt-6-sol", MaxContextLength: 524288},
			{Name: "deepseek-v4-flash", MaxContextLength: 1048576},
		},
	}

	newMap := map[string][]config.OAuthModelSetting{
		"codex": {
			{Name: "deepseek-v4-flash", MaxContextLength: 1048576},
			{Name: "gpt-6-sol", MaxContextLength: 524288},
		},
	}

	changes, affected := DiffOAuthSettingsChanges(oldMap, newMap)
	if len(changes) == 0 {
		t.Fatalf("expected changes when rules are reordered, got none")
	}
	if len(affected) == 0 || affected[0] != "codex" {
		t.Fatalf("expected codex to be affected by reordering: %#v", affected)
	}
}