Merge pull request #340 from ydb-platform/repeater

fix bug with nil repeater + minor fixes

Author: asmyasnikov

connection_e2e_test.go

@@ -25,7 +25,6 @@ import (
 	"google.golang.org/protobuf/types/known/durationpb"
 
 	"github.com/ydb-platform/ydb-go-sdk/v3"
-	"github.com/ydb-platform/ydb-go-sdk/v3/balancers"
 	"github.com/ydb-platform/ydb-go-sdk/v3/config"
 	"github.com/ydb-platform/ydb-go-sdk/v3/internal/meta"
 	"github.com/ydb-platform/ydb-go-sdk/v3/log"
@@ -72,7 +71,6 @@ func TestConnection(t *testing.T) {
 			config.WithOperationTimeout(time.Second*2),
 			config.WithOperationCancelAfter(time.Second*2),
 		),
-		ydb.WithBalancer(balancers.SingleConn()),
 		ydb.WithConnectionTTL(time.Millisecond*10000),
 		ydb.WithMinTLSVersion(tls.VersionTLS10),
 		ydb.WithLogger(
@@ -117,7 +115,7 @@ func TestConnection(t *testing.T) {
 	defer func() {
 		// cleanup connection
 		if e := db.Close(ctx); e != nil {
-			t.Fatalf("balancer close failed: %+v", e)
+			t.Fatalf("close failed: %+v", e)
 		}
 	}()
 	t.Run("discovery.WhoAmI", func(t *testing.T) {

internal/balancer/balancer.go

@@ -287,7 +287,7 @@ func (b *balancer) getConn(ctx context.Context) (c conn.Conn, err error) {
 	)
 
 	defer func() {
-		if failedCount*2 > state.PreferredCount() {
+		if failedCount*2 > state.PreferredCount() && b.discoveryRepeater != nil {
 			b.discoveryRepeater.Force()
 		}
 	}()

topic/client_e2e_test.go

@@ -46,7 +46,7 @@ func connect(t testing.TB) ydb.Connection {
 	}
 	db, err := ydb.Open(context.Background(), connectionString,
 		ydb.WithDialTimeout(time.Second),
-		ydb.WithAccessTokenCredentials(os.Getenv("YDB_TOKEN")),
+		ydb.WithAccessTokenCredentials(os.Getenv("YDB_ACCESS_TOKEN_CREDENTIALS")),
 	)
 	require.NoError(t, err)
 	return db