storj/private/dbutil/mapping_test.go
JT Olio 0ba516d405 satellite: support pointing db components at different databases
the immediate need is to be able to move the repair queue back out
of cockroach if we can't save it.

Change-Id: If26001a4e6804f6bb8713b4aee7e4fd6254dc326
2020-11-28 18:39:16 +00:00

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)
}
}
}