Merge pull request #1152 from ydb-platform/parking-integration-test

revert integration tests

Author: asmyasnikov

tests/integration/discovery_test.go

@@ -46,7 +46,8 @@ func TestDiscovery(t *testing.T) {
 				t.Fatalf("unknown request type: %s", requestTypes[0])
 			}
 		}
-		ctx = xtest.Context(t)
+		parking = make(chan struct{})
+		ctx     = xtest.Context(t)
 	)
 
 	db, err := ydb.Open(ctx,
@@ -59,6 +60,7 @@ func TestDiscovery(t *testing.T) {
 			config.WithOperationCancelAfter(time.Second*2),
 		),
 		ydb.WithBalancer(balancers.SingleConn()),
+		ydb.WithConnectionTTL(time.Second*1),
 		ydb.WithMinTLSVersion(tls.VersionTLS10),
 		ydb.WithLogger(
 			newLoggerWithMinLevel(t, log.WARN),
@@ -92,6 +94,13 @@ func TestDiscovery(t *testing.T) {
 				}),
 			),
 		),
+		ydb.WithTraceDriver(trace.Driver{
+			OnConnPark: func(info trace.DriverConnParkStartInfo) func(trace.DriverConnParkDoneInfo) {
+				return func(info trace.DriverConnParkDoneInfo) {
+					parking <- struct{}{}
+				}
+			},
+		}),
 	)
 	if err != nil {
 		t.Fatal(err)
@@ -108,5 +117,17 @@ func TestDiscovery(t *testing.T) {
 		} else {
 			t.Log(endpoints)
 		}
+		t.Run("wait", func(t *testing.T) {
+			t.Run("parking", func(t *testing.T) {
+				<-parking // wait for parking conn
+				t.Run("re-discover", func(t *testing.T) {
+					if endpoints, err := db.Discovery().Discover(ctx); err != nil {
+						t.Fatal(err)
+					} else {
+						t.Log(endpoints)
+					}
+				})
+			})
+		})
 	})
 }

tests/integration/helpers_test.go

@@ -85,9 +85,9 @@ func (scope *scopeT) Driver(opts ...ydb.Option) *ydb.Driver {
 
 		token := scope.AuthToken()
 		if token == "" {
-			scope.Logf("Change empty auth token")
+			scope.Logf("With empty auth token")
 		} else {
-			scope.Logf("Change auth token")
+			scope.Logf("With auth token")
 		}
 
 		connectionContext, cancel := context.WithTimeout(scope.Ctx, time.Second*10)

tests/integration/query_execute_test.go

@@ -31,6 +31,7 @@ func TestQueryExecute(t *testing.T) {
 	db, err := ydb.Open(ctx,
 		os.Getenv("YDB_CONNECTION_STRING"),
 		ydb.WithAccessTokenCredentials(os.Getenv("YDB_ACCESS_TOKEN_CREDENTIALS")),
+		ydb.WithSessionPoolSizeLimit(10),
 		ydb.WithTraceQuery(
 			log.Query(
 				log.Default(os.Stdout,
@@ -43,6 +44,11 @@ func TestQueryExecute(t *testing.T) {
 		),
 	)
 	require.NoError(t, err)
+	t.Run("Stats", func(t *testing.T) {
+		s, err := query.Stats(db.Query())
+		require.NoError(t, err)
+		require.EqualValues(t, 10, s.Limit)
+	})
 	t.Run("Scan", func(t *testing.T) {
 		var (
 			p1 string

tests/integration/topic_grpc_stopper_helper_test.go

@@ -19,8 +19,8 @@ import (
 //
 //		db, err := ydb.Open(context.Background(), connectionString,
 //	     ...
-//			ydb.Change(config.WithGrpcOptions(grpc.WithChainUnaryInterceptor(grpcStopper.UnaryClientInterceptor)),
-//			ydb.Change(config.WithGrpcOptions(grpc.WithStreamInterceptor(grpcStopper.StreamClientInterceptor)),
+//			ydb.With(config.WithGrpcOptions(grpc.WithChainUnaryInterceptor(grpcStopper.UnaryClientInterceptor)),
+//			ydb.With(config.WithGrpcOptions(grpc.WithStreamInterceptor(grpcStopper.StreamClientInterceptor)),
 //		),
 //
 //		grpcStopper.Stop(errors.New("test error"))