mirror of
https://github.com/rcourtman/Pulse.git
synced 2026-04-28 03:20:11 +00:00
a6a8efaa65
New test files with expanded coverage: API tests: - ai_handler_test.go: AI handler unit tests with mocking - agent_profiles_tools_test.go: Profile management tests - alerts_endpoints_test.go: Alert API endpoint tests - alerts_test.go: Updated for interface changes - audit_handlers_test.go: Audit handler tests - frontend_embed_test.go: Frontend embedding tests - metadata_handlers_test.go, metadata_provider_test.go: Metadata tests - notifications_test.go: Updated for interface changes - profile_suggestions_test.go: Profile suggestion tests - saml_service_test.go: SAML authentication tests - sensor_proxy_gate_test.go: Sensor proxy tests - updates_test.go: Updated for interface changes Agent tests: - dockeragent/signature_test.go: Docker agent signature tests - hostagent/agent_metrics_test.go: Host agent metrics tests - hostagent/commands_test.go: Command execution tests - hostagent/network_helpers_test.go: Network helper tests - hostagent/proxmox_setup_test.go: Updated setup tests - kubernetesagent/*_test.go: Kubernetes agent tests Core package tests: - monitoring/kubernetes_agents_test.go, reload_test.go - remoteconfig/client_test.go, signature_test.go - sensors/collector_test.go - updates/adapter_installsh_*_test.go: Install adapter tests - updates/manager_*_test.go: Update manager tests - websocket/hub_*_test.go: WebSocket hub tests Library tests: - pkg/audit/export_test.go: Audit export tests - pkg/metrics/store_test.go: Metrics store tests - pkg/proxmox/*_test.go: Proxmox client tests - pkg/reporting/reporting_test.go: Reporting tests - pkg/server/*_test.go: Server tests - pkg/tlsutil/extra_test.go: TLS utility tests Total: ~8000 lines of new test code
48 lines
1.3 KiB
Go
48 lines
1.3 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/rcourtman/pulse-go-rewrite/internal/config"
|
|
)
|
|
|
|
func TestSensorProxyGate(t *testing.T) {
|
|
r := &Router{config: &config.Config{EnableSensorProxy: true}}
|
|
if !r.isSensorProxyEnabled() {
|
|
t.Fatal("expected sensor proxy enabled")
|
|
}
|
|
|
|
allowed := r.requireSensorProxyEnabled(func(w http.ResponseWriter, _ *http.Request) {
|
|
w.WriteHeader(http.StatusNoContent)
|
|
})
|
|
req := httptest.NewRequest(http.MethodGet, "/api/sensor", nil)
|
|
resp := httptest.NewRecorder()
|
|
allowed(resp, req)
|
|
if resp.Code != http.StatusNoContent {
|
|
t.Fatalf("unexpected status: %d", resp.Code)
|
|
}
|
|
|
|
r.config.EnableSensorProxy = false
|
|
denied := r.requireSensorProxyEnabled(func(http.ResponseWriter, *http.Request) {
|
|
t.Fatal("handler should not be called")
|
|
})
|
|
resp = httptest.NewRecorder()
|
|
denied(resp, req)
|
|
if resp.Code != http.StatusGone {
|
|
t.Fatalf("unexpected status: %d", resp.Code)
|
|
}
|
|
if warning := resp.Header().Get("Warning"); warning == "" {
|
|
t.Fatal("expected Warning header")
|
|
}
|
|
|
|
var apiErr APIError
|
|
if err := json.Unmarshal(resp.Body.Bytes(), &apiErr); err != nil {
|
|
t.Fatalf("decode response: %v", err)
|
|
}
|
|
if apiErr.Code != "sensor_proxy_disabled" {
|
|
t.Fatalf("unexpected error code: %s", apiErr.Code)
|
|
}
|
|
}
|