forked from molecule-ai/molecule-core
24fec62d7f
Forked clean from public hackathon repo (Starfire-AgentTeam, BSL 1.1) with full rebrand to Molecule AI under github.com/Molecule-AI/molecule-monorepo. Brand: Starfire → Molecule AI. Slug: starfire / agent-molecule → molecule. Env vars: STARFIRE_* → MOLECULE_*. Go module: github.com/agent-molecule/platform → github.com/Molecule-AI/molecule-monorepo/platform. Python packages: starfire_plugin → molecule_plugin, starfire_agent → molecule_agent. DB: agentmolecule → molecule. History truncated; see public repo for prior commits and contributor attribution. Verified green: go test -race ./... (platform), pytest (workspace-template 1129 + sdk 132), vitest (canvas 352), build (mcp). Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
213 lines
7.7 KiB
Bash
213 lines
7.7 KiB
Bash
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
BASE="http://localhost:8080"
|
|
PASS=0
|
|
FAIL=0
|
|
TIMEOUT="${A2A_TIMEOUT:-120}" # seconds per A2A call (override via A2A_TIMEOUT env var)
|
|
|
|
check() {
|
|
local desc="$1"
|
|
local expected="$2"
|
|
local actual="$3"
|
|
if echo "$actual" | grep -qF -- "$expected"; then
|
|
echo "PASS: $desc"
|
|
PASS=$((PASS + 1))
|
|
else
|
|
echo "FAIL: $desc"
|
|
echo " expected to contain: $expected"
|
|
echo " got: $(echo "$actual" | head -5)"
|
|
FAIL=$((FAIL + 1))
|
|
fi
|
|
}
|
|
|
|
a2a_send() {
|
|
local ws_id="$1"
|
|
local text="$2"
|
|
curl -s --max-time "$TIMEOUT" -X POST "$BASE/workspaces/$ws_id/a2a" \
|
|
-H "Content-Type: application/json" \
|
|
-d "{
|
|
\"method\": \"message/send\",
|
|
\"params\": {
|
|
\"message\": {
|
|
\"role\": \"user\",
|
|
\"parts\": [{\"type\": \"text\", \"text\": \"$text\"}]
|
|
}
|
|
}
|
|
}"
|
|
}
|
|
|
|
echo "=== A2A End-to-End Tests (Free Model: google/gemini-2.5-flash) ==="
|
|
echo ""
|
|
|
|
# --- Setup: find or create workspaces ---
|
|
ECHO_ID=$(curl -s "$BASE/workspaces" | python3 -c "
|
|
import sys, json
|
|
ws = json.load(sys.stdin)
|
|
for w in ws:
|
|
if w['name'] == 'Echo Agent' and w['status'] == 'online':
|
|
print(w['id']); break
|
|
else:
|
|
print('')
|
|
")
|
|
SEO_ID=$(curl -s "$BASE/workspaces" | python3 -c "
|
|
import sys, json
|
|
ws = json.load(sys.stdin)
|
|
for w in ws:
|
|
if w['name'] == 'SEO Agent' and w['status'] == 'online':
|
|
print(w['id']); break
|
|
else:
|
|
print('')
|
|
")
|
|
|
|
if [ -z "$ECHO_ID" ] || [ -z "$SEO_ID" ]; then
|
|
echo "ERROR: Need both Echo Agent and SEO Agent online. Found echo=$ECHO_ID seo=$SEO_ID"
|
|
exit 1
|
|
fi
|
|
|
|
echo "Echo Agent: $ECHO_ID"
|
|
echo "SEO Agent: $SEO_ID"
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Test 1: Basic message/send — Echo Agent
|
|
# ========================================
|
|
echo "--- Test 1: Basic message/send ---"
|
|
R=$(a2a_send "$ECHO_ID" "Say hello back")
|
|
check "JSON-RPC response has result" '"result"' "$R"
|
|
check "Response has agent role" '"role":"agent"' "$R"
|
|
check "Response has text part" '"kind":"text"' "$R"
|
|
TEXT=$(echo "$R" | python3 -c "import sys,json; r=json.load(sys.stdin); print(r['result']['parts'][0]['text'][:200])" 2>/dev/null || echo "PARSE_ERROR")
|
|
echo " Agent said: $TEXT"
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Test 2: Basic message/send — SEO Agent
|
|
# ========================================
|
|
echo "--- Test 2: SEO Agent responds ---"
|
|
R=$(a2a_send "$SEO_ID" "What SEO skills do you have?")
|
|
check "SEO agent responds" '"result"' "$R"
|
|
check "SEO response has text" '"kind":"text"' "$R"
|
|
TEXT=$(echo "$R" | python3 -c "import sys,json; r=json.load(sys.stdin); print(r['result']['parts'][0]['text'][:200])" 2>/dev/null || echo "PARSE_ERROR")
|
|
echo " SEO Agent said: $TEXT"
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Test 3: JSON-RPC envelope wrapping
|
|
# ========================================
|
|
echo "--- Test 3: Auto JSON-RPC envelope wrapping ---"
|
|
# Send bare method+params (no jsonrpc/id fields) — proxy should wrap it
|
|
R=$(curl -s --max-time "$TIMEOUT" -X POST "$BASE/workspaces/$ECHO_ID/a2a" \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"method":"message/send","params":{"message":{"role":"user","parts":[{"type":"text","text":"Bare request test"}]}}}')
|
|
check "Bare request gets wrapped and works" '"result"' "$R"
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Test 4: Full JSON-RPC 2.0 envelope
|
|
# ========================================
|
|
echo "--- Test 4: Full JSON-RPC 2.0 envelope ---"
|
|
R=$(curl -s --max-time "$TIMEOUT" -X POST "$BASE/workspaces/$ECHO_ID/a2a" \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"jsonrpc":"2.0","id":"custom-id-123","method":"message/send","params":{"message":{"role":"user","parts":[{"type":"text","text":"Full envelope test"}]}}}')
|
|
check "Full envelope returns result" '"result"' "$R"
|
|
check "Preserves custom request ID" '"id":"custom-id-123"' "$R"
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Test 5: Invalid method returns error
|
|
# ========================================
|
|
echo "--- Test 5: Invalid method ---"
|
|
R=$(curl -s --max-time 10 -X POST "$BASE/workspaces/$ECHO_ID/a2a" \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"method":"nonexistent/method","params":{}}')
|
|
check "Invalid method returns JSON-RPC error" '"error"' "$R"
|
|
check "Error code is method not found" '-32601' "$R"
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Test 6: Offline workspace returns error
|
|
# ========================================
|
|
echo "--- Test 6: Offline workspace ---"
|
|
# Create a workspace but don't provision it
|
|
R=$(curl -s -X POST "$BASE/workspaces" -H "Content-Type: application/json" -d '{"name":"Offline Test","tier":1}')
|
|
OFFLINE_ID=$(echo "$R" | python3 -c "import sys,json; print(json.load(sys.stdin)['id'])")
|
|
R=$(curl -s --max-time 10 -X POST "$BASE/workspaces/$OFFLINE_ID/a2a" \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"method":"message/send","params":{"message":{"role":"user","parts":[{"type":"text","text":"test"}]}}}')
|
|
check "Offline workspace returns error" '"error"' "$R"
|
|
# Clean up
|
|
curl -s -X DELETE "$BASE/workspaces/$OFFLINE_ID" >/dev/null
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Test 7: Nonexistent workspace returns 404
|
|
# ========================================
|
|
echo "--- Test 7: Nonexistent workspace ---"
|
|
R=$(curl -s -o /dev/null -w "%{http_code}" --max-time 10 -X POST "$BASE/workspaces/00000000-0000-0000-0000-000000000000/a2a" \
|
|
-H "Content-Type: application/json" \
|
|
-d '{"method":"message/send","params":{"message":{"role":"user","parts":[{"type":"text","text":"test"}]}}}')
|
|
check "Nonexistent workspace returns 404" "404" "$R"
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Test 8: Multi-turn conversation
|
|
# ========================================
|
|
echo "--- Test 8: Multi-turn conversation ---"
|
|
R1=$(a2a_send "$ECHO_ID" "My name is Alice. Remember that.")
|
|
check "Turn 1 succeeds" '"result"' "$R1"
|
|
R2=$(a2a_send "$ECHO_ID" "What is my name?")
|
|
check "Turn 2 succeeds" '"result"' "$R2"
|
|
TEXT2=$(echo "$R2" | python3 -c "import sys,json; r=json.load(sys.stdin); print(r['result']['parts'][0]['text'])" 2>/dev/null || echo "PARSE_ERROR")
|
|
echo " Turn 2 response: $(echo "$TEXT2" | head -3)"
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Test 9: Long input handling
|
|
# ========================================
|
|
echo "--- Test 9: Long input ---"
|
|
LONG_TEXT=$(python3 -c "print('This is a test sentence. ' * 50)")
|
|
R=$(a2a_send "$ECHO_ID" "$LONG_TEXT")
|
|
check "Long input returns result" '"result"' "$R"
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Test 10: Peers can discover each other
|
|
# ========================================
|
|
echo "--- Test 10: Peer discovery ---"
|
|
R=$(curl -s "$BASE/registry/$ECHO_ID/peers")
|
|
check "Echo sees SEO as peer" 'SEO Agent' "$R"
|
|
R=$(curl -s "$BASE/registry/$SEO_ID/peers")
|
|
check "SEO sees Echo as peer" 'Echo Agent' "$R"
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Test 11: Agent card reflects skills
|
|
# ========================================
|
|
echo "--- Test 11: Agent cards ---"
|
|
R=$(curl -s "$BASE/workspaces/$ECHO_ID")
|
|
check "Echo agent has agent_card" '"agent_card"' "$R"
|
|
check "Echo has skills" '"skills"' "$R"
|
|
|
|
R=$(curl -s "$BASE/workspaces/$SEO_ID")
|
|
check "SEO agent has agent_card" '"agent_card"' "$R"
|
|
check "SEO has skills" '"skills"' "$R"
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Test 12: Heartbeat updates
|
|
# ========================================
|
|
echo "--- Test 12: Heartbeat ---"
|
|
sleep 2
|
|
R=$(curl -s "$BASE/workspaces/$ECHO_ID")
|
|
UPTIME=$(echo "$R" | python3 -c "import sys,json; print(json.load(sys.stdin)['uptime_seconds'])")
|
|
check "Echo has uptime > 0" "true" "$([ "$UPTIME" -gt 0 ] 2>/dev/null && echo true || echo false)"
|
|
echo " Echo uptime: ${UPTIME}s"
|
|
echo ""
|
|
|
|
# ========================================
|
|
# Summary
|
|
# ========================================
|
|
echo "=== Results: $PASS passed, $FAIL failed ==="
|
|
exit $FAIL
|