First commit

vendor/github.com/tomasen/realip/realip_test.go

@@ -0,0 +1,86 @@
+package realip
+
+import (
+	"net/http"
+	"strings"
+	"testing"
+)
+
+func TestIsLocalAddr(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 {
+		if isLocalAddress(addr) != isLocal {
+			format := "%s should "
+			if !isLocal {
+				format += "not "
+			}
+			format += "be local address"
+
+			t.Errorf(format, addr)
+		}
+	}
+}
+
+func TestIpAddrFromRemoteAddr(t *testing.T) {
+	testData := map[string]string{
+		"127.0.0.1:8888": "127.0.0.1",
+		"ip:port":        "ip",
+		"ip":             "ip",
+		"12:34::0":       "12:34:",
+	}
+
+	for remoteAddr, expectedAddr := range testData {
+		if actualAddr := ipAddrFromRemoteAddr(remoteAddr); actualAddr != expectedAddr {
+			t.Errorf("ipAddrFromRemoteAddr of %s should be %s but get %s", remoteAddr, expectedAddr, actualAddr)
+		}
+	}
+}
+
+func TestRealIP(t *testing.T) {
+	newRequest := func(remoteAddr, hdrRealIP, hdrForwardedFor string) *http.Request {
+		h := http.Header{}
+		h["X-Real-Ip"] = []string{hdrRealIP}
+		h["X-Forwarded-For"] = []string{hdrForwardedFor}
+		return &http.Request{
+			RemoteAddr: remoteAddr,
+			Header:     h,
+		}
+	}
+
+	remoteAddr := "144.12.54.87"
+	anotherRemoteAddr := "119.14.55.11"
+	localAddr := "127.0.0.0"
+
+	testData := []struct {
+		expected string
+		request  *http.Request
+	}{
+		{remoteAddr, newRequest(remoteAddr, "", "")}, // no header
+		{remoteAddr, newRequest("", "", remoteAddr)}, // X-Forwarded-For: remoteAddr
+		{remoteAddr, newRequest("", remoteAddr, "")}, // X-RealIP: remoteAddr
+
+		// X-Forwarded-For: localAddr, remoteAddr, anotherRemoteAddr
+		{remoteAddr, newRequest("", "", strings.Join([]string{localAddr, remoteAddr, anotherRemoteAddr}, ", "))},
+	}
+
+	for _, v := range testData {
+		if actual := RealIP(v.request); v.expected != actual {
+			t.Errorf("expected %s but get %s", v.expected, actual)
+		}
+	}
+}