From 8d1318fbe41ae456dc4afaef4a3dcde9b0a35986 Mon Sep 17 00:00:00 2001 From: Grigoriy Pisarenko Date: Mon, 12 Feb 2024 10:56:08 +0000 Subject: [PATCH] Fixed insert restarts tests --- .github/config/muted_ya.txt | 1 - .../yql/dq/actors/task_runner/task_runner_actor_local.cpp | 1 - ydb/tests/fq/restarts/test_insert_restarts.py | 1 + 3 files changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/config/muted_ya.txt b/.github/config/muted_ya.txt index 9a2bed82b377..e23f2b5c87c3 100644 --- a/.github/config/muted_ya.txt +++ b/.github/config/muted_ya.txt @@ -81,7 +81,6 @@ ydb/tests/fq/generic * ydb/tests/fq/mem_alloc * ydb/tests/fq/multi_plane * ydb/tests/fq/plans test_stats_mode.py.TestStatsMode.test_mode[v1-STATS_MODE_FULL-client0] -ydb/tests/fq/restarts test_insert_restarts.py.TestS3.test_atomic_upload_commit* ydb/tests/fq/s3 * ydb/tests/fq/yds * ydb/tests/functional/audit * diff --git a/ydb/library/yql/dq/actors/task_runner/task_runner_actor_local.cpp b/ydb/library/yql/dq/actors/task_runner/task_runner_actor_local.cpp index 2cb8e72eeb0b..47b80e7999d7 100644 --- a/ydb/library/yql/dq/actors/task_runner/task_runner_actor_local.cpp +++ b/ydb/library/yql/dq/actors/task_runner/task_runner_actor_local.cpp @@ -123,7 +123,6 @@ class TLocalTaskRunnerActor if (MemoryQuota) { MemoryQuota->TryReleaseQuota(); } - TaskRunner.Reset(); TActor::PassAway(); } diff --git a/ydb/tests/fq/restarts/test_insert_restarts.py b/ydb/tests/fq/restarts/test_insert_restarts.py index 287d2ae24175..43abc891775a 100644 --- a/ydb/tests/fq/restarts/test_insert_restarts.py +++ b/ydb/tests/fq/restarts/test_insert_restarts.py @@ -93,6 +93,7 @@ def test_atomic_upload_commit(self, kikimr, s3, client): assert time.time() < deadline, f"Insert not finished for already {time.time() - start} seconds" time.sleep(0.001) + kikimr.compute_plane.wait_bootstrap() client.wait_query(query_id, statuses=[fq.QueryMeta.COMPLETED, fq.QueryMeta.ABORTED_BY_SYSTEM, fq.QueryMeta.FAILED], timeout=timeout) final_number_rows = self.run_atomic_upload_check_query(client, bucket, "ibucket", "insert/", "csv_with_names")