コード例 #1
0
ファイル: test_container.py プロジェクト: shubham90/pulp
    def test_run(self, fake_feed):
        # test
        queue = RequestQueue(Mock())
        queue.run()

        # validation
        queue.downloader.download.assert_called_with(fake_feed())
コード例 #2
0
ファイル: test_container.py プロジェクト: shubham90/pulp
    def test_halt(self):
        # test
        queue = RequestQueue(Mock())
        queue.halt()

        # validation
        self.assertTrue(queue._halted)
コード例 #3
0
ファイル: test_container.py プロジェクト: shubham90/pulp
    def test_put(self, fake_queue):
        # test
        item = Mock()
        queue = RequestQueue(Mock())
        queue.put(item)

        # validation
        fake_queue().put.assert_called_with(item, timeout=3)
コード例 #4
0
ファイル: test_container.py プロジェクト: shubham90/pulp
    def test_run_with_exception(self, fake_feed, fake_drain):
        # test
        queue = RequestQueue(Mock())
        queue.downloader.download.side_effect = ValueError()
        queue.run()

        # validation
        queue.downloader.download.assert_called_with(fake_feed())
        fake_drain.assert_called_with()
コード例 #5
0
ファイル: test_container.py プロジェクト: shubham90/pulp
    def test_get_halted(self, fake_queue):
        # test
        queue = RequestQueue(Mock())
        queue.halt()
        item = queue.get()

        # validation
        self.assertFalse(fake_queue().get.called)
        self.assertEqual(item, None)
コード例 #6
0
ファイル: test_container.py プロジェクト: shubham90/pulp
    def test_put_halted(self, fake_queue):
        # test
        item = Mock()
        queue = RequestQueue(Mock())
        queue.halt()
        queue.put(item)

        # validation
        self.assertFalse(fake_queue().put.called)
コード例 #7
0
    def test_run(self, fake_feed):
        canceled = Mock()
        canceled.is_set.return_value = False

        # test
        queue = RequestQueue(canceled, Mock())
        queue.run()

        # validation
        queue.downloader.download.assert_called_with(fake_feed())
コード例 #8
0
ファイル: test_container.py プロジェクト: shubham90/pulp
    def test_get_empty(self, fake_queue):
        fake_queue().get.side_effect = SideEffect([Empty(), Empty(), 123])

        # test
        queue = RequestQueue(Mock())
        item = queue.get()

        # validation
        self.assertEqual(fake_queue().get.call_count, 3)
        self.assertEqual(item, 123)
コード例 #9
0
ファイル: test_container.py プロジェクト: shubham90/pulp
    def test_get(self, fake_queue):
        fake_queue().get.return_value = 123

        # test
        queue = RequestQueue(Mock())
        item = queue.get()

        # validation
        fake_queue().get.assert_called_with(timeout=3)
        self.assertEqual(item, 123)
コード例 #10
0
ファイル: test_container.py プロジェクト: shubham90/pulp
    def test_put_full(self, fake_queue):
        fake_queue().put.side_effect = SideEffect([Full(), Full(), None])

        # test
        item = Mock()
        queue = RequestQueue(Mock())
        queue.put(item)

        # validation
        self.assertEqual(fake_queue().put.call_count, 3)
コード例 #11
0
    def test_halt(self):
        canceled = Mock()
        canceled.is_set.return_value = False

        # test
        queue = RequestQueue(canceled, Mock())
        queue.halt()

        # validation
        self.assertTrue(queue._halted)
コード例 #12
0
    def test_put(self, fake_queue):
        canceled = Mock()
        canceled.is_set.return_value = False

        # test
        item = Mock()
        queue = RequestQueue(canceled, Mock())
        queue.put(item)

        # validation
        fake_queue().put.assert_called_with(item, timeout=3)
コード例 #13
0
    def test_put_full(self, fake_queue):
        canceled = Mock()
        canceled.is_set.return_value = False
        fake_queue().put.side_effect = SideEffect([Full(), Full(), None])

        # test
        item = Mock()
        queue = RequestQueue(canceled, Mock())
        queue.put(item)

        # validation
        self.assertEqual(fake_queue().put.call_count, 3)
コード例 #14
0
    def test_put_halted(self, fake_queue):
        canceled = Mock()
        canceled.is_set.return_value = False

        # test
        item = Mock()
        queue = RequestQueue(canceled, Mock())
        queue.halt()
        queue.put(item)

        # validation
        self.assertFalse(fake_queue().put.called)
コード例 #15
0
    def test_get_empty(self, fake_queue):
        canceled = Mock()
        canceled.is_set.return_value = False
        fake_queue().get.side_effect = SideEffect([Empty(), Empty(), 123])

        # test
        queue = RequestQueue(canceled, Mock())
        item = queue.get()

        # validation
        self.assertEqual(fake_queue().get.call_count, 3)
        self.assertEqual(item, 123)
コード例 #16
0
    def test_get(self, fake_queue):
        canceled = Mock()
        canceled.is_set.return_value = False
        fake_queue().get.return_value = 123

        # test
        queue = RequestQueue(canceled, Mock())
        item = queue.get()

        # validation
        fake_queue().get.assert_called_with(timeout=3)
        self.assertEqual(item, 123)
コード例 #17
0
    def test_get_halted(self, fake_queue):
        canceled = Mock()
        canceled.is_set.return_value = False

        # test
        queue = RequestQueue(canceled, Mock())
        queue.halt()
        item = queue.get()

        # validation
        self.assertFalse(fake_queue().get.called)
        self.assertEqual(item, None)
コード例 #18
0
    def test_run_with_exception(self, fake_feed, fake_drain):
        canceled = Mock()
        canceled.is_set.return_value = False

        # test
        queue = RequestQueue(canceled, Mock())
        queue.downloader.download.side_effect = ValueError()
        queue.run()

        # validation
        queue.downloader.download.assert_called_with(fake_feed())
        fake_drain.assert_called_with()
コード例 #19
0
ファイル: test_container.py プロジェクト: shubham90/pulp
    def test_init(self, fake_queue, fake_setDaemon):
        source = Mock()
        source.id = 'fake_id'
        source.max_concurrent = 10

        # test
        queue = RequestQueue(source)
        queue.setDaemon = Mock()

        # validation
        fake_queue.assert_called_with(source.max_concurrent)
        fake_setDaemon.assert_called_with(True)
        self.assertEqual(queue._halted, False)
        self.assertEqual(queue.queue, fake_queue())
        self.assertEqual(queue.downloader, source.get_downloader())
コード例 #20
0
ファイル: test_container.py プロジェクト: shubham90/pulp
    def test_drain(self, fake_feed, fake_from):
        queued = [1, 2, 3]
        fake_feed.return_value = queued
        fake_from.side_effect = queued

        # test
        queue = RequestQueue(Mock())
        queue.drain()

        # validation
        calls = fake_from.call_args_list
        self.assertEqual(len(calls), len(queued))
        for i, request in enumerate(queued):
            self.assertEqual(calls[i][0][0], request)
        calls = queue.downloader.fire_download_failed.call_args_list
        for i, request in enumerate(queued):
            self.assertEqual(calls[i][0][0], request)
コード例 #21
0
    def test_construction(self, fake_queue, fake_setDaemon):
        source = Mock()
        source.id = 'fake_id'
        source.max_concurrent = 10

        # test
        canceled = Mock()
        canceled.is_set.return_value = False
        queue = RequestQueue(canceled, source)
        queue.setDaemon = Mock()

        # validation
        fake_queue.assert_called_with(source.max_concurrent)
        fake_setDaemon.assert_called_with(True)
        self.assertEqual(queue._halted, False)
        self.assertEqual(queue.canceled, canceled)
        self.assertEqual(queue.queue, fake_queue())
        self.assertEqual(queue.downloader, source.get_downloader())
コード例 #22
0
    def test__run(self):
        canceled = Mock()
        canceled.is_set.return_value = False

        # test
        queue = RequestQueue(canceled, Mock())

        # validation
        # all good
        queue._halted = False
        canceled.is_set.return_value = False
        self.assertTrue(queue._run)
        # halted only
        queue._halted = True
        canceled.is_set.return_value = False
        self.assertFalse(queue._run)
        # canceled only
        queue._halted = False
        canceled.is_set.return_value = True
        self.assertFalse(queue._run)
        # both
        queue._halted = True
        canceled.is_set.return_value = True
        self.assertFalse(queue._run)