Example #1
0
class TestPayload(DustyTestCase):
    def setUp(self):
        super(TestPayload, self).setUp()
        self.test_payload = Payload(_fn, 'arg1', arg2='arg2value')
        self.serialized_payload = {'fn': _fn, 'args': ('arg1',), 'kwargs': (('arg2', 'arg2value'),)}

    def test_serialize(self):
        result = cPickle.loads(self.test_payload.serialize().decode('string_escape'))
        self.assertItemsEqual(result, self.serialized_payload)

    def test_deserialize(self):
        fn, args, kwargs = Payload.deserialize(self.test_payload.serialize())
        self.assertEqual(fn, _fn)
        self.assertEqual(args, ('arg1',))
        self.assertItemsEqual(kwargs, {'arg2': 'arg2value'})

    def test_equality_matches(self):
        self.assertEqual(self.test_payload, Payload(_fn, 'arg1', arg2='arg2value'))

    def test_equality_fails_bad_fn(self):
        self.assertNotEqual(self.test_payload, Payload(_fn2, 'arg1', arg2='arg2value'))

    def test_equality_fails_bad_args(self):
        self.assertNotEqual(self.test_payload, Payload(_fn, 'arg3', arg2='arg2value'))

    def test_equality_fails_bad_kwargs(self):
        self.assertNotEqual(self.test_payload, Payload(_fn, 'arg1', arg2='wrongvalue'))

    def test_equality_fails_wrong_class(self):
        self.assertNotEqual(self.test_payload, object())
Example #2
0
 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'), )
     }
Example #3
0
class TestPayload(DustyTestCase):
    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_serialize(self):
        result = json.loads(self.test_payload.serialize().decode('string_escape'))
        self.assertItemsEqual(result, self.serialized_payload)

    def test_deserialize(self):
        payload = Payload.deserialize(self.test_payload.serialize())
        fn_key, client_version, suppress_warnings, args, kwargs = payload['fn_key'], payload['client_version'], payload['suppress_warnings'], payload['args'], payload['kwargs']
        self.assertEqual(get_payload_function(fn_key), _fn)
        self.assertEqual(client_version, VERSION)
        self.assertEqual(set(args), set(('arg1',)))
        self.assertItemsEqual(kwargs, {'arg2': 'arg2value'})
        self.assertEqual(suppress_warnings, False)

    def test_equality_matches(self):
        self.assertEqual(self.test_payload, Payload(_fn, 'arg1', arg2='arg2value'))

    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_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_fn(self):
        self.assertNotEqual(self.test_payload, Payload(_fn2, 'arg1', arg2='arg2value'))

    def test_equality_fails_bad_args(self):
        self.assertNotEqual(self.test_payload, Payload(_fn, 'arg3', arg2='arg2value'))

    def test_equality_fails_bad_kwargs(self):
        self.assertNotEqual(self.test_payload, Payload(_fn, 'arg1', arg2='wrongvalue'))

    def test_equality_fails_wrong_class(self):
        self.assertNotEqual(self.test_payload, object())

    def test_get_payload_function_succeeds(self):
        self.assertEqual(_fn, get_payload_function(_function_key(_fn)))

    def test_get_payload_function_raises(self):
        with self.assertRaises(RuntimeError):
            get_payload_function(_function_key(_fn2))
Example #4
0
 def test_deserialize(self):
     payload = Payload.deserialize(self.test_payload.serialize())
     fn_key, client_version, suppress_warnings, args, kwargs = payload['fn_key'], payload['client_version'], payload['suppress_warnings'], payload['args'], payload['kwargs']
     self.assertEqual(get_payload_function(fn_key), _fn)
     self.assertEqual(client_version, VERSION)
     self.assertEqual(set(args), set(('arg1',)))
     self.assertItemsEqual(kwargs, {'arg2': 'arg2value'})
     self.assertEqual(suppress_warnings, False)
Example #5
0
 def test_deserialize(self):
     payload = Payload.deserialize(self.test_payload.serialize())
     fn_key, client_version, suppress_warnings, args, kwargs = payload[
         'fn_key'], payload['client_version'], payload[
             'suppress_warnings'], payload['args'], payload['kwargs']
     self.assertEqual(get_payload_function(fn_key), _fn)
     self.assertEqual(client_version, VERSION)
     self.assertEqual(set(args), set(('arg1', )))
     self.assertItemsEqual(kwargs, {'arg2': 'arg2value'})
     self.assertEqual(suppress_warnings, False)
Example #6
0
 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'),)}
Example #7
0
 def test_deserialize(self):
     fn, args, kwargs = Payload.deserialize(self.test_payload.serialize())
     self.assertEqual(fn, _fn)
     self.assertEqual(args, ('arg1',))
     self.assertItemsEqual(kwargs, {'arg2': 'arg2value'})
Example #8
0
 def setUp(self):
     super(TestPayload, self).setUp()
     self.test_payload = Payload(_fn, 'arg1', arg2='arg2value')
     self.serialized_payload = {'fn': _fn, 'args': ('arg1',), 'kwargs': (('arg2', 'arg2value'),)}
Example #9
0
 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'))
Example #10
0
 def test_equality_fails_bad_kwargs(self):
     self.assertNotEqual(self.test_payload,
                         Payload(_fn, 'arg1', arg2='wrongvalue'))
Example #11
0
 def test_equality_fails_bad_args(self):
     self.assertNotEqual(self.test_payload,
                         Payload(_fn, 'arg3', arg2='arg2value'))
Example #12
0
 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)
Example #13
0
 def test_equality_fails_bad_suppress(self):
     payload = Payload(_fn, 'arg1', arg2='arg2value')
     payload.suppress_warnings = True
     self.assertNotEqual(self.test_payload, payload)
Example #14
0
 def test_equality_matches(self):
     self.assertEqual(self.test_payload,
                      Payload(_fn, 'arg1', arg2='arg2value'))
Example #15
0
 def test_equality_fails_bad_suppress(self):
     payload = Payload(_fn, 'arg1', arg2='arg2value')
     payload.suppress_warnings = True
     self.assertNotEqual(self.test_payload, payload)
Example #16
0
 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'))
Example #17
0
 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)
Example #18
0
class TestPayload(DustyTestCase):
    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_serialize(self):
        result = json.loads(
            self.test_payload.serialize().decode('string_escape'))
        self.assertItemsEqual(result, self.serialized_payload)

    def test_deserialize(self):
        payload = Payload.deserialize(self.test_payload.serialize())
        fn_key, client_version, suppress_warnings, args, kwargs = payload[
            'fn_key'], payload['client_version'], payload[
                'suppress_warnings'], payload['args'], payload['kwargs']
        self.assertEqual(get_payload_function(fn_key), _fn)
        self.assertEqual(client_version, VERSION)
        self.assertEqual(set(args), set(('arg1', )))
        self.assertItemsEqual(kwargs, {'arg2': 'arg2value'})
        self.assertEqual(suppress_warnings, False)

    def test_equality_matches(self):
        self.assertEqual(self.test_payload,
                         Payload(_fn, 'arg1', arg2='arg2value'))

    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_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_fn(self):
        self.assertNotEqual(self.test_payload,
                            Payload(_fn2, 'arg1', arg2='arg2value'))

    def test_equality_fails_bad_args(self):
        self.assertNotEqual(self.test_payload,
                            Payload(_fn, 'arg3', arg2='arg2value'))

    def test_equality_fails_bad_kwargs(self):
        self.assertNotEqual(self.test_payload,
                            Payload(_fn, 'arg1', arg2='wrongvalue'))

    def test_equality_fails_wrong_class(self):
        self.assertNotEqual(self.test_payload, object())

    def test_get_payload_function_succeeds(self):
        self.assertEqual(_fn, get_payload_function(function_key(_fn)))

    def test_get_payload_function_raises(self):
        with self.assertRaises(RuntimeError):
            get_payload_function(function_key(_fn2))