From ade2378f6af5bf652c6c3282e4ea41a93c9490cc Mon Sep 17 00:00:00 2001
From: Michelle Ark <michelle.ark@dbtlabs.com>
Date: Fri, 12 Apr 2024 14:50:51 -0700
Subject: [PATCH] add test for unit test that depends on ephemeral model

---
 .../unit_testing/test_ut_ephemeral.py         | 84 +++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 tests/functional/unit_testing/test_ut_ephemeral.py

diff --git a/tests/functional/unit_testing/test_ut_ephemeral.py b/tests/functional/unit_testing/test_ut_ephemeral.py
new file mode 100644
index 00000000000..2898633ec40
--- /dev/null
+++ b/tests/functional/unit_testing/test_ut_ephemeral.py
@@ -0,0 +1,84 @@
+import pytest
+from dbt.tests.util import run_dbt, write_file
+from dbt.contracts.results import RunStatus, TestStatus
+
+
+ephemeral_model_sql = """
+{{ config(materialized="ephemeral") }}
+select 1 as id, 'Emily' as first_name
+"""
+
+nested_ephemeral_model_sql = """
+{{ config(materialized="ephemeral") }}
+select * from {{ ref('ephemeral_model') }}
+"""
+
+customers_sql = """
+select * from {{ ref('nested_ephemeral_model') }}
+"""
+
+test_sql_format_yml = """
+unit_tests:
+  - name: test_customers
+    model: customers
+    given:
+      - input: ref('nested_ephemeral_model')
+        format: sql
+        rows: |
+          select 1 as id, 'Emily' as first_name
+    expect:
+      rows:
+        - {id: 1, first_name: Emily}
+"""
+
+failing_test_sql_format_yml = """
+  - name: fail_test_customers
+    model: customers
+    given:
+      - input: ref('nested_ephemeral_model')
+        format: sql
+        rows: |
+          select 1 as id, 'Emily' as first_name
+    expect:
+      rows:
+        - {id: 1, first_name: Joan}
+"""
+
+
+class TestUnitTestEphemeralInput:
+    @pytest.fixture(scope="class")
+    def models(self):
+        return {
+            "customers.sql": customers_sql,
+            "ephemeral_model.sql": ephemeral_model_sql,
+            "nested_ephemeral_model.sql": nested_ephemeral_model_sql,
+            "tests.yml": test_sql_format_yml,
+        }
+
+    def test_ephemeral_input(self, project):
+        results = run_dbt(["run"])
+        len(results) == 1
+
+        results = run_dbt(["test", "--select", "test_type:unit"])
+        assert len(results) == 1
+
+        results = run_dbt(["build"])
+        assert len(results) == 2
+        result_unique_ids = [result.node.unique_id for result in results]
+        assert len(result_unique_ids) == 2
+        assert "unit_test.test.customers.test_customers" in result_unique_ids
+
+        # write failing unit test
+        write_file(
+            test_sql_format_yml + failing_test_sql_format_yml,
+            project.project_root,
+            "models",
+            "tests.yml",
+        )
+        results = run_dbt(["build"], expect_pass=False)
+        for result in results:
+            if result.node.unique_id == "model.test.customers":
+                assert result.status == RunStatus.Skipped
+            elif result.node.unique_id == "unit_test.test.customers.fail_test_customers":
+                assert result.status == TestStatus.Fail
+        assert len(results) == 3