ITWorx Pulse release export
bd774932d5
Public source validation / validate (push) Failing after 3m8s
58 lines
2.2 KiB
Go
58 lines
2.2 KiB
Go
package reverseproxy
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type staticToken string
|
|
|
|
func (s staticToken) Token(context.Context) (string, error) { return string(s), nil }
|
|
|
|
func TestNPMClientReadsOnlyProxyHostsAndUsesExternalToken(t *testing.T) {
|
|
var method, authorization, path string
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
method, authorization, path = r.Method, r.Header.Get("Authorization"), r.URL.Path
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, _ = w.Write([]byte("[{\"id\":7,\"domain_names\":[\"pulse.example.test\"],\"forward_scheme\":\"http\",\"forward_host\":\"10.0.0.7\",\"forward_port\":8080,\"enabled\":true,\"meta\":{\"pulse_service_id\":\"svc-api\"}}]"))
|
|
}))
|
|
defer server.Close()
|
|
now := time.Date(2026, 8, 2, 12, 0, 0, 0, time.UTC)
|
|
routes, err := (NPMClient{BaseURL: server.URL, Token: staticToken("external-token"), Now: func() time.Time { return now }}).ListRoutes(context.Background())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if method != http.MethodGet || path != "/api/nginx/proxy-hosts" || authorization != "Bearer external-token" {
|
|
t.Fatalf("unexpected request method=%s path=%s authorization=%s", method, path, authorization)
|
|
}
|
|
if len(routes) != 1 || routes[0].ID != "7:pulse.example.test" || routes[0].TargetServiceID != "svc-api" || routes[0].SourceID != "npm" {
|
|
t.Fatalf("unexpected routes: %+v", routes)
|
|
}
|
|
}
|
|
|
|
func TestNPMClientRejectsMutationAndOversizedResponses(t *testing.T) {
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != http.MethodGet {
|
|
t.Fatalf("unexpected method %s", r.Method)
|
|
}
|
|
_, _ = w.Write([]byte("[]"))
|
|
}))
|
|
defer server.Close()
|
|
client := NPMClient{BaseURL: server.URL, MaxBody: 1024}
|
|
if _, err := client.ListRoutes(context.Background()); err != nil {
|
|
t.Fatalf("small valid response should pass: %v", err)
|
|
}
|
|
if _, err := (NPMClient{BaseURL: server.URL, MaxBody: 9 << 20}).ListRoutes(context.Background()); err == nil {
|
|
t.Fatal("expected unsafe body bound")
|
|
}
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel()
|
|
if _, err := client.ListRoutes(ctx); !errors.Is(err, context.Canceled) {
|
|
t.Fatalf("expected cancellation, got %v", err)
|
|
}
|
|
}
|