Name: pytest-mp
Owner: Ansible
Description: multiprocessing.Process(target=pytest_runtest_protocol, args=(your_test, None))
Created: 2017-11-09 17:23:33.0
Updated: 2018-01-04 16:50:46.0
Pushed: 2018-01-02 17:49:32.0
Size: 44
Language: Python
GitHub Committers
| User | Most Recent Commit | # Commits |
|---|
Other Committers
| User | Most Recent Commit | # Commits |
|---|
pytest-mp is a minimalist approach to distributing and segregating pytest tests across processes using python's multiprocessing library and is heavily inspired by pytest-concurrent and pytest-xdist. As a very early beta, it doesn't pledge or intend to support the majority of platforms or use cases. Design is based on supporting slow, io-bound testing with often tedious system under test configuration that can benefit from running several tests at one time.
clone https://github.com/ansible/pytest-mp
install pytest-mp/
y_test_dir
st basic invocation that will spin up to as many test runner processes as multiprocessing.cpu_count() indicates.
st --mp
eate up to 4 concurrent child processes.
st --mp --np 4
st --multiprocessing --num-processes 4 # Same as above but with more informative option names.
pytest-mp provides four test segregation strategies that come in handy for common test and fixture patterns. Each strategy has its own performance (dis)advantages and caveats in terms of fixture scoping and invocations.
rt pytest
est.mark.mp_group('SomeGroupName', 'free') # free, serial, isolated_free, or isolated_serial
s TestSomething(object):
def test_one(self, fixture_one):
assert True
def test_two(self, fixture_two):
assert True
est.mark.mp_group('SomeAdditionalGroupName') # uses free strategy by default
test_three():
assert True
est.mark.mp_group(group='SomeOtherGroupName', strategy='isolated_serial')
test_four(fixture_three):
assert True
est.mark.mp_group('SomeOtherGroupName') # still uses previously-defined strategy isolated_serial
test_five(fixture_three):
assert True
free strategy distributes each test in a group to a fresh pytest session in a child process that invokes all sourced fixtures (regardless of scope), runs that single test, and calls all registered finalizers before joining. This is the default test strategy for grouped and ungrouped tests.serial strategy distributes each group of tests to a fresh pytest session in a child process that invokes all sourced fixtures (regardless of scope) and runs each test serially in the same process before tearing down. This group is best suited for tests that require shared, highly-scoped fixtures that won't affect the state of the system under test for other tests.isolated_free strategy is the same as free, but all tests in this group will be run separately in time from any other test group. Best suited for tests with noisy or destructive fixtures that would affect the requirements of other tests, but that don't require a shared process.isolated_serial strategy is the same as serial, but all tests in this group will be run separate in time from any other test group, essentially like a regular pytest invocation. Best suited for tests with shared, noisy, or destructive fixtures. Absolute pytest execution will be limited to a single process while these tests are running.For example, of the tests defined above, TestSomething.test_one, TestSomething.test_two, and test_three could potentially be run at the same time among 3 processes, but test_four and test_five are guaranteed to run in the same process and with no other tests running in the background.
Given that tests generally run in child processes that emulate a fresh pytest session and that by nature pytest fixtures of class or greater scope are designed to be shared and invoked once by the test runner, some synchronization between test processes is needed to provide idempotency. pytest-mp provides two session-scoped synchronization fixtures: mp_message_board and mp_lock, a multiprocesssing.Manager.dict() and multiprocessing.Manager.Lock() instance, respectively.
rt pytest
est.fixture(scope='session')
must_be_idempotent_without_teardown(mp_lock, mp_message_board, some_resource):
with mp_lock:
if mp_message_board.get('some_flag_your_fixture_creates'):
return
mp_message_board['some_flag_your_fixture_creates'] = True
some_resource.destructive_call()
return
est.fixture(scope='session')
must_be_idempotent_with_teardown(mp_lock, mp_message_board, some_resource):
with mp_lock:
if mp_message_board.get('must_be_idempotent_setup'):
yielded = mp_message_board['must_be_idempotent_val']
else:
mp_message_board['must_be_idempotent_setup'] = True
yielded = some_resource.destructive_call()
mp_message_board['must_be_idempotent_val'] = yielded # pickle!!!
yield yielded
with mp_lock:
if mp_message_board.get('must_be_idempotent_teardown'):
return
mp_message_board['must_be_idempotent_teardown'] = True
# Notice there is no synchronization w/ this teardown and active fixture consumers!
some_resource.cleanup()
A helper fixture mp_trail() that internally uses mp_lock and mp_message_board is provided to assist in the assurance that a single setup and teardown invocation of shared fixtures and test logic occurs with multiple test runners. A trail is any named, shared path with single 'start' and 'finish' events made available as context manager values.
est.fixture(scope='session')
must_be_idempotent_with_teardown(mp_trail, some_resource):
with mp_trail('TrailName', 'start') as start: # mp_trail('TrailName') defaults to 'start'
if start: # First invocation of `must_be_idempotent_with_teardown` detected.
yielded = some_resource.destructive_call()
else: # setup has already occured by another test runner
yielded = some_resource.get_current_state()
yield yielded
with mp_trail('TrailName', 'finish') as finish:
if finish: # Last finalizer/teardown detected.
# There are no other tests using this fixture at the moment
some_resource.cleanup()