File: test_worker_thread.py

package info (click to toggle)
circuits 3.2.3-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 2,980 kB
  • sloc: python: 17,583; javascript: 3,226; makefile: 100
file content (51 lines) | stat: -rw-r--r-- 838 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
"""Workers Tests"""

import pytest

from circuits import Worker, task


task.complete = True


@pytest.fixture()
def worker(request, manager, watcher):
    worker = Worker().register(manager)
    assert watcher.wait('registered')

    def finalizer():
        worker.unregister()
        assert watcher.wait('unregistered')

    request.addfinalizer(finalizer)

    return worker


def f():
    x = 0
    i = 0
    while i < 1000000:
        x += 1
        i += 1
    return x


def add(a, b):
    return a + b


def test(manager, watcher, worker):
    x = manager.fire(task(f))
    assert watcher.wait('task_complete')

    assert x.result
    assert x.value == 1000000


def test_args(manager, watcher, worker):
    x = manager.fire(task(add, 1, 2))
    assert watcher.wait('task_complete')

    assert x.result
    assert x.value == 3