Skip to content

Commit

Permalink
Expand unit test list test
Browse files Browse the repository at this point in the history
  • Loading branch information
gshank committed Jan 29, 2024
1 parent 6254cea commit fd45bbb
Showing 1 changed file with 47 additions and 0 deletions.
47 changes: 47 additions & 0 deletions tests/functional/unit_testing/test_ut_list.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import pytest
from dbt.tests.util import run_dbt
import json

from fixtures import ( # noqa: F401
my_model_vars_sql,
Expand All @@ -9,6 +10,7 @@
datetime_test,
)


class TestUnitTestList:
@pytest.fixture(scope="class")
def models(self):
Expand All @@ -31,4 +33,49 @@ def test_unit_test_list(self, project):
assert len(results) == 5

results = run_dbt(["list"])
expected = [
"test.my_model",
"test.my_model_a",
"test.my_model_b",
"unit_test:test.test_my_model",
"unit_test:test.test_my_model_datetime",
"unit_test:test.test_my_model_empty",
"unit_test:test.test_my_model_overrides",
"unit_test:test.test_my_model_string_concat",
]
assert sorted(results) == sorted(expected)

results = run_dbt(["list", "--select", "test_type:unit"])
assert len(results) == 5

# Check json result
results = run_dbt(["list", "--select", "test_type:unit", "--output", "json"])
expected_test_my_model = {
"name": "test_my_model",
"resource_type": "unit_test",
"package_name": "test",
"original_file_path": "models/test_my_model.yml",
"unique_id": "unit_test.test.my_model.test_my_model",
"depends_on": {"macros": [], "nodes": ["model.test.my_model"]},
"config": {"tags": [], "meta": {}},
}
for result in results:
json_result = json.loads(result)
if "name" in json_result and json_result["name"] == "test_my_model":
assert json_result == expected_test_my_model

results = run_dbt(
[
"list",
"--select",
"test_type:unit",
"--output",
"json",
"--output-keys",
"unique_id",
"model",
]
)
for result in results:
json_result = json.loads(result)
assert json_result["model"] == "my_model"

0 comments on commit fd45bbb

Please sign in to comment.