Cleaner IMO (#1361)
This commit is contained in:
parent
bbbd08edc1
commit
d349f059af
1 changed files with 40 additions and 10 deletions
|
@ -11,15 +11,45 @@ import (
|
|||
)
|
||||
|
||||
func Test_parsePostgreSQLHostPort(t *testing.T) {
|
||||
test := func(input, expectedHost, expectedPort string) {
|
||||
host, port := parsePostgreSQLHostPort(input)
|
||||
assert.Equal(t, expectedHost, host)
|
||||
assert.Equal(t, expectedPort, port)
|
||||
tests := []struct {
|
||||
HostPort string
|
||||
Host string
|
||||
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")
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue