Adding test for project dirs

godatadriven · Jul 19, 2024 · e2cf7af · e2cf7af
1 parent db0db71
commit e2cf7af
Show file tree

Hide file tree

Showing 10 changed files with 84 additions and 6 deletions.
diff --git a/.env copy → .env.example b/.env copy → .env.example
diff --git a/dbt_bouncer/runner.py b/dbt_bouncer/runner.py
@@ -1,4 +1,5 @@
 import inspect
+from pathlib import Path
 from typing import Dict, List
 
 import pytest
@@ -25,8 +26,8 @@ def tests(self):
 
 # Inspiration: https://github.com/pytest-dev/pytest-xdist/discussions/957#discussioncomment-7335007
 class MyFunctionItem(pytest.Function):
-    def __init__(self, model, *args, **kwargs):
-        self.model: Dict[str, str] = model
+    def __init__(self, model=None, *args, **kwargs):
+        self.model: Dict[str, str] | None = model
         super().__init__(*args, **kwargs)
 
 
@@ -46,12 +47,21 @@ def pytest_pycollect_makeitem(self, collector, name, obj):
             fixture_info = pytest.Function.from_parent(
                 collector, name=name, callobj=obj
             )._fixtureinfo
-            for model in self.models:
+
+            if "request" in fixture_info.argnames:
+                for model in self.models:
+                    item = MyFunctionItem.from_parent(
+                        parent=collector,
+                        name=name,
+                        fixtureinfo=fixture_info,
+                        model=model,
+                    )
+                    items.append(item)
+            else:
                 item = MyFunctionItem.from_parent(
                     parent=collector,
                     name=name,
                     fixtureinfo=fixture_info,
-                    model=model,
                 )
                 items.append(item)
 
@@ -74,6 +84,6 @@ def runner(
 
     # Run the tests, if one fails then pytest will raise an exception
     pytest.main(
-        ["dbt_bouncer/tests"],
+        [(Path(__file__).parent / "tests").__str__(), "-s"],
         plugins=[fixtures, GenerateTestsPlugin(models)],
     )
diff --git a/dbt_bouncer/tests/test_project_directories.py b/dbt_bouncer/tests/test_project_directories.py
@@ -0,0 +1,8 @@
+def test_top_level_directories(models):
+    for model in models:
+        top_level_dir = model["path"].split("/")[0]
+        assert top_level_dir in [
+            "staging",
+            "intermediate",
+            "marts",
+        ], f"{model['unique_id']} is located in `{top_level_dir}`, this is not a valid top-level directory."
diff --git a/dbt_project/models/_schema.yml → dbt_project/models/staging/_schema.yml b/dbt_project/models/_schema.yml → dbt_project/models/staging/_schema.yml
diff --git a/dbt_project/models/_source.yml → dbt_project/models/staging/_source.yml b/dbt_project/models/_source.yml → dbt_project/models/staging/_source.yml
diff --git a/dbt_project/models/model_1.sql → dbt_project/models/staging/model_1.sql b/dbt_project/models/model_1.sql → dbt_project/models/staging/model_1.sql
diff --git a/dbt_project/models/model_2.sql → dbt_project/models/staging/model_2.sql b/dbt_project/models/model_2.sql → dbt_project/models/staging/model_2.sql
diff --git a/dbt_project/target/manifest.json b/dbt_project/target/manifest.json
diff --git a/dist/dbt-bouncer.pex b/dist/dbt-bouncer.pex
diff --git a/tests/tests/test_project_directories.py b/tests/tests/test_project_directories.py
@@ -0,0 +1,60 @@
+from contextlib import nullcontext as does_not_raise
+
+import pytest
+
+from dbt_bouncer.tests.test_project_directories import test_top_level_directories
+
+
+@pytest.mark.parametrize(
+    "models, expectation",
+    [
+        (
+            [
+                {
+                    "path": "staging/model_1.sql",
+                    "unique_id": "model.package_name.model_1",
+                }
+            ],
+            does_not_raise(),
+        ),
+        (
+            [
+                {
+                    "path": "intermediate/model_1.sql",
+                    "unique_id": "model.package_name.model_1",
+                }
+            ],
+            does_not_raise(),
+        ),
+        (
+            [
+                {
+                    "path": "marts/model_1.sql",
+                    "unique_id": "model.package_name.model_1",
+                }
+            ],
+            does_not_raise(),
+        ),
+        (
+            [
+                {
+                    "path": "model_1.sql",
+                    "unique_id": "model.package_name.model_1",
+                }
+            ],
+            pytest.raises(AssertionError),
+        ),
+        (
+            [
+                {
+                    "path": "aggregation/model_1.sql",
+                    "unique_id": "model.package_name.model_1",
+                }
+            ],
+            pytest.raises(AssertionError),
+        ),
+    ],
+)
+def test_test_top_level_directories(models, expectation):
+    with expectation:
+        test_top_level_directories(models=models)