Skip to content

Commit 58ef377

Browse files
committed
Fix Py37 mock call args
1 parent b4be988 commit 58ef377

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_config.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -244,7 +244,7 @@ def test_athena_wait_delay_config(wr: ModuleType, glue_database: str, polling_de
244244

245245
mock_wait_query.assert_called_once()
246246

247-
assert mock_wait_query.call_args.kwargs["athena_query_wait_polling_delay"] == polling_delay
247+
assert mock_wait_query.call_args[1]["athena_query_wait_polling_delay"] == polling_delay
248248

249249

250250
def test_athena_wait_delay_config_override(wr: ModuleType, glue_database: str) -> None:
@@ -258,4 +258,4 @@ def test_athena_wait_delay_config_override(wr: ModuleType, glue_database: str) -
258258

259259
mock_wait_query.assert_called_once()
260260

261-
assert mock_wait_query.call_args.kwargs["athena_query_wait_polling_delay"] == polling_delay_argument
261+
assert mock_wait_query.call_args[1]["athena_query_wait_polling_delay"] == polling_delay

0 commit comments

Comments
 (0)