def test_purge(self, _find, _reader):
name = 'test'
_find.return_value = Mock()
queued = [Mock(), Mock(), None]
def _open():
_reader.return_value.open()
return _reader.return_value
def _close(*unused):
_reader.return_value.close()
_reader.return_value.__enter__.side_effect = _open
_reader.return_value.__exit__.side_effect = _close
_reader.return_value.get.side_effect = queued
# test
queue = Queue(name)
queue.purge(TEST_URL)
# validation
_reader.assert_called_once_with(queue, url=TEST_URL)
_reader.return_value.open.assert_called_once_with()
_reader.return_value.close.assert_called_once_with()
queued[0].ack.assert_called_once_with()
queued[1].ack.assert_called_once_with()
def test_purge(self, _find, _reader):
name = 'test'
_find.return_value = Mock()
queued = [
Mock(),
Mock(),
None
]
def _open():
_reader.return_value.open()
return _reader.return_value
def _close(*unused):
_reader.return_value.close()
_reader.return_value.__enter__.side_effect = _open
_reader.return_value.__exit__.side_effect = _close
_reader.return_value.get.side_effect = queued
# test
queue = Queue(name)
queue.purge(TEST_URL)
# validation
_reader.assert_called_once_with(queue, url=TEST_URL)
_reader.return_value.open.assert_called_once_with()
_reader.return_value.close.assert_called_once_with()
queued[0].ack.assert_called_once_with()
queued[1].ack.assert_called_once_with()
