-
Notifications
You must be signed in to change notification settings - Fork 1.6k
/
model_deprecations.py
106 lines (84 loc) · 3.37 KB
/
model_deprecations.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
import pytest
from dbt.cli.main import dbtRunner
from dbt.exceptions import EventCompilationError
from dbt.tests.util import run_dbt
deprecated_model__yml = """
version: 2
models:
- name: my_model
description: deprecated
deprecation_date: 1999-01-01
"""
deprecating_model__yml = """
version: 2
models:
- name: my_model
description: deprecating in the future
deprecation_date: 2999-01-01
"""
model__sql = """
select 1 as Id
"""
dependant_model__sql = """
select * from {{ ref("my_model") }}
"""
class TestModelDeprecationWarning:
@pytest.fixture(scope="class")
def models(self):
return {"my_model.sql": model__sql, "my_schema.yml": deprecated_model__yml}
def test_deprecation_warning(self, project):
events = []
dbtRunner(callbacks=[events.append]).invoke(["parse"])
matches = list([e for e in events if e.info.name == "DeprecatedModel"])
assert len(matches) == 1
assert matches[0].data.model_name == "my_model"
def test_deprecation_warning_error(self, project):
with pytest.raises(EventCompilationError):
run_dbt(["--warn-error", "parse"])
def test_deprecation_warning_error_options(self, project):
with pytest.raises(EventCompilationError):
run_dbt(["--warn-error-options", '{"include": ["DeprecatedModel"]}', "parse"])
class TestUpcomingReferenceDeprecatingWarning:
@pytest.fixture(scope="class")
def models(self):
return {
"my_model.sql": model__sql,
"my_dependant_model.sql": dependant_model__sql,
"my_schema.yml": deprecating_model__yml,
}
def test_deprecation_warning(self, project):
events = []
dbtRunner(callbacks=[events.append]).invoke(["parse"])
matches = list([e for e in events if e.info.name == "UpcomingReferenceDeprecation"])
assert len(matches) == 1
assert matches[0].data.model_name == "my_dependant_model"
assert matches[0].data.ref_model_name == "my_model"
def test_deprecation_warning_error(self, project):
with pytest.raises(EventCompilationError):
run_dbt(["--warn-error", "parse"])
def test_deprecation_warning_error_options(self, project):
with pytest.raises(EventCompilationError):
run_dbt(
["--warn-error-options", '{"include": ["UpcomingReferenceDeprecation"]}', "parse"]
)
class TestDeprecatedReferenceWarning:
@pytest.fixture(scope="class")
def models(self):
return {
"my_model.sql": model__sql,
"my_dependant_model.sql": dependant_model__sql,
"my_schema.yml": deprecated_model__yml,
}
def test_deprecation_warning(self, project):
events = []
dbtRunner(callbacks=[events.append]).invoke(["parse"])
matches = list([e for e in events if e.info.name == "DeprecatedReference"])
assert len(matches) == 1
assert matches[0].data.model_name == "my_dependant_model"
assert matches[0].data.ref_model_name == "my_model"
def test_deprecation_warning_error(self, project):
with pytest.raises(EventCompilationError):
run_dbt(["--warn-error", "parse"])
def test_deprecation_warning_error_options(self, project):
with pytest.raises(EventCompilationError):
run_dbt(["--warn-error-options", '{"include": ["DeprecatedReference"]}', "parse"])