fix: Change type of BaseConfig.state to Path

Author: tomasfarias

airflow_dbt_python/hooks/dbt.py

@@ -108,14 +108,16 @@ class BaseConfig:
     log_cache_events: Optional[bool] = None
     defer: Optional[bool] = None
     no_defer: Optional[bool] = None
-    state: Optional[str] = None
+    state: Optional[Path] = None
     threads: Optional[int] = None
     compiled_target: Optional[Union["os.PathLike[str]", str]] = None
 
     def __post_init__(self):
         """Support dictionary args by casting them to str after setting."""
         if isinstance(self.vars, dict):
             self.vars = json.dumps(self.vars)
+        if isinstance(self.state, str):
+            self.state = Path(self.state)
 
     @property
     def dbt_task(self) -> BaseTask:

tests/operators/test_dbt_build.py

@@ -1,5 +1,6 @@
 """Unit test module for DbtBuildOperator."""
 import json
+from pathlib import Path
 from unittest.mock import patch
 
 import pytest
@@ -62,7 +63,7 @@ def test_dbt_build_mocked_all_args():
     ]
     assert config.exclude == ["/path/to/model/to/exclude.sql"]
     assert config.selector_name == ["a-selector"]
-    assert config.state == "/path/to/state/"
+    assert config.state == Path("/path/to/state/")
     assert config.data is True
     assert config.schema is True
     assert config.show is True

tests/operators/test_dbt_compile.py

@@ -1,4 +1,6 @@
 """Unit test module for DbtCompileOperator."""
+from pathlib import Path
+
 from dbt.contracts.results import RunStatus
 
 from airflow_dbt_python.hooks.dbt import CompileTaskConfig
@@ -48,7 +50,7 @@ def test_dbt_compile_mocked_all_args():
     ]
     assert config.exclude == ["/path/to/data/to/exclude.sql"]
     assert config.selector_name == ["a-selector"]
-    assert config.state == "/path/to/state/"
+    assert config.state == Path("/path/to/state/")
 
 
 def test_dbt_compile_non_existent_model(profiles_file, dbt_project_file, model_files):

tests/operators/test_dbt_run.py

@@ -1,5 +1,6 @@
 """Unit test module for DbtRunOperator."""
 import json
+from pathlib import Path
 
 import pytest
 from dbt.contracts.results import RunStatus
@@ -54,7 +55,7 @@ def test_dbt_run_mocked_all_args():
     assert config.select == ["/path/to/model.sql", "+/another/model.sql+2"]
     assert config.exclude == ["/path/to/model/to/exclude.sql"]
     assert config.selector_name == ["a-selector"]
-    assert config.state == "/path/to/state/"
+    assert config.state == Path("/path/to/state/")
 
 
 def test_dbt_run_non_existent_model(profiles_file, dbt_project_file, model_files):

tests/operators/test_dbt_seed.py

@@ -1,5 +1,6 @@
 """Unit test module for DbtSeedOperator."""
 import json
+from pathlib import Path
 from unittest.mock import patch
 
 import pytest
@@ -55,7 +56,7 @@ def test_dbt_seed_mocked_all_args():
     assert config.show is True
     assert config.exclude == ["/path/to/data/to/exclude.csv"]
     assert config.selector_name == ["a-selector"]
-    assert config.state == "/path/to/state/"
+    assert config.state == Path("/path/to/state/")
 
 
 def test_dbt_seed_non_existent_file(profiles_file, dbt_project_file, seed_files):

tests/operators/test_dbt_snapshot.py

@@ -1,5 +1,5 @@
 """Unit test module for DbtSnapshotOperator."""
-from unittest.mock import patch
+from pathlib import Path
 
 import pytest
 from dbt.contracts.results import RunStatus
@@ -41,7 +41,7 @@ def test_dbt_snapshot_mocked_all_args():
     assert config.select == ["/path/to/models"]
     assert config.exclude == ["/path/to/data/to/exclude.sql"]
     assert config.selector_name == ["a-selector"]
-    assert config.state == "/path/to/state/"
+    assert config.state == Path("/path/to/state/")
 
 
 def test_dbt_snapshot(profiles_file, dbt_project_file, snapshot_files):

tests/operators/test_dbt_test.py

@@ -1,4 +1,6 @@
 """Unit test module for DbtTestOperator."""
+from pathlib import Path
+
 import pytest
 from dbt.contracts.results import TestStatus
 
@@ -55,7 +57,7 @@ def test_dbt_test_configuration_all_args():
     assert config.select == ["/path/to/models"]
     assert config.exclude == ["/path/to/data/to/exclude.sql"]
     assert config.selector_name == ["a-selector"]
-    assert config.state == "/path/to/state/"
+    assert config.state == Path("/path/to/state/")
     assert config.no_defer is True