From ed23125aee0df2fbecd8b5f1a107a9f3ca3cee7d Mon Sep 17 00:00:00 2001 From: avogar Date: Wed, 18 Sep 2024 19:37:10 +0000 Subject: [PATCH] Fix flaky tests --- tests/integration/test_storage_postgresql/test.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/integration/test_storage_postgresql/test.py b/tests/integration/test_storage_postgresql/test.py index cb7451e1cad..16ce4797778 100644 --- a/tests/integration/test_storage_postgresql/test.py +++ b/tests/integration/test_storage_postgresql/test.py @@ -526,6 +526,7 @@ def test_postgres_distributed(started_cluster): result = node2.query("SELECT DISTINCT(name) FROM test_shards ORDER BY name") started_cluster.unpause_container("postgres1") assert result == "host2\nhost4\n" or result == "host3\nhost4\n" + node2.query("DROP TABLE test_shards2") node2.query("DROP TABLE test_shards") node2.query("DROP TABLE test_replicas") @@ -817,6 +818,9 @@ def test_auto_close_connection(started_cluster): # Connection from python + pg_stat table also has a connection at the moment of current query assert count == 2 + node2.query("DROP TABLE test.stat") + node2.query("DROP TABLE test.test_table") + def test_literal_escaping(started_cluster): cursor = started_cluster.postgres_conn.cursor() @@ -832,6 +836,7 @@ def test_literal_escaping(started_cluster): node1.query("SELECT * FROM escaping WHERE text like '%a''a%'") # %a'a% -> %a''a% node1.query("SELECT * FROM escaping WHERE text like '%a\\'a%'") # %a'a% -> %a''a% cursor.execute(f"DROP TABLE escaping") + node1.query("DROP TABLE default.escaping") def test_filter_pushdown(started_cluster): @@ -886,6 +891,10 @@ def test_filter_pushdown(started_cluster): ) cursor.execute("DROP SCHEMA test_filter_pushdown CASCADE") + node1.query("DROP TABLE test_filter_pushdown_local_table") + node1.query("DROP TABLE test_filter_pushdown_pg_table") + + def test_fixed_string_type(started_cluster):