65 lines
2.4 KiB
Go
65 lines
2.4 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 TestWorkerStatusReportsLocalFallback(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-fallback" || !value.LocalFallback {
|
|
t.Fatalf("unexpected status: code=%d value=%+v", response.Code, 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)
|
|
}
|
|
}
|