Add test

tests/integration/test_log_query_probability/test.py

@@ -0,0 +1,46 @@
+import pytest
+
+from helpers.cluster import ClickHouseCluster
+
+cluster = ClickHouseCluster(__file__, name="log_quries_probability")
+node1 = cluster.add_instance('node1', with_zookeeper=False)
+node2 = cluster.add_instance('node2', with_zookeeper=False)
+
+
+@pytest.fixture(scope="module")
+def start_cluster():
+    try:
+        cluster.start()
+        yield cluster
+
+    finally:
+        cluster.shutdown()
+
+
+def test_log_quries_probability_one(start_cluster):
+    for i in range(100):
+        node1.query("SELECT 12345", settings={"log_queries_probability":0.5})
+
+    node1.query("SYSTEM FLUSH LOGS")
+
+    assert node1.query("SELECT count() < (2 * 100) FROM system.query_log WHERE query LIKE '%12345%' AND query NOT LIKE '%system.query_log%'") == "1\n"
+    assert node1.query("SELECT count() > 0         FROM system.query_log WHERE query LIKE '%12345%' AND query NOT LIKE '%system.query_log%'") == "1\n"
+    assert node1.query("SELECT count() % 2         FROM system.query_log WHERE query LIKE '%12345%' AND query NOT LIKE '%system.query_log%'") == "0\n"
+
+    node1.query("TRUNCATE TABLE system.query_log")
+
+
+def test_log_quries_probability_two(start_cluster):
+    for i in range(100):
+        node1.query("SELECT 12345 FROM remote('node2', system, one)", settings={"log_queries_probability":0.5})
+
+        node1.query("SYSTEM FLUSH LOGS")
+        node2.query("SYSTEM FLUSH LOGS")
+
+        ans1 = node1.query("SELECT count() FROM system.query_log WHERE query LIKE '%12345%' AND query NOT LIKE '%system.query_log%'")
+        ans2 = node2.query("SELECT count() FROM system.query_log WHERE query LIKE '%12345%' AND query NOT LIKE '%system.query_log%'")
+
+        assert ans1 == ans2
+
+        node1.query("TRUNCATE TABLE system.query_log")
+        node2.query("TRUNCATE TABLE system.query_log")