Cleaner IMO (#1361)

This commit is contained in:
Kim "BKC" Carlbäcker 2017-03-22 15:30:40 +01:00 committed by Lunny Xiao
parent bbbd08edc1
commit d349f059af
1 changed files with 40 additions and 10 deletions

View File

@ -11,15 +11,45 @@ import (
) )
func Test_parsePostgreSQLHostPort(t *testing.T) { func Test_parsePostgreSQLHostPort(t *testing.T) {
test := func(input, expectedHost, expectedPort string) { tests := []struct {
host, port := parsePostgreSQLHostPort(input) HostPort string
assert.Equal(t, expectedHost, host) Host string
assert.Equal(t, expectedPort, port) Port string
}{
{
HostPort: "127.0.0.1:1234",
Host: "127.0.0.1",
Port: "1234",
},
{
HostPort: "127.0.0.1",
Host: "127.0.0.1",
Port: "5432",
},
{
HostPort: "[::1]:1234",
Host: "[::1]",
Port: "1234",
},
{
HostPort: "[::1]",
Host: "[::1]",
Port: "5432",
},
{
HostPort: "/tmp/pg.sock:1234",
Host: "/tmp/pg.sock",
Port: "1234",
},
{
HostPort: "/tmp/pg.sock",
Host: "/tmp/pg.sock",
Port: "5432",
},
}
for _, test := range tests {
host, port := parsePostgreSQLHostPort(test.HostPort)
assert.Equal(t, test.Host, host)
assert.Equal(t, test.Port, port)
} }
test("127.0.0.1:1234", "127.0.0.1", "1234")
test("127.0.0.1", "127.0.0.1", "5432")
test("[::1]:1234", "[::1]", "1234")
test("[::1]", "[::1]", "5432")
test("/tmp/pg.sock:1234", "/tmp/pg.sock", "1234")
test("/tmp/pg.sock", "/tmp/pg.sock", "5432")
} }