Update test.py

ClickHouse · Feb 28, 2024 · b032624 · b032624
1 parent ffd69e0
commit b032624
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/integration/test_filesystem_cache/test.py b/tests/integration/test_filesystem_cache/test.py
@@ -94,12 +94,16 @@ def test_parallel_cache_loading_on_startup(cluster, node_name):
     cache_state = node.query(
         "SELECT key, file_segment_range_begin, size FROM system.filesystem_cache WHERE size > 0 ORDER BY key, file_segment_range_begin, size"
     )
+    keys = node.query(
+        "SELECT key FROM system.filesystem_cache WHERE size > 0 ORDER BY key, file_segment_range_begin, size"
+    ).strip().splitlines()
 
     node.restart_clickhouse()
 
-    assert cache_count == int(node.query("SELECT count() FROM system.filesystem_cache"))
+    # < because of additional files loaded into cache on server startup.
+    assert cache_count <= int(node.query("SELECT count() FROM system.filesystem_cache"))
     assert cache_state == node.query(
-        "SELECT key, file_segment_range_begin, size FROM system.filesystem_cache ORDER BY key, file_segment_range_begin, size"
+        f"SELECT key, file_segment_range_begin, size FROM system.filesystem_cache ORDER BY key, file_segment_range_begin, size WHERE key in ({",".join(("'" + x + "'" for x in keys))})"
     )
 
     assert node.contains_in_log("Loading filesystem cache with 30 threads")