From a5b95a6b427ac9deeca4062b2fbbc107933a265c Mon Sep 17 00:00:00 2001 From: "shane.mazur" Date: Fri, 7 Feb 2025 13:17:22 -0500 Subject: [PATCH] Add end to end test for auto liquid clustering with optimize post hook --- .../adapter/liquid_clustering/fixtures.py | 5 +++++ .../liquid_clustering/test_liquid_clustering.py | 13 +++++++++++++ 2 files changed, 18 insertions(+) diff --git a/tests/functional/adapter/liquid_clustering/fixtures.py b/tests/functional/adapter/liquid_clustering/fixtures.py index 951210ea6..6c7302227 100644 --- a/tests/functional/adapter/liquid_clustering/fixtures.py +++ b/tests/functional/adapter/liquid_clustering/fixtures.py @@ -2,3 +2,8 @@ {{ config(materialized='incremental', liquid_clustered_by='id') }} select 1 as id, 'Joe' as name """ + +auto_liquid_cluster_sql = """ +{{ config(materialized='incremental', auto_liquid_cluster=true) }} +select 1 as id, 'Joe' as name +""" diff --git a/tests/functional/adapter/liquid_clustering/test_liquid_clustering.py b/tests/functional/adapter/liquid_clustering/test_liquid_clustering.py index a9cc0ee09..45c7dfe2c 100644 --- a/tests/functional/adapter/liquid_clustering/test_liquid_clustering.py +++ b/tests/functional/adapter/liquid_clustering/test_liquid_clustering.py @@ -15,3 +15,16 @@ def models(self): def test_liquid_clustering(self, project): _, logs = util.run_dbt_and_capture(["--debug", "run"]) assert "optimize" in logs + + +class TestAutoLiquidClustering: + @pytest.fixture(scope="class") + def models(self): + return { + "liquid_clustering.sql": fixtures.liquid_cluster_sql, + } + + @pytest.mark.skip_profile("databricks_uc_cluster", "databricks_cluster") + def test_liquid_clustering(self, project): + _, logs = util.run_dbt_and_capture(["--debug", "run"]) + assert "optimize" in logs