first commit

bouncer_test.go

@@ -0,0 +1,49 @@
+package crowdsec_bouncer_traefik_plugin_test
+
+import (
+	"context"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+
+	"github.com/maxlerebourg/crowdsec-bouncer-traefik-plugin"
+)
+
+func TestDemo(t *testing.T) {
+	cfg := crowdsec_bouncer_traefik_plugin.CreateConfig()
+	cfg.Headers["X-Host"] = "[[.Host]]"
+	cfg.Headers["X-Method"] = "[[.Method]]"
+	cfg.Headers["X-URL"] = "[[.URL]]"
+	cfg.Headers["X-URL"] = "[[.URL]]"
+	cfg.Headers["X-Demo"] = "test"
+
+	ctx := context.Background()
+	next := http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) {})
+
+	handler, err := crowdsec_bouncer_traefik_plugin.New(ctx, next, cfg, "demo-plugin")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	recorder := httptest.NewRecorder()
+
+	req, err := http.NewRequestWithContext(ctx, http.MethodGet, "http://localhost", nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	handler.ServeHTTP(recorder, req)
+
+	assertHeader(t, req, "X-Host", "localhost")
+	assertHeader(t, req, "X-URL", "http://localhost")
+	assertHeader(t, req, "X-Method", "GET")
+	assertHeader(t, req, "X-Demo", "test")
+}
+
+func assertHeader(t *testing.T, req *http.Request, key, expected string) {
+	t.Helper()
+
+	if req.Header.Get(key) != expected {
+		t.Errorf("invalid header value: %s", req.Header.Get(key))
+	}
+}