def test_close(self): connection = Mock() channel = Mock() receiver = Mock() # test reader = Reader(None, '') reader.connection = connection reader.channel = channel reader.receiver = receiver reader.is_open = Mock(return_value=True) reader.close() # validation receiver.close.assert_called_once_with() channel.close.assert_called_once_with() self.assertFalse(connection.close.called)
def test_repair(self, receiver, connection): url = 'test-url' queue = Queue('test-queue') receiver.return_value.open.return_value = receiver.return_value # test reader = Reader(queue, 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.channel.assert_called_once_with() receiver.assert_called_once_with(reader) self.assertEqual(reader.channel, connection.return_value.channel.return_value) self.assertEqual(reader.receiver, receiver.return_value)