Function TestMultipleIpAddressesReverse

geoblock_test.go:334–363

(t *testing.T)

}

func TestMultipleIpAddressesReverse(t *testing.T) {
	mockServer := createMockAPIServer(t, map[string][]byte{caExampleIP: []byte(`CA`), chExampleIP: []byte(`CH`)})
	defer mockServer.Close()

	cfg := createTesterConfig()

	cfg.Countries = append(cfg.Countries, "CH")
	cfg.API = mockServer.URL + "/{ip}"

	ctx := context.Background()
	next := http.HandlerFunc(func(_ http.ResponseWriter, _ *http.Request) {})

	handler, err := geoblock.New(ctx, next, cfg, "GeoBlock")
	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)
	}

	req.Header.Add(xForwardedFor, strings.Join([]string{caExampleIP, chExampleIP}, ","))

	handler.ServeHTTP(recorder, req)

	assertStatusCode(t, recorder.Result(), http.StatusForbidden)
}

func TestMultipleIpAddressesProxy(t *testing.T) {
	mockServer := createMockAPIServer(t, map[string][]byte{caExampleIP: []byte(`CA`)})