package providers_test

import (
	"testing"

	"github.com/salvacybersec/keyhunter/pkg/providers"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"gopkg.in/yaml.v3"
)

func TestRegistryLoad(t *testing.T) {
	reg, err := providers.NewRegistry()
	require.NoError(t, err)
	assert.GreaterOrEqual(t, len(reg.List()), 3, "expected at least 3 providers")
}

func TestRegistryGet(t *testing.T) {
	reg, err := providers.NewRegistry()
	require.NoError(t, err)

	p, ok := reg.Get("openai")
	assert.True(t, ok)
	assert.Equal(t, "openai", p.Name)
	assert.Equal(t, 1, p.Tier)

	_, notOk := reg.Get("nonexistent-provider")
	assert.False(t, notOk)
}

func TestRegistryStats(t *testing.T) {
	reg, err := providers.NewRegistry()
	require.NoError(t, err)

	stats := reg.Stats()
	assert.GreaterOrEqual(t, stats.Total, 3)
	assert.GreaterOrEqual(t, stats.ByTier[1], 2)
}

func TestAhoCorasickBuild(t *testing.T) {
	reg, err := providers.NewRegistry()
	require.NoError(t, err)

	ac := reg.AC()
	matches := ac.FindAll("export OPENAI_API_KEY=sk-proj-abc")
	assert.NotEmpty(t, matches)

	noMatches := ac.FindAll("hello world nothing here")
	assert.Empty(t, noMatches)
}

func TestProviderSchemaValidation(t *testing.T) {
	invalid := []byte("format_version: 0\nname: invalid\nlast_verified: \"\"\n")
	var p providers.Provider
	err := yaml.Unmarshal(invalid, &p)
	assert.Error(t, err)
	assert.Contains(t, err.Error(), "format_version")
}