Python TaskManagerの例

コード例 #1

ファイル: test_task_manager.py プロジェクト: snorp/socorro

 def test_executor_identity(self):
     config = DotDict()
     config.logger = self.logger
     tm = TaskManager(
         config,
         job_source_iterator=range(1),
     )
     tm._pid = 666
     eq_(tm.executor_identity(), '666-MainThread')

コード例 #2

ファイル: test_task_manager.py プロジェクト: adngdb/socorro

    def test_executor_identity(self):
        config = DotDict()
        config.logger = self.logger
        tm = TaskManager(
            config,
            job_source_iterator=range(1),

        )
        tm._pid = 666
        eq_(tm.executor_identity(), '666-MainThread')

コード例 #3

ファイル: test_task_manager.py プロジェクト: snorp/socorro

    def test_get_iterator(self):
        config = DotDict()
        config.logger = self.logger
        config.quit_on_empty_queue = False

        tm = TaskManager(
            config,
            job_source_iterator=range(1),
        )
        eq_(tm._get_iterator(), [0])

        def an_iter(self):
            for i in range(5):
                yield i

        tm = TaskManager(
            config,
            job_source_iterator=an_iter,
        )
        eq_([x for x in tm._get_iterator()], [0, 1, 2, 3, 4])

        class X(object):
            def __init__(self, config):
                self.config = config

            def __iter__(self):
                for key in self.config:
                    yield key

        tm = TaskManager(config, job_source_iterator=X(config))
        eq_([x for x in tm._get_iterator()], [y for y in config.keys()])

コード例 #4

ファイル: test_task_manager.py プロジェクト: kellpossible/socorro

    def test_get_iterator(self):
        config = DotDict()
        config.logger = self.logger
        config.quit_on_empty_queue = False

        tm = TaskManager(
            config,
            job_source_iterator=range(1),
        )
        assert tm._get_iterator() == [0]

        def an_iter(self):
            for i in range(5):
                yield i

        tm = TaskManager(
            config,
            job_source_iterator=an_iter,
        )
        assert list(tm._get_iterator()) == [0, 1, 2, 3, 4]

        class X(object):
            def __init__(self, config):
                self.config = config

            def __iter__(self):
                for key in self.config:
                    yield key

        tm = TaskManager(config, job_source_iterator=X(config))
        assert list(tm._get_iterator()) == list(config.keys())

コード例 #5

ファイル: test_task_manager.py プロジェクト: mozilla/socorro

    def test_blocking_start_with_quit_on_empty(self):
        config = DotDict()
        config.idle_delay = 1
        config.quit_on_empty_queue = True

        tm = TaskManager(config, task_func=Mock())

        waiting_func = Mock()

        tm.blocking_start(waiting_func=waiting_func)

        assert tm.task_func.call_count == 10
        assert waiting_func.call_count == 0

コード例 #6

ファイル: test_task_manager.py プロジェクト: glandium/socorro

    def test_blocking_start_with_quit_on_empty(self):
        config = DotDict()
        config.idle_delay = 1
        config.quit_on_empty_queue = True

        tm = TaskManager(config, task_func=Mock())

        waiting_func = Mock()

        tm.blocking_start(waiting_func=waiting_func)

        assert tm.task_func.call_count == 10
        assert waiting_func.call_count == 0

コード例 #7

ファイル: test_task_manager.py プロジェクト: snorp/socorro

    def test_blocking_start_with_quit_on_empty(self):
        config = DotDict()
        config.logger = self.logger
        config.idle_delay = 1
        config.quit_on_empty_queue = True

        tm = TaskManager(config, task_func=Mock())

        waiting_func = Mock()

        tm.blocking_start(waiting_func=waiting_func)

        eq_(tm.task_func.call_count, 10)
        eq_(waiting_func.call_count, 0)

コード例 #8

ファイル: test_task_manager.py プロジェクト: snorp/socorro

    def test_blocking_start_with_quit_on_empty(self):
        config = DotDict()
        config.logger = self.logger
        config.idle_delay = 1
        config.quit_on_empty_queue = True

        tm = TaskManager(config, task_func=Mock())

        waiting_func = Mock()

        tm.blocking_start(waiting_func=waiting_func)

        eq_(tm.task_func.call_count, 10)
        eq_(waiting_func.call_count, 0)

コード例 #9

ファイル: test_task_manager.py プロジェクト: glandium/socorro

    def test_constuctor1(self):
        config = DotDict()
        config.quit_on_empty_queue = False

        tm = TaskManager(config)
        assert tm.config == config
        assert tm.task_func == default_task_func
        assert tm.quit is False

コード例 #10

ファイル: test_task_manager.py プロジェクト: walikhan/socorro

 def test_constuctor1(self):
     config = DotDict()
     config.logger = self.logger
     tm = TaskManager(config)
     ok_(tm.config == config)
     ok_(tm.logger == self.logger)
     ok_(tm.task_func == default_task_func)
     ok_(tm.quit == False)

コード例 #11

ファイル: test_task_manager.py プロジェクト: snorp/socorro

    def test_get_iterator(self):
        config = DotDict()
        config.logger = self.logger
        config.quit_on_empty_queue = False

        tm = TaskManager(config, job_source_iterator=range(1))
        eq_(tm._get_iterator(), [0])

        def an_iter(self):
            for i in range(5):
                yield i

        tm = TaskManager(config, job_source_iterator=an_iter)
        eq_([x for x in tm._get_iterator()], [0, 1, 2, 3, 4])

        class X(object):
            def __init__(self, config):
                self.config = config

            def __iter__(self):
                for key in self.config:
                    yield key

        tm = TaskManager(config, job_source_iterator=X(config))
        eq_([x for x in tm._get_iterator()], [y for y in config.keys()])

コード例 #12

ファイル: test_task_manager.py プロジェクト: m8ttyB/socorro

    def test_get_iterator(self):
        config = DotDict()
        config.logger = self.logger
        config.quit_on_empty_queue = False

        tm = TaskManager(
            config,
            job_source_iterator=range(1),
        )
        assert tm._get_iterator() == [0]

        def an_iter(self):
            for i in range(5):
                yield i

        tm = TaskManager(
            config,
            job_source_iterator=an_iter,
        )
        assert list(tm._get_iterator()) == [0, 1, 2, 3, 4]

        class X(object):

            def __init__(self, config):
                self.config = config

            def __iter__(self):
                for key in self.config:
                    yield key

        tm = TaskManager(
            config,
            job_source_iterator=X(config)
        )
        assert list(tm._get_iterator()) == list(config.keys())

コード例 #13

    def test_get_iterator(self):
        config = DotDict()
        config.quit_on_empty_queue = False

        tm = TaskManager(config, job_source_iterator=range(1))
        assert list(tm._get_iterator()) == [0]

        def an_iter(self):
            yield from range(5)

        tm = TaskManager(config, job_source_iterator=an_iter)
        assert list(tm._get_iterator()) == [0, 1, 2, 3, 4]

        class X:
            def __init__(self, config):
                self.config = config

            def __iter__(self):
                yield from self.config

        tm = TaskManager(config, job_source_iterator=X(config))
        assert list(tm._get_iterator()) == list(config.keys())

コード例 #14

ファイル: test_task_manager.py プロジェクト: FishingCactus/socorro

    def test_get_iterator(self):
        config = DotDict()
        config.logger = self.logger
        tm = TaskManager(
            config,
            job_source_iterator=range(1),
        )
        self.assertEqual(tm._get_iterator(), [0])

        def an_iter(self):
            for i in range(5):
                yield i

        tm = TaskManager(
            config,
            job_source_iterator=an_iter,
        )
        self.assertEqual(
            [x for x in tm._get_iterator()],
            [0, 1, 2, 3, 4]
        )

        class X(object):
            def __init__(self, config):
                self.config = config

            def __iter__(self):
                for key in self.config:
                    yield key

        tm = TaskManager(
            config,
            job_source_iterator=X(config)
        )
        self.assertEqual(
            [x for x in tm._get_iterator()],
            [y for y in config.keys()]
        )