def test_is_open(self): url = 'test-url' node = Mock(address='test') reader = Reader(node, url=url) # closed self.assertFalse(reader.is_open()) # open reader.receiver = Mock() self.assertTrue(reader.is_open())
def test_ack(self): node = Mock(address='test') url = 'test-url' # test reader = Reader(node, url=url) reader.receiver = Mock() reader.ack(None) # validation reader.receiver.accept.assert_called_once_with()
def test_open_already(self): url = 'test-url' node = Mock(address='test') # test reader = Reader(node, url) reader.is_open = Mock(return_value=True) reader.open() # validation self.assertFalse(reader.connection.open.called)
def test_reject_queued(self): node = Mock(address='test') url = 'test-url' # test reader = Reader(node, url=url) reader.receiver = Mock() reader.reject(None, requeue=True) # validation reader.receiver.release.assert_called_once_with()
def test_get_empty(self): node = Mock(address='test') url = 'test-url' # test reader = Reader(node, url=url) reader.receiver = Mock() reader.receiver.receive.side_effect = Timeout message = reader.get(10) # validation reader.receiver.receive.assert_called_once_with(10) self.assertEqual(message, None)
def test_open(self, connection): url = 'test-url' node = Mock(address='test') # test reader = Reader(node, url) reader.is_open = Mock(return_value=False) reader.open() # validation connection.return_value.open.assert_called_once_with() connection.return_value.receiver.assert_called_once_with(node.address) self.assertEqual(reader.receiver, reader.connection.receiver.return_value)
def test_repair(self, connection): url = 'test-url' node = Mock(address='test') # test reader = Reader(node, url) reader.close = Mock() reader.repair() # validation reader.close.assert_called_once_with() reader.connection.close.assert_called_once_with() connection.return_value.open.assert_called_once_with() connection.return_value.receiver.assert_called_once_with(node.address) self.assertEqual(reader.receiver, reader.connection.receiver.return_value)
def test_get(self): node = Mock(address='test') received = Mock(body='<body/>') url = 'test-url' # test reader = Reader(node, url=url) reader.receiver = Mock() reader.receiver.receive.return_value = received message = reader.get(10) # validation reader.receiver.receive.assert_called_once_with(10) self.assertTrue(isinstance(message, Message)) self.assertEqual(message._reader, reader) self.assertEqual(message._impl, received) self.assertEqual(message._body, received.body)
def test_init(self, connection): node = Mock(address='test') url = 'test-url' # test reader = Reader(node, url=url) # validation connection.assert_called_once_with(url) self.assertTrue(isinstance(reader, BaseReader)) self.assertEqual(reader.url, url) self.assertEqual(reader.connection, connection.return_value) self.assertEqual(reader.node, node) self.assertEqual(reader.receiver, None)
def test_close(self): connection = Mock() receiver = Mock() receiver.close.side_effect = KeyError node = Mock(address='test') # test reader = Reader(node, '') reader.connection = connection reader.receiver = receiver reader.is_open = Mock(return_value=True) reader.close() # validation receiver.close.assert_called_once_with() self.assertFalse(connection.close.called)