mirror of
https://github.com/hanzoai/agents.git
synced 2026-10-30 18:48:36 +00:00
zeekayandHanzo Dev
93d83605da
The note URL replaced the first /v1 in its base, which is wrong for any base carrying its own version. A client test named an endpoint the /v1 sweep missed. The rest is gofmt. Co-authored-by: Hanzo Dev <[email protected]>
123 lines
3.6 KiB
Go
123 lines
3.6 KiB
Go
package agent
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDiscoverJSON(t *testing.T) {
|
|
body := `{
|
|
"discovered_at": "2025-01-01T00:00:00Z",
|
|
"total_agents": 1,
|
|
"total_reasoners": 1,
|
|
"total_skills": 0,
|
|
"pagination": {"limit": 50, "offset": 0, "has_more": false},
|
|
"capabilities": [{
|
|
"agent_id": "agent-1",
|
|
"base_url": "http://agent",
|
|
"version": "1.0.0",
|
|
"health_status": "active",
|
|
"deployment_type": "long_running",
|
|
"last_heartbeat": "2025-01-01T00:00:00Z",
|
|
"reasoners": [{"id": "deep_research", "invocation_target": "agent-1:deep_research"}],
|
|
"skills": []
|
|
}]
|
|
}`
|
|
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
query := r.URL.Query()
|
|
assert.Equal(t, "agent-1", query.Get("agent"))
|
|
assert.Equal(t, "true", query.Get("include_input_schema"))
|
|
assert.Equal(t, "json", query.Get("format"))
|
|
w.Header().Set("Content-Type", "application/json")
|
|
fmt.Fprint(w, body)
|
|
}))
|
|
defer server.Close()
|
|
|
|
a, err := New(Config{
|
|
NodeID: "node-1",
|
|
Version: "1.0.0",
|
|
HanzoAgentsURL: server.URL,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
result, err := a.Discover(context.Background(), WithAgent("agent-1"), WithDiscoveryInputSchema(true), WithLimit(50))
|
|
require.NoError(t, err)
|
|
require.NotNil(t, result)
|
|
require.NotNil(t, result.JSON)
|
|
|
|
assert.Equal(t, "json", result.Format)
|
|
assert.Equal(t, 1, result.JSON.TotalAgents)
|
|
assert.Equal(t, "agent-1", result.JSON.Capabilities[0].AgentID)
|
|
assert.Equal(t, "agent-1:deep_research", result.JSON.Capabilities[0].Reasoners[0].InvocationTarget)
|
|
assert.Equal(t, body, result.Raw)
|
|
}
|
|
|
|
func TestDiscoverCompactAndXML(t *testing.T) {
|
|
compact := `{"discovered_at":"2025-01-01T00:00:00Z","reasoners":[{"id":"r1","agent_id":"a1","target":"a1:r1"}],"skills":[]}`
|
|
xml := "<discovery><summary total_agents=\"0\" /></discovery>"
|
|
|
|
compactServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, "compact", r.URL.Query().Get("format"))
|
|
fmt.Fprint(w, compact)
|
|
}))
|
|
defer compactServer.Close()
|
|
|
|
xmlServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
assert.Equal(t, "xml", r.URL.Query().Get("format"))
|
|
w.Header().Set("Content-Type", "application/xml")
|
|
fmt.Fprint(w, xml)
|
|
}))
|
|
defer xmlServer.Close()
|
|
|
|
aCompact, err := New(Config{
|
|
NodeID: "node-1",
|
|
Version: "1.0.0",
|
|
HanzoAgentsURL: compactServer.URL,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
compactResult, err := aCompact.Discover(context.Background(), WithFormat("compact"))
|
|
require.NoError(t, err)
|
|
require.NotNil(t, compactResult.Compact)
|
|
assert.Equal(t, "compact", compactResult.Format)
|
|
assert.Equal(t, "a1", compactResult.Compact.Reasoners[0].AgentID)
|
|
|
|
aXML, err := New(Config{
|
|
NodeID: "node-1",
|
|
Version: "1.0.0",
|
|
HanzoAgentsURL: xmlServer.URL,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
xmlResult, err := aXML.Discover(context.Background(), WithFormat("xml"), WithTags([]string{"ml"}))
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "xml", xmlResult.Format)
|
|
assert.Equal(t, xml, xmlResult.XML)
|
|
assert.Nil(t, xmlResult.JSON)
|
|
assert.Nil(t, xmlResult.Compact)
|
|
}
|
|
|
|
func TestDiscoverRejectsInvalidFormat(t *testing.T) {
|
|
a, err := New(Config{
|
|
NodeID: "node-1",
|
|
Version: "1.0.0",
|
|
HanzoAgentsURL: "http://localhost:8080",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
_, err = a.Discover(context.Background(), WithFormat("yaml"))
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestDedupeHelper(t *testing.T) {
|
|
values := []string{"a", "b", "a", "", "c"}
|
|
assert.Equal(t, []string{"a", "b", "c"}, dedupe(values))
|
|
}
|