Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 28 additions & 0 deletions tests/dbt/test_docs.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
from pathlib import Path
import pytest

from sqlmesh.core.config.model import ModelDefaultsConfig
from sqlmesh.dbt.context import DbtContext
from sqlmesh.dbt.manifest import ManifestHelper
from sqlmesh.dbt.profile import Profile


pytestmark = pytest.mark.dbt


@pytest.mark.xdist_group("dbt_manifest")
def test_docs_inline():
project_path = Path("tests/fixtures/dbt/sushi_test")
profile = Profile.load(DbtContext(project_path))

helper = ManifestHelper(
project_path,
project_path,
"sushi",
profile.target,
model_defaults=ModelDefaultsConfig(start="2020-01-01"),
)
# Inline description in yaml
assert helper.models()["waiters"].description == "waiters docs block"
# Docs block from .md file
assert helper.models()["top_waiters"].description == "description of top waiters"
2 changes: 2 additions & 0 deletions tests/fixtures/dbt/sushi_test/models/schema.yml
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@ version: 2

models:
- name: top_waiters
description: description of top waiters
columns:
- name: waiter_id
data_type: int
Expand All @@ -18,6 +19,7 @@ models:
warn_after: {count: 8, period: hour}
error_after: {count: 9, period: hour}
- name: waiters
description: '{{ doc("waiters") }}'
- name: waiter_as_customer_by_day
- name: waiter_revenue_by_day
versions:
Expand Down
4 changes: 4 additions & 0 deletions tests/fixtures/dbt/sushi_test/models/waiters_doc_block.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
{% docs waiters %}
waiters docs block
{% enddocs %}