featurebase/pilosa_test.go
2022-01-21 10:57:05 -07:00

44 lines
1.3 KiB
Go

// Copyright 2021 Molecula Corp. All rights reserved.
package pilosa_test
import (
"strings"
"testing"
"github.com/molecula/featurebase/v3"
_ "github.com/molecula/featurebase/v3/test"
)
func TestAddressWithDefaults(t *testing.T) {
tests := []struct {
addr string
expected string
err string
}{
{addr: "", expected: "localhost:10101"},
{addr: ":", expected: "localhost:10101"},
{addr: "localhost", expected: "localhost:10101"},
{addr: "localhost:", expected: "localhost:10101"},
{addr: "127.0.0.1:10101", expected: "127.0.0.1:10101"},
{addr: "127.0.0.1:", expected: "127.0.0.1:10101"},
{addr: ":10101", expected: "localhost:10101"},
{addr: ":55555", expected: "localhost:55555"},
{addr: "1.2.3.4", expected: "1.2.3.4:10101"},
{addr: "1.2.3.4:", expected: "1.2.3.4:10101"},
{addr: "1.2.3.4:55555", expected: "1.2.3.4:55555"},
// The following tests check the error conditions.
{addr: "[invalid][addr]:port", err: pilosa.ErrInvalidAddress.Error()},
}
for _, test := range tests {
actual, err := pilosa.AddressWithDefaults(test.addr)
if err != nil {
if !strings.Contains(err.Error(), test.err) {
t.Errorf("expected error: %v, but got: %v", test.err, err)
}
} else {
if actual.HostPort() != test.expected {
t.Errorf("expected: %v, but got: %v", test.expected, actual)
}
}
}
}