Function TestGetSourceIPs

middleware/source_ips_test.go:14–203

(t *testing.T)

)

func TestGetSourceIPs(t *testing.T) {
	tests := []struct {
		name string
		req  *http.Request
		want string
	}{
		{
			name: "no header",
			req:  &http.Request{RemoteAddr: "192.168.1.100:3454"},
			want: "192.168.1.100",
		},
		{
			name: "no header and remote has no port",
			req:  &http.Request{RemoteAddr: "192.168.1.100"},
			want: "192.168.1.100",
		},
		{
			name: "no header, remote address is invalid",
			req:  &http.Request{RemoteAddr: "192.168.100"},
			want: "192.168.100",
		},
		{
			name: "X-Forwarded-For and single forward address",
			req: &http.Request{
				RemoteAddr: "192.168.1.100:3454",
				Header: map[string][]string{
					http.CanonicalHeaderKey(xForwardedFor): {"172.16.1.1"},
				},
			},
			want: "172.16.1.1, 192.168.1.100",
		},
		{
			name: "X-Forwarded-For and single forward address which is same as remote",
			req: &http.Request{
				RemoteAddr: "192.168.1.100:3454",
				Header: map[string][]string{
					http.CanonicalHeaderKey(xForwardedFor): {"192.168.1.100"},
				},
			},
			want: "192.168.1.100",
		},
		{
			name: "single IPv6 X-Forwarded-For address",
			req: &http.Request{
				RemoteAddr: "[2001:db9::1]:3454",
				Header: map[string][]string{
					http.CanonicalHeaderKey(xForwardedFor): {"2001:db8::1"},
				},
			},
			want: "2001:db8::1, 2001:db9::1",
		},
		{
			name: "single X-Forwarded-For address no RemoteAddr",
			req: &http.Request{
				Header: map[string][]string{
					http.CanonicalHeaderKey(xForwardedFor): {"172.16.1.1"},
				},
			},