45 lines
1.3 KiB
Go
45 lines
1.3 KiB
Go
// SPDX-License-Identifier: EUPL-1.2
|
|
|
|
package agentic
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestProviderManager_NewProviderManager_Good_RegistersBuiltIns(t *testing.T) {
|
|
manager := NewProviderManager(func(context.Context, string, map[string]any) (string, error) {
|
|
return "Draft ready", nil
|
|
})
|
|
|
|
require.NotNil(t, manager)
|
|
assert.Equal(t, []string{"claude", "gemini", "openai"}, manager.Names())
|
|
|
|
provider, ok := manager.Provider("claude")
|
|
require.True(t, ok)
|
|
assert.Equal(t, "claude", provider.Name())
|
|
assert.Equal(t, "claude-3.7-sonnet", provider.DefaultModel())
|
|
|
|
text, err := provider.Generate(context.Background(), "Write a release note", nil)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "Draft ready", text)
|
|
}
|
|
|
|
func TestProviderManager_Provider_Bad_UnknownNameReturnsFalse(t *testing.T) {
|
|
manager := NewProviderManager(nil)
|
|
|
|
provider, ok := manager.Provider("unknown")
|
|
assert.False(t, ok)
|
|
assert.Nil(t, provider)
|
|
}
|
|
|
|
func TestProviderManager_ContentProvider_Ugly_NoGeneratorReturnsError(t *testing.T) {
|
|
provider := newContentProvider("claude", "claude-3.7-sonnet", true, nil)
|
|
|
|
_, err := provider.Generate(context.Background(), "Draft a release note", nil)
|
|
require.Error(t, err)
|
|
assert.Contains(t, err.Error(), "provider not configured")
|
|
}
|