80 lines
3.1 KiB
Go
80 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
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 TestWorkerStatusReportsLocalExecutor(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 TestWorkerReadyzRequiresExecutor(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(true).ServeHTTP(response, httptest.NewRequest(http.MethodGet, "/readyz", nil))
|
|
if response.Code != http.StatusOK {
|
|
t.Fatalf("configured readyz status=%d", response.Code)
|
|
}
|
|
}
|