def test_send(self, _find, auth, uuid4, document): _impl = Mock() _impl.send.return_value = '456' plugin = Mock() plugin.Sender.return_value = _impl _find.return_value = plugin uuid4.return_value = '<uuid>' address = 'amq.direct/bar' ttl = 234 body = {'A': 1, 'B': 2} # test producer = Producer(TEST_URL) producer.authenticator = Mock() sn = producer.send(address, ttl=ttl, **body) # validation document.assert_called_once_with( sn=str(uuid4.return_value), version=VERSION, routing=(None, address) ) unsigned = document.return_value auth.sign.assert_called_once_with( producer.authenticator, unsigned.__iadd__.return_value.dump.return_value) _impl.send.assert_called_once_with(address, auth.sign.return_value, ttl) self.assertEqual(sn, uuid4.return_value)
def test_send(self, _find, auth, uuid4, document): _impl = Mock() _impl.send.return_value = '456' plugin = Mock() plugin.Sender.return_value = _impl _find.return_value = plugin uuid4.return_value = '<uuid>' address = 'amq.direct/bar' ttl = 234 body = {'A': 1, 'B': 2} # test producer = Producer(TEST_URL) producer.authenticator = Mock() sn = producer.send(address, ttl=ttl, **body) # validation document.assert_called_once_with(sn=str(uuid4.return_value), version=VERSION, routing=(None, address)) unsigned = document.return_value auth.sign.assert_called_once_with( producer.authenticator, unsigned.__iadd__.return_value.dump.return_value) _impl.send.assert_called_once_with(address, auth.sign.return_value, ttl) self.assertEqual(sn, uuid4.return_value)
def test_repair(self, _find): _impl = Mock() plugin = Mock() plugin.Sender.return_value = _impl _find.return_value = plugin url = TEST_URL producer = Producer(url) producer.repair() _impl.repair.assert_called_with()
def test_close(self, _find): _impl = Mock() plugin = Mock() plugin.Sender.return_value = _impl _find.return_value = plugin url = TEST_URL producer = Producer(url) # soft producer.close() _impl.close.assert_called_with()
def test_is_open(self, _find): _impl = Mock() plugin = Mock() plugin.Sender.return_value = _impl _find.return_value = plugin url = TEST_URL producer = Producer(url) is_open = producer.is_open() _impl.is_open.assert_called_with() self.assertEqual(is_open, _impl.is_open.return_value)
def test_init(self, _find): _impl = Mock() plugin = Mock() plugin.Sender.return_value = _impl _find.return_value = plugin url = TEST_URL producer = Producer(url) _find.assert_called_with(url) self.assertEqual(producer.url, url) self.assertEqual(producer.authenticator, None) self.assertEqual(producer._impl, _impl) self.assertTrue(isinstance(producer, Messenger))