Function TestFormatHost

calnex/api/api_test.go:914–949

(t *testing.T)

	f, err := calnexAPI.FetchSystemMonitor()
	require.NoError(t, err)
	require.Equal(t, expected, f)
}

func TestFormatHost(t *testing.T) {
	tests := []struct {
		name    string
		input   string
		want    string
		wantErr bool
	}{
		{name: "IPv6 full", input: "2401:db00:2715:f002:face:0000:021a:0000", want: "[2401:db00:2715:f002:face:0000:021a:0000]"},
		{name: "IPv6 loopback short", input: "::1", want: "[::1]"},
		{name: "IPv6 already bracketed", input: "[2401:db00:2715:f002:face:0000:021a:0000]", want: "[2401:db00:2715:f002:face:0000:021a:0000]"},
		{name: "IPv4 plain", input: "10.128.64.5", want: "10.128.64.5"},
		{name: "FQDN", input: "calnex01.dc1.facebook.com", want: "calnex01.dc1.facebook.com"},
		{name: "localhost", input: "localhost", want: "localhost"},

		{name: "empty", input: "", wantErr: true},
		{name: "spaces", input: "not valid", wantErr: true},
		{name: "garbage in brackets", input: "[garbage]", wantErr: true},
		{name: "IPv4 in brackets", input: "[10.128.64.5]", wantErr: true},
		{name: "path injection", input: "host/extra", wantErr: true},
		{name: "userinfo injection", input: "user@host", wantErr: true},
		{name: "query injection", input: "host?x=1", wantErr: true},
		{name: "fragment injection", input: "host#frag", wantErr: true},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got, err := formatHost(tt.input)
			if tt.wantErr {
				require.Error(t, err)
				return
			}
			require.NoError(t, err)
			require.Equal(t, tt.want, got)
		})
	}
}