From 8dad3f1b22a5f76893f01d0a2cd3dc6d5c6b2427 Mon Sep 17 00:00:00 2001
From: vdimir <vdimir@clickhouse.com>
Date: Mon, 8 Jan 2024 10:58:26 +0000
Subject: [PATCH] Add test 02962_max_joined_block_rows

---
 .../02962_max_joined_block_rows.reference     | 32 ++++++++++++++++
 .../02962_max_joined_block_rows.sql           | 38 +++++++++++++++++++
 2 files changed, 70 insertions(+)
 create mode 100644 tests/queries/0_stateless/02962_max_joined_block_rows.reference
 create mode 100644 tests/queries/0_stateless/02962_max_joined_block_rows.sql

diff --git a/tests/queries/0_stateless/02962_max_joined_block_rows.reference b/tests/queries/0_stateless/02962_max_joined_block_rows.reference
new file mode 100644
index 00000000000..8bc1bad225b
--- /dev/null
+++ b/tests/queries/0_stateless/02962_max_joined_block_rows.reference
@@ -0,0 +1,32 @@
+1	0
+1	1
+1	2
+1	3
+1	4
+1	5
+1	6
+1	7
+1	8
+1	9
+--
+1	0
+1	1
+1	2
+1	3
+1	4
+1	5
+1	6
+1	7
+1	8
+1	9
+--
+1	0
+1	1
+1	2
+1	3
+1	4
+1	5
+1	6
+1	7
+1	8
+1	9
diff --git a/tests/queries/0_stateless/02962_max_joined_block_rows.sql b/tests/queries/0_stateless/02962_max_joined_block_rows.sql
new file mode 100644
index 00000000000..c31ab5e1132
--- /dev/null
+++ b/tests/queries/0_stateless/02962_max_joined_block_rows.sql
@@ -0,0 +1,38 @@
+DROP TABLE IF EXISTS t1;
+DROP TABLE IF EXISTS t2;
+
+CREATE table t1 (a UInt64, b UInt64) ENGINE = Memory;
+INSERT INTO t1 SELECT number % 2, number FROM numbers(10);
+
+CREATE table t2 (a UInt64) ENGINE = Memory;
+
+INSERT INTO t2 SELECT number % 2 FROM numbers(10);
+
+-- block size is always multiple of 5 because we have 5 rows for each key in right table
+-- we do not split rows corresponding to the same key
+
+SELECT max(bs) <= 5, b FROM (
+    SELECT blockSize() as bs, * FROM t1 JOIN t2 ON t1.a = t2.a
+) GROUP BY b
+ORDER BY b
+SETTINGS max_joined_block_size_rows = 5;
+
+SELECT '--';
+
+SELECT max(bs) <= 10, b FROM (
+    SELECT blockSize() as bs, * FROM t1 JOIN t2 ON t1.a = t2.a
+) GROUP BY b
+ORDER BY b
+SETTINGS max_joined_block_size_rows = 10;
+
+SELECT '--';
+
+-- parallel_hash doen't support max_joined_block_size_rows
+
+SET join_algorithm = 'parallel_hash';
+
+SELECT max(bs) > 10, b FROM (
+    SELECT blockSize() as bs, * FROM t1 JOIN t2 ON t1.a = t2.a
+) GROUP BY b
+ORDER BY b
+SETTINGS max_joined_block_size_rows = 10;