コード例 #1
0
ファイル: test_control.py プロジェクト: Scalr/celery
    def test_stop(self):
        parent = Mock()
        g = gPidbox(parent)
        stopped = g._node_stopped = Mock()
        shutdown = g._node_shutdown = Mock()
        close_chan = g._close_channel = Mock()

        g.stop(parent)
        shutdown.set.assert_called_with()
        stopped.wait.assert_called_with()
        close_chan.assert_called_with(parent)
        assert g._node_stopped is None
        assert g._node_shutdown is None

        close_chan.reset()
        g.stop(parent)
        close_chan.assert_called_with(parent)
コード例 #2
0
    def test_stop(self):
        parent = Mock()
        g = gPidbox(parent)
        stopped = g._node_stopped = Mock()
        shutdown = g._node_shutdown = Mock()
        close_chan = g._close_channel = Mock()

        g.stop(parent)
        shutdown.set.assert_called_with()
        stopped.wait.assert_called_with()
        close_chan.assert_called_with(parent)
        self.assertIsNone(g._node_stopped)
        self.assertIsNone(g._node_shutdown)

        close_chan.reset()
        g.stop(parent)
        close_chan.assert_called_with(parent)
コード例 #3
0
ファイル: test_control.py プロジェクト: Gatevin/celery
    def test_loop(self):
        parent = Mock()
        conn = parent.connect.return_value = self.app.connection_for_read()
        drain = conn.drain_events = Mock()
        g = gPidbox(parent)
        parent.connection = Mock()
        do_reset = g._do_reset = Mock()

        call_count = [0]

        def se(*args, **kwargs):
            if call_count[0] > 2:
                g._node_shutdown.set()
            g.reset()
            call_count[0] += 1
        drain.side_effect = se
        g.loop(parent)

        self.assertEqual(do_reset.call_count, 4)
コード例 #4
0
ファイル: test_control.py プロジェクト: xn8x/celery
    def test_loop(self):
        parent = Mock()
        conn = parent.connect.return_value = self.app.connection()
        drain = conn.drain_events = Mock()
        g = gPidbox(parent)
        parent.connection = Mock()
        do_reset = g._do_reset = Mock()

        call_count = [0]

        def se(*args, **kwargs):
            if call_count[0] > 2:
                g._node_shutdown.set()
            g.reset()
            call_count[0] += 1
        drain.side_effect = se
        g.loop(parent)

        self.assertEqual(do_reset.call_count, 4)
コード例 #5
0
 def test_resets(self):
     parent = Mock()
     g = gPidbox(parent)
     g._resets = 100
     g.reset()
     self.assertEqual(g._resets, 101)
コード例 #6
0
ファイル: test_control.py プロジェクト: Scalr/celery
 def test_resets(self):
     parent = Mock()
     g = gPidbox(parent)
     g._resets = 100
     g.reset()
     assert g._resets == 101
コード例 #7
0
ファイル: test_control.py プロジェクト: Gatevin/celery
 def test_resets(self):
     parent = Mock()
     g = gPidbox(parent)
     g._resets = 100
     g.reset()
     self.assertEqual(g._resets, 101)
コード例 #8
0
ファイル: test_control.py プロジェクト: yyz940922/celery
 def test_resets(self):
     parent = Mock()
     g = gPidbox(parent)
     g._resets = 100
     g.reset()
     assert g._resets == 101