add test for sugar.DSN

Author: asmyasnikov

internal/dsn/dsn_test.go

@@ -31,34 +31,34 @@ func TestParseConnectionString(t *testing.T) {
 		error            error
 	}{
 		{
-			"grpc://ydb-ru.yandex.xnet:2135/?" +
+			"grpc://ydb-ru.yandex.net:2135/?" +
 				"database=/ru/home/gvit/mydb&token=123",
 			false,
-			"ydb-ru.yandex.xnet:2135",
+			"ydb-ru.yandex.net:2135",
 			"/ru/home/gvit/mydb",
 			"123",
 			nil,
 		},
 		{
-			"grpcs://ydb.serverless.yandexcloud.xnet:2135/?" +
+			"grpcs://ydb.serverless.yandexcloud.net:2135/?" +
 				"database=/ru-central1/b1g8skpblkos03malf3s/etn02qso4v3isjb00te1&token=123",
 			true,
-			"ydb.serverless.yandexcloud.xnet:2135",
+			"ydb.serverless.yandexcloud.net:2135",
 			"/ru-central1/b1g8skpblkos03malf3s/etn02qso4v3isjb00te1",
 			"123",
 			nil,
 		},
 		{
-			"grpcs://lb.etn03r9df42nb631unbv.ydb.mdb.yandexcloud.xnet:2135/?" +
+			"grpcs://lb.etn03r9df42nb631unbv.ydb.mdb.yandexcloud.net:2135/?" +
 				"database=/ru-central1/b1g8skpblkos03malf3s/etn03r9df42nb631unbv&token=123",
 			true,
-			"lb.etn03r9df42nb631unbv.ydb.mdb.yandexcloud.xnet:2135",
+			"lb.etn03r9df42nb631unbv.ydb.mdb.yandexcloud.net:2135",
 			"/ru-central1/b1g8skpblkos03malf3s/etn03r9df42nb631unbv",
 			"123",
 			nil,
 		},
 		{
-			"abcd://ydb-ru.yandex.xnet:2135/?database=/ru/home/gvit/mydb",
+			"abcd://ydb-ru.yandex.net:2135/?database=/ru/home/gvit/mydb",
 			true,
 			"",
 			"",

sugar/dsn_test.go

@@ -0,0 +1,47 @@
+package sugar
+
+import (
+	"fmt"
+	"testing"
+)
+
+func TestDSN(t *testing.T) {
+	for _, test := range []struct {
+		endpoint string
+		database string
+		secure   bool
+		dsn      string
+	}{
+		{
+			"localhost:2135",
+			"/local",
+			false,
+			"grpc://localhost:2135?database=%2Flocal",
+		},
+		{
+			"ydb-ru.yandex.net:2135",
+			"/ru/home/gvit/mydb",
+			false,
+			"grpc://ydb-ru.yandex.net:2135?database=%2Fru%2Fhome%2Fgvit%2Fmydb",
+		},
+		{
+			"ydb.serverless.yandexcloud.net:2135",
+			"/ru-central1/b1g8skpblkos03malf3s/etn02qso4v3isjb00te1",
+			true,
+			"grpcs://ydb.serverless.yandexcloud.net:2135?database=%2Fru-central1%2Fb1g8skpblkos03malf3s%2Fetn02qso4v3isjb00te1",
+		},
+		{
+			"lb.etn03r9df42nb631unbv.ydb.mdb.yandexcloud.net:2135",
+			"/ru-central1/b1g8skpblkos03malf3s/etn03r9df42nb631unbv",
+			true,
+			"grpcs://lb.etn03r9df42nb631unbv.ydb.mdb.yandexcloud.net:2135?database=%2Fru-central1%2Fb1g8skpblkos03malf3s%2Fetn03r9df42nb631unbv",
+		},
+	} {
+		t.Run(test.dsn, func(t *testing.T) {
+			dsn := DSN(test.endpoint, test.database, test.secure)
+			if dsn != test.dsn {
+				t.Fatal(fmt.Sprintf("Unexpected result: %s, exp: %s", dsn, test.dsn))
+			}
+		})
+	}
+}