mock.sentinel.queue

Here are the examples of the python api mock.sentinel.queue taken from open source projects. By voting up you can indicate which examples are most useful and appropriate.

14 Examples 7

Example 1

Project: pyon Source File: test_procs.py
    def test__cleanup_method(self):
        ep = Mock()
        ep._chan._queue_auto_delete = False

        self.pm._cleanup_method(sentinel.queue, ep)

        ch = self.container.node.channel.return_value
        ch._destroy_queue.assert_called_once_with()
        self.assertIsInstance(ch._recv_name, NameTrio)
        self.assertIn(str(sentinel.queue), str(ch._recv_name))

Example 2

Project: pyon Source File: test_procs.py
    @patch('pyon.container.procs.log')
    def test__cleanup_method_raises_error(self, mocklog):
        ep = Mock()
        ep._chan._queue_auto_delete = False
        ch = self.container.node.channel.return_value
        ch._destroy_queue.side_effect = TransportError

        self.pm._cleanup_method(sentinel.queue, ep)

        self.assertEquals(mocklog.warn.call_count, 1)

Example 3

Project: pyon Source File: test_transport.py
    def test_declare_queue_impl(self):
        retqueue = self.tp.declare_queue_impl(sentinel.queue)

        self.tp._sync_call.assert_called_once_with(self.tp._client.queue_declare,
                                                   'callback',
                                                   queue=sentinel.queue,
                                                   auto_delete=True,
                                                   durable=False,
                                                   arguments={})

        self.assertEquals(retqueue, self.tp._sync_call.return_value.method.queue)

Example 4

Project: pyon Source File: test_transport.py
Function: test_delete_queue
    def test_delete_queue(self):
        self.tp.delete_queue_impl(sentinel.queue)

        self.tp._sync_call.assert_called_once_with(self.tp._client.queue_delete,
                                                   'callback',
                                                   queue=sentinel.queue)

Example 5

Project: pyon Source File: test_transport.py
    def test_get_stats_impl(self):
        mo = Mock()
        self.tp._sync_call.return_value = mo
        ret = self.tp.get_stats_impl(sentinel.queue)

        self.tp._sync_call.assert_called_once_with(self.tp._client.queue_declare,
                                                   'callback',
                                                   queue=sentinel.queue,
                                                   passive=True)

        self.assertEquals(ret, (mo.method.message_count, mo.method.consumer_count))

Example 6

Project: pyon Source File: test_transport.py
    def test_purge_impl(self):
        self.tp.purge_impl(sentinel.queue)

        self.tp._sync_call.assert_called_once_with(self.tp._client.queue_purge,
                                                   'callback',
                                                   queue=sentinel.queue)

Example 7

Project: pyon Source File: test_transport.py
Function: test_start_consume
    def test_start_consume(self):
        self.lr._gl_pool = Mock()
        self.lr.declare_queue(sentinel.queue)
        ctag = self.lr.start_consume(sentinel.callback, sentinel.queue)

        self.assertIn(sentinel.queue, self.lr._consumers)
        self.assertIn(ctag, self.lr._consumers_by_ctag)

        self.assertEquals(self.lr._consumers[sentinel.queue], [(ctag,
                                                                sentinel.callback,
                                                                False,
                                                                False,
                                                                self.lr._gl_pool.spawn.return_value)])

Example 8

Project: pyon Source File: test_transport.py
Function: test_stop_consume
    def test_stop_consume(self):
        self.lr._gl_pool = Mock()
        self.lr.declare_queue(sentinel.queue)
        ctag = self.lr.start_consume(sentinel.callback, sentinel.queue)

        self.lr.stop_consume(ctag)

        self.assertNotIn(ctag, self.lr._consumers_by_ctag)
        self.lr._gl_pool.spawn().join.assert_called_once_with(timeout=5)
        self.lr._gl_pool.spawn().kill.assert_called_once_with()
        self.assertEquals(len(self.lr._consumers[sentinel.queue]), 0)

Example 9

Project: pyon Source File: test_transport.py
    def test_reject_requeue(self):
        q = Mock()
        self.lr._queues[sentinel.queue] = q
        self.lr._unacked[sentinel.dtag] = (None, sentinel.queue, sentinel.m)

        self.lr.reject(sentinel.dtag, requeue=True)
        self.assertEquals(len(self.lr._unacked), 0)
        q.put.assert_called_once_with(sentinel.m)

Example 10

Project: pyon Source File: test_transport.py
Function: test_get_stats
    def test_get_stats(self):
        self.lr.declare_queue(sentinel.queue)
        self.lr._queues[sentinel.queue].put(sentinel.m)

        mc, cc = self.lr.get_stats(sentinel.queue)

        self.assertEquals((mc, cc), (1, 0))

Example 11

Project: pyon Source File: test_transport.py
    def test_get_stats_with_consumers(self):
        self.lr.declare_queue(sentinel.queue)
        self.lr._queues[sentinel.queue].put(sentinel.m)
        self.lr._consumers[sentinel.queue] = [sentinel.ctag]

        mc, cc = self.lr.get_stats(sentinel.queue)

        self.assertEquals((mc, cc), (1, 1))

Example 12

Project: pyon Source File: test_transport.py
Function: test_purge
    def test_purge(self):
        self.lr.declare_queue(sentinel.queue)
        self.lr._queues[sentinel.queue].put(sentinel.m)
        self.assertEquals(self.lr._queues[sentinel.queue].qsize(), 1)

        self.lr.purge(sentinel.queue)

        self.assertEquals(self.lr._queues[sentinel.queue].qsize(), 0)

Example 13

Project: pyon Source File: test_transport.py
Function: test_declare_queue
    def test_declare_queue(self):
        self.lr.declare_queue(sentinel.queue)
        self.assertIn(sentinel.queue, self.lr._queues)

Example 14

Project: pyon Source File: test_transport.py
Function: test_delete_queue
    def test_delete_queue(self):
        self.lr.declare_queue(sentinel.queue)
        self.lr.delete_queue(sentinel.queue)
        self.assertNotIn(sentinel.queue, self.lr._queues)