-
Notifications
You must be signed in to change notification settings - Fork 14
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
This probably should have been part of #148, but we forgot. I'm adding these tests here because I plan to add `label` to the `SavedQuery` protocol in the coming commits. With that, I'll want to updated the parsing tests to check it. To do that, the tests need to exist.
- Loading branch information
Showing
1 changed file
with
132 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,132 @@ | ||
import textwrap | ||
|
||
from dbt_semantic_interfaces.parsing.dir_to_model import ( | ||
parse_yaml_files_to_semantic_manifest, | ||
) | ||
from dbt_semantic_interfaces.parsing.objects import YamlConfigFile | ||
from tests.example_project_configuration import ( | ||
EXAMPLE_PROJECT_CONFIGURATION_YAML_CONFIG_FILE, | ||
) | ||
|
||
|
||
def test_saved_query_metadata_parsing() -> None: | ||
"""Test for asserting that internal metadata is parsed into the SavedQuery object.""" | ||
yaml_contents = textwrap.dedent( | ||
"""\ | ||
saved_query: | ||
name: metadata_test | ||
metrics: | ||
- test_metric | ||
""" | ||
) | ||
file = YamlConfigFile(filepath="test_dir/inline_for_test", contents=yaml_contents) | ||
|
||
build_result = parse_yaml_files_to_semantic_manifest(files=[file, EXAMPLE_PROJECT_CONFIGURATION_YAML_CONFIG_FILE]) | ||
|
||
assert len(build_result.semantic_manifest.saved_queries) == 1 | ||
saved_query = build_result.semantic_manifest.saved_queries[0] | ||
assert saved_query.metadata is not None | ||
assert saved_query.metadata.repo_file_path == "test_dir/inline_for_test" | ||
assert saved_query.metadata.file_slice.filename == "inline_for_test" | ||
expected_metadata_content = textwrap.dedent( | ||
"""\ | ||
name: metadata_test | ||
metrics: | ||
- test_metric | ||
""" | ||
) | ||
assert saved_query.metadata.file_slice.content == expected_metadata_content | ||
|
||
|
||
def test_saved_query_base_parsing() -> None: | ||
"""Test for base parsing a saved query.""" | ||
name = "base_parsing_test" | ||
description = "the base saved query parsing test" | ||
yaml_contents = textwrap.dedent( | ||
f"""\ | ||
saved_query: | ||
name: {name} | ||
description: {description} | ||
metrics: | ||
- test_metric | ||
""" | ||
) | ||
file = YamlConfigFile(filepath="inline_for_test", contents=yaml_contents) | ||
|
||
build_result = parse_yaml_files_to_semantic_manifest(files=[file, EXAMPLE_PROJECT_CONFIGURATION_YAML_CONFIG_FILE]) | ||
|
||
assert len(build_result.semantic_manifest.saved_queries) == 1 | ||
saved_query = build_result.semantic_manifest.saved_queries[0] | ||
assert saved_query.name == name | ||
assert saved_query.description == description | ||
|
||
|
||
def test_saved_query_metrics_parsing() -> None: | ||
"""Test for parsing metrics referenced in a saved query.""" | ||
yaml_contents = textwrap.dedent( | ||
"""\ | ||
saved_query: | ||
name: test_saved_query_metrics | ||
metrics: | ||
- test_metric_a | ||
- test_metric_b | ||
- test_metric_c | ||
""" | ||
) | ||
file = YamlConfigFile(filepath="inline_for_test", contents=yaml_contents) | ||
|
||
build_result = parse_yaml_files_to_semantic_manifest(files=[file, EXAMPLE_PROJECT_CONFIGURATION_YAML_CONFIG_FILE]) | ||
|
||
assert len(build_result.semantic_manifest.saved_queries) == 1 | ||
saved_query = build_result.semantic_manifest.saved_queries[0] | ||
assert len(saved_query.metrics) == 3 | ||
assert {"test_metric_a", "test_metric_b", "test_metric_c"} == set(saved_query.metrics) | ||
|
||
|
||
def test_saved_query_group_bys() -> None: | ||
"""Test for parsing group_bys in a saved query.""" | ||
yaml_contents = textwrap.dedent( | ||
"""\ | ||
saved_query: | ||
name: test_saved_query_group_bys | ||
metrics: | ||
- test_metric_a | ||
group_bys: | ||
- Dimension(test_entity__test_dimension_a) | ||
- Dimension(test_entity__test_dimension_b) | ||
""" | ||
) | ||
file = YamlConfigFile(filepath="test_dir/inline_for_test", contents=yaml_contents) | ||
|
||
build_result = parse_yaml_files_to_semantic_manifest(files=[file, EXAMPLE_PROJECT_CONFIGURATION_YAML_CONFIG_FILE]) | ||
|
||
assert len(build_result.semantic_manifest.saved_queries) == 1 | ||
saved_query = build_result.semantic_manifest.saved_queries[0] | ||
assert len(saved_query.group_bys) == 2 | ||
assert {"Dimension(test_entity__test_dimension_a)", "Dimension(test_entity__test_dimension_b)"} == set( | ||
saved_query.group_bys | ||
) | ||
|
||
|
||
def test_saved_query_where() -> None: | ||
"""Test for parsing where clause in a saved query.""" | ||
where = "Dimension(test_entity__test_dimension) == true" | ||
yaml_contents = textwrap.dedent( | ||
f"""\ | ||
saved_query: | ||
name: test_saved_query_where_clause | ||
metrics: | ||
- test_metric_a | ||
where: | ||
- '{where}' | ||
""" | ||
) | ||
file = YamlConfigFile(filepath="test_dir/inline_for_test", contents=yaml_contents) | ||
|
||
build_result = parse_yaml_files_to_semantic_manifest(files=[file, EXAMPLE_PROJECT_CONFIGURATION_YAML_CONFIG_FILE]) | ||
assert len(build_result.semantic_manifest.saved_queries) == 1 | ||
saved_query = build_result.semantic_manifest.saved_queries[0] | ||
assert len(saved_query.where) == 1 | ||
assert where == saved_query.where[0].where_sql_template |