Function TestLogSourceIPs

server/server_test.go:783–852

(t *testing.T)

}

func TestLogSourceIPs(t *testing.T) {
	var level log.Level
	require.NoError(t, level.Set("info"))
	cfg := Config{
		HTTPMiddleware:   []middleware.Interface{middleware.Log{Log: log.Global()}},
		MetricsNamespace: "testing_mux",
		LogLevel:         level,
		LogSourceIPs:     true,
	}
	setAutoAssignedPorts(DefaultNetwork, &cfg)

	startServer := func(cfg Config) *Server {
		prometheus.DefaultRegisterer = prometheus.NewRegistry()
		server, err := New(cfg)
		require.NoError(t, err)

		server.HTTP.HandleFunc("/error500", func(w http.ResponseWriter, _ *http.Request) {
			w.WriteHeader(500)
		})

		go func() {
			require.NoError(t, server.Run())
		}()

		return server
	}

	t.Run("without PROXY protocol", func(t *testing.T) {
		logger := newFakeLogger()
		cfg.Log = logger

		server := startServer(cfg)
		defer server.Shutdown()

		logger.assertNotContains(t, "sourceIPs")

		req, err := http.NewRequest("GET", httpTarget(server, "/error500"), nil)
		require.NoError(t, err)
		_, err = http.DefaultClient.Do(req)
		require.NoError(t, err)

		logger.assertContains(t, "sourceIPs=127.0.0.1")
	})

	t.Run("with PROXY protocol", func(t *testing.T) {
		logger := newFakeLogger()
		cfg.Log = logger
		cfg.ProxyProtocolEnabled = true

		server := startServer(cfg)
		defer server.Shutdown()

		logger.assertNotContains(t, "sourceIPs")

		fakeSourceIP := "1.2.3.4"
		proxyHeader := fmt.Sprintf("PROXY TCP4 %s 192.168.0.1 51234 80\r\n", fakeSourceIP)
		client := &http.Client{
			Transport: &http.Transport{