From 133947ff17c7aa138ce44c7085bc9c5e126635e7 Mon Sep 17 00:00:00 2001
From: Ben King <9087625+benfdking@users.noreply.github.com>
Date: Mon, 29 Sep 2025 16:25:36 +0100
Subject: [PATCH] chore: add dbt unit tests to fixture

---
 .../fixtures/dbt/sushi_test/models/schema.yml | 49 +++++++++++++++++++
 1 file changed, 49 insertions(+)

diff --git a/tests/fixtures/dbt/sushi_test/models/schema.yml b/tests/fixtures/dbt/sushi_test/models/schema.yml
index 87a201c418..21985f19ff 100644
--- a/tests/fixtures/dbt/sushi_test/models/schema.yml
+++ b/tests/fixtures/dbt/sushi_test/models/schema.yml
@@ -1,6 +1,55 @@
 version: 2
 
 models:
+  - name: simple_model_a
+    description: A simple model for testing
+    columns:
+      - name: a
+        data_type: int
+    unit_tests:
+      - name: test_simple_model_a_outputs_one
+        description: Test that simple_model_a outputs 1 as column a
+        model: simple_model_a
+        given: []  # No input models needed
+        expect:
+          format: csv
+          rows: |
+            a
+            1
+  - name: simple_model_b
+    description: Model that references simple_model_a
+    columns:
+      - name: a
+        data_type: int
+    unit_tests:
+      - name: test_simple_model_b_with_mock_input
+        description: Test simple_model_b with mocked simple_model_a input
+        model: simple_model_b
+        given:
+          - input: ref('simple_model_a')
+            format: csv
+            rows: |
+              a
+              10
+              20
+              30
+        expect:
+          format: csv
+          rows: |
+            a
+            10
+            20
+            30
+      - name: test_simple_model_b_with_sql_input
+        description: Test simple_model_b with SQL-defined input data
+        model: simple_model_b
+        given:
+          - input: ref('simple_model_a')
+            format: sql
+            rows: SELECT 42 AS a
+        expect:
+          format: sql
+          rows: SELECT 42 AS a
   - name: top_waiters
     description: description of top waiters
     columns: