override test sql macro and add limit-placer macro

dbt/include/clickhouse/macros/utils/utils.sql

@@ -1,3 +1,33 @@
+{% macro clickhouse__get_test_sql(main_sql, fail_calc, warn_if, error_if, limit) -%}
+    {% set main_sql_formatted = clickhouse__place_limit(main_sql, limit) if limit !=None else main_sql%}
+    select
+      {{ fail_calc }} as failures,
+      {{ fail_calc }} {{ warn_if }} as should_warn,
+      {{ fail_calc }} {{ error_if }} as should_error
+    from (
+      {{ main_sql_formatted }}
+    ) dbt_internal_test
+
+{%- endmacro %}
+
+
+-- This macro is designed to add a LIMIT clause to a ClickHouse SQL query while preserving any ClickHouse settings specified in the query.
+-- When multiple queries are nested, the limit will be attached to the outer query
+{% macro clickhouse__place_limit(query, limit) -%}
+   {% if 'settings' in query.lower()%}
+        {% if '-- settings_section' not in query.lower()%}
+            {{exceptions.raise_compiler_error("-- settings_section must be set when using ClickHouse settings")}}
+        {% endif %}
+        {% set split_by_settings_sections = query.split("-- settings_section")%}
+        {% set split_by_settings_sections_with_limit = split_by_settings_sections[-2] + "\n LIMIT " + limit|string  + "\n" %}
+        {% set query_with_limit = "-- settings_section".join(split_by_settings_sections[:-2] + [split_by_settings_sections_with_limit, split_by_settings_sections[-1]])%}
+        {{query_with_limit}}
+    {% else %}
+    {{query}}
+    {{"limit " ~ limit}}
+    {% endif %}
+{%- endmacro %}
+
 {% macro clickhouse__any_value(expression) -%}
     any({{ expression }})
 {%- endmacro %}