db-migrator.go

1
package connection
2

3
import (
4
	"testing"
5

6
	"github.com/stretchr/testify/assert"
7
)
8

9
func TestNormalizeClickhouseDSN(t *testing.T) {
10
	var err error
11
	var actualDSN, originDSN, expectedDSN string
12

13
	for _, data := range getDataProvider() {
14
		originDSN, expectedDSN = data[0], data[1]
15
		actualDSN, err = NormalizeClickhouseDSN(originDSN)
16
		assert.NoError(t, err)
17
		assert.Equal(t, expectedDSN, actualDSN)
18
	}
19
}
20

21
func getDataProvider() [][]string {
22
	return [][]string{
23
		{
24
			"clickhouse://default:@clickhouse:9000/docker?sslmode=disable&compress=true&debug=false&cluster=test_cluster",
25
			"tcp://clickhouse:9000?username=default&password=&database=docker&sslmode=disable&compress=true&debug=false&cluster=test_cluster",
26
		},
27
		{
28
			"clickhouse://user:pass@clickhouse/docker?sslmode=disable&compress=true&debug=false&cluster=test_cluster",
29
			"tcp://clickhouse:9000?username=user&password=pass&database=docker&sslmode=disable&compress=true&debug=false&cluster=test_cluster",
30
		},
31
	}
32
}
33