ITWorx Pulse release export
bd774932d5
Public source validation / validate (push) Failing after 3m8s
57 lines
2.4 KiB
Go
57 lines
2.4 KiB
Go
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)
|
|
}
|
|
}
|