Repository navigation
Expand file tree
/
Copy pathtest_agent_flow.sh
More file actions
140 lines (116 loc) · 4.85 KB
/
Copy pathtest_agent_flow.sh
File metadata and controls
140 lines (116 loc) · 4.85 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
#!/bin/bash
set -e
# Manual test script for agent/command flow
# Run this after starting the server with: make dev
BASE_URL="${BASE_URL:-http://localhost:8000}"
DEPLOYMENT_HASH="test_deployment_$(uuidgen | tr '[:upper:]' '[:lower:]')"
echo "=========================================="
echo "Testing Agent/Command Flow"
echo "Deployment Hash: $DEPLOYMENT_HASH"
echo "=========================================="
# Step 1: Register an agent
echo -e "\n=== Step 1: Register Agent ==="
REGISTER_RESPONSE=$(curl -s -X POST "$BASE_URL/api/v1/agent/register" \
-H "Content-Type: application/json" \
-d "{
\"deployment_hash\": \"$DEPLOYMENT_HASH\",
\"agent_version\": \"1.0.0\",
\"capabilities\": [\"docker\", \"compose\", \"logs\"],
\"system_info\": {
\"os\": \"linux\",
\"arch\": \"x86_64\",
\"memory_gb\": 8
}
}")
echo "Register Response:"
echo "$REGISTER_RESPONSE" | jq '.'
# Extract agent_id and token
AGENT_ID=$(echo "$REGISTER_RESPONSE" | jq -r '.item.agent_id // .data.item.agent_id // empty')
AGENT_TOKEN=$(echo "$REGISTER_RESPONSE" | jq -r '.item.agent_token // .data.item.agent_token // empty')
if [ -z "$AGENT_ID" ] || [ -z "$AGENT_TOKEN" ]; then
echo "ERROR: Failed to register agent or extract credentials"
echo "Response was: $REGISTER_RESPONSE"
exit 1
fi
echo "Agent ID: $AGENT_ID"
echo "Agent Token: ${AGENT_TOKEN:0:20}..."
# Step 2: Create a command (requires authentication - will likely fail without OAuth)
echo -e "\n=== Step 2: Create Command (may fail without auth) ==="
CREATE_CMD_RESPONSE=$(curl -s -w "\nHTTP_STATUS:%{http_code}" -X POST "$BASE_URL/api/v1/commands" \
-H "Content-Type: application/json" \
-d "{
\"deployment_hash\": \"$DEPLOYMENT_HASH\",
\"type\": \"restart_service\",
\"priority\": \"high\",
\"parameters\": {
\"service\": \"web\",
\"graceful\": true
},
\"timeout_seconds\": 300
}" 2>&1)
HTTP_STATUS=$(echo "$CREATE_CMD_RESPONSE" | grep "HTTP_STATUS:" | cut -d: -f2)
BODY=$(echo "$CREATE_CMD_RESPONSE" | sed '/HTTP_STATUS:/d')
echo "Create Command Response (Status: $HTTP_STATUS):"
echo "$BODY" | jq '.' 2>/dev/null || echo "$BODY"
if [ "$HTTP_STATUS" != "200" ] && [ "$HTTP_STATUS" != "201" ]; then
echo "WARNING: Command creation failed (expected - requires OAuth)"
echo "You can manually create a command in the database to test the wait/report flow"
echo ""
echo "SQL to insert test command:"
echo "INSERT INTO command (deployment_hash, type, priority, parameters, timeout_seconds, status)"
echo "VALUES ('$DEPLOYMENT_HASH', 'restart_service', 'high', '{\"service\": \"web\"}'::jsonb, 300, 'pending');"
echo ""
read -p "Press Enter after inserting the command manually, or Ctrl+C to exit..."
fi
COMMAND_ID=$(echo "$BODY" | jq -r '.item.command_id // .data.item.command_id // empty')
echo "Command ID: $COMMAND_ID"
# Step 3: Agent polls for commands
echo -e "\n=== Step 3: Agent Polls for Commands ==="
echo "Waiting for commands (timeout: 35s)..."
WAIT_RESPONSE=$(curl -s -w "\nHTTP_STATUS:%{http_code}" \
-X GET "$BASE_URL/api/v1/agent/commands/wait/$DEPLOYMENT_HASH" \
-H "X-Agent-Id: $AGENT_ID" \
-H "Authorization: Bearer $AGENT_TOKEN" \
--max-time 35 2>&1)
HTTP_STATUS=$(echo "$WAIT_RESPONSE" | grep "HTTP_STATUS:" | cut -d: -f2)
BODY=$(echo "$WAIT_RESPONSE" | sed '/HTTP_STATUS:/d')
echo "Wait Response (Status: $HTTP_STATUS):"
echo "$BODY" | jq '.' 2>/dev/null || echo "$BODY"
RECEIVED_COMMAND_ID=$(echo "$BODY" | jq -r '.item.command_id // .data.item.command_id // empty')
if [ -z "$RECEIVED_COMMAND_ID" ]; then
echo "No command received (timeout or no commands in queue)"
exit 0
fi
echo "Received Command ID: $RECEIVED_COMMAND_ID"
# Step 4: Agent reports command result
echo -e "\n=== Step 4: Agent Reports Command Result ==="
REPORT_RESPONSE=$(curl -s -w "\nHTTP_STATUS:%{http_code}" \
-X POST "$BASE_URL/api/v1/agent/commands/report" \
-H "Content-Type: application/json" \
-H "X-Agent-Id: $AGENT_ID" \
-H "Authorization: Bearer $AGENT_TOKEN" \
-d "{
\"command_id\": \"$RECEIVED_COMMAND_ID\",
\"deployment_hash\": \"$DEPLOYMENT_HASH\",
\"status\": \"completed\",
\"result\": {
\"service_restarted\": true,
\"restart_time_seconds\": 5.2,
\"final_status\": \"running\"
},
\"metadata\": {
\"execution_node\": \"worker-1\"
}
}" 2>&1)
HTTP_STATUS=$(echo "$REPORT_RESPONSE" | grep "HTTP_STATUS:" | cut -d: -f2)
BODY=$(echo "$REPORT_RESPONSE" | sed '/HTTP_STATUS:/d')
echo "Report Response (Status: $HTTP_STATUS):"
echo "$BODY" | jq '.' 2>/dev/null || echo "$BODY"
echo -e "\n=========================================="
echo "Test Flow Complete!"
echo "=========================================="
echo "Summary:"
echo " - Agent registered: $AGENT_ID"
echo " - Command created: ${COMMAND_ID:-N/A (auth required)}"
echo " - Command received: ${RECEIVED_COMMAND_ID:-N/A}"
echo " - Report status: $HTTP_STATUS"