def test_nominal(self, m_worker_cls):
m_worker = m_worker_cls.return_value
m_worker.run.side_effect = [None, None, KeyboardInterrupt('KILLTEST')]
args = [
'-d', 'DOMAIN', '-m', 'caravan.tests.fixtures', '-t', 'TASKLIST',
]
with mock_args(args):
with self.assertRaises(SystemExit) as exc:
Command.main()
self.assertEqual(exc.exception.code, 1)
self.assertEqual(m_worker.run.call_count, 3)
args, kwargs = m_worker_cls.call_args
self.assertEqual(kwargs['domain'], 'DOMAIN')
self.assertEqual(kwargs['task_list'], 'TASKLIST')
self.assertEqual(kwargs['workflows'],
[fixtures.TestWorkflow1, fixtures.TestWorkflow2])
# Boto3 client object are dynamically forged...
swf_connection = kwargs['connection']
self.assertEqual(swf_connection.__class__.__name__, 'SWF')
self.assertEqual(swf_connection.__class__.__module__, 'botocore.client')
def test_register_workflows(self, m_get_conn, m_register, m_worker_cls):
m_worker = m_worker_cls.return_value
m_worker.run.side_effect = KeyboardInterrupt('KILLTEST')
m_conn = m_get_conn.return_value
args = [
'-d', 'DOMAIN', '-m', 'caravan.tests.fixtures', '-t', 'TASKLIST',
'--register-workflows',
]
with mock_args(args):
with self.assertRaises(SystemExit):
Command.main()
expected = [
mock.call(connection=m_conn,
domain='DOMAIN',
workflow=fixtures.TestWorkflow1),
mock.call(connection=m_conn,
domain='DOMAIN',
workflow=fixtures.TestWorkflow2),
]
self.assertEqual(m_register.call_args_list, expected)