From d5443293a3cfa26793429b63ab58ca21e195d4b5 Mon Sep 17 00:00:00 2001
From: Alexey Milovidov <milovidov@yandex-team.ru>
Date: Mon, 8 Jun 2020 01:49:26 +0300
Subject: [PATCH] Added a test

---
 ...regate_state_exception_memory_leak.reference |  2 ++
 ...301_aggregate_state_exception_memory_leak.sh | 17 +++++++++++++++++
 2 files changed, 19 insertions(+)
 create mode 100644 tests/queries/0_stateless/01301_aggregate_state_exception_memory_leak.reference
 create mode 100755 tests/queries/0_stateless/01301_aggregate_state_exception_memory_leak.sh

diff --git a/tests/queries/0_stateless/01301_aggregate_state_exception_memory_leak.reference b/tests/queries/0_stateless/01301_aggregate_state_exception_memory_leak.reference
new file mode 100644
index 00000000000..b20e7415f52
--- /dev/null
+++ b/tests/queries/0_stateless/01301_aggregate_state_exception_memory_leak.reference
@@ -0,0 +1,2 @@
+Memory limit (for query) exceeded
+Ok
diff --git a/tests/queries/0_stateless/01301_aggregate_state_exception_memory_leak.sh b/tests/queries/0_stateless/01301_aggregate_state_exception_memory_leak.sh
new file mode 100755
index 00000000000..633fa5ce315
--- /dev/null
+++ b/tests/queries/0_stateless/01301_aggregate_state_exception_memory_leak.sh
@@ -0,0 +1,17 @@
+#!/usr/bin/env bash
+
+CURDIR=$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)
+. $CURDIR/../shell_config.sh
+
+function test()
+{
+    for i in {1..1000}; do
+        $CLICKHOUSE_CLIENT --max_memory_usage 1G <<< "SELECT uniqExactState(number) FROM system.numbers_mt GROUP BY number % 10";
+    done
+}
+
+export -f test;
+
+# If the memory leak exists, it will lead to OOM fairly quickly.
+timeout 30 bash -c test 2>&1 | grep -o -F 'Memory limit (for query) exceeded' | uniq
+echo 'Ok'