diff --git a/tests/activities/test_activities.py b/tests/activities/test_activities.py index 2740d18..ecb5c8d 100644 --- a/tests/activities/test_activities.py +++ b/tests/activities/test_activities.py @@ -133,5 +133,5 @@ def test_shutdown(mock_mongodb_close, mock_postgres_close, _mock_mongodb_init, activities.shutdown() - mock_postgres_close.assert_called_once() - mock_mongodb_close.assert_called_once() + mock_postgres_close.assert_called() + mock_mongodb_close.assert_called()