Function TestGetSourceIPsWithCustomRegex

middleware/source_ips_test.go:406–458

(t *testing.T)

}

func TestGetSourceIPsWithCustomRegex(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 matching entry in the header",
			req: &http.Request{
				RemoteAddr: "192.168.1.100:3454",
				Header: map[string][]string{
					http.CanonicalHeaderKey("SomeHeader"): {"not matching"},
				},
			},
			want: "192.168.1.100",
		},
		{
			name: "one matching entry in the header",
			req: &http.Request{
				RemoteAddr: "192.168.1.100:3454",
				Header: map[string][]string{
					http.CanonicalHeaderKey("SomeHeader"): {"172.16.1.1"},
				},
			},
			want: "172.16.1.1, 192.168.1.100",
		},
		{
			name: "multiple matching entries in the header, only first used",
			req: &http.Request{
				RemoteAddr: "192.168.1.100:3454",
				Header: map[string][]string{
					http.CanonicalHeaderKey("SomeHeader"): {"172.16.1.1,172.16.2.1"},
				},
			},
			want: "172.16.1.1, 192.168.1.100",
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			sourceIPs, err := NewSourceIPs("SomeHeader", "((?:[0-9]{1,3}\\.){3}[0-9]{1,3})", false)
			require.NoError(t, err)

			if got := sourceIPs.Get(tt.req); got != tt.want {
				t.Errorf("GetSource() = %v, want %v", got, tt.want)
			}
		})
	}
}

func TestGetSourceIPsWithCustomRegexFullExtract(t *testing.T) {
	tests := []struct {