File tree Expand file tree Collapse file tree 1 file changed +11
-0
lines changed
experimental/python/databricks_tests/core Expand file tree Collapse file tree 1 file changed +11
-0
lines changed Original file line number Diff line number Diff line change 44import pytest
55
66from databricks .bundles .core import Location , Resources , Severity
7+ from databricks .bundles .core ._bundle import Bundle
78from databricks .bundles .core ._resource import Resource
89from databricks .bundles .core ._resource_mutator import (
910 ResourceMutator ,
@@ -201,9 +202,19 @@ def test_mutator(tc: TestCase, tpe: _ResourceType):
201202 def my_func (bundle , resource ):
202203 return resource
203204
205+ bundle = Bundle (target = "default" )
206+
204207 assert isinstance (my_func , ResourceMutator )
208+ assert tc .mutator .__name__ == tpe .singular_name + "_mutator"
205209 assert my_func .resource_type == tpe .resource_type
210+ assert my_func .function (bundle , tc .dataclass_example ) is tc .dataclass_example
211+
212+
213+ @pytest .mark .parametrize ("tc,tpe" , test_cases , ids = test_case_ids )
214+ def test_mutator_export (tc : TestCase , tpe : _ResourceType ):
215+ import databricks .bundles .core
206216
217+ assert tc .mutator .__name__ in databricks .bundles .core .__all__ , "mutator is not in databricks.bundles.core.__all__"
207218
208219@pytest .mark .parametrize ("tc,tpe" , test_cases , ids = test_case_ids )
209220def test_add_duplicate_resource (tc : TestCase , tpe : _ResourceType ):
You can’t perform that action at this time.
0 commit comments