add TestRegister

Author: asmyasnikov

internal/dsn/dsn_test.go

@@ -1,100 +1,81 @@
 package dsn
 
 import (
-	"context"
+	"strconv"
 	"testing"
 
+	"github.com/stretchr/testify/require"
+
 	"github.com/ydb-platform/ydb-go-sdk/v3/config"
-	"github.com/ydb-platform/ydb-go-sdk/v3/internal/credentials"
 	"github.com/ydb-platform/ydb-go-sdk/v3/testutil"
 )
 
-func init() {
-	_ = Register("token", func(token string) ([]config.Option, error) {
-		return []config.Option{
-			config.WithCredentials(
-				credentials.NewAccessTokenCredentials(token, ""),
-			),
-		}, nil
-	})
-}
-
 func TestParseConnectionString(t *testing.T) {
 	for _, test := range []struct {
 		connectionString string
 		secure           bool
 		endpoint         string
 		database         string
-		token            string
 	}{
 		{
 			"grpc://ydb-ru.yandex.net:2135/?" +
-				"database=/ru/home/gvit/mydb&token=123",
+				"database=/ru/home/gvit/mydb",
 			false,
 			"ydb-ru.yandex.net:2135",
 			"/ru/home/gvit/mydb",
-			"123",
 		},
 		{
-			"grpc://ydb-ru.yandex.net:2135/ru/home/gvit/mydb?token=123",
+			"grpc://ydb-ru.yandex.net:2135/ru/home/gvit/mydb",
 			false,
 			"ydb-ru.yandex.net:2135",
 			"/ru/home/gvit/mydb",
-			"123",
 		},
 		{
 			"grpcs://ydb.serverless.yandexcloud.net:2135/?" +
-				"database=/ru-central1/b1g8skpblkos03malf3s/etn02qso4v3isjb00te1&token=123",
+				"database=/ru-central1/b1g8skpblkos03malf3s/etn02qso4v3isjb00te1",
 			true,
 			"ydb.serverless.yandexcloud.net:2135",
 			"/ru-central1/b1g8skpblkos03malf3s/etn02qso4v3isjb00te1",
-			"123",
 		},
 		{
 			"grpcs://ydb.serverless.yandexcloud.net:2135" +
-				"/ru-central1/b1g8skpblkos03malf3s/etn02qso4v3isjb00te1?token=123",
+				"/ru-central1/b1g8skpblkos03malf3s/etn02qso4v3isjb00te1",
 			true,
 			"ydb.serverless.yandexcloud.net:2135",
 			"/ru-central1/b1g8skpblkos03malf3s/etn02qso4v3isjb00te1",
-			"123",
 		},
 		{
 			"grpcs://ydb.serverless.yandexcloud.net:2135" +
-				"/ru-central1/b1g8skpblkos03malf3s/etn02qso4v3isjb00te1?database=/ru/home/gvit/mydb&token=123",
+				"/ru-central1/b1g8skpblkos03malf3s/etn02qso4v3isjb00te1?database=/ru/home/gvit/mydb",
 			true,
 			"ydb.serverless.yandexcloud.net:2135",
 			"/ru/home/gvit/mydb",
-			"123",
 		},
 		{
 			"grpcs://lb.etn03r9df42nb631unbv.ydb.mdb.yandexcloud.net:2135/?" +
-				"database=/ru-central1/b1g8skpblkos03malf3s/etn03r9df42nb631unbv&token=123",
+				"database=/ru-central1/b1g8skpblkos03malf3s/etn03r9df42nb631unbv",
 			true,
 			"lb.etn03r9df42nb631unbv.ydb.mdb.yandexcloud.net:2135",
 			"/ru-central1/b1g8skpblkos03malf3s/etn03r9df42nb631unbv",
-			"123",
 		},
 		{
 			"grpcs://lb.etn03r9df42nb631unbv.ydb.mdb.yandexcloud.net:2135" +
-				"/ru-central1/b1g8skpblkos03malf3s/etn03r9df42nb631unbv?token=123",
+				"/ru-central1/b1g8skpblkos03malf3s/etn03r9df42nb631unbv",
 			true,
 			"lb.etn03r9df42nb631unbv.ydb.mdb.yandexcloud.net:2135",
 			"/ru-central1/b1g8skpblkos03malf3s/etn03r9df42nb631unbv",
-			"123",
 		},
 		{
 			"abcd://ydb-ru.yandex.net:2135/?database=/ru/home/gvit/mydb",
 			true,
 			"ydb-ru.yandex.net:2135",
 			"/ru/home/gvit/mydb",
-			"",
 		},
 		{
 			"abcd://ydb-ru.yandex.net:2135/ru/home/gvit/mydb",
 			true,
 			"ydb-ru.yandex.net:2135",
 			"/ru/home/gvit/mydb",
-			"",
 		},
 	} {
 		t.Run(test.connectionString, func(t *testing.T) {
@@ -106,16 +87,29 @@ func TestParseConnectionString(t *testing.T) {
 			testutil.Equal(t, test.secure, config.Secure())
 			testutil.Equal(t, test.endpoint, config.Endpoint())
 			testutil.Equal(t, test.database, config.Database())
-			var token string
-			if credentials := config.Credentials(); credentials != nil {
-				token, err = credentials.Token(context.Background())
-				if err != nil {
-					t.Fatalf("Received unexpected error:\n%+v", err)
-				}
-			} else {
-				token = ""
-			}
-			testutil.Equal(t, test.token, token)
 		})
 	}
 }
+
+func TestRegister(t *testing.T) {
+	var test1, test2, test3 int
+	_ = Register("test1", func(value string) (_ []config.Option, err error) {
+		test1, err = strconv.Atoi(value)
+		if err != nil {
+			return nil, err
+		}
+		return []config.Option{}, nil
+	})
+	_ = Register("test2", func(value string) (_ []config.Option, err error) {
+		test2, err = strconv.Atoi(value)
+		if err != nil {
+			return nil, err
+		}
+		return []config.Option{}, nil
+	})
+	_, err := Parse("grpc://ydb-ru.yandex.net:2135/ru/home/gvit/mydb?test1=1&test2=2&test3=3")
+	require.NoError(t, err, "")
+	require.Equal(t, 1, test1, "")
+	require.Equal(t, 2, test2, "")
+	require.NotEqualf(t, 3, test3, "")
+}