tinybird-toolset 2.3.0__tar.gz → 2.4.0__tar.gz

{tinybird_toolset-2.3.0/src/tinybird_toolset.egg-info → tinybird_toolset-2.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tinybird-toolset
-Version: 2.3.0
+Version: 2.4.0
 Home-page: https://gitlab.com/tinybird/clickhouse-toolset
 Author: Tinybird.co
 Author-email: support@tinybird.co

{tinybird_toolset-2.3.0 → tinybird_toolset-2.4.0}/setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup, Extension
 
 NAME = 'tinybird-toolset'
-VERSION = '2.3.0'
+VERSION = '2.4.0'
 
 try:
     from conf import *

{tinybird_toolset-2.3.0 → tinybird_toolset-2.4.0/src/tinybird_toolset.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tinybird-toolset
-Version: 2.3.0
+Version: 2.4.0
 Home-page: https://gitlab.com/tinybird/clickhouse-toolset
 Author: Tinybird.co
 Author-email: support@tinybird.co

{tinybird_toolset-2.3.0 → tinybird_toolset-2.4.0}/tests/test_replace_tables.py

@@ -1144,6 +1144,25 @@ class TestDisabledFunctions(unittest.TestCase):
         expected = "SELECT * FROM (SELECT count() from system.numbers LIMIT 2) AS a, (SELECT count() from system.numbers LIMIT 2) AS a SETTINGS join_use_nulls = 1"
         self.assertEqual(chquery.format(expected), chquery.format(replaced))
 
+    def test_allow_analytics_query_settings(self):
+        sql = """SELECT * FROM a SETTINGS
+            max_threads = 1,
+            max_memory_usage = 1000000,
+            max_execution_time = 60,
+            use_query_condition_cache = 0
+        """
+        replacement = {("default", "a"): ("", "(SELECT count() FROM system.numbers LIMIT 1)")}
+        replaced = chquery.replace_tables(sql, replacement, default_database="default")
+        out = chquery.format(replaced)
+        self.assertIn("system.numbers", out)
+        for key in (
+            "max_threads",
+            "max_memory_usage",
+            "max_execution_time",
+            "use_query_condition_cache",
+        ):
+            self.assertIn(key, out)
+
     def test_disallow_enabled_settings(self):
         sql = "SELECT * FROM a, a SETTINGS join_use_nulls=1"
         replacement = {

{tinybird_toolset-2.3.0 → tinybird_toolset-2.4.0}/tests/test_tables.py

@@ -2025,6 +2025,23 @@ SELECT *
         self.assertEqual(chquery.tables("SELECT count() from system.numbers SETTINGS aggregate_functions_null_for_empty=1"),
                          [('system', 'numbers', '')])
 
+    def test_allow_analytics_query_settings(self):
+        sql = """SELECT count() FROM system.numbers LIMIT 1 SETTINGS
+            max_threads = 1,
+            max_memory_usage = 1000000,
+            max_execution_time = 60,
+            optimize_aggregation_in_order = 1,
+            use_skip_indexes = 1,
+            use_skip_indexes_if_final = 0,
+            optimize_move_to_prewhere = 1,
+            query_plan_optimize_lazy_materialization = 0,
+            min_bytes_to_use_direct_io = 0,
+            enable_filesystem_cache = 1,
+            use_query_cache = 0,
+            use_query_condition_cache = 0
+        """
+        self.assertEqual(chquery.tables(sql), [('system', 'numbers', '')])
+
     def test_disallow_blocked_settings(self):
         with self.assertRaisesRegex(ValueError, """DB::Exception: Usage of setting 'aggregate_functions_null_for_empty' is restricted. Contact support@tinybird.co if you require access to this feature"""):
             chquery.tables("SELECT count() from system.numbers SETTINGS aggregate_functions_null_for_empty=1",