Function TestContext_RealIP

context_test.go:1338–1376

(t *testing.T)

}

func TestContext_RealIP(t *testing.T) {
	_, ipv6ForRemoteAddrExternalRange, _ := net.ParseCIDR("2001:db8::/64")

	var testCases = []struct {
		name            string
		givenIPExtrator IPExtractor
		whenReq         *http.Request
		expect          string
	}{
		{
			name:            "ip from remote addr",
			givenIPExtrator: nil,
			whenReq:         &http.Request{RemoteAddr: "89.89.89.89:1654"},
			expect:          "89.89.89.89",
		},
		{
			name:            "ip from ip extractor",
			givenIPExtrator: ExtractIPFromRealIPHeader(TrustIPRange(ipv6ForRemoteAddrExternalRange)),
			whenReq: &http.Request{
				Header: http.Header{
					HeaderXRealIP:       []string{"[2001:db8::113:199]"},
					HeaderXForwardedFor: []string{"[2001:db8::113:198], [2001:db8::113:197]"}, // <-- should not affect anything
				},
				RemoteAddr: "[2001:db8::113:1]:8080",
			},
			expect: "2001:db8::113:199",
		},
	}
	for _, tc := range testCases {
		t.Run(tc.name, func(t *testing.T) {
			e := New()
			c := e.NewContext(tc.whenReq, nil)
			if tc.givenIPExtrator != nil {
				e.IPExtractor = tc.givenIPExtrator
			}
			assert.Equal(t, tc.expect, c.RealIP())
		})
	}
}

func TestContext_File(t *testing.T) {
	var testCases = []struct {