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_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_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))
