File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -61,8 +61,10 @@ class CustomIterator:
61
61
def __init__ (self , values ):
62
62
self ._values = iter (values )
63
63
self ._has_next = True
64
+ self ._next_value = None
64
65
65
- def hasNext (self ): # pylint: disable=invalid-name
66
+ # pylint: disable=invalid-name
67
+ def hasNext (self ):
66
68
try :
67
69
self ._next_value = next (self ._values )
68
70
self ._has_next = True
@@ -154,9 +156,11 @@ def inner(cb, **replace) -> RuntimeContext:
154
156
ctx .tables_crawler ._spark ._jsparkSession .sharedState ().externalCatalog ().listDatabases .return_value = (
155
157
mock_list_databases_iterator
156
158
)
159
+ # pylint: disable=protected-access
157
160
ctx .tables_crawler ._spark ._jsparkSession .sharedState ().externalCatalog ().listTables .return_value = (
158
161
mock_list_tables_iterator
159
162
)
163
+ # pylint: disable=protected-access
160
164
ctx .tables_crawler ._spark ._jsparkSession .sharedState ().externalCatalog ().getTable .return_value = (
161
165
get_table_mock
162
166
)
@@ -169,8 +173,9 @@ def inner(cb, **replace) -> RuntimeContext:
169
173
170
174
@pytest .fixture
171
175
def acc_client ():
172
- acc = create_autospec (AccountClient ) # pylint: disable=mock-no-usage
176
+ acc = create_autospec (AccountClient )
173
177
acc .config = Config (host = "https://accounts.cloud.databricks.com" , account_id = "123" , token = "123" )
178
+ acc .asset_not_called ()
174
179
return acc
175
180
176
181
You can’t perform that action at this time.
0 commit comments