Fix test and unmute (#10021)

nshestakov · web-flow · commit a4dca6dd1ca3 · 2024-10-04T12:37:02.000+03:00
diff --git a/.github/config/muted_ya.txt b/.github/config/muted_ya.txt
@@ -31,17 +31,6 @@ ydb/core/kqp/ut/service KqpQueryService.QueryOnClosedSession
 ydb/core/kqp/ut/service KqpService.CloseSessionsWithLoad
 ydb/core/persqueue/ut [*/*]*
 ydb/core/persqueue/ut TPQTest.*DirectRead*
-ydb/core/persqueue/ut/ut_with_sdk [*/*]*
-ydb/core/persqueue/ut/ut_with_sdk TopicAutoscaling.PartitionMerge_PreferedPartition_BeforeAutoscaleAwareSDK
-ydb/core/persqueue/ut/ut_with_sdk TopicAutoscaling.PartitionSplit_ManySession_AutoscaleAwareSDK
-ydb/core/persqueue/ut/ut_with_sdk TopicAutoscaling.PartitionSplit_ManySession_BeforeAutoscaleAwareSDK
-ydb/core/persqueue/ut/ut_with_sdk TopicAutoscaling.PartitionSplit_ManySession_existed_AutoscaleAwareSDK
-ydb/core/persqueue/ut/ut_with_sdk TopicAutoscaling.PartitionSplit_PreferedPartition_AutoscaleAwareSDK
-ydb/core/persqueue/ut/ut_with_sdk TopicAutoscaling.PartitionSplit_ReadNotEmptyPartitions_AutoscaleAwareSDK
-ydb/core/persqueue/ut/ut_with_sdk TopicAutoscaling.PartitionSplit_ReadNotEmptyPartitions_BeforeAutoscaleAwareSDK
-ydb/core/persqueue/ut/ut_with_sdk TopicAutoscaling.ReadingAfterSplitTest_AutoscaleAwareSDK_AutoCommit
-ydb/core/persqueue/ut/ut_with_sdk TopicAutoscaling.ReadingAfterSplitTest_PreferedPartition_AutoscaleAwareSDK
-ydb/core/persqueue/ut/ut_with_sdk TopicAutoscaling.ReadingAfterSplitTest_PreferedPartition_BeforeAutoscaleAwareSDK
 ydb/core/quoter/ut QuoterWithKesusTest.PrefetchCoefficient
 ydb/core/tx/coordinator/ut Coordinator.RestoreTenantConfiguration
 ydb/core/tx/datashard/ut_change_exchange Cdc.InitialScanDebezium
diff --git a/ydb/core/persqueue/ut/common/autoscaling_ut_common.cpp b/ydb/core/persqueue/ut/common/autoscaling_ut_common.cpp
@@ -520,7 +520,7 @@ template<SdkVersion Sdk>
 void TTestReadSession<Sdk>::Close() {
     Run();
     Cerr << ">>>>> " << Impl->Name << " Closing reading session " << Endl << Flush;
-    Session->Close();
+    Session->Close(TDuration::Seconds(5));
     Session.reset();
 }
 
diff --git a/ydb/core/persqueue/ut/ut_with_sdk/balancing_ut.cpp b/ydb/core/persqueue/ut/ut_with_sdk/balancing_ut.cpp
@@ -143,6 +143,9 @@ Y_UNIT_TEST_SUITE(Balancing) {
             UNIT_ASSERT_VALUES_EQUAL(5, p[TEST_TOPIC].size());
             UNIT_ASSERT_VALUES_EQUAL(5, p["other-test-topic"].size());
         }
+
+        readSession0->Close();
+        readSession1->Close();
     }
 
     Y_UNIT_TEST(Balancing_ManyTopics_TopicApi) {

Original file line number	Diff line number	Diff line change
`@@ -520,7 +520,7 @@ template<SdkVersion Sdk>`
`520`	`520`	`void TTestReadSession<Sdk>::Close() {`
`521`	`521`	`Run();`
`522`	`522`	`Cerr << ">>>>> " << Impl->Name << " Closing reading session " << Endl << Flush;`
`523`		`- Session->Close();`
	`523`	`+ Session->Close(TDuration::Seconds(5));`
`524`	`524`	`Session.reset();`
`525`	`525`	`}`
`526`	`526`
Original file line number	Diff line number	Diff line change
`@@ -143,6 +143,9 @@ Y_UNIT_TEST_SUITE(Balancing) {`
`143`	`143`	`UNIT_ASSERT_VALUES_EQUAL(5, p[TEST_TOPIC].size());`
`144`	`144`	`UNIT_ASSERT_VALUES_EQUAL(5, p["other-test-topic"].size());`
`145`	`145`	`}`
	`146`	`+`
	`147`	`+ readSession0->Close();`
	`148`	`+ readSession1->Close();`
`146`	`149`	`}`
`147`	`150`
`148`	`151`	`Y_UNIT_TEST(Balancing_ManyTopics_TopicApi) {`