115 lines
4.7 KiB
Go
115 lines
4.7 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestLoadProtectedBridgeURL(t *testing.T) {
|
|
root := t.TempDir()
|
|
path := filepath.Join(root, "agent_execution_bridge_url")
|
|
if err := os.WriteFile(path, []byte("http://bridge.local\n"), 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if value, configured := loadProtectedBridgeURLFrom(root, path); !configured || value != "http://bridge.local" {
|
|
t.Fatalf("protected bridge was not loaded: %q %v", value, configured)
|
|
}
|
|
if value, configured := loadProtectedBridgeURLFrom(root, filepath.Join(root, "missing")); configured || value != "" {
|
|
t.Fatalf("missing bridge was accepted: %q %v", value, configured)
|
|
}
|
|
if value, configured := loadProtectedBridgeURLFrom(root, "/tmp/not-protected"); configured || value != "" {
|
|
t.Fatalf("path outside protected root was accepted: %q %v", value, configured)
|
|
}
|
|
}
|
|
|
|
func TestWorkerStatusReportsRecoveryOnlyWithoutExecutor(t *testing.T) {
|
|
response := httptest.NewRecorder()
|
|
workerStatusHandler(false).ServeHTTP(response, httptest.NewRequest(http.MethodGet, "/status", nil))
|
|
var value workerStatusResponse
|
|
if err := json.NewDecoder(response.Body).Decode(&value); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if response.Code != http.StatusOK || value.Ready || value.Mode != "recovery-only" || value.ExecutorConfigured {
|
|
t.Fatalf("unexpected status: code=%d value=%+v", response.Code, value)
|
|
}
|
|
}
|
|
|
|
func TestWorkerStatusReportsLocalExecutorCapabilityReady(t *testing.T) {
|
|
response := httptest.NewRecorder()
|
|
workerStatusHandler(false, true).ServeHTTP(response, httptest.NewRequest(http.MethodGet, "/status", nil))
|
|
var value workerStatusResponse
|
|
if err := json.NewDecoder(response.Body).Decode(&value); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if response.Code != http.StatusOK || !value.Ready || value.Mode != "local-executor" || !value.LocalFallback || !value.LocalExecutor {
|
|
t.Fatalf("unexpected status: code=%d value=%+v", response.Code, value)
|
|
}
|
|
if len(value.Capabilities) != 1 || value.Capabilities[0] != "task-evaluation" {
|
|
t.Fatalf("unexpected local capabilities: %+v", value.Capabilities)
|
|
}
|
|
}
|
|
|
|
func TestWorkerStatusDoesNotCallLocalFallbackAnExternalExecutor(t *testing.T) {
|
|
response := httptest.NewRecorder()
|
|
workerStatusHandler(false, true).ServeHTTP(response, httptest.NewRequest(http.MethodGet, "/status", nil))
|
|
var value workerStatusResponse
|
|
if err := json.NewDecoder(response.Body).Decode(&value); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if value.ExecutorConfigured || !value.Ready || !value.LocalFallback || !value.LocalExecutor || value.Mode != "local-executor" {
|
|
t.Fatalf("local fallback was presented as external execution: %+v", value)
|
|
}
|
|
}
|
|
|
|
func TestWorkerReadyzRequiresAnAdvertisedCapability(t *testing.T) {
|
|
response := httptest.NewRecorder()
|
|
workerStatusHandler(false).ServeHTTP(response, httptest.NewRequest(http.MethodGet, "/readyz", nil))
|
|
if response.Code != http.StatusServiceUnavailable {
|
|
t.Fatalf("readyz status=%d", response.Code)
|
|
}
|
|
response = httptest.NewRecorder()
|
|
workerStatusHandler(false, true).ServeHTTP(response, httptest.NewRequest(http.MethodGet, "/readyz", nil))
|
|
if response.Code != http.StatusOK {
|
|
t.Fatalf("local capability readyz status=%d", response.Code)
|
|
}
|
|
response = httptest.NewRecorder()
|
|
workerStatusHandler(true).ServeHTTP(response, httptest.NewRequest(http.MethodGet, "/readyz", nil))
|
|
if response.Code != http.StatusOK {
|
|
t.Fatalf("configured readyz status=%d", response.Code)
|
|
}
|
|
}
|
|
|
|
func TestWorkerStatusReportsRecoveryCycle(t *testing.T) {
|
|
runtime := newAgentWorkerRuntimeStatus()
|
|
runtime.cycleStarted(time.Date(2026, 8, 23, 10, 0, 0, 0, time.UTC))
|
|
runtime.cycleFinished(time.Date(2026, 8, 23, 10, 0, 1, 0, time.UTC), 3, "dispatch-1", "recovered")
|
|
response := httptest.NewRecorder()
|
|
workerStatusHandler(false, true, runtime).ServeHTTP(response, httptest.NewRequest(http.MethodGet, "/status", nil))
|
|
var value workerStatusResponse
|
|
if err := json.NewDecoder(response.Body).Decode(&value); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if value.LastRecoveredClaims != 3 || value.TotalRecoveredClaims != 3 || value.LastDispatchID != "dispatch-1" || value.LastResult != "recovered" {
|
|
t.Fatalf("recovery evidence missing: %+v", value)
|
|
}
|
|
}
|
|
|
|
func TestWorkerStatusReportsCycleError(t *testing.T) {
|
|
runtime := newAgentWorkerRuntimeStatus()
|
|
runtime.cycleFailed(time.Date(2026, 8, 23, 10, 0, 1, 0, time.UTC), os.ErrClosed)
|
|
response := httptest.NewRecorder()
|
|
workerStatusHandler(false, runtime).ServeHTTP(response, httptest.NewRequest(http.MethodGet, "/status", nil))
|
|
var value workerStatusResponse
|
|
if err := json.NewDecoder(response.Body).Decode(&value); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if value.LastResult != "failed" || value.LastError == "" {
|
|
t.Fatalf("cycle error missing: %+v", value)
|
|
}
|
|
}
|