89 lines
2.6 KiB
Elixir
89 lines
2.6 KiB
Elixir
defmodule AgentCoordinator.MetadataTest do
|
|
use ExUnit.Case, async: true
|
|
|
|
describe "Agent registration with metadata" do
|
|
test "register agent with metadata through TaskRegistry" do
|
|
# Use the existing TaskRegistry (started by application)
|
|
|
|
# Test metadata structure
|
|
metadata = %{
|
|
client_type: "github_copilot",
|
|
session_id: "test_session_123",
|
|
vs_code_version: "1.85.0",
|
|
auto_registered: true
|
|
}
|
|
|
|
agent_name = "MetadataTestAgent_#{:rand.uniform(1000)}"
|
|
|
|
# Register agent with metadata
|
|
result =
|
|
AgentCoordinator.TaskRegistry.register_agent(
|
|
agent_name,
|
|
["coding", "testing", "vscode_integration"],
|
|
metadata: metadata
|
|
)
|
|
|
|
assert :ok = result
|
|
|
|
# Retrieve agent and verify metadata
|
|
{:ok, agent} = AgentCoordinator.TaskRegistry.get_agent_by_name(agent_name)
|
|
|
|
assert agent.metadata[:client_type] == "github_copilot"
|
|
assert agent.metadata[:session_id] == "test_session_123"
|
|
assert agent.metadata[:vs_code_version] == "1.85.0"
|
|
assert agent.metadata[:auto_registered] == true
|
|
|
|
# Verify capabilities are preserved
|
|
assert "coding" in agent.capabilities
|
|
assert "testing" in agent.capabilities
|
|
assert "vscode_integration" in agent.capabilities
|
|
end
|
|
|
|
test "register agent without metadata (legacy compatibility)" do
|
|
# Use the existing TaskRegistry (started by application)
|
|
|
|
agent_name = "LegacyTestAgent_#{:rand.uniform(1000)}"
|
|
|
|
# Register agent without metadata (old way)
|
|
result =
|
|
AgentCoordinator.TaskRegistry.register_agent(
|
|
agent_name,
|
|
["coding", "testing"]
|
|
)
|
|
|
|
assert :ok = result
|
|
|
|
# Retrieve agent and verify empty metadata
|
|
{:ok, agent} = AgentCoordinator.TaskRegistry.get_agent_by_name(agent_name)
|
|
|
|
assert agent.metadata == %{}
|
|
assert "coding" in agent.capabilities
|
|
assert "testing" in agent.capabilities
|
|
end
|
|
|
|
test "Agent.new creates proper metadata structure" do
|
|
# Test metadata handling in Agent.new
|
|
metadata = %{
|
|
test_key: "test_value",
|
|
number: 42,
|
|
boolean: true
|
|
}
|
|
|
|
agent =
|
|
AgentCoordinator.Agent.new(
|
|
"TestAgent",
|
|
["capability1"],
|
|
metadata: metadata
|
|
)
|
|
|
|
assert agent.metadata[:test_key] == "test_value"
|
|
assert agent.metadata[:number] == 42
|
|
assert agent.metadata[:boolean] == true
|
|
|
|
# Test default empty metadata
|
|
agent_no_metadata = AgentCoordinator.Agent.new("NoMetadataAgent", ["capability1"])
|
|
assert agent_no_metadata.metadata == %{}
|
|
end
|
|
end
|
|
end
|