59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
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")
|
|
}
|