mirror of
https://github.com/hanzoai/agent.git
synced 2026-09-11 21:08:51 +00:00
zeekay
63bde13037
The bulk rename from agentfield to hanzo-agents introduced hyphens into Go variable names (e.g. hanzo-agentsHome), which are not valid Go identifiers. Fixed 361 occurrences across 36 files to use camelCase (e.g. hanzoAgentsHome). String literals like ".hanzo-agents" are preserved as-is.
106 lines
3.0 KiB
Go
106 lines
3.0 KiB
Go
package services
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hanzoai/agents/control-plane/internal/storage"
|
|
"github.com/hanzoai/agents/control-plane/pkg/types"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func setupTestStorage(t *testing.T) (storage.StorageProvider, context.Context) {
|
|
t.Helper()
|
|
|
|
ctx := context.Background()
|
|
tempDir := t.TempDir()
|
|
cfg := storage.StorageConfig{
|
|
Mode: "local",
|
|
Local: storage.LocalStorageConfig{
|
|
DatabasePath: filepath.Join(tempDir, "hanzo-agents.db"),
|
|
KVStorePath: filepath.Join(tempDir, "hanzo-agents.bolt"),
|
|
},
|
|
}
|
|
|
|
provider := storage.NewLocalStorage(storage.LocalStorageConfig{})
|
|
if err := provider.Initialize(ctx, cfg); err != nil {
|
|
if strings.Contains(strings.ToLower(err.Error()), "fts5") {
|
|
t.Skip("sqlite3 compiled without FTS5; skipping DID registry test")
|
|
}
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
t.Cleanup(func() {
|
|
_ = provider.Close(ctx)
|
|
})
|
|
|
|
return provider, ctx
|
|
}
|
|
|
|
func TestDIDRegistryInitializeAndLookup(t *testing.T) {
|
|
provider, ctx := setupTestStorage(t)
|
|
|
|
hanzoAgentsID := "hanzo-agents-1"
|
|
now := time.Now().UTC().Truncate(time.Second)
|
|
|
|
require.NoError(t, provider.StoreHanzoAgentsServerDID(ctx, hanzoAgentsID, "did:hanzo-agents:root", []byte("seed"), now, now))
|
|
|
|
components := []storage.ComponentDIDRequest{
|
|
{
|
|
ComponentDID: "did:reasoner:1",
|
|
ComponentType: "reasoner",
|
|
ComponentName: "reasoner.fn",
|
|
PublicKeyJWK: "{}",
|
|
DerivationIndex: 1,
|
|
},
|
|
{
|
|
ComponentDID: "did:skill:1",
|
|
ComponentType: "skill",
|
|
ComponentName: "skill.fn",
|
|
PublicKeyJWK: "{}",
|
|
DerivationIndex: 2,
|
|
},
|
|
}
|
|
|
|
require.NoError(t, provider.StoreAgentDIDWithComponents(ctx, "agent-1", "did:agent:1", hanzoAgentsID, "{}", 0, components))
|
|
|
|
registry := NewDIDRegistryWithStorage(provider)
|
|
require.NoError(t, registry.Initialize())
|
|
|
|
loaded, err := registry.GetRegistry(hanzoAgentsID)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, loaded)
|
|
require.Contains(t, loaded.AgentNodes, "agent-1")
|
|
|
|
// Validate reasoner lookup
|
|
reasonerID, err := registry.FindDIDByComponent(hanzoAgentsID, "reasoner", "reasoner.fn")
|
|
require.NoError(t, err)
|
|
require.Equal(t, "did:reasoner:1", reasonerID.DID)
|
|
|
|
// Validate skill lookup
|
|
skillID, err := registry.FindDIDByComponent(hanzoAgentsID, "skill", "skill.fn")
|
|
require.NoError(t, err)
|
|
require.Equal(t, "did:skill:1", skillID.DID)
|
|
|
|
// Update status and ensure it is persisted in-memory
|
|
require.NoError(t, registry.UpdateAgentStatus(hanzoAgentsID, "agent-1", types.AgentDIDStatusActive))
|
|
|
|
loadedAfterUpdate, err := registry.GetRegistry(hanzoAgentsID)
|
|
require.NoError(t, err)
|
|
require.Equal(t, types.AgentDIDStatusActive, loadedAfterUpdate.AgentNodes["agent-1"].Status)
|
|
|
|
packageResult, err := registry.GetAgentDIDs(hanzoAgentsID, "agent-1")
|
|
require.NoError(t, err)
|
|
require.Equal(t, "did:agent:1", packageResult.AgentDID.DID)
|
|
require.Contains(t, packageResult.ReasonerDIDs, "reasoner.fn")
|
|
require.Contains(t, packageResult.SkillDIDs, "skill.fn")
|
|
|
|
registries, err := registry.ListRegistries()
|
|
require.NoError(t, err)
|
|
require.Len(t, registries, 1)
|
|
}
|