Rough tests for the clientAddr function. (1ef6211c) · Commits · Cecylia Bocovich / Snowflake WebExtension

server/server_test.go

0 → 100644

+49 −0

Original line number	Diff line number	Diff line
		package main

		import (
		"net"
		"strconv"
		"testing"
		)

		func TestClientAddr(t *testing.T) {
		// good tests
		for _, test := range []struct {
		input string
		expected net.IP
		}{
		{"1.2.3.4", net.ParseIP("1.2.3.4")},
		{"1:2::3:4", net.ParseIP("1:2::3:4")},
		} {
		useraddr := clientAddr(test.input)
		host, port, err := net.SplitHostPort(useraddr)
		if err != nil {
		t.Errorf("clientAddr(%q) → SplitHostPort error %v", test.input, err)
		continue
		}
		if !test.expected.Equal(net.ParseIP(host)) {
		t.Errorf("clientAddr(%q) → host %q, not %v", test.input, host, test.expected)
		}
		portNo, err := strconv.Atoi(port)
		if err != nil {
		t.Errorf("clientAddr(%q) → port %q", test.input, port)
		continue
		}
		if portNo == 0 {
		t.Errorf("clientAddr(%q) → port %d", test.input, portNo)
		}
		}

		// bad tests
		for _, input := range []string{
		"",
		"abc",
		"1.2.3.4.5",
		"[12::34]",
		} {
		useraddr := clientAddr(input)
		if useraddr != "" {
		t.Errorf("clientAddr(%q) → %q, not %q", input, useraddr, "")
		}
		}
		}