From 73f71cc9fc872941442d3615e5df044005ca24e7 Mon Sep 17 00:00:00 2001
From: Bharat Nallan Chakravarthy <bharatnc@gmail.com>
Date: Mon, 18 Dec 2023 11:50:54 -0800
Subject: [PATCH] add a test

---
 .../__init__.py                               |  0
 .../asynchronous_metrics_update_period_s.xml  |  3 +
 .../test.py                                   | 57 +++++++++++++++++++
 3 files changed, 60 insertions(+)
 create mode 100644 tests/integration/test_asynchronous_metrics_pk_bytes_fields/__init__.py
 create mode 100644 tests/integration/test_asynchronous_metrics_pk_bytes_fields/configs/asynchronous_metrics_update_period_s.xml
 create mode 100644 tests/integration/test_asynchronous_metrics_pk_bytes_fields/test.py

diff --git a/tests/integration/test_asynchronous_metrics_pk_bytes_fields/__init__.py b/tests/integration/test_asynchronous_metrics_pk_bytes_fields/__init__.py
new file mode 100644
index 00000000000..e69de29bb2d
diff --git a/tests/integration/test_asynchronous_metrics_pk_bytes_fields/configs/asynchronous_metrics_update_period_s.xml b/tests/integration/test_asynchronous_metrics_pk_bytes_fields/configs/asynchronous_metrics_update_period_s.xml
new file mode 100644
index 00000000000..47e88730482
--- /dev/null
+++ b/tests/integration/test_asynchronous_metrics_pk_bytes_fields/configs/asynchronous_metrics_update_period_s.xml
@@ -0,0 +1,3 @@
+<clickhouse>
+    <asynchronous_metrics_update_period_s>1</asynchronous_metrics_update_period_s>
+</clickhouse>
diff --git a/tests/integration/test_asynchronous_metrics_pk_bytes_fields/test.py b/tests/integration/test_asynchronous_metrics_pk_bytes_fields/test.py
new file mode 100644
index 00000000000..23123daf805
--- /dev/null
+++ b/tests/integration/test_asynchronous_metrics_pk_bytes_fields/test.py
@@ -0,0 +1,57 @@
+import time
+
+import pytest
+from helpers.cluster import ClickHouseCluster
+
+cluster = ClickHouseCluster(__file__)
+node1 = cluster.add_instance(
+    "node1",
+    with_zookeeper=True,
+    main_configs=["configs/asynchronous_metrics_update_period_s.xml"],
+)
+
+
+@pytest.fixture(scope="module")
+def started_cluster():
+    try:
+        cluster.start()
+        yield cluster
+
+    finally:
+        cluster.shutdown()
+
+
+def test_total_pk_bytes_in_memory_fields(started_cluster):
+    try:
+        cluster.start()
+        node1.query("SET log_queries = 1;")
+        node1.query("CREATE DATABASE replica;")
+        query_create = """CREATE TABLE replica.test
+        (
+           id Int64,
+           event_time DateTime
+        )
+        Engine=MergeTree()
+        PARTITION BY toYYYYMMDD(event_time)
+        ORDER BY id;"""
+        time.sleep(2)
+        node1.query(query_create)
+        node1.query("""INSERT INTO replica.test VALUES (1, now())""")
+        node1.query("SYSTEM FLUSH LOGS;")
+
+        # query system.asynchronous_metrics
+        test_query = "SELECT count() > 0 ? 'ok' : 'fail' FROM system.asynchronous_metrics WHERE metric='TotalPrimaryKeyBytesInMemory';"
+        assert "ok\n" in node1.query(test_query)
+
+        test_query = "SELECT count() > 0 ? 'ok' : 'fail' FROM system.asynchronous_metrics WHERE metric='TotalPrimaryKeyBytesInMemoryAllocated';"
+        assert "ok\n" in node1.query(test_query)
+
+        # query system.asynchronous_metric_log
+        test_query = "SELECT count() > 0 ? 'ok' : 'fail' FROM system.asynchronous_metric_log WHERE metric='TotalPrimaryKeyBytesInMemory';"
+        assert "ok\n" in node1.query(test_query)
+
+        test_query = "SELECT count() > 0 ? 'ok' : 'fail' FROM system.asynchronous_metric_log WHERE metric='TotalPrimaryKeyBytesInMemoryAllocated';"
+        assert "ok\n" in node1.query(test_query)
+
+    finally:
+        cluster.shutdown()