def setUp(self):
super(TestPayload, self).setUp()
self.test_payload = Payload(_fn, 'arg1', arg2='arg2value')
self.serialized_payload = {
'fn_key': function_key(_fn),
'client_version': VERSION,
'suppress_warnings': False,
'args': ('arg1', ),
'kwargs': (('arg2', 'arg2value'), )
}
def test_main_container_to_container(self):
result = main(['website:/tmp/a', 'api:/tmp/b'])
self.assertEqual(
result,
Payload(copy_between_containers, 'website', '/tmp/a', 'api',
'/tmp/b'))
def test_main_container_to_local(self):
result = main(['website:/tmp/a', '/tmp/b'])
self.assertEqual(result,
Payload(copy_to_local, '/tmp/b', 'website', '/tmp/a'))
def test_equality_fails_bad_kwargs(self):
self.assertNotEqual(self.test_payload,
Payload(_fn, 'arg1', arg2='wrongvalue'))
def test_equality_fails_bad_args(self):
self.assertNotEqual(self.test_payload,
Payload(_fn, 'arg3', arg2='arg2value'))
def test_equality_fails_bad_run_on_daemon(self):
payload = Payload(_fn, 'arg1', arg2='arg2value')
payload.run_on_daemon = False
self.assertNotEqual(self.test_payload, payload)
def test_equality_fails_bad_suppress(self):
payload = Payload(_fn, 'arg1', arg2='arg2value')
payload.suppress_warnings = True
self.assertNotEqual(self.test_payload, payload)
def test_equality_matches(self):
self.assertEqual(self.test_payload,
Payload(_fn, 'arg1', arg2='arg2value'))
