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()
def test_delete(self, _find):
plugin = Mock()
_find.return_value = plugin
name = 'test'
# test
queue = Queue(name)
queue.delete(TEST_URL)
# validation
plugin.Queue.assert_called_with(name)
impl = plugin.Queue()
impl.delete.assert_called_with(TEST_URL)
def test_delete(self, _find):
plugin = Mock()
_find.return_value = plugin
name = 'test'
# test
queue = Queue(name)
queue.delete(TEST_URL)
# validation
plugin.Queue.assert_called_with(name)
impl = plugin.Queue()
impl.delete.assert_called_with(TEST_URL)
def test_init(self):
name = 'test'
queue = Queue(name)
self.assertEqual(queue.name, name)
self.assertEqual(queue.exclusive, False)
self.assertEqual(queue.auto_delete, False)
self.assertEqual(queue.expiration, 0)
def test_declare(self, _find):
plugin = Mock()
_find.return_value = plugin
name = 'test'
queue = Queue(name)
queue.durable = 1
queue.auto_delete = 2
queue.expiration = 3
queue.exclusive = 4
# test
queue.declare(TEST_URL)
# validation
plugin.Queue.assert_called_with(name)
impl = plugin.Queue()
impl.declare.assert_called_with(TEST_URL)
self.assertEqual(impl.durable, queue.durable)
self.assertEqual(impl.auto_delete, queue.auto_delete)
self.assertEqual(impl.expiration, queue.expiration)
self.assertEqual(impl.exclusive, queue.exclusive)
def test_all(self):
cat = Node('cat')
dog = Queue('dog')
domain = _Domain()
domain.add(cat)
# add
domain.add(dog)
self.assertEqual(domain.content, {'Node::cat': cat, 'Queue::dog': dog})
# contains
self.assertTrue(domain.contains(dog))
self.assertFalse(domain.contains(Node('dog')))
self.assertTrue(dog in domain)
# find
self.assertEqual(domain.find(dog.domain_id), dog)
# not found
self.assertRaises(NotFound, domain.find, '123')
# len
self.assertEqual(len(domain), 2)
# delete
domain.delete(dog)
self.assertEqual(domain.content, {'Node::cat': cat})
def test_declare(self, _find):
plugin = Mock()
_find.return_value = plugin
name = 'test'
queue = Queue(name)
queue.durable = 1
queue.auto_delete = 2
queue.expiration = 3
queue.exclusive = 4
# test
queue.declare(TEST_URL)
# validation
plugin.Queue.assert_called_with(name)
impl = plugin.Queue()
impl.declare.assert_called_with(TEST_URL)
self.assertEqual(impl.durable, queue.durable)
self.assertEqual(impl.auto_delete, queue.auto_delete)
self.assertEqual(impl.expiration, queue.expiration)
self.assertEqual(impl.exclusive, queue.exclusive)