IP_test.go

// Copyright (c) 2018 Shen Sheng
// Copyright (c) 2019 Eduard Urbach

package aero

import (
	"net/http"
	"testing"
)

func TestIsPrivateAddress(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 the test
	for _, v := range testData {
		if actual := realIP(v.request); v.expected != actual {
			t.Errorf("%s: expected %s but get %s", v.name, v.expected, actual)
		}
	}
}