diff options
Diffstat (limited to 'vendor/github.com/tomasen/realip/realip_test.go')
-rw-r--r-- | vendor/github.com/tomasen/realip/realip_test.go | 95 |
1 files changed, 0 insertions, 95 deletions
diff --git a/vendor/github.com/tomasen/realip/realip_test.go b/vendor/github.com/tomasen/realip/realip_test.go deleted file mode 100644 index e80efe0b..00000000 --- a/vendor/github.com/tomasen/realip/realip_test.go +++ /dev/null @@ -1,95 +0,0 @@ -package realip - -import ( - "net/http" - "testing" -) - -func TestIsPrivateAddr(t *testing.T) { - testData := map[string]bool{ - "127.0.0.0": true, - "10.0.0.0": true, - "169.254.0.0": true, - "192.168.0.0": true, - "::1": true, - "fc00::": true, - - "172.15.0.0": false, - "172.16.0.0": true, - "172.31.0.0": true, - "172.32.0.0": false, - - "147.12.56.11": false, - } - - for addr, isLocal := range testData { - isPrivate, err := isPrivateAddress(addr) - if err != nil { - t.Errorf("fail processing %s: %v", addr, err) - } - - if isPrivate != isLocal { - format := "%s should " - if !isLocal { - format += "not " - } - format += "be local address" - - t.Errorf(format, addr) - } - } -} - -func TestRealIP(t *testing.T) { - // Create type and function for testing - type testIP struct { - name string - request *http.Request - expected string - } - - newRequest := func(remoteAddr, xRealIP string, xForwardedFor ...string) *http.Request { - h := http.Header{} - h.Set("X-Real-IP", xRealIP) - for _, address := range xForwardedFor { - h.Set("X-Forwarded-For", address) - } - - return &http.Request{ - RemoteAddr: remoteAddr, - Header: h, - } - } - - // Create test data - publicAddr1 := "144.12.54.87" - publicAddr2 := "119.14.55.11" - localAddr := "127.0.0.0" - - testData := []testIP{ - { - name: "No header", - request: newRequest(publicAddr1, ""), - expected: publicAddr1, - }, { - name: "Has X-Forwarded-For", - request: newRequest("", "", publicAddr1), - expected: publicAddr1, - }, { - name: "Has multiple X-Forwarded-For", - request: newRequest("", "", localAddr, publicAddr1, publicAddr2), - expected: publicAddr2, - }, { - name: "Has X-Real-IP", - request: newRequest("", publicAddr1), - expected: publicAddr1, - }, - } - - // Run test - for _, v := range testData { - if actual := FromRequest(v.request); v.expected != actual { - t.Errorf("%s: expected %s but get %s", v.name, v.expected, actual) - } - } -} |