Function TestExtractAddress

coderd/httpmw/realip_test.go:19–223

TestExtractAddress checks the ExtractAddress function.

(t *testing.T)

// TestExtractAddress checks the ExtractAddress function.
func TestExtractAddress(t *testing.T) {
	t.Parallel()

	tests := []struct {
		Name               string
		Config             *httpmw.RealIPConfig
		Header             http.Header
		RemoteAddr         string
		TLS                bool
		ExpectedRemoteAddr string
		ExpectedTLS        bool
	}{
		{
			Name:               "default-nil-config",
			RemoteAddr:         "123.45.67.89",
			ExpectedRemoteAddr: "123.45.67.89",
		},
		{
			Name:               "default-empty-config",
			RemoteAddr:         "123.45.67.89",
			ExpectedRemoteAddr: "123.45.67.89",
			Config:             &httpmw.RealIPConfig{},
		},
		{
			Name: "default-filter-headers",
			Config: &httpmw.RealIPConfig{
				TrustedOrigins: []*net.IPNet{
					{
						IP:   net.ParseIP("10.0.0.0"),
						Mask: net.CIDRMask(8, 32),
					},
				},
			},
			RemoteAddr: "123.45.67.89",
			Header: http.Header{
				"X-Forwarded-For": []string{
					"127.0.0.1",
					"10.0.0.5",
					"10.0.0.5,4.4.4.4",
				},
			},
			ExpectedRemoteAddr: "123.45.67.89",
		},
		{
			Name: "multiple-x-forwarded-for",
			Config: &httpmw.RealIPConfig{
				TrustedOrigins: []*net.IPNet{
					{
						IP:   net.ParseIP("0.0.0.0"),
						Mask: net.CIDRMask(0, 32),
					},
				},
				TrustedHeaders: []string{
					"X-Forwarded-For",
				},
			},
			RemoteAddr: "123.45.67.89",
			Header: http.Header{