Publish ITWorx Pulse source

internal/reverseproxyapi/handler_test.go

@@ -0,0 +1,56 @@
+package reverseproxyapi
+
+import (
+	"context"
+	"errors"
+	"net/http"
+	"net/http/httptest"
+	"strings"
+	"testing"
+
+	"github.com/itworx/pulse/internal/auth"
+	"github.com/itworx/pulse/internal/reverseproxy"
+)
+
+type staticProvider struct {
+	snapshot reverseproxy.Snapshot
+	err      error
+}
+
+func (p staticProvider) Snapshot(context.Context) (reverseproxy.Snapshot, error) {
+	return p.snapshot, p.err
+}
+
+func authenticatedRequest(method, path string) *http.Request {
+	request := httptest.NewRequest(method, path, nil)
+	return request.WithContext(auth.WithPrincipal(request.Context(), auth.Principal{Subject: "viewer", Role: auth.RoleViewer}))
+}
+
+func TestHandlerRequiresAuthAndReturnsDisabledSnapshot(t *testing.T) {
+	handler := Handler{}
+	unauthorized := httptest.NewRecorder()
+	handler.ServeHTTP(unauthorized, httptest.NewRequest(http.MethodGet, "/api/v1/reverse-proxy", nil))
+	if unauthorized.Code != http.StatusUnauthorized {
+		t.Fatalf("unauthorized status=%d", unauthorized.Code)
+	}
+	response := httptest.NewRecorder()
+	handler.ServeHTTP(response, authenticatedRequest(http.MethodGet, "/api/v1/reverse-proxy"))
+	if response.Code != http.StatusOK || !strings.Contains(response.Body.String(), "\"state\":\"disabled\"") || !strings.Contains(response.Body.String(), "\"routes\":[]") {
+		t.Fatalf("disabled response status=%d body=%s", response.Code, response.Body.String())
+	}
+}
+
+func TestHandlerPreservesRouteSourceAndMapsProviderError(t *testing.T) {
+	handler := Handler{Provider: staticProvider{snapshot: reverseproxy.Snapshot{Source: reverseproxy.Source{ID: "npm", State: reverseproxy.StateEnabled}, Routes: []reverseproxy.Route{{ID: "route-1", SourceID: "npm", Hostname: "pulse.example.test", URL: "https://pulse.example.test", TargetServiceID: "svc", Enabled: true}}, Total: 1}}}
+	response := httptest.NewRecorder()
+	handler.ServeHTTP(response, authenticatedRequest(http.MethodGet, "/api/v1/reverse-proxy"))
+	if response.Code != http.StatusOK || !strings.Contains(response.Body.String(), "\"sourceId\":\"npm\"") {
+		t.Fatalf("route response status=%d body=%s", response.Code, response.Body.String())
+	}
+	failed := Handler{Provider: staticProvider{err: errors.New("backend unavailable")}}
+	failure := httptest.NewRecorder()
+	failed.ServeHTTP(failure, authenticatedRequest(http.MethodGet, "/api/v1/reverse-proxy"))
+	if failure.Code != http.StatusServiceUnavailable {
+		t.Fatalf("failure status=%d", failure.Code)
+	}
+}