def test_open(self, connection): url = 'test-url' # test sender = Sender(url) sender.is_open = Mock(return_value=False) sender.open() # validation connection.return_value.open.assert_called_once_with()
def test_open_already(self): url = 'test-url' # test sender = Sender(url) sender.is_open = Mock(return_value=True) sender.open() # validation self.assertFalse(sender.connection.open.called)
def test_is_open(self): url = 'test-url' sender = Sender(url) sender.connection = Mock() sender.connection.is_open.return_value = False # closed self.assertFalse(sender.is_open()) # open sender.connection = Mock() sender.connection.is_open.return_value = True self.assertTrue(sender.is_open())
def test_init(self, connection): url = 'test-url' # test sender = Sender(url) # validation connection.assert_called_once_with(url) self.assertTrue(isinstance(sender, BaseSender)) self.assertEqual(sender.url, url) self.assertEqual(sender.connection, connection.return_value)
def test_repair(self, connection): url = 'test-url' # test sender = Sender(url) sender.close = Mock() sender.repair() # validation sender.connection.close.assert_called_once_with() sender.close.assert_called_once_with() connection.return_value.open.assert_called_once_with()
def test_close(self): connection = Mock() connection.close.side_effect = ValueError # test sender = Sender(None) sender.connection = connection sender.is_open = Mock(return_value=True) sender.close() # validation self.assertFalse(connection.close.called)
def test_send(self, builder): ttl = 10 address = 'q1' content = 'hello' # test sender = Sender('') sender.durable = 18 sender.connection = Mock() sender.send(address, content, ttl=ttl) # validation builder.assert_called_once_with(content, ttl, sender.durable) sender.connection.sender.assert_called_once_with(address) _sender = sender.connection.sender.return_value _sender.send.assert_called_once_with(builder.return_value) _sender.close.assert_called_once_with()