53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
|
// Copyright (C) 2020 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package dbutil
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestEscapableCommaSplit(t *testing.T) {
|
||
|
|
for _, testcase := range []struct {
|
||
|
|
input string
|
||
|
|
expected []string
|
||
|
|
}{
|
||
|
|
{"", []string{""}},
|
||
|
|
{",", []string{"", ""}},
|
||
|
|
{",hello", []string{"", "hello"}},
|
||
|
|
{"hello,", []string{"hello", ""}},
|
||
|
|
{"hello,there", []string{"hello", "there"}},
|
||
|
|
{"hello,,there", []string{"hello,there"}},
|
||
|
|
{",,hello", []string{",hello"}},
|
||
|
|
{"hello,,", []string{"hello,"}},
|
||
|
|
{"hello,,,there", []string{"hello,", "there"}},
|
||
|
|
{"hello,,,,there", []string{"hello,,there"}},
|
||
|
|
} {
|
||
|
|
require.Equal(t, testcase.expected, EscapableCommaSplit(testcase.input))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestParseDBMapping(t *testing.T) {
|
||
|
|
for _, testcase := range []struct {
|
||
|
|
input string
|
||
|
|
expected map[string]string
|
||
|
|
err error
|
||
|
|
}{
|
||
|
|
{"db://host", map[string]string{"": "db://host"}, nil},
|
||
|
|
{"db://host,override:db2://host2/db,,name",
|
||
|
|
map[string]string{"": "db://host", "override": "db2://host2/db,name"}, nil},
|
||
|
|
{"db://host,db2://host2", nil,
|
||
|
|
fmt.Errorf("invalid db mapping spec: %q", "db://host,db2://host2")},
|
||
|
|
} {
|
||
|
|
actual, err := ParseDBMapping(testcase.input)
|
||
|
|
if testcase.err != nil {
|
||
|
|
require.Equal(t, testcase.err, err)
|
||
|
|
} else {
|
||
|
|
require.Equal(t, testcase.expected, actual)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|