def test_main_method(ManagerWorker):
"""
Test creation of manager process from _main method
"""
_main(["email1", "email2"], concurrency=2)
ManagerWorker.assert_called_once_with(['email1', 'email2'], 2, region='us-east-1', secret_access_key=None, access_key_id=None)
ManagerWorker.return_value.start.assert_called_once_with()
def test_main_method(ManagerWorker):
"""
Test creation of manager process from _main method
"""
_main(["email1", "email2"], concurrency=2)
ManagerWorker.assert_called_once_with(['email1', 'email2'], 2, 1, 10, prefetch_multiplier=2, region='us-east-1', secret_access_key=None, access_key_id=None)
ManagerWorker.return_value.start.assert_called_once_with()
def test_main_method(SimpleManagerWorker):
"""
Test creation of simple manager process from _main method
"""
_main(["email1", "email2"], concurrency=2, simple_worker=True)
SimpleManagerWorker.assert_called_once_with(
['email1', 'email2'],
2,
1,
10,
region=None,
secret_access_key=None,
access_key_id=None,
)
SimpleManagerWorker.return_value.start.assert_called_once_with()