@@ -52,6 +52,7 @@ def invalid_value_yaml_file():
52
52
if temp_file .exists ():
53
53
temp_file .unlink ()
54
54
55
+
55
56
@pytest .fixture
56
57
def malformed_yaml_file ():
57
58
"""Creates a temporary malformed yaml file for testing."""
@@ -259,14 +260,17 @@ def test_dry_run_sync(self, mock_get_default_graph, test_yaml_file):
259
260
assert not mock_get_default_graph .emit .called
260
261
261
262
@patch ("datahub.cli.specific.dataset_cli.get_default_graph" )
262
- def test_dry_run_sync_fail_bad_type (self , mock_get_default_graph , invalid_value_yaml_file ):
263
+ def test_dry_run_sync_fail_bad_type (
264
+ self , mock_get_default_graph , invalid_value_yaml_file
265
+ ):
263
266
mock_graph = MagicMock ()
264
267
mock_graph .exists .return_value = True
265
268
mock_get_default_graph .return_value .__enter__ .return_value = mock_graph
266
269
267
270
runner = CliRunner ()
268
271
result = runner .invoke (
269
- dataset , ["sync" , "--dry-run" , "--to-datahub" , "-f" , str (invalid_value_yaml_file )]
272
+ dataset ,
273
+ ["sync" , "--dry-run" , "--to-datahub" , "-f" , str (invalid_value_yaml_file )],
270
274
)
271
275
272
276
# Verify
@@ -275,7 +279,9 @@ def test_dry_run_sync_fail_bad_type(self, mock_get_default_graph, invalid_value_
275
279
assert "Type bad_type is not a valid primitive type" in result .output
276
280
277
281
@patch ("datahub.cli.specific.dataset_cli.get_default_graph" )
278
- def test_dry_run_sync_fail_missing_ref (self , mock_get_default_graph , test_yaml_file ):
282
+ def test_dry_run_sync_fail_missing_ref (
283
+ self , mock_get_default_graph , test_yaml_file
284
+ ):
279
285
mock_graph = MagicMock ()
280
286
mock_graph .exists .return_value = False
281
287
mock_get_default_graph .return_value .__enter__ .return_value = mock_graph
@@ -297,7 +303,9 @@ def test_run_sync(self, mock_get_default_graph, test_yaml_file):
297
303
mock_get_default_graph .return_value .__enter__ .return_value = mock_graph
298
304
299
305
runner = CliRunner ()
300
- result = runner .invoke (dataset , ["sync" , "--to-datahub" , "-f" , str (test_yaml_file )])
306
+ result = runner .invoke (
307
+ dataset , ["sync" , "--to-datahub" , "-f" , str (test_yaml_file )]
308
+ )
301
309
302
310
# Verify
303
311
assert result .exit_code == 0
@@ -310,7 +318,9 @@ def test_run_sync_fail(self, mock_get_default_graph, invalid_value_yaml_file):
310
318
mock_get_default_graph .return_value .__enter__ .return_value = mock_graph
311
319
312
320
runner = CliRunner ()
313
- result = runner .invoke (dataset , ["sync" , "--to-datahub" , "-f" , str (invalid_value_yaml_file )])
321
+ result = runner .invoke (
322
+ dataset , ["sync" , "--to-datahub" , "-f" , str (invalid_value_yaml_file )]
323
+ )
314
324
315
325
# Verify
316
326
assert result .exit_code != 0
@@ -338,9 +348,10 @@ def test_sync_from_datahub_fail(self, mock_get_default_graph, test_yaml_file):
338
348
mock_get_default_graph .return_value .__enter__ .return_value = mock_graph
339
349
340
350
runner = CliRunner ()
341
- result = runner .invoke (dataset , ["sync" , "--dry-run" , "--from-datahub" , "-f" , str (test_yaml_file )])
351
+ result = runner .invoke (
352
+ dataset , ["sync" , "--dry-run" , "--from-datahub" , "-f" , str (test_yaml_file )]
353
+ )
342
354
343
355
# Verify
344
356
assert result .exit_code != 0
345
357
assert "does not exist" in result .output
346
-
0 commit comments