Compare commits
53 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
2ea07f73a7 | ||
|
|
faf2e0861f | ||
|
|
eb53a5f8a0 | ||
|
|
5da798f01f | ||
|
|
94a7723ba8 | ||
|
|
87be3b7582 | ||
|
|
d84f1f08d8 | ||
|
|
7a0cb35322 | ||
|
|
3a75b754d0 | ||
|
|
b418f0701c | ||
|
|
e3c22c65b8 | ||
|
|
c4e0327824 | ||
|
|
9e28a56101 | ||
|
|
76c297bdb0 | ||
|
|
fc335fb7dc | ||
|
|
f643991f7f | ||
|
|
969582187e | ||
|
|
e275ce0f8e | ||
|
|
f88275046f | ||
|
|
9d549a0c06 | ||
|
|
cc237b0d8e | ||
|
|
67f80b7c2c | ||
|
|
bfd00a58ff | ||
|
|
ed12f57193 | ||
|
|
de377c1001 | ||
|
|
eb639d19ec | ||
|
|
4441002e95 | ||
|
|
3f33a2d23e | ||
|
|
b0732f99a6 | ||
|
|
ab987d512c | ||
|
|
ffa156ced5 | ||
|
|
e26d4486fc | ||
|
|
18a30fab7d | ||
|
|
dd73d132b1 | ||
|
|
a22e0384a1 | ||
|
|
8eabc7d5df | ||
|
|
dbf7932514 | ||
|
|
6355c3e8d0 | ||
|
|
c505f00320 | ||
|
|
7e066be35e | ||
|
|
27f1d8049b | ||
|
|
bcf1c85f44 | ||
|
|
a3fac52e85 | ||
|
|
152f965a70 | ||
|
|
027b3f51ed | ||
|
|
f4b86f0127 | ||
|
|
8146b27671 | ||
|
|
a488cc5add | ||
|
|
aad1aace52 | ||
|
|
82d357ab1a | ||
|
|
ca54911481 | ||
|
|
8e88261c15 | ||
|
|
6045195f16 |
@@ -14,9 +14,17 @@ syntax:glob
|
|||||||
*.class
|
*.class
|
||||||
*.orig
|
*.orig
|
||||||
|
|
||||||
|
*.sublime-*
|
||||||
|
.Python
|
||||||
|
|
||||||
build/
|
build/
|
||||||
dist/
|
dist/
|
||||||
|
include/
|
||||||
|
lib/
|
||||||
|
bin/
|
||||||
pytest_xdist.egg-info
|
pytest_xdist.egg-info
|
||||||
issue/
|
issue/
|
||||||
3rdparty/
|
3rdparty/
|
||||||
|
pytestdebug.log
|
||||||
.tox
|
.tox
|
||||||
|
.cache
|
||||||
|
|||||||
3
.hgtags
3
.hgtags
@@ -13,3 +13,6 @@ cd44a941c833c098e4899fe3d42a96703754d0d5 1.5
|
|||||||
0d1c00018008433956aa7d93007bab6ea7de96e4 1.8
|
0d1c00018008433956aa7d93007bab6ea7de96e4 1.8
|
||||||
0d1c00018008433956aa7d93007bab6ea7de96e4 1.8
|
0d1c00018008433956aa7d93007bab6ea7de96e4 1.8
|
||||||
1d27987c267577899350a25ba5828d55d87083ad 1.8
|
1d27987c267577899350a25ba5828d55d87083ad 1.8
|
||||||
|
5c5cb6d59e12e566fbb0217aea718dc31578bee1 1.9
|
||||||
|
4406fc2a6427fadc021ed7e43e7aa5032b1ea91f 1.10
|
||||||
|
220f6e46eb71a6212ccbe6b67b9e6edcf8ee4fa5 1.11
|
||||||
|
|||||||
50
CHANGELOG
50
CHANGELOG
@@ -1,3 +1,49 @@
|
|||||||
|
1.12
|
||||||
|
-------------------------
|
||||||
|
|
||||||
|
- fix issue594: properly report errors when the test collection
|
||||||
|
is random. Thanks Bruno Oliveira.
|
||||||
|
|
||||||
|
- some internal test suite adaptation (to become forward
|
||||||
|
compatible with the upcoming pytest-2.8)
|
||||||
|
|
||||||
|
|
||||||
|
1.11
|
||||||
|
-------------------------
|
||||||
|
|
||||||
|
- fix pytest/xdist issue485 (also depends on py-1.4.22):
|
||||||
|
attach stdout/stderr on --boxed processes that die.
|
||||||
|
|
||||||
|
- fix pytest/xdist issue503: make sure that a node has usually
|
||||||
|
two items to execute to avoid scoped fixtures to be torn down
|
||||||
|
pre-maturely (fixture teardown/setup is "nextitem" sensitive).
|
||||||
|
Thanks to Andreas Pelme for bug analysis and failing test.
|
||||||
|
|
||||||
|
- restart crashed nodes by internally refactoring setup handling
|
||||||
|
of nodes. Also includes better code documentation.
|
||||||
|
Many thanks to Floris Bruynooghe for the complete PR.
|
||||||
|
|
||||||
|
|
||||||
|
1.10
|
||||||
|
-------------------------
|
||||||
|
|
||||||
|
- add glob support for rsyncignores, add command line option to pass
|
||||||
|
additional rsyncignores. Thanks Anatoly Bubenkov.
|
||||||
|
|
||||||
|
- fix pytest issue382 - produce "pytest_runtest_logstart" event again
|
||||||
|
in master. Thanks Aron Curzon.
|
||||||
|
|
||||||
|
- fix pytest issue419 by sending/receiving indices into the test
|
||||||
|
collection instead of node ids (which are not neccessarily unique
|
||||||
|
for functions parametrized with duplicate values)
|
||||||
|
|
||||||
|
- send multiple "to test" indices in one network message to a slave
|
||||||
|
and improve heuristics for sending chunks where the chunksize
|
||||||
|
depends on the number of remaining tests rather than fixed numbers.
|
||||||
|
This reduces the number of master -> node messages (but not the
|
||||||
|
reverse direction)
|
||||||
|
|
||||||
|
|
||||||
1.9
|
1.9
|
||||||
-------------------------
|
-------------------------
|
||||||
|
|
||||||
@@ -9,14 +55,14 @@
|
|||||||
- fix pytest issue41: re-run tests on all file changes, not just
|
- fix pytest issue41: re-run tests on all file changes, not just
|
||||||
randomly select ones like .py/.c.
|
randomly select ones like .py/.c.
|
||||||
|
|
||||||
- fix pytest issue347: slaves running on top of Python3.2
|
- fix pytest issue347: slaves running on top of Python3.2
|
||||||
will set PYTHONDONTWRITEYBTECODE to 1 to avoid import concurrency
|
will set PYTHONDONTWRITEYBTECODE to 1 to avoid import concurrency
|
||||||
bugs.
|
bugs.
|
||||||
|
|
||||||
1.8
|
1.8
|
||||||
-------------------------
|
-------------------------
|
||||||
|
|
||||||
- fix pytest-issue93 - use the refined pytest-2.2.1 runtestprotocol
|
- fix pytest-issue93 - use the refined pytest-2.2.1 runtestprotocol
|
||||||
interface to perform eager teardowns for test items.
|
interface to perform eager teardowns for test items.
|
||||||
|
|
||||||
1.7
|
1.7
|
||||||
|
|||||||
420
README.txt
420
README.txt
@@ -1,205 +1,215 @@
|
|||||||
xdist: pytest distributed testing plugin
|
.. image:: https://drone.io/bitbucket.org/pytest-dev/pytest-xdist/status.png
|
||||||
===============================================================
|
:target: https://drone.io/bitbucket.org/pytest-dev/pytest-xdist/latest
|
||||||
|
.. image:: https://pypip.in/v/pytest-xdist/badge.png
|
||||||
The `pytest-xdist`_ plugin extends py.test with some unique
|
:target: https://pypi.python.org/pypi/pytest-xdist
|
||||||
test execution modes:
|
|
||||||
|
xdist: pytest distributed testing plugin
|
||||||
* test run parallelization_: if you have multiple CPUs or hosts you can use
|
============================
|
||||||
those for a combined test run. This allows to speed up
|
|
||||||
development or to use special resources of `remote machines`_.
|
The `pytest-xdist`_ plugin extends py.test with some unique
|
||||||
|
test execution modes:
|
||||||
* ``--boxed``: (not available on Windows) run each test in a boxed_
|
|
||||||
subprocess to survive ``SEGFAULTS`` or otherwise dying processes
|
* test run parallelization_: if you have multiple CPUs or hosts you can use
|
||||||
|
those for a combined test run. This allows to speed up
|
||||||
* ``--looponfail``: run your tests repeatedly in a subprocess. After each run
|
development or to use special resources of `remote machines`_.
|
||||||
py.test waits until a file in your project changes and then re-runs
|
|
||||||
the previously failing tests. This is repeated until all tests pass
|
* ``--boxed``: (not available on Windows) run each test in a boxed_
|
||||||
after which again a full run is performed.
|
subprocess to survive ``SEGFAULTS`` or otherwise dying processes
|
||||||
|
|
||||||
* `Multi-Platform`_ coverage: you can specify different Python interpreters
|
* ``--looponfail``: run your tests repeatedly in a subprocess. After each run
|
||||||
or different platforms and run tests in parallel on all of them.
|
py.test waits until a file in your project changes and then re-runs
|
||||||
|
the previously failing tests. This is repeated until all tests pass
|
||||||
Before running tests remotely, ``py.test`` efficiently "rsyncs" your
|
after which again a full run is performed.
|
||||||
program source code to the remote place. All test results
|
|
||||||
are reported back and displayed to your local terminal.
|
* `Multi-Platform`_ coverage: you can specify different Python interpreters
|
||||||
You may specify different Python versions and interpreters.
|
or different platforms and run tests in parallel on all of them.
|
||||||
|
|
||||||
|
Before running tests remotely, ``py.test`` efficiently "rsyncs" your
|
||||||
Installation
|
program source code to the remote place. All test results
|
||||||
-----------------------
|
are reported back and displayed to your local terminal.
|
||||||
|
You may specify different Python versions and interpreters.
|
||||||
Install the plugin with::
|
|
||||||
|
|
||||||
easy_install pytest-xdist
|
Installation
|
||||||
|
-----------------------
|
||||||
# or
|
|
||||||
|
Install the plugin with::
|
||||||
pip install pytest-xdist
|
|
||||||
|
easy_install pytest-xdist
|
||||||
or use the package in develope/in-place mode with
|
|
||||||
a checkout of the `pytest-xdist repository`_ ::
|
# or
|
||||||
|
|
||||||
python setup.py develop
|
pip install pytest-xdist
|
||||||
|
|
||||||
Usage examples
|
or use the package in develope/in-place mode with
|
||||||
---------------------
|
a checkout of the `pytest-xdist repository`_ ::
|
||||||
|
|
||||||
.. _parallelization:
|
python setup.py develop
|
||||||
|
|
||||||
Speed up test runs by sending tests to multiple CPUs
|
Usage examples
|
||||||
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
---------------------
|
||||||
|
|
||||||
To send tests to multiple CPUs, type::
|
.. _parallelization:
|
||||||
|
|
||||||
py.test -n NUM
|
Speed up test runs by sending tests to multiple CPUs
|
||||||
|
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||||||
Especially for longer running tests or tests requiring
|
|
||||||
a lot of IO this can lead to considerable speed ups.
|
To send tests to multiple CPUs, type::
|
||||||
|
|
||||||
|
py.test -n NUM
|
||||||
Running tests in a Python subprocess
|
|
||||||
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
Especially for longer running tests or tests requiring
|
||||||
|
a lot of IO this can lead to considerable speed ups.
|
||||||
To instantiate a python2.5 sub process and send tests to it, you may type::
|
|
||||||
|
|
||||||
py.test -d --tx popen//python=python2.5
|
Running tests in a Python subprocess
|
||||||
|
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||||||
This will start a subprocess which is run with the "python2.5"
|
|
||||||
Python interpreter, found in your system binary lookup path.
|
To instantiate a python2.5 sub process and send tests to it, you may type::
|
||||||
|
|
||||||
If you prefix the --tx option value like this::
|
py.test -d --tx popen//python=python2.5
|
||||||
|
|
||||||
--tx 3*popen//python=python2.5
|
This will start a subprocess which is run with the "python2.5"
|
||||||
|
Python interpreter, found in your system binary lookup path.
|
||||||
then three subprocesses would be created and tests
|
|
||||||
will be load-balanced across these three processes.
|
If you prefix the --tx option value like this::
|
||||||
|
|
||||||
.. _boxed:
|
--tx 3*popen//python=python2.5
|
||||||
|
|
||||||
Running tests in a boxed subprocess
|
then three subprocesses would be created and tests
|
||||||
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
will be load-balanced across these three processes.
|
||||||
|
|
||||||
If you have tests involving C or C++ libraries you might have to deal
|
.. _boxed:
|
||||||
with tests crashing the process. For this case you max use the boxing
|
|
||||||
options::
|
Running tests in a boxed subprocess
|
||||||
|
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||||||
py.test --boxed
|
|
||||||
|
If you have tests involving C or C++ libraries you might have to deal
|
||||||
which will run each test in a subprocess and will report if a test
|
with tests crashing the process. For this case you may use the boxing
|
||||||
crashed the process. You can also combine this option with
|
options::
|
||||||
running multiple processes to speed up the test run and use your CPU cores::
|
|
||||||
|
py.test --boxed
|
||||||
py.test -n3 --boxed
|
|
||||||
|
which will run each test in a subprocess and will report if a test
|
||||||
this would run 3 testing subprocesses in parallel which each
|
crashed the process. You can also combine this option with
|
||||||
create new boxed subprocesses for each test.
|
running multiple processes to speed up the test run and use your CPU cores::
|
||||||
|
|
||||||
|
py.test -n3 --boxed
|
||||||
.. _`remote machines`:
|
|
||||||
|
this would run 3 testing subprocesses in parallel which each
|
||||||
Sending tests to remote SSH accounts
|
create new boxed subprocesses for each test.
|
||||||
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
|
||||||
|
|
||||||
Suppose you have a package ``mypkg`` which contains some
|
.. _`remote machines`:
|
||||||
tests that you can successfully run locally. And you
|
|
||||||
have a ssh-reachable machine ``myhost``. Then
|
Sending tests to remote SSH accounts
|
||||||
you can ad-hoc distribute your tests by typing::
|
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||||||
|
|
||||||
py.test -d --tx ssh=myhostpopen --rsyncdir mypkg mypkg
|
Suppose you have a package ``mypkg`` which contains some
|
||||||
|
tests that you can successfully run locally. And you
|
||||||
This will synchronize your ``mypkg`` package directory
|
have a ssh-reachable machine ``myhost``. Then
|
||||||
to an remote ssh account and then locally collect tests
|
you can ad-hoc distribute your tests by typing::
|
||||||
and send them to remote places for execution.
|
|
||||||
|
py.test -d --tx ssh=myhostpopen --rsyncdir mypkg mypkg
|
||||||
You can specify multiple ``--rsyncdir`` directories
|
|
||||||
to be sent to the remote side.
|
This will synchronize your ``mypkg`` package directory
|
||||||
|
to an remote ssh account and then locally collect tests
|
||||||
**NOTE:** For py.test to collect and send tests correctly
|
and send them to remote places for execution.
|
||||||
you not only need to make sure all code and tests
|
|
||||||
directories are rsynced, but that any test (sub) directory
|
You can specify multiple ``--rsyncdir`` directories
|
||||||
also has an ``__init__.py`` file because internally
|
to be sent to the remote side.
|
||||||
py.test references tests as a fully qualified python
|
|
||||||
module path. **You will otherwise get strange errors**
|
**NOTE:** For py.test to collect and send tests correctly
|
||||||
during setup of the remote side.
|
you not only need to make sure all code and tests
|
||||||
|
directories are rsynced, but that any test (sub) directory
|
||||||
Sending tests to remote Socket Servers
|
also has an ``__init__.py`` file because internally
|
||||||
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
py.test references tests as a fully qualified python
|
||||||
|
module path. **You will otherwise get strange errors**
|
||||||
Download the single-module `socketserver.py`_ Python program
|
during setup of the remote side.
|
||||||
and run it like this::
|
|
||||||
|
You can specify multiple ``--rsyncignore`` glob-patterns
|
||||||
python socketserver.py
|
to be ignored when file are sent to the remote side.
|
||||||
|
There are also internal ignores: .*, *.pyc, *.pyo, *~
|
||||||
It will tell you that it starts listening on the default
|
Those you cannot override using rsyncignore command-line or
|
||||||
port. You can now on your home machine specify this
|
ini-file option(s).
|
||||||
new socket host with something like this::
|
|
||||||
|
|
||||||
py.test -d --tx socket=192.168.1.102:8888 --rsyncdir mypkg mypkg
|
Sending tests to remote Socket Servers
|
||||||
|
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||||||
|
|
||||||
.. _`atonce`:
|
Download the single-module `socketserver.py`_ Python program
|
||||||
.. _`Multi-Platform`:
|
and run it like this::
|
||||||
|
|
||||||
|
python socketserver.py
|
||||||
Running tests on many platforms at once
|
|
||||||
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
It will tell you that it starts listening on the default
|
||||||
|
port. You can now on your home machine specify this
|
||||||
The basic command to run tests on multiple platforms is::
|
new socket host with something like this::
|
||||||
|
|
||||||
py.test --dist=each --tx=spec1 --tx=spec2
|
py.test -d --tx socket=192.168.1.102:8888 --rsyncdir mypkg mypkg
|
||||||
|
|
||||||
If you specify a windows host, an OSX host and a Linux
|
|
||||||
environment this command will send each tests to all
|
.. _`atonce`:
|
||||||
platforms - and report back failures from all platforms
|
.. _`Multi-Platform`:
|
||||||
at once. The specifications strings use the `xspec syntax`_.
|
|
||||||
|
|
||||||
.. _`xspec syntax`: http://codespeak.net/execnet/trunk/basics.html#xspec
|
Running tests on many platforms at once
|
||||||
|
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||||||
.. _`socketserver.py`: http://bitbucket.org/hpk42/execnet/raw/2af991418160/execnet/script/socketserver.py
|
|
||||||
|
The basic command to run tests on multiple platforms is::
|
||||||
.. _`execnet`: http://codespeak.net/execnet
|
|
||||||
|
py.test --dist=each --tx=spec1 --tx=spec2
|
||||||
Specifying test exec environments in an ini file
|
|
||||||
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
If you specify a windows host, an OSX host and a Linux
|
||||||
|
environment this command will send each tests to all
|
||||||
pytest (since version 2.0) supports ini-style cofiguration.
|
platforms - and report back failures from all platforms
|
||||||
You can for example make running with three subprocesses
|
at once. The specifications strings use the `xspec syntax`_.
|
||||||
your default like this::
|
|
||||||
|
.. _`xspec syntax`: http://codespeak.net/execnet/trunk/basics.html#xspec
|
||||||
[pytest]
|
|
||||||
addopts = -n3
|
.. _`socketserver.py`: http://bitbucket.org/hpk42/execnet/raw/2af991418160/execnet/script/socketserver.py
|
||||||
|
|
||||||
You can also add default environments like this::
|
.. _`execnet`: http://codespeak.net/execnet
|
||||||
|
|
||||||
[pytest]
|
Specifying test exec environments in an ini file
|
||||||
addopts = --tx ssh=myhost//python=python2.5 --tx ssh=myhost//python=python2.6
|
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||||||
|
|
||||||
and then just type::
|
pytest (since version 2.0) supports ini-style cofiguration.
|
||||||
|
You can for example make running with three subprocesses
|
||||||
py.test --dist=each
|
your default like this::
|
||||||
|
|
||||||
to run tests in each of the environments.
|
[pytest]
|
||||||
|
addopts = -n3
|
||||||
Specifying "rsync" dirs in an ini-file
|
|
||||||
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
You can also add default environments like this::
|
||||||
|
|
||||||
In a ``tox.ini`` or ``setup.cfg`` file in your root project directory
|
[pytest]
|
||||||
you may specify directories to include or to exclude in synchronisation::
|
addopts = --tx ssh=myhost//python=python2.5 --tx ssh=myhost//python=python2.6
|
||||||
|
|
||||||
[pytest]
|
and then just type::
|
||||||
rsyncdirs = . mypkg helperpkg
|
|
||||||
rsyncignore = .hg
|
py.test --dist=each
|
||||||
|
|
||||||
These directory specifications are relative to the directory
|
to run tests in each of the environments.
|
||||||
where the configuration file was found.
|
|
||||||
|
Specifying "rsync" dirs in an ini-file
|
||||||
.. _`pytest-xdist`: http://pypi.python.org/pypi/pytest-xdist
|
+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
|
||||||
.. _`pytest-xdist repository`: http://bitbucket.org/hpk42/pytest-xdist
|
|
||||||
.. _`pytest`: http://pytest.org
|
In a ``tox.ini`` or ``setup.cfg`` file in your root project directory
|
||||||
|
you may specify directories to include or to exclude in synchronisation::
|
||||||
Issue and Bug Tracker
|
|
||||||
------------------------
|
[pytest]
|
||||||
|
rsyncdirs = . mypkg helperpkg
|
||||||
Please use the pytest issue tracker for bugs in this plugin, see https://bitbucket.org/hpk42/pytest/issues .
|
rsyncignore = .hg
|
||||||
|
|
||||||
|
These directory specifications are relative to the directory
|
||||||
|
where the configuration file was found.
|
||||||
|
|
||||||
|
.. _`pytest-xdist`: http://pypi.python.org/pypi/pytest-xdist
|
||||||
|
.. _`pytest-xdist repository`: http://bitbucket.org/pytest-dev/pytest-xdist
|
||||||
|
.. _`pytest`: http://pytest.org
|
||||||
|
|
||||||
|
Issue and Bug Tracker
|
||||||
|
------------------------
|
||||||
|
|
||||||
|
Please use the pytest issue tracker for bugs in this plugin, see https://bitbucket.org/hpk42/pytest/issues .
|
||||||
4
setup.py
4
setup.py
@@ -2,7 +2,7 @@ from setuptools import setup
|
|||||||
|
|
||||||
setup(
|
setup(
|
||||||
name="pytest-xdist",
|
name="pytest-xdist",
|
||||||
version='1.9',
|
version='1.12',
|
||||||
description='py.test xdist plugin for distributed testing and loop-on-failing modes',
|
description='py.test xdist plugin for distributed testing and loop-on-failing modes',
|
||||||
long_description=open('README.txt').read(),
|
long_description=open('README.txt').read(),
|
||||||
license='MIT',
|
license='MIT',
|
||||||
@@ -13,7 +13,7 @@ setup(
|
|||||||
packages = ['xdist'],
|
packages = ['xdist'],
|
||||||
entry_points = {'pytest11': ['xdist = xdist.plugin'],},
|
entry_points = {'pytest11': ['xdist = xdist.plugin'],},
|
||||||
zip_safe=False,
|
zip_safe=False,
|
||||||
install_requires = ['execnet>=1.1', 'pytest>=2.3.5'],
|
install_requires = ['execnet>=1.1', 'pytest>=2.4.2', 'py>=1.4.22'],
|
||||||
classifiers=[
|
classifiers=[
|
||||||
'Development Status :: 5 - Production/Stable',
|
'Development Status :: 5 - Production/Stable',
|
||||||
'Intended Audience :: Developers',
|
'Intended Audience :: Developers',
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
import py
|
import py
|
||||||
import sys
|
import pytest
|
||||||
|
|
||||||
|
|
||||||
class TestDistribution:
|
class TestDistribution:
|
||||||
def test_n1_pass(self, testdir):
|
def test_n1_pass(self, testdir):
|
||||||
@@ -112,7 +113,7 @@ class TestDistribution:
|
|||||||
import py
|
import py
|
||||||
assert tmpdir.relto(py.path.local(%r)), tmpdir
|
assert tmpdir.relto(py.path.local(%r)), tmpdir
|
||||||
""" % str(testdir.tmpdir))
|
""" % str(testdir.tmpdir))
|
||||||
result = testdir.runpytest(p1, "-n1")
|
result = testdir.runpytest_subprocess(p1, "-n1")
|
||||||
assert result.ret == 0
|
assert result.ret == 0
|
||||||
result.stdout.fnmatch_lines([
|
result.stdout.fnmatch_lines([
|
||||||
"*1 passed*",
|
"*1 passed*",
|
||||||
@@ -193,7 +194,7 @@ class TestDistribution:
|
|||||||
assert dest.join(subdir.basename).check(dir=1)
|
assert dest.join(subdir.basename).check(dir=1)
|
||||||
|
|
||||||
def test_data_exchange(self, testdir):
|
def test_data_exchange(self, testdir):
|
||||||
c1 = testdir.makeconftest("""
|
testdir.makeconftest("""
|
||||||
# This hook only called on master.
|
# This hook only called on master.
|
||||||
def pytest_configure_node(node):
|
def pytest_configure_node(node):
|
||||||
node.slaveinput['a'] = 42
|
node.slaveinput['a'] = 42
|
||||||
@@ -242,7 +243,7 @@ class TestDistribution:
|
|||||||
print ("s2call-finished")
|
print ("s2call-finished")
|
||||||
""")
|
""")
|
||||||
args = ["-n1", "--debug"]
|
args = ["-n1", "--debug"]
|
||||||
result = testdir.runpytest(*args)
|
result = testdir.runpytest_subprocess(*args)
|
||||||
s = result.stdout.str()
|
s = result.stdout.str()
|
||||||
assert result.ret == 2
|
assert result.ret == 2
|
||||||
assert 's2call' in s
|
assert 's2call' in s
|
||||||
@@ -250,13 +251,13 @@ class TestDistribution:
|
|||||||
|
|
||||||
def test_keyboard_interrupt_dist(self, testdir):
|
def test_keyboard_interrupt_dist(self, testdir):
|
||||||
# xxx could be refined to check for return code
|
# xxx could be refined to check for return code
|
||||||
p = testdir.makepyfile("""
|
testdir.makepyfile("""
|
||||||
def test_sleep():
|
def test_sleep():
|
||||||
import time
|
import time
|
||||||
time.sleep(10)
|
time.sleep(10)
|
||||||
""")
|
""")
|
||||||
child = testdir.spawn_pytest("-n1")
|
child = testdir.spawn_pytest("-n1 -v")
|
||||||
child.expect(".*test session starts.*")
|
child.expect(".*test_sleep.*")
|
||||||
child.kill(2) # keyboard interrupt
|
child.kill(2) # keyboard interrupt
|
||||||
child.expect(".*KeyboardInterrupt.*")
|
child.expect(".*KeyboardInterrupt.*")
|
||||||
#child.expect(".*seconds.*")
|
#child.expect(".*seconds.*")
|
||||||
@@ -269,7 +270,7 @@ class TestDistEach:
|
|||||||
def test_hello():
|
def test_hello():
|
||||||
pass
|
pass
|
||||||
""")
|
""")
|
||||||
result = testdir.runpytest("--debug", "--dist=each", "--tx=2*popen")
|
result = testdir.runpytest_subprocess("--debug", "--dist=each", "--tx=2*popen")
|
||||||
assert not result.ret
|
assert not result.ret
|
||||||
result.stdout.fnmatch_lines(["*2 pass*"])
|
result.stdout.fnmatch_lines(["*2 pass*"])
|
||||||
|
|
||||||
@@ -299,7 +300,7 @@ class TestDistEach:
|
|||||||
|
|
||||||
class TestTerminalReporting:
|
class TestTerminalReporting:
|
||||||
def test_pass_skip_fail(self, testdir):
|
def test_pass_skip_fail(self, testdir):
|
||||||
p = testdir.makepyfile("""
|
testdir.makepyfile("""
|
||||||
import py
|
import py
|
||||||
def test_ok():
|
def test_ok():
|
||||||
pass
|
pass
|
||||||
@@ -310,9 +311,9 @@ class TestTerminalReporting:
|
|||||||
""")
|
""")
|
||||||
result = testdir.runpytest("-n1", "-v")
|
result = testdir.runpytest("-n1", "-v")
|
||||||
result.stdout.fnmatch_lines_random([
|
result.stdout.fnmatch_lines_random([
|
||||||
"*PASS*test_pass_skip_fail.py:2: *test_ok*",
|
"*PASS*test_pass_skip_fail.py*test_ok*",
|
||||||
"*SKIP*test_pass_skip_fail.py:4: *test_skip*",
|
"*SKIP*test_pass_skip_fail.py*test_skip*",
|
||||||
"*FAIL*test_pass_skip_fail.py:6: *test_func*",
|
"*FAIL*test_pass_skip_fail.py*test_func*",
|
||||||
])
|
])
|
||||||
result.stdout.fnmatch_lines([
|
result.stdout.fnmatch_lines([
|
||||||
"*def test_func():",
|
"*def test_func():",
|
||||||
@@ -321,13 +322,13 @@ class TestTerminalReporting:
|
|||||||
])
|
])
|
||||||
|
|
||||||
def test_fail_platinfo(self, testdir):
|
def test_fail_platinfo(self, testdir):
|
||||||
p = testdir.makepyfile("""
|
testdir.makepyfile("""
|
||||||
def test_func():
|
def test_func():
|
||||||
assert 0
|
assert 0
|
||||||
""")
|
""")
|
||||||
result = testdir.runpytest("-n1", "-v")
|
result = testdir.runpytest("-n1", "-v")
|
||||||
result.stdout.fnmatch_lines([
|
result.stdout.fnmatch_lines([
|
||||||
"*FAIL*test_fail_platinfo.py:1: *test_func*",
|
"*FAIL*test_fail_platinfo.py*test_func*",
|
||||||
"*0*Python*",
|
"*0*Python*",
|
||||||
"*def test_func():",
|
"*def test_func():",
|
||||||
"> assert 0",
|
"> assert 0",
|
||||||
@@ -350,7 +351,7 @@ def test_teardownfails_one_function(testdir):
|
|||||||
@py.test.mark.xfail
|
@py.test.mark.xfail
|
||||||
def test_terminate_on_hangingnode(testdir):
|
def test_terminate_on_hangingnode(testdir):
|
||||||
p = testdir.makeconftest("""
|
p = testdir.makeconftest("""
|
||||||
def pytest_sessionfinishes(session):
|
def pytest_sessionfinish(session):
|
||||||
if session.nodeid == "my": # running on slave
|
if session.nodeid == "my": # running on slave
|
||||||
import time
|
import time
|
||||||
time.sleep(3)
|
time.sleep(3)
|
||||||
@@ -362,7 +363,7 @@ def test_terminate_on_hangingnode(testdir):
|
|||||||
])
|
])
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.xfail(reason="works if run outside test suite", run=False)
|
||||||
def test_session_hooks(testdir):
|
def test_session_hooks(testdir):
|
||||||
testdir.makeconftest("""
|
testdir.makeconftest("""
|
||||||
import sys
|
import sys
|
||||||
@@ -406,7 +407,7 @@ def test_funcarg_teardown_failure(testdir):
|
|||||||
def test_hello(myarg):
|
def test_hello(myarg):
|
||||||
pass
|
pass
|
||||||
""")
|
""")
|
||||||
result = testdir.runpytest("--debug", p) # , "-n1")
|
result = testdir.runpytest_subprocess("--debug", p) # , "-n1")
|
||||||
result.stdout.fnmatch_lines([
|
result.stdout.fnmatch_lines([
|
||||||
"*ValueError*42*",
|
"*ValueError*42*",
|
||||||
"*1 passed*1 error*",
|
"*1 passed*1 error*",
|
||||||
@@ -453,8 +454,110 @@ def test_issue34_pluginloading_in_subprocess(testdir):
|
|||||||
def test_hello():
|
def test_hello():
|
||||||
assert pytest.sample_variable == "testing"
|
assert pytest.sample_variable == "testing"
|
||||||
""")
|
""")
|
||||||
result = testdir.runpytest("-n1", "-p", "plugin123")
|
result = testdir.runpytest_subprocess("-n1", "-p", "plugin123")
|
||||||
assert result.ret == 0
|
assert result.ret == 0
|
||||||
result.stdout.fnmatch_lines([
|
result.stdout.fnmatch_lines([
|
||||||
"*1 passed*",
|
"*1 passed*",
|
||||||
])
|
])
|
||||||
|
|
||||||
|
|
||||||
|
def test_fixture_scope_caching_issue503(testdir):
|
||||||
|
p1 = testdir.makepyfile("""
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
@pytest.fixture(scope='session')
|
||||||
|
def fix():
|
||||||
|
assert fix.counter == 0, 'session fixture was invoked multiple times'
|
||||||
|
fix.counter += 1
|
||||||
|
fix.counter = 0
|
||||||
|
|
||||||
|
def test_a(fix):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def test_b(fix):
|
||||||
|
pass
|
||||||
|
""")
|
||||||
|
result = testdir.runpytest(p1, '-v', '-n1')
|
||||||
|
assert result.ret == 0
|
||||||
|
result.stdout.fnmatch_lines([
|
||||||
|
"*2 passed*",
|
||||||
|
])
|
||||||
|
|
||||||
|
|
||||||
|
def test_issue_594_random_parametrize(testdir):
|
||||||
|
"""
|
||||||
|
Make sure that tests that are randomly parametrized display an appropriate
|
||||||
|
error message, instead of silently skipping the entire test run.
|
||||||
|
"""
|
||||||
|
p1 = testdir.makepyfile("""
|
||||||
|
import pytest
|
||||||
|
import random
|
||||||
|
|
||||||
|
xs = list(range(10))
|
||||||
|
random.shuffle(xs)
|
||||||
|
@pytest.mark.parametrize('x', xs)
|
||||||
|
def test_foo(x):
|
||||||
|
assert 1
|
||||||
|
""")
|
||||||
|
result = testdir.runpytest(p1, '-v', '-n4')
|
||||||
|
assert result.ret == 1
|
||||||
|
result.stdout.fnmatch_lines([
|
||||||
|
"Different tests were collected between gw* and gw*",
|
||||||
|
])
|
||||||
|
|
||||||
|
|
||||||
|
class TestNodeFailure:
|
||||||
|
|
||||||
|
def test_load_single(self, testdir):
|
||||||
|
f = testdir.makepyfile("""
|
||||||
|
import os
|
||||||
|
def test_a(): os._exit(1)
|
||||||
|
def test_b(): pass
|
||||||
|
""")
|
||||||
|
res = testdir.runpytest(f, '-n1')
|
||||||
|
res.stdout.fnmatch_lines([
|
||||||
|
"*Replacing failed node*",
|
||||||
|
"*Slave*crashed while running*",
|
||||||
|
"*1 failed*1 passed*",
|
||||||
|
])
|
||||||
|
|
||||||
|
def test_load_multiple(self, testdir):
|
||||||
|
f = testdir.makepyfile("""
|
||||||
|
import os
|
||||||
|
def test_a(): pass
|
||||||
|
def test_b(): os._exit(1)
|
||||||
|
def test_c(): pass
|
||||||
|
def test_d(): pass
|
||||||
|
""")
|
||||||
|
res = testdir.runpytest(f, '-n2')
|
||||||
|
res.stdout.fnmatch_lines([
|
||||||
|
"*Replacing failed node*",
|
||||||
|
"*Slave*crashed while running*",
|
||||||
|
"*1 failed*3 passed*",
|
||||||
|
])
|
||||||
|
|
||||||
|
def test_each_single(self, testdir):
|
||||||
|
f = testdir.makepyfile("""
|
||||||
|
import os
|
||||||
|
def test_a(): os._exit(1)
|
||||||
|
def test_b(): pass
|
||||||
|
""")
|
||||||
|
res = testdir.runpytest(f, '--dist=each', '--tx=popen')
|
||||||
|
res.stdout.fnmatch_lines([
|
||||||
|
"*Replacing failed node*",
|
||||||
|
"*Slave*crashed while running*",
|
||||||
|
"*1 failed*1 passed*",
|
||||||
|
])
|
||||||
|
|
||||||
|
def test_each_multiple(self, testdir):
|
||||||
|
f = testdir.makepyfile("""
|
||||||
|
import os
|
||||||
|
def test_a(): os._exit(1)
|
||||||
|
def test_b(): pass
|
||||||
|
""")
|
||||||
|
res = testdir.runpytest(f, '--dist=each', '--tx=2*popen')
|
||||||
|
res.stdout.fnmatch_lines([
|
||||||
|
"*Replacing failed node*",
|
||||||
|
"*Slave*crashed while running*",
|
||||||
|
"*2 failed*2 passed*",
|
||||||
|
])
|
||||||
|
|||||||
@@ -1,20 +1,43 @@
|
|||||||
import py
|
import py
|
||||||
|
import pytest
|
||||||
import execnet
|
import execnet
|
||||||
|
|
||||||
|
@pytest.fixture(scope="session", autouse=True)
|
||||||
|
def _ensure_imports():
|
||||||
|
# we import some modules because pytest-2.8's testdir fixture
|
||||||
|
# will unload all modules after each test and this cause
|
||||||
|
# (unknown) problems with execnet.Group()
|
||||||
|
execnet.Group
|
||||||
|
execnet.makegateway
|
||||||
|
|
||||||
pytest_plugins = "pytester"
|
pytest_plugins = "pytester"
|
||||||
|
|
||||||
#rsyncdirs = ['.', '../xdist', py.path.local(execnet.__file__).dirpath()]
|
#rsyncdirs = ['.', '../xdist', py.path.local(execnet.__file__).dirpath()]
|
||||||
|
|
||||||
|
@pytest.fixture(autouse=True)
|
||||||
|
def _divert_atexit(request, monkeypatch):
|
||||||
|
import atexit
|
||||||
|
l = []
|
||||||
|
def finish():
|
||||||
|
while l:
|
||||||
|
l.pop()()
|
||||||
|
monkeypatch.setattr(atexit, "register", l.append)
|
||||||
|
request.addfinalizer(finish)
|
||||||
|
|
||||||
def pytest_addoption(parser):
|
def pytest_addoption(parser):
|
||||||
parser.addoption('--gx',
|
parser.addoption('--gx',
|
||||||
action="append", dest="gspecs", default=None,
|
action="append", dest="gspecs",
|
||||||
help=("add a global test environment, XSpec-syntax. "))
|
help=("add a global test environment, XSpec-syntax. "))
|
||||||
|
|
||||||
def pytest_funcarg__specssh(request):
|
def pytest_funcarg__specssh(request):
|
||||||
return getspecssh(request.config)
|
return getspecssh(request.config)
|
||||||
def getgspecs(config):
|
|
||||||
return [execnet.XSpec(spec)
|
@pytest.fixture
|
||||||
for spec in config.getvalueorskip("gspecs")]
|
def testdir(testdir):
|
||||||
|
# pytest before 2.8 did not have a runpytest_subprocess
|
||||||
|
if not hasattr(testdir, "runpytest_subprocess"):
|
||||||
|
testdir.runpytest_subprocess = testdir.runpytest
|
||||||
|
return testdir
|
||||||
|
|
||||||
# configuration information for tests
|
# configuration information for tests
|
||||||
def getgspecs(config):
|
def getgspecs(config):
|
||||||
|
|||||||
@@ -1,6 +1,10 @@
|
|||||||
import py
|
import pytest
|
||||||
|
import os
|
||||||
|
|
||||||
@py.test.mark.skipif("not hasattr(os, 'fork')")
|
needsfork = pytest.mark.skipif(not hasattr(os, "fork"),
|
||||||
|
reason="os.fork required")
|
||||||
|
|
||||||
|
@needsfork
|
||||||
def test_functional_boxed(testdir):
|
def test_functional_boxed(testdir):
|
||||||
p1 = testdir.makepyfile("""
|
p1 = testdir.makepyfile("""
|
||||||
import os
|
import os
|
||||||
@@ -13,12 +17,36 @@ def test_functional_boxed(testdir):
|
|||||||
"*1 failed*"
|
"*1 failed*"
|
||||||
])
|
])
|
||||||
|
|
||||||
|
@needsfork
|
||||||
|
@pytest.mark.parametrize("capmode", [
|
||||||
|
"no",
|
||||||
|
pytest.mark.xfail("sys", reason="capture cleanup needed"),
|
||||||
|
pytest.mark.xfail("fd", reason="capture cleanup needed")])
|
||||||
|
def test_functional_boxed_capturing(testdir, capmode):
|
||||||
|
p1 = testdir.makepyfile("""
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
def test_function():
|
||||||
|
sys.stdout.write("hello\\n")
|
||||||
|
sys.stderr.write("world\\n")
|
||||||
|
os.kill(os.getpid(), 15)
|
||||||
|
""")
|
||||||
|
result = testdir.runpytest(p1, "--boxed", "--capture=%s" % capmode)
|
||||||
|
result.stdout.fnmatch_lines("""
|
||||||
|
*CRASHED*
|
||||||
|
*stdout*
|
||||||
|
hello
|
||||||
|
*stderr*
|
||||||
|
world
|
||||||
|
*1 failed*
|
||||||
|
""")
|
||||||
|
|
||||||
class TestOptionEffects:
|
class TestOptionEffects:
|
||||||
def test_boxed_option_default(self, testdir):
|
def test_boxed_option_default(self, testdir):
|
||||||
tmpdir = testdir.tmpdir.ensure("subdir", dir=1)
|
tmpdir = testdir.tmpdir.ensure("subdir", dir=1)
|
||||||
config = testdir.parseconfig()
|
config = testdir.parseconfig()
|
||||||
assert not config.option.boxed
|
assert not config.option.boxed
|
||||||
py.test.importorskip("execnet")
|
pytest.importorskip("execnet")
|
||||||
config = testdir.parseconfig('-d', tmpdir)
|
config = testdir.parseconfig('-d', tmpdir)
|
||||||
assert not config.option.boxed
|
assert not config.option.boxed
|
||||||
|
|
||||||
|
|||||||
@@ -4,8 +4,8 @@ from xdist.dsession import (
|
|||||||
EachScheduling,
|
EachScheduling,
|
||||||
report_collection_diff,
|
report_collection_diff,
|
||||||
)
|
)
|
||||||
from _pytest import main as outcome
|
|
||||||
import py
|
import py
|
||||||
|
import pytest
|
||||||
import execnet
|
import execnet
|
||||||
|
|
||||||
XSpec = execnet.XSpec
|
XSpec = execnet.XSpec
|
||||||
@@ -28,15 +28,12 @@ class MockNode:
|
|||||||
self.sent = []
|
self.sent = []
|
||||||
self.gateway = MockGateway()
|
self.gateway = MockGateway()
|
||||||
|
|
||||||
def send_runtest(self, nodeid):
|
def send_runtest_some(self, indices):
|
||||||
self.sent.append(nodeid)
|
self.sent.extend(indices)
|
||||||
|
|
||||||
def send_runtest_all(self):
|
def send_runtest_all(self):
|
||||||
self.sent.append("ALL")
|
self.sent.append("ALL")
|
||||||
|
|
||||||
def sendlist(self, items):
|
|
||||||
self.sent.extend(items)
|
|
||||||
|
|
||||||
def shutdown(self):
|
def shutdown(self):
|
||||||
self._shutdown=True
|
self._shutdown=True
|
||||||
|
|
||||||
@@ -63,9 +60,9 @@ class TestEachScheduling:
|
|||||||
assert sched.tests_finished()
|
assert sched.tests_finished()
|
||||||
assert node1.sent == ['ALL']
|
assert node1.sent == ['ALL']
|
||||||
assert node2.sent == ['ALL']
|
assert node2.sent == ['ALL']
|
||||||
sched.remove_item(node1, collection[0])
|
sched.remove_item(node1, 0)
|
||||||
assert sched.tests_finished()
|
assert sched.tests_finished()
|
||||||
sched.remove_item(node2, collection[0])
|
sched.remove_item(node2, 0)
|
||||||
assert sched.tests_finished()
|
assert sched.tests_finished()
|
||||||
|
|
||||||
def test_schedule_remove_node(self):
|
def test_schedule_remove_node(self):
|
||||||
@@ -86,11 +83,10 @@ class TestEachScheduling:
|
|||||||
|
|
||||||
class TestLoadScheduling:
|
class TestLoadScheduling:
|
||||||
def test_schedule_load_simple(self):
|
def test_schedule_load_simple(self):
|
||||||
node1 = MockNode()
|
|
||||||
node2 = MockNode()
|
|
||||||
sched = LoadScheduling(2)
|
sched = LoadScheduling(2)
|
||||||
sched.addnode(node1)
|
sched.addnode(MockNode())
|
||||||
sched.addnode(node2)
|
sched.addnode(MockNode())
|
||||||
|
node1, node2 = sched.nodes
|
||||||
collection = ["a.py::test_1", "a.py::test_2"]
|
collection = ["a.py::test_1", "a.py::test_2"]
|
||||||
assert not sched.collection_is_completed
|
assert not sched.collection_is_completed
|
||||||
sched.addnode_collection(node1, collection)
|
sched.addnode_collection(node1, collection)
|
||||||
@@ -100,39 +96,40 @@ class TestLoadScheduling:
|
|||||||
assert sched.node2collection[node1] == collection
|
assert sched.node2collection[node1] == collection
|
||||||
assert sched.node2collection[node2] == collection
|
assert sched.node2collection[node2] == collection
|
||||||
sched.init_distribute()
|
sched.init_distribute()
|
||||||
assert sched.tests_finished()
|
|
||||||
assert len(node1.sent) == 1
|
|
||||||
assert len(node2.sent) == 1
|
|
||||||
x = sorted(node1.sent + node2.sent)
|
|
||||||
assert x == collection
|
|
||||||
sched.remove_item(node1, node1.sent[0])
|
|
||||||
sched.remove_item(node2, node2.sent[0])
|
|
||||||
assert sched.tests_finished()
|
|
||||||
assert not sched.pending
|
assert not sched.pending
|
||||||
|
assert not sched.tests_finished()
|
||||||
|
assert len(node1.sent) == 2
|
||||||
|
assert len(node2.sent) == 0
|
||||||
|
assert node1.sent == [0, 1]
|
||||||
|
sched.remove_item(node1, node1.sent[0])
|
||||||
|
assert sched.tests_finished()
|
||||||
|
sched.remove_item(node1, node1.sent[1])
|
||||||
|
assert sched.tests_finished()
|
||||||
|
|
||||||
def test_init_distribute_chunksize(self):
|
def test_init_distribute_chunksize(self):
|
||||||
sched = LoadScheduling(2)
|
sched = LoadScheduling(2)
|
||||||
node1 = MockNode()
|
sched.addnode(MockNode())
|
||||||
node2 = MockNode()
|
sched.addnode(MockNode())
|
||||||
sched.addnode(node1)
|
node1, node2 = sched.nodes
|
||||||
sched.addnode(node2)
|
col = ["xyz"] * (6)
|
||||||
sched.ITEM_CHUNKSIZE = 2
|
|
||||||
col = ["xyz"] * (2*sched.ITEM_CHUNKSIZE +1)
|
|
||||||
sched.addnode_collection(node1, col)
|
sched.addnode_collection(node1, col)
|
||||||
sched.addnode_collection(node2, col)
|
sched.addnode_collection(node2, col)
|
||||||
sched.init_distribute()
|
sched.init_distribute()
|
||||||
#assert not sched.tests_finished()
|
#assert not sched.tests_finished()
|
||||||
sent1 = node1.sent
|
sent1 = node1.sent
|
||||||
sent2 = node2.sent
|
sent2 = node2.sent
|
||||||
chunkitems = col[:sched.ITEM_CHUNKSIZE]
|
assert sent1 == [0, 1]
|
||||||
assert sent1 == chunkitems
|
assert sent2 == [2, 3]
|
||||||
assert sent2 == chunkitems
|
assert sched.pending == [4, 5]
|
||||||
assert sched.node2pending[node1] == sent1
|
assert sched.node2pending[node1] == sent1
|
||||||
assert sched.node2pending[node2] == sent2
|
assert sched.node2pending[node2] == sent2
|
||||||
assert len(sched.pending) == 1
|
assert len(sched.pending) == 2
|
||||||
for node in (node1, node2):
|
sched.remove_item(node1, 0)
|
||||||
for i in range(sched.ITEM_CHUNKSIZE):
|
assert node1.sent == [0, 1, 4]
|
||||||
sched.remove_item(node, "xyz")
|
assert sched.pending == [5]
|
||||||
|
assert node2.sent == [2, 3]
|
||||||
|
sched.remove_item(node1, 1)
|
||||||
|
assert node1.sent == [0, 1, 4, 5]
|
||||||
assert not sched.pending
|
assert not sched.pending
|
||||||
|
|
||||||
def test_add_remove_node(self):
|
def test_add_remove_node(self):
|
||||||
@@ -147,6 +144,37 @@ class TestLoadScheduling:
|
|||||||
crashitem = sched.remove_node(node)
|
crashitem = sched.remove_node(node)
|
||||||
assert crashitem == collection[0]
|
assert crashitem == collection[0]
|
||||||
|
|
||||||
|
def test_different_tests_collected(self, testdir):
|
||||||
|
"""
|
||||||
|
Test that LoadScheduling is reporting collection errors when
|
||||||
|
different test ids are collected by slaves.
|
||||||
|
"""
|
||||||
|
class CollectHook(object):
|
||||||
|
"""
|
||||||
|
Dummy hook that stores collection reports.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self.reports = []
|
||||||
|
|
||||||
|
def pytest_collectreport(self, report):
|
||||||
|
self.reports.append(report)
|
||||||
|
|
||||||
|
collect_hook = CollectHook()
|
||||||
|
config = testdir.parseconfig()
|
||||||
|
config.pluginmanager.register(collect_hook, "collect_hook")
|
||||||
|
node1 = MockNode()
|
||||||
|
node2 = MockNode()
|
||||||
|
sched = LoadScheduling(2, config=config)
|
||||||
|
sched.addnode(node1)
|
||||||
|
sched.addnode(node2)
|
||||||
|
sched.addnode_collection(node1, ["a.py::test_1"])
|
||||||
|
sched.addnode_collection(node2, ["a.py::test_2"])
|
||||||
|
sched.init_distribute()
|
||||||
|
assert len(collect_hook.reports) == 1
|
||||||
|
rep = collect_hook.reports[0]
|
||||||
|
assert 'Different tests were collected between' in rep.longrepr
|
||||||
|
|
||||||
|
|
||||||
class TestDistReporter:
|
class TestDistReporter:
|
||||||
|
|
||||||
@@ -182,7 +210,7 @@ class TestDistReporter:
|
|||||||
def test_report_collection_diff_equal():
|
def test_report_collection_diff_equal():
|
||||||
"""Test reporting of equal collections."""
|
"""Test reporting of equal collections."""
|
||||||
from_collection = to_collection = ['aaa', 'bbb', 'ccc']
|
from_collection = to_collection = ['aaa', 'bbb', 'ccc']
|
||||||
assert report_collection_diff(from_collection, to_collection, 1, 2)
|
assert report_collection_diff(from_collection, to_collection, 1, 2) is None
|
||||||
|
|
||||||
|
|
||||||
def test_report_collection_diff_different():
|
def test_report_collection_diff_different():
|
||||||
@@ -205,7 +233,18 @@ def test_report_collection_diff_different():
|
|||||||
'-YYY'
|
'-YYY'
|
||||||
)
|
)
|
||||||
|
|
||||||
try:
|
msg = report_collection_diff(from_collection, to_collection, 1, 2)
|
||||||
report_collection_diff(from_collection, to_collection, 1, 2)
|
assert msg == error_message
|
||||||
except AssertionError as e:
|
|
||||||
assert py.builtin._totext(e) == error_message
|
@pytest.mark.xfail(reason="duplicate test ids not supported yet")
|
||||||
|
def test_pytest_issue419(testdir):
|
||||||
|
testdir.makepyfile("""
|
||||||
|
import pytest
|
||||||
|
|
||||||
|
@pytest.mark.parametrize('birth_year', [1988, 1988, ])
|
||||||
|
def test_2011_table(birth_year):
|
||||||
|
pass
|
||||||
|
""")
|
||||||
|
reprec = testdir.inline_run("-n1")
|
||||||
|
reprec.assertoutcome(passed=2)
|
||||||
|
assert 0
|
||||||
|
|||||||
@@ -14,6 +14,10 @@ class TestStatRecorder:
|
|||||||
changed = sd.check()
|
changed = sd.check()
|
||||||
assert changed
|
assert changed
|
||||||
|
|
||||||
|
(hello + "c").write("hello")
|
||||||
|
changed = sd.check()
|
||||||
|
assert not changed
|
||||||
|
|
||||||
p = tmp.ensure("new.py")
|
p = tmp.ensure("new.py")
|
||||||
changed = sd.check()
|
changed = sd.check()
|
||||||
assert changed
|
assert changed
|
||||||
@@ -36,6 +40,12 @@ class TestStatRecorder:
|
|||||||
changed = sd.check()
|
changed = sd.check()
|
||||||
assert changed
|
assert changed
|
||||||
|
|
||||||
|
def test_dirchange(self, tmpdir):
|
||||||
|
tmp = tmpdir
|
||||||
|
tmp.ensure("dir", "hello.py")
|
||||||
|
sd = StatRecorder([tmp])
|
||||||
|
assert not sd.fil(tmp.join("dir"))
|
||||||
|
|
||||||
def test_filechange_deletion_race(self, tmpdir, monkeypatch):
|
def test_filechange_deletion_race(self, tmpdir, monkeypatch):
|
||||||
tmp = tmpdir
|
tmp = tmpdir
|
||||||
sd = StatRecorder([tmp])
|
sd = StatRecorder([tmp])
|
||||||
@@ -63,12 +73,10 @@ class TestStatRecorder:
|
|||||||
|
|
||||||
pycfile = hello + "c"
|
pycfile = hello + "c"
|
||||||
pycfile.ensure()
|
pycfile.ensure()
|
||||||
changed = sd.check()
|
|
||||||
assert changed
|
|
||||||
|
|
||||||
hello.write("world")
|
hello.write("world")
|
||||||
changed = sd.check()
|
changed = sd.check()
|
||||||
assert changed
|
assert changed
|
||||||
|
assert not pycfile.check()
|
||||||
|
|
||||||
def test_waitonchange(self, tmpdir, monkeypatch):
|
def test_waitonchange(self, tmpdir, monkeypatch):
|
||||||
tmp = tmpdir
|
tmp = tmpdir
|
||||||
|
|||||||
@@ -46,6 +46,11 @@ class TestDistOptions:
|
|||||||
assert nm.roots
|
assert nm.roots
|
||||||
assert testdir.tmpdir in nm.roots
|
assert testdir.tmpdir in nm.roots
|
||||||
|
|
||||||
|
def test_getrsyncignore(self, testdir):
|
||||||
|
config = testdir.parseconfigure('--rsyncignore=fo*')
|
||||||
|
nm = NodeManager(config, specs=[execnet.XSpec("popen//chdir=qwe")])
|
||||||
|
assert 'fo*' in nm.rsyncoptions['ignores']
|
||||||
|
|
||||||
def test_getrsyncdirs_with_conftest(self, testdir):
|
def test_getrsyncdirs_with_conftest(self, testdir):
|
||||||
p = py.path.local()
|
p = py.path.local()
|
||||||
for bn in 'x y z'.split():
|
for bn in 'x y z'.split():
|
||||||
|
|||||||
@@ -3,7 +3,6 @@ from xdist.slavemanage import SlaveController, unserialize_report
|
|||||||
from xdist.remote import serialize_report
|
from xdist.remote import serialize_report
|
||||||
import execnet
|
import execnet
|
||||||
queue = py.builtin._tryimport("queue", "Queue")
|
queue = py.builtin._tryimport("queue", "Queue")
|
||||||
from py.builtin import print_
|
|
||||||
import marshal
|
import marshal
|
||||||
|
|
||||||
WAIT_TIMEOUT = 10.0
|
WAIT_TIMEOUT = 10.0
|
||||||
@@ -26,7 +25,7 @@ class SlaveSetup:
|
|||||||
use_callback = False
|
use_callback = False
|
||||||
|
|
||||||
def __init__(self, request):
|
def __init__(self, request):
|
||||||
self.testdir = testdir = request.getfuncargvalue("testdir")
|
self.testdir = request.getfuncargvalue("testdir")
|
||||||
self.request = request
|
self.request = request
|
||||||
self.events = queue.Queue()
|
self.events = queue.Queue()
|
||||||
|
|
||||||
@@ -140,7 +139,7 @@ class TestReportSerialization:
|
|||||||
|
|
||||||
class TestSlaveInteractor:
|
class TestSlaveInteractor:
|
||||||
def test_basic_collect_and_runtests(self, slave):
|
def test_basic_collect_and_runtests(self, slave):
|
||||||
p = slave.testdir.makepyfile("""
|
slave.testdir.makepyfile("""
|
||||||
def test_func():
|
def test_func():
|
||||||
pass
|
pass
|
||||||
""")
|
""")
|
||||||
@@ -154,8 +153,11 @@ class TestSlaveInteractor:
|
|||||||
assert ev.kwargs['topdir'] == slave.testdir.tmpdir
|
assert ev.kwargs['topdir'] == slave.testdir.tmpdir
|
||||||
ids = ev.kwargs['ids']
|
ids = ev.kwargs['ids']
|
||||||
assert len(ids) == 1
|
assert len(ids) == 1
|
||||||
slave.sendcommand("runtests", ids=ids)
|
slave.sendcommand("runtests", indices=list(range(len(ids))))
|
||||||
slave.sendcommand("shutdown")
|
slave.sendcommand("shutdown")
|
||||||
|
ev = slave.popevent("logstart")
|
||||||
|
assert ev.kwargs["nodeid"].endswith("test_func")
|
||||||
|
assert len(ev.kwargs["location"]) == 3
|
||||||
ev = slave.popevent("testreport") # setup
|
ev = slave.popevent("testreport") # setup
|
||||||
ev = slave.popevent("testreport")
|
ev = slave.popevent("testreport")
|
||||||
assert ev.name == "testreport"
|
assert ev.name == "testreport"
|
||||||
@@ -167,7 +169,7 @@ class TestSlaveInteractor:
|
|||||||
assert 'slaveoutput' in ev.kwargs
|
assert 'slaveoutput' in ev.kwargs
|
||||||
|
|
||||||
def test_remote_collect_skip(self, slave):
|
def test_remote_collect_skip(self, slave):
|
||||||
p = slave.testdir.makepyfile("""
|
slave.testdir.makepyfile("""
|
||||||
import py
|
import py
|
||||||
py.test.skip("hello")
|
py.test.skip("hello")
|
||||||
""")
|
""")
|
||||||
@@ -184,7 +186,7 @@ class TestSlaveInteractor:
|
|||||||
assert not ev.kwargs['ids']
|
assert not ev.kwargs['ids']
|
||||||
|
|
||||||
def test_remote_collect_fail(self, slave):
|
def test_remote_collect_fail(self, slave):
|
||||||
p = slave.testdir.makepyfile("""aasd qwe""")
|
slave.testdir.makepyfile("""aasd qwe""")
|
||||||
slave.setup()
|
slave.setup()
|
||||||
ev = slave.popevent("collectionstart")
|
ev = slave.popevent("collectionstart")
|
||||||
assert not ev.kwargs
|
assert not ev.kwargs
|
||||||
@@ -198,7 +200,7 @@ class TestSlaveInteractor:
|
|||||||
assert not ev.kwargs['ids']
|
assert not ev.kwargs['ids']
|
||||||
|
|
||||||
def test_runtests_all(self, slave):
|
def test_runtests_all(self, slave):
|
||||||
p = slave.testdir.makepyfile("""
|
slave.testdir.makepyfile("""
|
||||||
def test_func(): pass
|
def test_func(): pass
|
||||||
def test_func2(): pass
|
def test_func2(): pass
|
||||||
""")
|
""")
|
||||||
@@ -225,7 +227,7 @@ class TestSlaveInteractor:
|
|||||||
def test_happy_run_events_converted(self, testdir, slave):
|
def test_happy_run_events_converted(self, testdir, slave):
|
||||||
py.test.xfail("implement a simple test for event production")
|
py.test.xfail("implement a simple test for event production")
|
||||||
assert not slave.use_callback
|
assert not slave.use_callback
|
||||||
p = slave.testdir.makepyfile("""
|
slave.testdir.makepyfile("""
|
||||||
def test_func():
|
def test_func():
|
||||||
pass
|
pass
|
||||||
""")
|
""")
|
||||||
|
|||||||
@@ -1,28 +1,35 @@
|
|||||||
import py
|
import py
|
||||||
import os
|
import pytest
|
||||||
import execnet
|
import execnet
|
||||||
|
from _pytest.pytester import HookRecorder
|
||||||
|
from xdist import slavemanage, newhooks
|
||||||
from xdist.slavemanage import HostRSync, NodeManager
|
from xdist.slavemanage import HostRSync, NodeManager
|
||||||
|
|
||||||
pytest_plugins = "pytester",
|
pytest_plugins = "pytester"
|
||||||
|
|
||||||
def pytest_funcarg__hookrecorder(request):
|
def pytest_funcarg__hookrecorder(request, config):
|
||||||
_pytest = request.getfuncargvalue('_pytest')
|
hookrecorder = HookRecorder(config.pluginmanager)
|
||||||
config = request.getfuncargvalue('config')
|
if hasattr(hookrecorder, "start_recording"):
|
||||||
return _pytest.gethookrecorder(config.hook)
|
hookrecorder.start_recording(newhooks)
|
||||||
|
request.addfinalizer(hookrecorder.finish_recording)
|
||||||
|
return hookrecorder
|
||||||
|
|
||||||
def pytest_funcarg__config(request):
|
def pytest_funcarg__config(testdir):
|
||||||
testdir = request.getfuncargvalue("testdir")
|
return testdir.parseconfig()
|
||||||
config = testdir.parseconfig()
|
|
||||||
return config
|
|
||||||
|
|
||||||
def pytest_funcarg__mysetup(request):
|
def pytest_funcarg__mysetup(tmpdir):
|
||||||
class mysetup:
|
class mysetup:
|
||||||
def __init__(self, request):
|
source = tmpdir.mkdir("source")
|
||||||
temp = request.getfuncargvalue("tmpdir")
|
dest = tmpdir.mkdir("dest")
|
||||||
self.source = temp.mkdir("source")
|
return mysetup()
|
||||||
self.dest = temp.mkdir("dest")
|
|
||||||
request.getfuncargvalue("_pytest")
|
@pytest.fixture
|
||||||
return mysetup(request)
|
def slavecontroller(monkeypatch):
|
||||||
|
class MockController(object):
|
||||||
|
def __init__(self, *args): pass
|
||||||
|
def setup(self): pass
|
||||||
|
monkeypatch.setattr(slavemanage, 'SlaveController', MockController)
|
||||||
|
return MockController
|
||||||
|
|
||||||
class TestNodeManagerPopen:
|
class TestNodeManagerPopen:
|
||||||
def test_popen_no_default_chdir(self, config):
|
def test_popen_no_default_chdir(self, config):
|
||||||
@@ -36,9 +43,9 @@ class TestNodeManagerPopen:
|
|||||||
for spec in NodeManager(config, l, defaultchdir="abc").specs:
|
for spec in NodeManager(config, l, defaultchdir="abc").specs:
|
||||||
assert spec.chdir == "abc"
|
assert spec.chdir == "abc"
|
||||||
|
|
||||||
def test_popen_makegateway_events(self, config, hookrecorder, _pytest):
|
def test_popen_makegateway_events(self, config, hookrecorder, slavecontroller):
|
||||||
hm = NodeManager(config, ["popen"] * 2)
|
hm = NodeManager(config, ["popen"] * 2)
|
||||||
hm.makegateways()
|
hm.setup_nodes(None)
|
||||||
call = hookrecorder.popcall("pytest_xdist_setupnodes")
|
call = hookrecorder.popcall("pytest_xdist_setupnodes")
|
||||||
assert len(call.specs) == 2
|
assert len(call.specs) == 2
|
||||||
|
|
||||||
@@ -51,10 +58,10 @@ class TestNodeManagerPopen:
|
|||||||
hm.teardown_nodes()
|
hm.teardown_nodes()
|
||||||
assert not len(hm.group)
|
assert not len(hm.group)
|
||||||
|
|
||||||
def test_popens_rsync(self, config, mysetup):
|
def test_popens_rsync(self, config, mysetup, slavecontroller):
|
||||||
source = mysetup.source
|
source = mysetup.source
|
||||||
hm = NodeManager(config, ["popen"] * 2)
|
hm = NodeManager(config, ["popen"] * 2)
|
||||||
hm.makegateways()
|
hm.setup_nodes(None)
|
||||||
assert len(hm.group) == 2
|
assert len(hm.group) == 2
|
||||||
for gw in hm.group:
|
for gw in hm.group:
|
||||||
class pseudoexec:
|
class pseudoexec:
|
||||||
@@ -65,19 +72,21 @@ class TestNodeManagerPopen:
|
|||||||
pass
|
pass
|
||||||
gw.remote_exec = pseudoexec
|
gw.remote_exec = pseudoexec
|
||||||
l = []
|
l = []
|
||||||
hm.rsync(source, notify=lambda *args: l.append(args))
|
for gw in hm.group:
|
||||||
|
hm.rsync(gw, source, notify=lambda *args: l.append(args))
|
||||||
assert not l
|
assert not l
|
||||||
hm.teardown_nodes()
|
hm.teardown_nodes()
|
||||||
assert not len(hm.group)
|
assert not len(hm.group)
|
||||||
assert "sys.path.insert" in gw.remote_exec.args[0]
|
assert "sys.path.insert" in gw.remote_exec.args[0]
|
||||||
|
|
||||||
def test_rsync_popen_with_path(self, config, mysetup):
|
def test_rsync_popen_with_path(self, config, mysetup, slavecontroller):
|
||||||
source, dest = mysetup.source, mysetup.dest
|
source, dest = mysetup.source, mysetup.dest
|
||||||
hm = NodeManager(config, ["popen//chdir=%s" %dest] * 1)
|
hm = NodeManager(config, ["popen//chdir=%s" % dest] * 1)
|
||||||
hm.makegateways()
|
hm.setup_nodes(None)
|
||||||
source.ensure("dir1", "dir2", "hello")
|
source.ensure("dir1", "dir2", "hello")
|
||||||
l = []
|
l = []
|
||||||
hm.rsync(source, notify=lambda *args: l.append(args))
|
for gw in hm.group:
|
||||||
|
hm.rsync(gw, source, notify=lambda *args: l.append(args))
|
||||||
assert len(l) == 1
|
assert len(l) == 1
|
||||||
assert l[0] == ("rsyncrootready", hm.group['gw0'].spec, source)
|
assert l[0] == ("rsyncrootready", hm.group['gw0'].spec, source)
|
||||||
hm.teardown_nodes()
|
hm.teardown_nodes()
|
||||||
@@ -86,12 +95,15 @@ class TestNodeManagerPopen:
|
|||||||
assert dest.join("dir1", "dir2").check()
|
assert dest.join("dir1", "dir2").check()
|
||||||
assert dest.join("dir1", "dir2", 'hello').check()
|
assert dest.join("dir1", "dir2", 'hello').check()
|
||||||
|
|
||||||
def test_rsync_same_popen_twice(self, config, mysetup, hookrecorder):
|
def test_rsync_same_popen_twice(self, config, mysetup,
|
||||||
|
hookrecorder, slavecontroller):
|
||||||
source, dest = mysetup.source, mysetup.dest
|
source, dest = mysetup.source, mysetup.dest
|
||||||
hm = NodeManager(config, ["popen//chdir=%s" %dest] * 2)
|
hm = NodeManager(config, ["popen//chdir=%s" % dest] * 2)
|
||||||
hm.makegateways()
|
hm.roots = []
|
||||||
|
hm.setup_nodes(None)
|
||||||
source.ensure("dir1", "dir2", "hello")
|
source.ensure("dir1", "dir2", "hello")
|
||||||
hm.rsync(source)
|
gw = hm.group[0]
|
||||||
|
hm.rsync(gw, source)
|
||||||
call = hookrecorder.popcall("pytest_xdist_rsyncstart")
|
call = hookrecorder.popcall("pytest_xdist_rsyncstart")
|
||||||
assert call.source == source
|
assert call.source == source
|
||||||
assert len(call.gateways) == 1
|
assert len(call.gateways) == 1
|
||||||
@@ -99,21 +111,13 @@ class TestNodeManagerPopen:
|
|||||||
call = hookrecorder.popcall("pytest_xdist_rsyncfinish")
|
call = hookrecorder.popcall("pytest_xdist_rsyncfinish")
|
||||||
|
|
||||||
class TestHRSync:
|
class TestHRSync:
|
||||||
def pytest_funcarg__mysetup(self, request):
|
|
||||||
class mysetup:
|
|
||||||
def __init__(self, request):
|
|
||||||
tmp = request.getfuncargvalue('tmpdir')
|
|
||||||
self.source = tmp.mkdir("source")
|
|
||||||
self.dest = tmp.mkdir("dest")
|
|
||||||
return mysetup(request)
|
|
||||||
|
|
||||||
def test_hrsync_filter(self, mysetup):
|
def test_hrsync_filter(self, mysetup):
|
||||||
source, dest = mysetup.source, mysetup.dest
|
source, _ = mysetup.source, mysetup.dest # noqa
|
||||||
source.ensure("dir", "file.txt")
|
source.ensure("dir", "file.txt")
|
||||||
source.ensure(".svn", "entries")
|
source.ensure(".svn", "entries")
|
||||||
source.ensure(".somedotfile", "moreentries")
|
source.ensure(".somedotfile", "moreentries")
|
||||||
source.ensure("somedir", "editfile~")
|
source.ensure("somedir", "editfile~")
|
||||||
syncer = HostRSync(source)
|
syncer = HostRSync(source, ignores=NodeManager.DEFAULT_IGNORES)
|
||||||
l = list(source.visit(rec=syncer.filter,
|
l = list(source.visit(rec=syncer.filter,
|
||||||
fil=syncer.filter))
|
fil=syncer.filter))
|
||||||
assert len(l) == 3
|
assert len(l) == 3
|
||||||
@@ -136,10 +140,10 @@ class TestHRSync:
|
|||||||
|
|
||||||
|
|
||||||
class TestNodeManager:
|
class TestNodeManager:
|
||||||
@py.test.mark.xfail
|
@py.test.mark.xfail(run=False)
|
||||||
def test_rsync_roots_no_roots(self, testdir, mysetup):
|
def test_rsync_roots_no_roots(self, testdir, mysetup):
|
||||||
mysetup.source.ensure("dir1", "file1").write("hello")
|
mysetup.source.ensure("dir1", "file1").write("hello")
|
||||||
config = testdir.parseconfig(source)
|
config = testdir.parseconfig(mysetup.source)
|
||||||
nodemanager = NodeManager(config, ["popen//chdir=%s" % mysetup.dest])
|
nodemanager = NodeManager(config, ["popen//chdir=%s" % mysetup.dest])
|
||||||
#assert nodemanager.config.topdir == source == config.topdir
|
#assert nodemanager.config.topdir == source == config.topdir
|
||||||
nodemanager.makegateways()
|
nodemanager.makegateways()
|
||||||
@@ -152,7 +156,7 @@ class TestNodeManager:
|
|||||||
assert p.join("dir1").check()
|
assert p.join("dir1").check()
|
||||||
assert p.join("dir1", "file1").check()
|
assert p.join("dir1", "file1").check()
|
||||||
|
|
||||||
def test_popen_rsync_subdir(self, testdir, mysetup):
|
def test_popen_rsync_subdir(self, testdir, mysetup, slavecontroller):
|
||||||
source, dest = mysetup.source, mysetup.dest
|
source, dest = mysetup.source, mysetup.dest
|
||||||
dir1 = mysetup.source.mkdir("dir1")
|
dir1 = mysetup.source.mkdir("dir1")
|
||||||
dir2 = dir1.mkdir("dir2")
|
dir2 = dir1.mkdir("dir2")
|
||||||
@@ -164,8 +168,7 @@ class TestNodeManager:
|
|||||||
"--rsyncdir", rsyncroot,
|
"--rsyncdir", rsyncroot,
|
||||||
source,
|
source,
|
||||||
))
|
))
|
||||||
nodemanager.makegateways()
|
nodemanager.setup_nodes(None) # calls .rsync_roots()
|
||||||
nodemanager.rsync_roots()
|
|
||||||
if rsyncroot == source:
|
if rsyncroot == source:
|
||||||
dest = dest.join("source")
|
dest = dest.join("source")
|
||||||
assert dest.join("dir1").check()
|
assert dest.join("dir1").check()
|
||||||
@@ -173,7 +176,7 @@ class TestNodeManager:
|
|||||||
assert dest.join("dir1", "dir2", 'hello').check()
|
assert dest.join("dir1", "dir2", 'hello').check()
|
||||||
nodemanager.teardown_nodes()
|
nodemanager.teardown_nodes()
|
||||||
|
|
||||||
def test_init_rsync_roots(self, testdir, mysetup):
|
def test_init_rsync_roots(self, testdir, mysetup, slavecontroller):
|
||||||
source, dest = mysetup.source, mysetup.dest
|
source, dest = mysetup.source, mysetup.dest
|
||||||
dir2 = source.ensure("dir1", "dir2", dir=1)
|
dir2 = source.ensure("dir1", "dir2", dir=1)
|
||||||
source.ensure("dir1", "somefile", dir=1)
|
source.ensure("dir1", "somefile", dir=1)
|
||||||
@@ -185,41 +188,43 @@ class TestNodeManager:
|
|||||||
"""))
|
"""))
|
||||||
config = testdir.parseconfig(source)
|
config = testdir.parseconfig(source)
|
||||||
nodemanager = NodeManager(config, ["popen//chdir=%s" % dest])
|
nodemanager = NodeManager(config, ["popen//chdir=%s" % dest])
|
||||||
nodemanager.makegateways()
|
nodemanager.setup_nodes(None) # calls .rsync_roots()
|
||||||
nodemanager.rsync_roots()
|
|
||||||
assert dest.join("dir2").check()
|
assert dest.join("dir2").check()
|
||||||
assert not dest.join("dir1").check()
|
assert not dest.join("dir1").check()
|
||||||
assert not dest.join("bogus").check()
|
assert not dest.join("bogus").check()
|
||||||
|
|
||||||
def test_rsyncignore(self, testdir, mysetup):
|
def test_rsyncignore(self, testdir, mysetup, slavecontroller):
|
||||||
source, dest = mysetup.source, mysetup.dest
|
source, dest = mysetup.source, mysetup.dest
|
||||||
dir2 = source.ensure("dir1", "dir2", dir=1)
|
dir2 = source.ensure("dir1", "dir2", dir=1)
|
||||||
dir5 = source.ensure("dir5", "dir6", "bogus")
|
source.ensure("dir5", "dir6", "bogus")
|
||||||
dirf = source.ensure("dir5", "file")
|
source.ensure("dir5", "file")
|
||||||
dir2.ensure("hello")
|
dir2.ensure("hello")
|
||||||
|
source.ensure("foo", "bar")
|
||||||
|
source.ensure("bar", "foo")
|
||||||
source.join("tox.ini").write(py.std.textwrap.dedent("""
|
source.join("tox.ini").write(py.std.textwrap.dedent("""
|
||||||
[pytest]
|
[pytest]
|
||||||
rsyncdirs = dir1 dir5
|
rsyncdirs = dir1 dir5
|
||||||
rsyncignore = dir1/dir2 dir5/dir6
|
rsyncignore = dir1/dir2 dir5/dir6 foo*
|
||||||
"""))
|
"""))
|
||||||
config = testdir.parseconfig(source)
|
config = testdir.parseconfig(source)
|
||||||
|
config.option.rsyncignore = ['bar']
|
||||||
nodemanager = NodeManager(config, ["popen//chdir=%s" % dest])
|
nodemanager = NodeManager(config, ["popen//chdir=%s" % dest])
|
||||||
nodemanager.makegateways()
|
nodemanager.setup_nodes(None) # calls .rsync_roots()
|
||||||
nodemanager.rsync_roots()
|
|
||||||
assert dest.join("dir1").check()
|
assert dest.join("dir1").check()
|
||||||
assert not dest.join("dir1", "dir2").check()
|
assert not dest.join("dir1", "dir2").check()
|
||||||
assert dest.join("dir5","file").check()
|
assert dest.join("dir5", "file").check()
|
||||||
assert not dest.join("dir6").check()
|
assert not dest.join("dir6").check()
|
||||||
|
assert not dest.join('foo').check()
|
||||||
|
assert not dest.join('bar').check()
|
||||||
|
|
||||||
def test_optimise_popen(self, testdir, mysetup):
|
def test_optimise_popen(self, testdir, mysetup, slavecontroller):
|
||||||
source, dest = mysetup.source, mysetup.dest
|
source = mysetup.source
|
||||||
specs = ["popen"] * 3
|
specs = ["popen"] * 3
|
||||||
source.join("conftest.py").write("rsyncdirs = ['a']")
|
source.join("conftest.py").write("rsyncdirs = ['a']")
|
||||||
source.ensure('a', dir=1)
|
source.ensure('a', dir=1)
|
||||||
config = testdir.parseconfig(source)
|
config = testdir.parseconfig(source)
|
||||||
nodemanager = NodeManager(config, specs)
|
nodemanager = NodeManager(config, specs)
|
||||||
nodemanager.makegateways()
|
nodemanager.setup_nodes(None) # calls .rysnc_roots()
|
||||||
nodemanager.rsync_roots()
|
|
||||||
for gwspec in nodemanager.specs:
|
for gwspec in nodemanager.specs:
|
||||||
assert gwspec._samefilesystem()
|
assert gwspec._samefilesystem()
|
||||||
assert not gwspec.chdir
|
assert not gwspec.chdir
|
||||||
@@ -230,8 +235,6 @@ class TestNodeManager:
|
|||||||
pass
|
pass
|
||||||
""")
|
""")
|
||||||
reprec = testdir.inline_run("-d", "--rsyncdir=%s" % testdir.tmpdir,
|
reprec = testdir.inline_run("-d", "--rsyncdir=%s" % testdir.tmpdir,
|
||||||
"--tx", specssh, testdir.tmpdir)
|
"--tx", specssh, testdir.tmpdir)
|
||||||
rep, = reprec.getreports("pytest_runtest_logreport")
|
rep, = reprec.getreports("pytest_runtest_logreport")
|
||||||
assert rep.passed
|
assert rep.passed
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
35
tox.ini
35
tox.ini
@@ -1,25 +1,42 @@
|
|||||||
[tox]
|
[tox]
|
||||||
envlist=py26,py32,py33,py27,py26,py26-old,py33-old
|
envlist=py26,py33,py34,py27,py27-pexpect,py33-pexpect,py26-old,py33-old,flakes
|
||||||
|
|
||||||
[testenv]
|
[testenv]
|
||||||
changedir=testing
|
changedir=testing
|
||||||
deps=pytest>=2.4.2
|
deps=pytest>=2.5.1
|
||||||
commands= py.test --junitxml={envlogdir}/junit-{envname}.xml []
|
commands= py.test {posargs}
|
||||||
|
|
||||||
[testenv:py27]
|
[testenv:py27-pexpect]
|
||||||
deps=
|
deps={[testenv]deps}
|
||||||
pytest>=2.4.2
|
|
||||||
pexpect
|
pexpect
|
||||||
|
[testenv:py33-pexpect]
|
||||||
|
deps={[testenv]deps}
|
||||||
|
pexpect
|
||||||
|
|
||||||
|
[testenv:flakes]
|
||||||
|
changedir=
|
||||||
|
deps = pytest-flakes>=0.2
|
||||||
|
commands = py.test --flakes -m flakes testing xdist
|
||||||
|
|
||||||
[testenv:py26-old]
|
[testenv:py26-old]
|
||||||
|
basepython = python2.6
|
||||||
deps=
|
deps=
|
||||||
pytest==2.3.5
|
pytest==2.5.2
|
||||||
pexpect
|
pycmd
|
||||||
|
|
||||||
|
commands=
|
||||||
|
py.cleanup -a
|
||||||
|
py.test {posargs}
|
||||||
|
|
||||||
[testenv:py33-old]
|
[testenv:py33-old]
|
||||||
basepython = python3.3
|
basepython = python3.3
|
||||||
deps=
|
deps=
|
||||||
pytest==2.3.5
|
pytest==2.5.2
|
||||||
|
pycmd
|
||||||
|
|
||||||
|
commands=
|
||||||
|
py.cleanup -a
|
||||||
|
py.test {posargs}
|
||||||
|
|
||||||
[pytest]
|
[pytest]
|
||||||
addopts = -rsfxX
|
addopts = -rsfxX
|
||||||
|
|||||||
@@ -1,2 +1,2 @@
|
|||||||
#
|
#
|
||||||
__version__ = '1.9'
|
__version__ = '1.12'
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import sys
|
|
||||||
import difflib
|
import difflib
|
||||||
|
from _pytest.runner import CollectReport
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
import py
|
import py
|
||||||
@@ -10,166 +10,409 @@ queue = py.builtin._tryimport('queue', 'Queue')
|
|||||||
|
|
||||||
|
|
||||||
class EachScheduling:
|
class EachScheduling:
|
||||||
|
"""Implement scheduling of test items on all nodes
|
||||||
|
|
||||||
|
If a node gets added after the test run is started then it is
|
||||||
|
assumed to replace a node which got removed before it finished
|
||||||
|
it's collection. In this case it will only be used if a a node
|
||||||
|
with the same spec got removed earlier.
|
||||||
|
|
||||||
|
Any nodes added after the run is started will only get items
|
||||||
|
assigned if a node with matching spec was removed before it
|
||||||
|
finished all it's pending items. The new node will then be
|
||||||
|
assigned the remaining items from the removed node.
|
||||||
|
"""
|
||||||
|
|
||||||
def __init__(self, numnodes, log=None):
|
def __init__(self, numnodes, log=None):
|
||||||
self.numnodes = numnodes
|
self.numnodes = numnodes
|
||||||
self.node2collection = {}
|
self.node2collection = {}
|
||||||
self.node2pending = {}
|
self.node2pending = {}
|
||||||
|
self._started = []
|
||||||
|
self._removed2pending = {}
|
||||||
if log is None:
|
if log is None:
|
||||||
self.log = py.log.Producer("eachsched")
|
self.log = py.log.Producer("eachsched")
|
||||||
else:
|
else:
|
||||||
self.log = log.loadsched
|
self.log = log.eachsched
|
||||||
self.collection_is_completed = False
|
self.collection_is_completed = False
|
||||||
|
|
||||||
|
@property
|
||||||
|
def nodes(self):
|
||||||
|
"""A list of all nodes in the scheduler."""
|
||||||
|
return list(self.node2pending.keys())
|
||||||
|
|
||||||
def hasnodes(self):
|
def hasnodes(self):
|
||||||
return bool(self.node2pending)
|
return bool(self.node2pending)
|
||||||
|
|
||||||
|
def haspending(self):
|
||||||
|
"""Return True if there are pending test items
|
||||||
|
|
||||||
|
This indicates that collection has finished and nodes are
|
||||||
|
still processing test items, so can be thought of as "the
|
||||||
|
scheduler is active".
|
||||||
|
"""
|
||||||
|
for pending in self.node2pending.values():
|
||||||
|
if pending:
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
def addnode(self, node):
|
def addnode(self, node):
|
||||||
self.node2collection[node] = None
|
assert node not in self.node2pending
|
||||||
|
self.node2pending[node] = []
|
||||||
|
|
||||||
def tests_finished(self):
|
def tests_finished(self):
|
||||||
if not self.collection_is_completed:
|
if not self.collection_is_completed:
|
||||||
return False
|
return False
|
||||||
|
if self._removed2pending:
|
||||||
|
return False
|
||||||
|
for pending in self.node2pending.values():
|
||||||
|
if len(pending) >= 2:
|
||||||
|
return False
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def addnode_collection(self, node, collection):
|
def addnode_collection(self, node, collection):
|
||||||
assert not self.collection_is_completed
|
"""Add the collected test items from a node
|
||||||
assert self.node2collection[node] is None
|
|
||||||
self.node2collection[node] = list(collection)
|
|
||||||
self.node2pending[node] = []
|
|
||||||
if len(self.node2pending) >= self.numnodes:
|
|
||||||
self.collection_is_completed = True
|
|
||||||
|
|
||||||
def remove_item(self, node, item):
|
Collection is complete once all nodes have submitted their
|
||||||
self.node2pending[node].remove(item)
|
collection. In this case it's peding list is set to an empty
|
||||||
|
list. When the collection is already completed this
|
||||||
|
submission is from a node which was restarted to replace a
|
||||||
|
dead node. In this case we already assing the pending items
|
||||||
|
here. In either case ``.init_distribute()`` will instruct the
|
||||||
|
node to start running the required tests.
|
||||||
|
"""
|
||||||
|
assert node in self.node2pending
|
||||||
|
if not self.collection_is_completed:
|
||||||
|
self.node2collection[node] = list(collection)
|
||||||
|
self.node2pending[node] = []
|
||||||
|
if len(self.node2collection) >= self.numnodes:
|
||||||
|
self.collection_is_completed = True
|
||||||
|
elif self._removed2pending:
|
||||||
|
for deadnode in self._removed2pending:
|
||||||
|
if deadnode.gateway.spec == node.gateway.spec:
|
||||||
|
dead_collection = self.node2collection[deadnode]
|
||||||
|
if collection != dead_collection:
|
||||||
|
msg = report_collection_diff(dead_collection,
|
||||||
|
collection,
|
||||||
|
deadnode.gateway.id,
|
||||||
|
node.gateway.id)
|
||||||
|
self.log(msg)
|
||||||
|
return
|
||||||
|
pending = self._removed2pending.pop(deadnode)
|
||||||
|
self.node2pending[node] = pending
|
||||||
|
break
|
||||||
|
|
||||||
|
def remove_item(self, node, item_index, duration=0):
|
||||||
|
self.node2pending[node].remove(item_index)
|
||||||
|
|
||||||
def remove_node(self, node):
|
def remove_node(self, node):
|
||||||
# KeyError if we didn't get an addnode() yet
|
# KeyError if we didn't get an addnode() yet
|
||||||
pending = self.node2pending.pop(node)
|
pending = self.node2pending.pop(node)
|
||||||
if not pending:
|
if not pending:
|
||||||
return
|
return
|
||||||
crashitem = pending.pop(0)
|
crashitem = self.node2collection[node][pending.pop(0)]
|
||||||
# XXX what about the rest of pending?
|
if pending:
|
||||||
|
self._removed2pending[node] = pending
|
||||||
return crashitem
|
return crashitem
|
||||||
|
|
||||||
def init_distribute(self):
|
def init_distribute(self):
|
||||||
|
"""Schedule the test items on the nodes
|
||||||
|
|
||||||
|
If the node's pending list is empty it is a new node which
|
||||||
|
needs to run all the tests. If the pending list is already
|
||||||
|
populated (by ``.addnode_collection()``) then it replaces a
|
||||||
|
died node and we only need to run those tests.
|
||||||
|
"""
|
||||||
assert self.collection_is_completed
|
assert self.collection_is_completed
|
||||||
for node, pending in self.node2pending.items():
|
for node, pending in self.node2pending.items():
|
||||||
node.send_runtest_all()
|
if node in self._started:
|
||||||
pending[:] = self.node2collection[node]
|
continue
|
||||||
|
if not pending:
|
||||||
|
pending[:] = range(len(self.node2collection[node]))
|
||||||
|
node.send_runtest_all()
|
||||||
|
else:
|
||||||
|
node.send_runtest_some(pending)
|
||||||
|
self._started.append(node)
|
||||||
|
|
||||||
|
|
||||||
class LoadScheduling:
|
class LoadScheduling:
|
||||||
LOAD_THRESHOLD_NEWITEMS = 5
|
"""Implement load scheduling accross nodes.
|
||||||
ITEM_CHUNKSIZE = 10
|
|
||||||
|
|
||||||
def __init__(self, numnodes, log=None):
|
This distributes the tests collected across all nodes so each test
|
||||||
|
is run just once. All nodes collect and submit the test suit and
|
||||||
|
when all collections are received it is verified they are
|
||||||
|
identical collections. Then the collection gets devided up in
|
||||||
|
chunks and chunks get submitted to nodes. Whenver a node finishes
|
||||||
|
an item they call ``.remove_item()`` which will trigger the
|
||||||
|
scheduler to assign more tests if the number of pending tests for
|
||||||
|
the node falls below a low-watermark.
|
||||||
|
|
||||||
|
When created ``numnodes`` defines how many nodes are expected to
|
||||||
|
submit a collection, this is used to know when all nodes have
|
||||||
|
finished collection or how large the chunks need to be created.
|
||||||
|
|
||||||
|
Attributes:
|
||||||
|
|
||||||
|
:numnodes: The expected number of nodes taking part. The actual
|
||||||
|
number of nodes will vary during the scheduler's lifetime as
|
||||||
|
nodes are added by the DSession as they are brought up and
|
||||||
|
removed either because of a died node or normal shutdown. This
|
||||||
|
number is primarily used to know when the initial collection is
|
||||||
|
completed.
|
||||||
|
|
||||||
|
:node2collection: Map of nodes and their test collection. All
|
||||||
|
collections should always be identical.
|
||||||
|
|
||||||
|
:node2pending: Map of nodes and the indices of their pending
|
||||||
|
tests. The indices are an index into ``.pending`` (which is
|
||||||
|
identical to their own collection stored in
|
||||||
|
``.node2collection``).
|
||||||
|
|
||||||
|
:collection: The one collection once it is validated to be
|
||||||
|
identical between all the nodes. It is initialised to None
|
||||||
|
until ``.init_distribute()`` is called.
|
||||||
|
|
||||||
|
:pending: List of indices of globally pending tests. These are
|
||||||
|
tests which have not yet been allocated to a chunk for a node
|
||||||
|
to process.
|
||||||
|
|
||||||
|
:log: A py.log.Producer instance.
|
||||||
|
|
||||||
|
:config: Config object, used for handling hooks.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, numnodes, log=None, config=None):
|
||||||
self.numnodes = numnodes
|
self.numnodes = numnodes
|
||||||
self.node2pending = {}
|
|
||||||
self.node2collection = {}
|
self.node2collection = {}
|
||||||
|
self.node2pending = {}
|
||||||
self.pending = []
|
self.pending = []
|
||||||
|
self.collection = None
|
||||||
if log is None:
|
if log is None:
|
||||||
self.log = py.log.Producer("loadsched")
|
self.log = py.log.Producer("loadsched")
|
||||||
else:
|
else:
|
||||||
self.log = log.loadsched
|
self.log = log.loadsched
|
||||||
self.collection_is_completed = False
|
self.config = config
|
||||||
|
|
||||||
|
@property
|
||||||
|
def nodes(self):
|
||||||
|
"""A list of all nodes in the scheduler."""
|
||||||
|
return list(self.node2pending.keys())
|
||||||
|
|
||||||
|
@property
|
||||||
|
def collection_is_completed(self):
|
||||||
|
"""Boolean indication initial test collection is complete.
|
||||||
|
|
||||||
|
This is a boolean indicating all initial participating nodes
|
||||||
|
have finished collection. The required number of initial
|
||||||
|
nodes is defined by ``.numnodes``.
|
||||||
|
"""
|
||||||
|
return len(self.node2collection) >= self.numnodes
|
||||||
|
|
||||||
|
def haspending(self):
|
||||||
|
"""Return True if there are pending test items
|
||||||
|
|
||||||
|
This indicates that collection has finished and nodes are
|
||||||
|
still processing test items, so can be thought of as "the
|
||||||
|
scheduler is active".
|
||||||
|
"""
|
||||||
|
if self.pending:
|
||||||
|
return True
|
||||||
|
for pending in self.node2pending.values():
|
||||||
|
if pending:
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
def hasnodes(self):
|
def hasnodes(self):
|
||||||
|
"""Return True if nodes exist in the scheduler."""
|
||||||
return bool(self.node2pending)
|
return bool(self.node2pending)
|
||||||
|
|
||||||
def addnode(self, node):
|
def addnode(self, node):
|
||||||
|
"""Add a new node in the scheduler.
|
||||||
|
|
||||||
|
From now on the node will be allocated chunks of tests to
|
||||||
|
execute.
|
||||||
|
|
||||||
|
Called by the ``DSession.slave_slaveready`` hook when it
|
||||||
|
sucessfully bootstrapped a new node.
|
||||||
|
"""
|
||||||
|
assert node not in self.node2pending
|
||||||
self.node2pending[node] = []
|
self.node2pending[node] = []
|
||||||
|
|
||||||
def tests_finished(self):
|
def tests_finished(self):
|
||||||
if not self.collection_is_completed or self.pending:
|
"""Return True if all tests have been executed by the nodes."""
|
||||||
|
if not self.collection_is_completed:
|
||||||
return False
|
return False
|
||||||
#for items in self.node2pending.values():
|
if self.pending:
|
||||||
# if items:
|
return False
|
||||||
# return False
|
for pending in self.node2pending.values():
|
||||||
|
if len(pending) >= 2:
|
||||||
|
return False
|
||||||
return True
|
return True
|
||||||
|
|
||||||
def addnode_collection(self, node, collection):
|
def addnode_collection(self, node, collection):
|
||||||
assert not self.collection_is_completed
|
"""Add the collected test items from a node
|
||||||
assert node in self.node2pending
|
|
||||||
self.node2collection[node] = list(collection)
|
|
||||||
if len(self.node2collection) >= self.numnodes:
|
|
||||||
self.collection_is_completed = True
|
|
||||||
|
|
||||||
def remove_item(self, node, item):
|
The collection is stored in the ``.node2collection`` map.
|
||||||
if item not in self.item2nodes:
|
Called by the ``DSession.slave_collectionfinish`` hook.
|
||||||
raise AssertionError(item, self.item2nodes)
|
"""
|
||||||
nodes = self.item2nodes[item]
|
assert node in self.node2pending
|
||||||
if node in nodes: # the node might have gone down already
|
if self.collection_is_completed:
|
||||||
nodes.remove(node)
|
# A new node has been added later, perhaps an original one died.
|
||||||
#if not nodes:
|
assert self.collection # .init_distribute() should have
|
||||||
# del self.item2nodes[item]
|
# been called by now
|
||||||
pending = self.node2pending[node]
|
if collection != self.collection:
|
||||||
pending.remove(item)
|
other_node = next(iter(self.node2collection.keys()))
|
||||||
# pre-load items-to-test if the node may become ready
|
msg = report_collection_diff(self.collection,
|
||||||
if self.pending and len(pending) < self.LOAD_THRESHOLD_NEWITEMS:
|
collection,
|
||||||
item = self.pending.pop(0)
|
other_node.gateway.id,
|
||||||
pending.append(item)
|
node.gateway.id)
|
||||||
self.item2nodes.setdefault(item, []).append(node)
|
self.log(msg)
|
||||||
node.send_runtest(item)
|
return
|
||||||
self.log("items waiting for node: %d" %(len(self.pending)))
|
self.node2collection[node] = list(collection)
|
||||||
#self.log("item2pending still executing: %s" %(self.item2nodes,))
|
|
||||||
#self.log("node2pending: %s" %(self.node2pending,))
|
def remove_item(self, node, item_index, duration=0):
|
||||||
|
"""Mark test item as completed by node
|
||||||
|
|
||||||
|
The duration it took to execute the item is used as a hint to
|
||||||
|
the scheduler.
|
||||||
|
|
||||||
|
This is called by the ``DSession.slave_testreport`` hook.
|
||||||
|
"""
|
||||||
|
self.node2pending[node].remove(item_index)
|
||||||
|
self.check_schedule(node, duration=duration)
|
||||||
|
|
||||||
|
def check_schedule(self, node, duration=0):
|
||||||
|
"""Maybe schedule new items on the node
|
||||||
|
|
||||||
|
If there are any globally pending nodes left then this will
|
||||||
|
check if the given node should be given any more tests. The
|
||||||
|
``duration`` of the last test is optionally used as a
|
||||||
|
heuristic to influence how many tests the node is assigned.
|
||||||
|
"""
|
||||||
|
if self.pending:
|
||||||
|
# how many nodes do we have?
|
||||||
|
num_nodes = len(self.node2pending)
|
||||||
|
# if our node goes below a heuristic minimum, fill it out to
|
||||||
|
# heuristic maximum
|
||||||
|
items_per_node_min = max(2, len(self.pending) // num_nodes // 4)
|
||||||
|
items_per_node_max = max(2, len(self.pending) // num_nodes // 2)
|
||||||
|
node_pending = self.node2pending[node]
|
||||||
|
if len(node_pending) < items_per_node_min:
|
||||||
|
if duration >= 0.1 and len(node_pending) >= 2:
|
||||||
|
# seems the node is doing long-running tests
|
||||||
|
# and has enough items to continue
|
||||||
|
# so let's rather wait with sending new items
|
||||||
|
return
|
||||||
|
num_send = items_per_node_max - len(node_pending)
|
||||||
|
self._send_tests(node, num_send)
|
||||||
|
self.log("num items waiting for node:", len(self.pending))
|
||||||
|
|
||||||
def remove_node(self, node):
|
def remove_node(self, node):
|
||||||
|
"""Remove an node from the scheduler
|
||||||
|
|
||||||
|
This should be called either when the node crashed or at
|
||||||
|
shutdown time. In the former case any pending items assigned
|
||||||
|
to the node will be re-scheduled. Called by the
|
||||||
|
``DSession.slave_slavefinished`` and
|
||||||
|
``DSession.slave_errordown`` hooks.
|
||||||
|
|
||||||
|
Return the item which was being executing while the node
|
||||||
|
crashed or None if the node has no more pending items.
|
||||||
|
|
||||||
|
"""
|
||||||
pending = self.node2pending.pop(node)
|
pending = self.node2pending.pop(node)
|
||||||
# KeyError if we didn't get an addnode() yet
|
|
||||||
for item in pending:
|
|
||||||
l = self.item2nodes[item]
|
|
||||||
l.remove(node)
|
|
||||||
if not l:
|
|
||||||
del self.item2nodes[item]
|
|
||||||
if not pending:
|
if not pending:
|
||||||
return
|
return
|
||||||
crashitem = pending.pop(0)
|
|
||||||
|
# The node crashed, reassing pending items
|
||||||
|
crashitem = self.collection[pending.pop(0)]
|
||||||
self.pending.extend(pending)
|
self.pending.extend(pending)
|
||||||
|
for node in self.node2pending:
|
||||||
|
self.check_schedule(node)
|
||||||
return crashitem
|
return crashitem
|
||||||
|
|
||||||
def init_distribute(self):
|
def init_distribute(self):
|
||||||
|
"""Initiate distribution of the test collection
|
||||||
|
|
||||||
|
Initiate scheduling of the items across the nodes. If this
|
||||||
|
gets called again later it behaves the same as calling
|
||||||
|
``.check_schedule()`` on all nodes so that newly added nodes
|
||||||
|
will start to be used.
|
||||||
|
|
||||||
|
This is called by the ``DSession.slave_collectionfinish`` hook
|
||||||
|
if ``.collection_is_completed`` is True.
|
||||||
|
|
||||||
|
XXX Perhaps this method should have been called ".schedule()".
|
||||||
|
"""
|
||||||
assert self.collection_is_completed
|
assert self.collection_is_completed
|
||||||
assert not hasattr(self, 'item2nodes')
|
|
||||||
self.item2nodes = {}
|
# Initial distribution already happend, reschedule on all nodes
|
||||||
|
if self.collection is not None:
|
||||||
|
for node in self.nodes:
|
||||||
|
self.check_schedule(node)
|
||||||
|
return
|
||||||
|
|
||||||
# XXX allow nodes to have different collections
|
# XXX allow nodes to have different collections
|
||||||
first_node, col = list(self.node2collection.items())[0]
|
if not self._check_nodes_have_same_collection():
|
||||||
for node, collection in self.node2collection.items():
|
self.log('**Different tests collected, aborting run**')
|
||||||
report_collection_diff(
|
return
|
||||||
|
|
||||||
|
# Collections are identical, create the index of pending items.
|
||||||
|
self.collection = list(self.node2collection.values())[0]
|
||||||
|
self.pending[:] = range(len(self.collection))
|
||||||
|
if not self.collection:
|
||||||
|
return
|
||||||
|
|
||||||
|
# how many items per node do we have about?
|
||||||
|
items_per_node = len(self.collection) // len(self.node2pending)
|
||||||
|
# take a fraction of tests for initial distribution
|
||||||
|
node_chunksize = max(items_per_node // 4, 2)
|
||||||
|
# and initialize each node with a chunk of tests
|
||||||
|
for node in self.nodes:
|
||||||
|
self._send_tests(node, node_chunksize)
|
||||||
|
|
||||||
|
def _send_tests(self, node, num):
|
||||||
|
tests_per_node = self.pending[:num]
|
||||||
|
if tests_per_node:
|
||||||
|
del self.pending[:num]
|
||||||
|
self.node2pending[node].extend(tests_per_node)
|
||||||
|
node.send_runtest_some(tests_per_node)
|
||||||
|
|
||||||
|
def _check_nodes_have_same_collection(self):
|
||||||
|
"""Return True if all nodes have collected the same items.
|
||||||
|
|
||||||
|
If collections differ, this method returns False while logging
|
||||||
|
the collection differences and posting collection errors to
|
||||||
|
pytest_collectreport hook.
|
||||||
|
"""
|
||||||
|
node_collection_items = list(self.node2collection.items())
|
||||||
|
first_node, col = node_collection_items[0]
|
||||||
|
same_collection = True
|
||||||
|
for node, collection in node_collection_items[1:]:
|
||||||
|
msg = report_collection_diff(
|
||||||
col,
|
col,
|
||||||
collection,
|
collection,
|
||||||
first_node.gateway.id,
|
first_node.gateway.id,
|
||||||
node.gateway.id,
|
node.gateway.id,
|
||||||
)
|
)
|
||||||
|
if msg:
|
||||||
|
same_collection = False
|
||||||
|
self.log(msg)
|
||||||
|
if self.config is not None:
|
||||||
|
rep = CollectReport(node.gateway.id, 'failed', longrepr=msg,
|
||||||
|
result=[])
|
||||||
|
self.config.hook.pytest_collectreport(report=rep)
|
||||||
|
|
||||||
self.pending = col
|
return same_collection
|
||||||
if not col:
|
|
||||||
return
|
|
||||||
available = list(self.node2pending.items())
|
|
||||||
num_available = self.numnodes
|
|
||||||
max_one_round = num_available * self.ITEM_CHUNKSIZE - 1
|
|
||||||
for i, item in enumerate(self.pending):
|
|
||||||
nodeindex = i % num_available
|
|
||||||
node, pending = available[nodeindex]
|
|
||||||
node.send_runtest(item)
|
|
||||||
self.item2nodes.setdefault(item, []).append(node)
|
|
||||||
pending.append(item)
|
|
||||||
if i >= max_one_round:
|
|
||||||
break
|
|
||||||
del self.pending[:i + 1]
|
|
||||||
|
|
||||||
|
|
||||||
def report_collection_diff(from_collection, to_collection, from_id, to_id):
|
def report_collection_diff(from_collection, to_collection, from_id, to_id):
|
||||||
"""Report the collected test difference between two nodes.
|
"""Report the collected test difference between two nodes.
|
||||||
|
|
||||||
:returns: True if collections are equal.
|
:returns: detailed message describing the difference between the given
|
||||||
|
collections, or None if they are equal.
|
||||||
:raises: AssertionError with a detailed error message describing the
|
|
||||||
difference between the collections.
|
|
||||||
|
|
||||||
"""
|
"""
|
||||||
if from_collection == to_collection:
|
if from_collection == to_collection:
|
||||||
return True
|
return None
|
||||||
|
|
||||||
diff = difflib.unified_diff(
|
diff = difflib.unified_diff(
|
||||||
from_collection,
|
from_collection,
|
||||||
@@ -183,13 +426,26 @@ def report_collection_diff(from_collection, to_collection, from_id, to_id):
|
|||||||
'{diff}'
|
'{diff}'
|
||||||
).format(from_id=from_id, to_id=to_id, diff='\n'.join(diff))
|
).format(from_id=from_id, to_id=to_id, diff='\n'.join(diff))
|
||||||
msg = "\n".join([x.rstrip() for x in error_message.split("\n")])
|
msg = "\n".join([x.rstrip() for x in error_message.split("\n")])
|
||||||
raise AssertionError(msg)
|
return msg
|
||||||
|
|
||||||
|
|
||||||
class Interrupted(KeyboardInterrupt):
|
class Interrupted(KeyboardInterrupt):
|
||||||
""" signals an immediate interruption. """
|
""" signals an immediate interruption. """
|
||||||
|
|
||||||
|
|
||||||
class DSession:
|
class DSession:
|
||||||
|
"""A py.test plugin which runs a distributed test session
|
||||||
|
|
||||||
|
At the beginning of the test session this creates a NodeManager
|
||||||
|
instance which creates and starts all nodes. Nodes then emit
|
||||||
|
events processed in the pytest_runtestloop hook using the slave_*
|
||||||
|
methods.
|
||||||
|
|
||||||
|
Once a node is started it will automatically start running the
|
||||||
|
py.test mainloop with some custom hooks. This means a node
|
||||||
|
automatically starts collecting tests. Once tests are collected
|
||||||
|
it will wait for instructions.
|
||||||
|
"""
|
||||||
def __init__(self, config):
|
def __init__(self, config):
|
||||||
self.config = config
|
self.config = config
|
||||||
self.log = py.log.Producer("dsession")
|
self.log = py.log.Producer("dsession")
|
||||||
@@ -200,6 +456,7 @@ class DSession:
|
|||||||
self.maxfail = config.getvalue("maxfail")
|
self.maxfail = config.getvalue("maxfail")
|
||||||
self.queue = queue.Queue()
|
self.queue = queue.Queue()
|
||||||
self._failed_collection_errors = {}
|
self._failed_collection_errors = {}
|
||||||
|
self._active_nodes = set()
|
||||||
try:
|
try:
|
||||||
self.terminal = config.pluginmanager.getplugin("terminalreporter")
|
self.terminal = config.pluginmanager.getplugin("terminalreporter")
|
||||||
except KeyError:
|
except KeyError:
|
||||||
@@ -208,18 +465,33 @@ class DSession:
|
|||||||
self.trdist = TerminalDistReporter(config)
|
self.trdist = TerminalDistReporter(config)
|
||||||
config.pluginmanager.register(self.trdist, "terminaldistreporter")
|
config.pluginmanager.register(self.trdist, "terminaldistreporter")
|
||||||
|
|
||||||
|
@property
|
||||||
|
def session_finished(self):
|
||||||
|
"""Return True if the distributed session has finished
|
||||||
|
|
||||||
|
This means all nodes have executed all test items. This is
|
||||||
|
used to by pytest_runtestloop to break out of it's loop.
|
||||||
|
"""
|
||||||
|
return bool(self.shuttingdown and not self._active_nodes)
|
||||||
|
|
||||||
def report_line(self, line):
|
def report_line(self, line):
|
||||||
if self.terminal and self.config.option.verbose >= 0:
|
if self.terminal and self.config.option.verbose >= 0:
|
||||||
self.terminal.write_line(line)
|
self.terminal.write_line(line)
|
||||||
|
|
||||||
@pytest.mark.trylast
|
@pytest.mark.trylast
|
||||||
def pytest_sessionstart(self, session):
|
def pytest_sessionstart(self, session):
|
||||||
|
"""Creates and starts the nodes.
|
||||||
|
|
||||||
|
The nodes are setup to put their events onto self.queue. As
|
||||||
|
soon as nodes start they will emit the slave_slaveready event.
|
||||||
|
"""
|
||||||
self.nodemanager = NodeManager(self.config)
|
self.nodemanager = NodeManager(self.config)
|
||||||
self.nodemanager.setup_nodes(putevent=self.queue.put)
|
nodes = self.nodemanager.setup_nodes(putevent=self.queue.put)
|
||||||
|
self._active_nodes.update(nodes)
|
||||||
|
|
||||||
def pytest_sessionfinish(self, session):
|
def pytest_sessionfinish(self, session):
|
||||||
""" teardown any resources after a test run. """
|
"""Shutdown all nodes."""
|
||||||
nm = getattr(self, 'nodemanager', None) # if not fully initialized
|
nm = getattr(self, 'nodemanager', None) # if not fully initialized
|
||||||
if nm is not None:
|
if nm is not None:
|
||||||
nm.teardown_nodes()
|
nm.teardown_nodes()
|
||||||
|
|
||||||
@@ -231,13 +503,13 @@ class DSession:
|
|||||||
numnodes = len(self.nodemanager.specs)
|
numnodes = len(self.nodemanager.specs)
|
||||||
dist = self.config.getvalue("dist")
|
dist = self.config.getvalue("dist")
|
||||||
if dist == "load":
|
if dist == "load":
|
||||||
self.sched = LoadScheduling(numnodes, log=self.log)
|
self.sched = LoadScheduling(numnodes, log=self.log,
|
||||||
|
config=self.config)
|
||||||
elif dist == "each":
|
elif dist == "each":
|
||||||
self.sched = EachScheduling(numnodes, log=self.log)
|
self.sched = EachScheduling(numnodes, log=self.log)
|
||||||
else:
|
else:
|
||||||
assert 0, dist
|
assert 0, dist
|
||||||
self.shouldstop = False
|
self.shouldstop = False
|
||||||
self.session_finished = False
|
|
||||||
while not self.session_finished:
|
while not self.session_finished:
|
||||||
self.loop_once()
|
self.loop_once()
|
||||||
if self.shouldstop:
|
if self.shouldstop:
|
||||||
@@ -245,7 +517,7 @@ class DSession:
|
|||||||
return True
|
return True
|
||||||
|
|
||||||
def loop_once(self):
|
def loop_once(self):
|
||||||
""" process one callback from one of the slaves. """
|
"""Process one callback from one of the slaves."""
|
||||||
while 1:
|
while 1:
|
||||||
try:
|
try:
|
||||||
eventcall = self.queue.get(timeout=2.0)
|
eventcall = self.queue.get(timeout=2.0)
|
||||||
@@ -256,7 +528,7 @@ class DSession:
|
|||||||
assert callname, kwargs
|
assert callname, kwargs
|
||||||
method = "slave_" + callname
|
method = "slave_" + callname
|
||||||
call = getattr(self, method)
|
call = getattr(self, method)
|
||||||
self.log("calling method: %s(**%s)" % (method, kwargs))
|
self.log("calling method", method, kwargs)
|
||||||
call(**kwargs)
|
call(**kwargs)
|
||||||
if self.sched.tests_finished():
|
if self.sched.tests_finished():
|
||||||
self.triggershutdown()
|
self.triggershutdown()
|
||||||
@@ -266,26 +538,40 @@ class DSession:
|
|||||||
#
|
#
|
||||||
|
|
||||||
def slave_slaveready(self, node, slaveinfo):
|
def slave_slaveready(self, node, slaveinfo):
|
||||||
|
"""Emitted when a node first starts up.
|
||||||
|
|
||||||
|
This adds the node to the scheduler, nodes continue with
|
||||||
|
collection without any further input.
|
||||||
|
"""
|
||||||
node.slaveinfo = slaveinfo
|
node.slaveinfo = slaveinfo
|
||||||
node.slaveinfo['id'] = node.gateway.id
|
node.slaveinfo['id'] = node.gateway.id
|
||||||
node.slaveinfo['spec'] = node.gateway.spec
|
node.slaveinfo['spec'] = node.gateway.spec
|
||||||
self.config.hook.pytest_testnodeready(node=node)
|
self.config.hook.pytest_testnodeready(node=node)
|
||||||
self.sched.addnode(node)
|
|
||||||
if self.shuttingdown:
|
if self.shuttingdown:
|
||||||
node.shutdown()
|
node.shutdown()
|
||||||
|
else:
|
||||||
|
self.sched.addnode(node)
|
||||||
|
|
||||||
def slave_slavefinished(self, node):
|
def slave_slavefinished(self, node):
|
||||||
|
"""Emitted when node executes its pytest_sessionfinish hook.
|
||||||
|
|
||||||
|
Removes the node from the scheduler.
|
||||||
|
|
||||||
|
The node might not be the scheduler if it had not emitted
|
||||||
|
slaveready before shutdown was triggered.
|
||||||
|
"""
|
||||||
self.config.hook.pytest_testnodedown(node=node, error=None)
|
self.config.hook.pytest_testnodedown(node=node, error=None)
|
||||||
if node.slaveoutput['exitstatus'] == 2: # keyboard-interrupt
|
if node.slaveoutput['exitstatus'] == 2: # keyboard-interrupt
|
||||||
self.shouldstop = "%s received keyboard-interrupt" % (node,)
|
self.shouldstop = "%s received keyboard-interrupt" % (node,)
|
||||||
self.slave_errordown(node, "keyboard-interrupt")
|
self.slave_errordown(node, "keyboard-interrupt")
|
||||||
return
|
return
|
||||||
crashitem = self.sched.remove_node(node)
|
if node in self.sched.nodes:
|
||||||
#assert not crashitem, (crashitem, node)
|
crashitem = self.sched.remove_node(node)
|
||||||
if self.shuttingdown and not self.sched.hasnodes():
|
assert not crashitem, (crashitem, node)
|
||||||
self.session_finished = True
|
self._active_nodes.remove(node)
|
||||||
|
|
||||||
def slave_errordown(self, node, error):
|
def slave_errordown(self, node, error):
|
||||||
|
"""Emitted by the SlaveController when a node dies."""
|
||||||
self.config.hook.pytest_testnodedown(node=node, error=error)
|
self.config.hook.pytest_testnodedown(node=node, error=error)
|
||||||
try:
|
try:
|
||||||
crashitem = self.sched.remove_node(node)
|
crashitem = self.sched.remove_node(node)
|
||||||
@@ -294,41 +580,70 @@ class DSession:
|
|||||||
else:
|
else:
|
||||||
if crashitem:
|
if crashitem:
|
||||||
self.handle_crashitem(crashitem, node)
|
self.handle_crashitem(crashitem, node)
|
||||||
#self.report_line("item crashed on node: %s" % crashitem)
|
self.report_line("Replacing failed node %s" % node.gateway.id)
|
||||||
if not self.sched.hasnodes():
|
self._clone_node(node)
|
||||||
self.session_finished = True
|
self._active_nodes.remove(node)
|
||||||
|
|
||||||
def slave_collectionfinish(self, node, ids):
|
def slave_collectionfinish(self, node, ids):
|
||||||
|
"""Slave has finished test collection.
|
||||||
|
|
||||||
|
This adds the collection for this node to the scheduler. If
|
||||||
|
the scheduler indicates collection is finished (i.e. all
|
||||||
|
initial nodes have submitted their collection), then tells the
|
||||||
|
scheduler to schedule the collected items. When initiating
|
||||||
|
scheduling the first time it logs which scheduler is in use.
|
||||||
|
"""
|
||||||
|
if self.shuttingdown:
|
||||||
|
return
|
||||||
self.sched.addnode_collection(node, ids)
|
self.sched.addnode_collection(node, ids)
|
||||||
if self.terminal:
|
if self.terminal:
|
||||||
self.trdist.setstatus(node.gateway.spec, "[%d]" %(len(ids)))
|
self.trdist.setstatus(node.gateway.spec, "[%d]" % (len(ids)))
|
||||||
|
|
||||||
if self.sched.collection_is_completed:
|
if self.sched.collection_is_completed:
|
||||||
if self.terminal:
|
if self.terminal and not self.sched.haspending():
|
||||||
self.trdist.ensure_show_status()
|
self.trdist.ensure_show_status()
|
||||||
self.terminal.write_line("")
|
self.terminal.write_line("")
|
||||||
self.terminal.write_line("scheduling tests via %s" %(
|
self.terminal.write_line("scheduling tests via %s" % (
|
||||||
self.sched.__class__.__name__))
|
self.sched.__class__.__name__))
|
||||||
|
|
||||||
self.sched.init_distribute()
|
self.sched.init_distribute()
|
||||||
|
|
||||||
def slave_logstart(self, node, nodeid, location):
|
def slave_logstart(self, node, nodeid, location):
|
||||||
|
"""Emitted when a node calls the pytest_runtest_logstart hook."""
|
||||||
self.config.hook.pytest_runtest_logstart(
|
self.config.hook.pytest_runtest_logstart(
|
||||||
nodeid=nodeid, location=location)
|
nodeid=nodeid, location=location)
|
||||||
|
|
||||||
def slave_testreport(self, node, rep):
|
def slave_testreport(self, node, rep):
|
||||||
if not (rep.passed and rep.when != "call"):
|
"""Emitted when a node calls the pytest_runtest_logreport hook.
|
||||||
if rep.when in ("setup", "call"):
|
|
||||||
self.sched.remove_item(node, rep.nodeid)
|
If the node indicates it is finished with a test item remove
|
||||||
|
the item from the pending list in the scheduler.
|
||||||
|
"""
|
||||||
|
if rep.when == "call" or (rep.when == "setup" and not rep.passed):
|
||||||
|
self.sched.remove_item(node, rep.item_index, rep.duration)
|
||||||
#self.report_line("testreport %s: %s" %(rep.id, rep.status))
|
#self.report_line("testreport %s: %s" %(rep.id, rep.status))
|
||||||
rep.node = node
|
rep.node = node
|
||||||
self.config.hook.pytest_runtest_logreport(report=rep)
|
self.config.hook.pytest_runtest_logreport(report=rep)
|
||||||
self._handlefailures(rep)
|
self._handlefailures(rep)
|
||||||
|
|
||||||
def slave_collectreport(self, node, rep):
|
def slave_collectreport(self, node, rep):
|
||||||
|
"""Emitted when a node calls the pytest_collectreport hook."""
|
||||||
if rep.failed:
|
if rep.failed:
|
||||||
self._failed_slave_collectreport(node, rep)
|
self._failed_slave_collectreport(node, rep)
|
||||||
|
|
||||||
|
def _clone_node(self, node):
|
||||||
|
"""Return new node based on an existing one.
|
||||||
|
|
||||||
|
This is normally for when a node died, this will copy the spec
|
||||||
|
of the existing node and create a new one with a new id. The
|
||||||
|
new node will have been setup so will start calling the
|
||||||
|
"slave_*" hooks and do work soon.
|
||||||
|
"""
|
||||||
|
spec = node.gateway.spec
|
||||||
|
spec.id = None
|
||||||
|
self.nodemanager.group.allocate_id(spec)
|
||||||
|
node = self.nodemanager.setup_node(spec, self.queue.put)
|
||||||
|
self._active_nodes.add(node)
|
||||||
|
return node
|
||||||
|
|
||||||
def _failed_slave_collectreport(self, node, rep):
|
def _failed_slave_collectreport(self, node, rep):
|
||||||
# Check we haven't already seen this report (from
|
# Check we haven't already seen this report (from
|
||||||
# another slave).
|
# another slave).
|
||||||
@@ -347,19 +662,21 @@ class DSession:
|
|||||||
def triggershutdown(self):
|
def triggershutdown(self):
|
||||||
self.log("triggering shutdown")
|
self.log("triggering shutdown")
|
||||||
self.shuttingdown = True
|
self.shuttingdown = True
|
||||||
for node in self.sched.node2pending:
|
for node in self.sched.nodes:
|
||||||
node.shutdown()
|
node.shutdown()
|
||||||
|
|
||||||
def handle_crashitem(self, nodeid, slave):
|
def handle_crashitem(self, nodeid, slave):
|
||||||
# XXX get more reporting info by recording pytest_runtest_logstart?
|
# XXX get more reporting info by recording pytest_runtest_logstart?
|
||||||
|
# XXX count no of failures and retry N times
|
||||||
runner = self.config.pluginmanager.getplugin("runner")
|
runner = self.config.pluginmanager.getplugin("runner")
|
||||||
fspath = nodeid.split("::")[0]
|
fspath = nodeid.split("::")[0]
|
||||||
msg = "Slave %r crashed while running %r" %(slave.gateway.id, nodeid)
|
msg = "Slave %r crashed while running %r" % (slave.gateway.id, nodeid)
|
||||||
rep = runner.TestReport(nodeid, (fspath, None, fspath), (),
|
rep = runner.TestReport(nodeid, (fspath, None, fspath),
|
||||||
"failed", msg, "???")
|
(), "failed", msg, "???")
|
||||||
rep.node = slave
|
rep.node = slave
|
||||||
self.config.hook.pytest_runtest_logreport(report=rep)
|
self.config.hook.pytest_runtest_logreport(report=rep)
|
||||||
|
|
||||||
|
|
||||||
class TerminalDistReporter:
|
class TerminalDistReporter:
|
||||||
def __init__(self, config):
|
def __init__(self, config):
|
||||||
self.config = config
|
self.config = config
|
||||||
|
|||||||
@@ -87,7 +87,7 @@ class RemoteControl(object):
|
|||||||
result = self.runsession()
|
result = self.runsession()
|
||||||
failures, reports, collection_failed = result
|
failures, reports, collection_failed = result
|
||||||
if collection_failed:
|
if collection_failed:
|
||||||
reports = ["Collection failed, keeping previous failure set"]
|
pass # "Collection failed, keeping previous failure set"
|
||||||
else:
|
else:
|
||||||
uniq_failures = []
|
uniq_failures = []
|
||||||
for failure in failures:
|
for failure in failures:
|
||||||
@@ -109,7 +109,6 @@ def repr_pytest_looponfailinfo(failreports, rootdirs):
|
|||||||
|
|
||||||
def init_slave_session(channel, args, option_dict):
|
def init_slave_session(channel, args, option_dict):
|
||||||
import os, sys
|
import os, sys
|
||||||
import py
|
|
||||||
outchannel = channel.gateway.newchannel()
|
outchannel = channel.gateway.newchannel()
|
||||||
sys.stdout = sys.stderr = outchannel.makefile('w')
|
sys.stdout = sys.stderr = outchannel.makefile('w')
|
||||||
channel.send(outchannel)
|
channel.send(outchannel)
|
||||||
@@ -188,7 +187,7 @@ class StatRecorder:
|
|||||||
self.check() # snapshot state
|
self.check() # snapshot state
|
||||||
|
|
||||||
def fil(self, p):
|
def fil(self, p):
|
||||||
return True # we are sensitive to all file changes since 1.9
|
return p.check(file=1, dotfile=0) and p.ext != ".pyc"
|
||||||
def rec(self, p):
|
def rec(self, p):
|
||||||
return p.check(dotfile=0)
|
return p.check(dotfile=0)
|
||||||
|
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import sys
|
import py
|
||||||
import py, pytest
|
import pytest
|
||||||
|
|
||||||
def pytest_addoption(parser):
|
def pytest_addoption(parser):
|
||||||
group = parser.getgroup("xdist", "distributed and subprocess testing")
|
group = parser.getgroup("xdist", "distributed and subprocess testing")
|
||||||
@@ -28,12 +28,14 @@ def pytest_addoption(parser):
|
|||||||
group._addoption('-d',
|
group._addoption('-d',
|
||||||
action="store_true", dest="distload", default=False,
|
action="store_true", dest="distload", default=False,
|
||||||
help="load-balance tests. shortcut for '--dist=load'")
|
help="load-balance tests. shortcut for '--dist=load'")
|
||||||
group.addoption('--rsyncdir', action="append", default=[], metavar="dir1",
|
group.addoption('--rsyncdir', action="append", default=[], metavar="DIR",
|
||||||
help="add directory for rsyncing to remote tx nodes.")
|
help="add directory for rsyncing to remote tx nodes.")
|
||||||
|
group.addoption('--rsyncignore', action="append", default=[], metavar="GLOB",
|
||||||
|
help="add expression for ignores when rsyncing to remote tx nodes.")
|
||||||
|
|
||||||
parser.addini('rsyncdirs', 'list of (relative) paths to be rsynced for'
|
parser.addini('rsyncdirs', 'list of (relative) paths to be rsynced for'
|
||||||
' remote distributed testing.', type="pathlist")
|
' remote distributed testing.', type="pathlist")
|
||||||
parser.addini('rsyncignore', 'list of (relative) paths to be ignored '
|
parser.addini('rsyncignore', 'list of (relative) glob-style paths to be ignored '
|
||||||
'for rsyncing.', type="pathlist")
|
'for rsyncing.', type="pathlist")
|
||||||
parser.addini("looponfailroots", type="pathlist",
|
parser.addini("looponfailroots", type="pathlist",
|
||||||
help="directories to check for changes", default=[py.path.local()])
|
help="directories to check for changes", default=[py.path.local()])
|
||||||
@@ -43,7 +45,11 @@ def pytest_addoption(parser):
|
|||||||
# -------------------------------------------------------------------------
|
# -------------------------------------------------------------------------
|
||||||
def pytest_addhooks(pluginmanager):
|
def pytest_addhooks(pluginmanager):
|
||||||
from xdist import newhooks
|
from xdist import newhooks
|
||||||
pluginmanager.addhooks(newhooks)
|
# avoid warnings with pytest-2.8
|
||||||
|
method = getattr(pluginmanager, "add_hookspecs", None)
|
||||||
|
if method is None:
|
||||||
|
method = pluginmanager.addhooks
|
||||||
|
method(newhooks)
|
||||||
|
|
||||||
# -------------------------------------------------------------------------
|
# -------------------------------------------------------------------------
|
||||||
# distributed testing initialization
|
# distributed testing initialization
|
||||||
@@ -51,17 +57,19 @@ def pytest_addhooks(pluginmanager):
|
|||||||
|
|
||||||
def pytest_cmdline_main(config):
|
def pytest_cmdline_main(config):
|
||||||
check_options(config)
|
check_options(config)
|
||||||
if config.getvalue("looponfail"):
|
if config.getoption("looponfail"):
|
||||||
from xdist.looponfail import looponfail_main
|
from xdist.looponfail import looponfail_main
|
||||||
looponfail_main(config)
|
looponfail_main(config)
|
||||||
return 2 # looponfail only can get stop with ctrl-C anyway
|
return 2 # looponfail only can get stop with ctrl-C anyway
|
||||||
|
|
||||||
def pytest_configure(config, __multicall__):
|
@pytest.mark.trylast
|
||||||
__multicall__.execute()
|
def pytest_configure(config):
|
||||||
if config.getvalue("dist") != "no":
|
if config.getoption("dist") != "no":
|
||||||
from xdist.dsession import DSession
|
from xdist.dsession import DSession
|
||||||
session = DSession(config)
|
session = DSession(config)
|
||||||
config.pluginmanager.register(session, "dsession")
|
config.pluginmanager.register(session, "dsession")
|
||||||
|
tr = config.pluginmanager.getplugin("terminalreporter")
|
||||||
|
tr.showfspath = False
|
||||||
|
|
||||||
def check_options(config):
|
def check_options(config):
|
||||||
if config.option.numprocesses:
|
if config.option.numprocesses:
|
||||||
@@ -114,10 +122,14 @@ def forked_run_report(item):
|
|||||||
|
|
||||||
def report_process_crash(item, result):
|
def report_process_crash(item, result):
|
||||||
path, lineno = item._getfslineno()
|
path, lineno = item._getfslineno()
|
||||||
info = "%s:%s: running the test CRASHED with signal %d" %(
|
info = ("%s:%s: running the test CRASHED with signal %d" %
|
||||||
path, lineno, result.signal)
|
(path, lineno, result.signal))
|
||||||
from _pytest import runner
|
from _pytest import runner
|
||||||
call = runner.CallInfo(lambda: 0/0, "???")
|
call = runner.CallInfo(lambda: 0/0, "???")
|
||||||
call.excinfo = info
|
call.excinfo = info
|
||||||
rep = runner.pytest_runtest_makereport(item, call)
|
rep = runner.pytest_runtest_makereport(item, call)
|
||||||
|
if result.out:
|
||||||
|
rep.sections.append(("captured stdout", result.out))
|
||||||
|
if result.err:
|
||||||
|
rep.sections.append(("captured stderr", result.err))
|
||||||
return rep
|
return rep
|
||||||
|
|||||||
@@ -24,7 +24,7 @@ class SlaveInteractor:
|
|||||||
|
|
||||||
def pytest_internalerror(self, excrepr):
|
def pytest_internalerror(self, excrepr):
|
||||||
for line in str(excrepr).split("\n"):
|
for line in str(excrepr).split("\n"):
|
||||||
self.log("IERROR> " + line)
|
self.log("IERROR>", line)
|
||||||
|
|
||||||
def pytest_sessionstart(self, session):
|
def pytest_sessionstart(self, session):
|
||||||
self.session = session
|
self.session = session
|
||||||
@@ -45,41 +45,44 @@ class SlaveInteractor:
|
|||||||
torun = []
|
torun = []
|
||||||
while 1:
|
while 1:
|
||||||
name, kwargs = self.channel.receive()
|
name, kwargs = self.channel.receive()
|
||||||
self.log("received command %s(**%s)" % (name, kwargs))
|
self.log("received command", name, kwargs)
|
||||||
if name == "runtests":
|
if name == "runtests":
|
||||||
ids = kwargs['ids']
|
torun.extend(kwargs['indices'])
|
||||||
for nodeid in ids:
|
|
||||||
torun.append(self._id2item[nodeid])
|
|
||||||
elif name == "runtests_all":
|
elif name == "runtests_all":
|
||||||
torun.extend(session.items)
|
torun.extend(range(len(session.items)))
|
||||||
self.log("items to run: %s" %(len(torun)))
|
self.log("items to run:", torun)
|
||||||
|
# only run if we have an item and a next item
|
||||||
while len(torun) >= 2:
|
while len(torun) >= 2:
|
||||||
item = torun.pop(0)
|
self.run_tests(torun)
|
||||||
nextitem = torun[0]
|
|
||||||
self.config.hook.pytest_runtest_protocol(item=item,
|
|
||||||
nextitem=nextitem)
|
|
||||||
if name == "shutdown":
|
if name == "shutdown":
|
||||||
while torun:
|
if torun:
|
||||||
self.config.hook.pytest_runtest_protocol(
|
self.run_tests(torun)
|
||||||
item=torun.pop(0), nextitem=None)
|
|
||||||
break
|
break
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
def run_tests(self, torun):
|
||||||
|
items = self.session.items
|
||||||
|
self.item_index = torun.pop(0)
|
||||||
|
if torun:
|
||||||
|
nextitem = items[torun[0]]
|
||||||
|
else:
|
||||||
|
nextitem = None
|
||||||
|
self.config.hook.pytest_runtest_protocol(
|
||||||
|
item=items[self.item_index],
|
||||||
|
nextitem=nextitem)
|
||||||
|
|
||||||
def pytest_collection_finish(self, session):
|
def pytest_collection_finish(self, session):
|
||||||
self._id2item = {}
|
|
||||||
ids = []
|
|
||||||
for item in session.items:
|
|
||||||
self._id2item[item.nodeid] = item
|
|
||||||
ids.append(item.nodeid)
|
|
||||||
self.sendevent("collectionfinish",
|
self.sendevent("collectionfinish",
|
||||||
topdir=str(session.fspath),
|
topdir=str(session.fspath),
|
||||||
ids=ids)
|
ids=[item.nodeid for item in session.items])
|
||||||
|
|
||||||
#def pytest_runtest_logstart(self, nodeid, location, fspath):
|
def pytest_runtest_logstart(self, nodeid, location):
|
||||||
# self.sendevent("logstart", nodeid=nodeid, location=location)
|
self.sendevent("logstart", nodeid=nodeid, location=location)
|
||||||
|
|
||||||
def pytest_runtest_logreport(self, report):
|
def pytest_runtest_logreport(self, report):
|
||||||
data = serialize_report(report)
|
data = serialize_report(report)
|
||||||
|
data["item_index"] = self.item_index
|
||||||
|
assert self.session.items[self.item_index].nodeid == report.nodeid
|
||||||
self.sendevent("testreport", data=data)
|
self.sendevent("testreport", data=data)
|
||||||
|
|
||||||
def pytest_collectreport(self, report):
|
def pytest_collectreport(self, report):
|
||||||
@@ -125,6 +128,7 @@ def remote_initconfig(option_dict, args):
|
|||||||
|
|
||||||
|
|
||||||
if __name__ == '__channelexec__':
|
if __name__ == '__channelexec__':
|
||||||
|
channel = channel # noqa
|
||||||
# python3.2 is not concurrent import safe, so let's play it safe
|
# python3.2 is not concurrent import safe, so let's play it safe
|
||||||
# https://bitbucket.org/hpk42/pytest/issue/347/pytest-xdist-and-python-32
|
# https://bitbucket.org/hpk42/pytest/issue/347/pytest-xdist-and-python-32
|
||||||
if sys.version_info[:2] == (3,2):
|
if sys.version_info[:2] == (3,2):
|
||||||
|
|||||||
@@ -1,5 +1,8 @@
|
|||||||
import py, pytest
|
import fnmatch
|
||||||
import sys, os
|
import os
|
||||||
|
|
||||||
|
import py
|
||||||
|
import pytest
|
||||||
import execnet
|
import execnet
|
||||||
import xdist.remote
|
import xdist.remote
|
||||||
|
|
||||||
@@ -7,6 +10,7 @@ from _pytest import runner # XXX load dynamically
|
|||||||
|
|
||||||
class NodeManager(object):
|
class NodeManager(object):
|
||||||
EXIT_TIMEOUT = 10
|
EXIT_TIMEOUT = 10
|
||||||
|
DEFAULT_IGNORES = ['.*', '*.pyc', '*.pyo', '*~']
|
||||||
def __init__(self, config, specs=None, defaultchdir="pyexecnetcache"):
|
def __init__(self, config, specs=None, defaultchdir="pyexecnetcache"):
|
||||||
self.config = config
|
self.config = config
|
||||||
self._nodesready = py.std.threading.Event()
|
self._nodesready = py.std.threading.Event()
|
||||||
@@ -23,38 +27,33 @@ class NodeManager(object):
|
|||||||
self.group.allocate_id(spec)
|
self.group.allocate_id(spec)
|
||||||
self.specs.append(spec)
|
self.specs.append(spec)
|
||||||
self.roots = self._getrsyncdirs()
|
self.roots = self._getrsyncdirs()
|
||||||
|
self.rsyncoptions = self._getrsyncoptions()
|
||||||
|
self._rsynced_specs = py.builtin.set()
|
||||||
|
|
||||||
def rsync_roots(self):
|
def rsync_roots(self, gateway):
|
||||||
""" make sure that all remote gateways
|
"""Rsync the set of roots to the node's gateway cwd."""
|
||||||
have the same set of roots in their
|
|
||||||
current directory.
|
|
||||||
"""
|
|
||||||
options = {
|
|
||||||
'ignores': self.config.getini("rsyncignore"),
|
|
||||||
'verbose': self.config.option.verbose,
|
|
||||||
}
|
|
||||||
if self.roots:
|
if self.roots:
|
||||||
# send each rsync root
|
|
||||||
for root in self.roots:
|
for root in self.roots:
|
||||||
self.rsync(root, **options)
|
self.rsync(gateway, root, **self.rsyncoptions)
|
||||||
|
|
||||||
def makegateways(self):
|
|
||||||
assert not list(self.group)
|
|
||||||
self.config.hook.pytest_xdist_setupnodes(config=self.config,
|
|
||||||
specs=self.specs)
|
|
||||||
for spec in self.specs:
|
|
||||||
gw = self.group.makegateway(spec)
|
|
||||||
self.config.hook.pytest_xdist_newgateway(gateway=gw)
|
|
||||||
|
|
||||||
def setup_nodes(self, putevent):
|
def setup_nodes(self, putevent):
|
||||||
self.makegateways()
|
self.config.hook.pytest_xdist_setupnodes(config=self.config,
|
||||||
self.rsync_roots()
|
specs=self.specs)
|
||||||
self.trace("setting up nodes")
|
self.trace("setting up nodes")
|
||||||
for gateway in self.group:
|
nodes = []
|
||||||
node = SlaveController(self, gateway, self.config, putevent)
|
for spec in self.specs:
|
||||||
gateway.node = node # to keep node alive
|
nodes.append(self.setup_node(spec, putevent))
|
||||||
node.setup()
|
return nodes
|
||||||
self.trace("started node %r" % node)
|
|
||||||
|
def setup_node(self, spec, putevent):
|
||||||
|
gw = self.group.makegateway(spec)
|
||||||
|
self.config.hook.pytest_xdist_newgateway(gateway=gw)
|
||||||
|
self.rsync_roots(gw)
|
||||||
|
node = SlaveController(self, gw, self.config, putevent)
|
||||||
|
gw.node = node # keep the node alive
|
||||||
|
node.setup()
|
||||||
|
self.trace("started node %r" % node)
|
||||||
|
return node
|
||||||
|
|
||||||
def teardown_nodes(self):
|
def teardown_nodes(self):
|
||||||
self.group.terminate(self.EXIT_TIMEOUT)
|
self.group.terminate(self.EXIT_TIMEOUT)
|
||||||
@@ -98,38 +97,47 @@ class NodeManager(object):
|
|||||||
roots.append(root)
|
roots.append(root)
|
||||||
return roots
|
return roots
|
||||||
|
|
||||||
def rsync(self, source, notify=None, verbose=False, ignores=None):
|
def _getrsyncoptions(self):
|
||||||
""" perform rsync to all remote hosts.
|
"""Get options to be passed for rsync."""
|
||||||
"""
|
ignores = list(self.DEFAULT_IGNORES)
|
||||||
|
ignores += self.config.option.rsyncignore
|
||||||
|
ignores += self.config.getini("rsyncignore")
|
||||||
|
|
||||||
|
return {
|
||||||
|
'ignores': ignores,
|
||||||
|
'verbose': self.config.option.verbose,
|
||||||
|
}
|
||||||
|
|
||||||
|
def rsync(self, gateway, source, notify=None, verbose=False, ignores=None):
|
||||||
|
"""Perform rsync to remote hosts for node."""
|
||||||
|
# XXX This changes the calling behaviour of
|
||||||
|
# pytest_xdist_rsyncstart and pytest_xdist_rsyncfinish to
|
||||||
|
# be called once per rsync target.
|
||||||
rsync = HostRSync(source, verbose=verbose, ignores=ignores)
|
rsync = HostRSync(source, verbose=verbose, ignores=ignores)
|
||||||
seen = py.builtin.set()
|
spec = gateway.spec
|
||||||
gateways = []
|
if spec.popen and not spec.chdir:
|
||||||
for gateway in self.group:
|
# XXX This assumes that sources are python-packages
|
||||||
spec = gateway.spec
|
# and that adding the basedir does not hurt.
|
||||||
if spec.popen and not spec.chdir:
|
gateway.remote_exec("""
|
||||||
# XXX this assumes that sources are python-packages
|
import sys ; sys.path.insert(0, %r)
|
||||||
# and that adding the basedir does not hurt
|
""" % os.path.dirname(str(source))).waitclose()
|
||||||
gateway.remote_exec("""
|
return
|
||||||
import sys ; sys.path.insert(0, %r)
|
if (spec, source) in self._rsynced_specs:
|
||||||
""" % os.path.dirname(str(source))).waitclose()
|
return
|
||||||
continue
|
def finished():
|
||||||
if spec not in seen:
|
if notify:
|
||||||
def finished():
|
notify("rsyncrootready", spec, source)
|
||||||
if notify:
|
rsync.add_target_host(gateway, finished=finished)
|
||||||
notify("rsyncrootready", spec, source)
|
self._rsynced_specs.add((spec, source))
|
||||||
rsync.add_target_host(gateway, finished=finished)
|
self.config.hook.pytest_xdist_rsyncstart(
|
||||||
seen.add(spec)
|
source=source,
|
||||||
gateways.append(gateway)
|
gateways=[gateway],
|
||||||
if seen:
|
)
|
||||||
self.config.hook.pytest_xdist_rsyncstart(
|
rsync.send()
|
||||||
source=source,
|
self.config.hook.pytest_xdist_rsyncfinish(
|
||||||
gateways=gateways,
|
source=source,
|
||||||
)
|
gateways=[gateway],
|
||||||
rsync.send()
|
)
|
||||||
self.config.hook.pytest_xdist_rsyncfinish(
|
|
||||||
source=source,
|
|
||||||
gateways=gateways,
|
|
||||||
)
|
|
||||||
|
|
||||||
class HostRSync(execnet.RSync):
|
class HostRSync(execnet.RSync):
|
||||||
""" RSyncer that filters out common files
|
""" RSyncer that filters out common files
|
||||||
@@ -144,14 +152,12 @@ class HostRSync(execnet.RSync):
|
|||||||
|
|
||||||
def filter(self, path):
|
def filter(self, path):
|
||||||
path = py.path.local(path)
|
path = py.path.local(path)
|
||||||
if not path.ext in ('.pyc', '.pyo'):
|
for x in self._ignores:
|
||||||
if not path.basename.endswith('~'):
|
x = getattr(x, 'strpath', x)
|
||||||
if path.check(dotfile=0):
|
if fnmatch.fnmatch(path.basename, x) or fnmatch.fnmatch(path.strpath, x):
|
||||||
for x in self._ignores:
|
return False
|
||||||
if path == x:
|
else:
|
||||||
break
|
return True
|
||||||
else:
|
|
||||||
return True
|
|
||||||
|
|
||||||
def add_target_host(self, gateway, finished=None):
|
def add_target_host(self, gateway, finished=None):
|
||||||
remotepath = os.path.basename(self._sourcedir)
|
remotepath = os.path.basename(self._sourcedir)
|
||||||
@@ -230,8 +236,8 @@ class SlaveController(object):
|
|||||||
self.gateway.exit()
|
self.gateway.exit()
|
||||||
#del self.gateway
|
#del self.gateway
|
||||||
|
|
||||||
def send_runtest(self, nodeid):
|
def send_runtest_some(self, indices):
|
||||||
self.sendcommand("runtests", ids=[nodeid])
|
self.sendcommand("runtests", indices=indices)
|
||||||
|
|
||||||
def send_runtest_all(self):
|
def send_runtest_all(self):
|
||||||
self.sendcommand("runtests_all",)
|
self.sendcommand("runtests_all",)
|
||||||
@@ -278,10 +284,13 @@ class SlaveController(object):
|
|||||||
self._down = True
|
self._down = True
|
||||||
self.slaveoutput = kwargs['slaveoutput']
|
self.slaveoutput = kwargs['slaveoutput']
|
||||||
self.notify_inproc("slavefinished", node=self)
|
self.notify_inproc("slavefinished", node=self)
|
||||||
#elif eventname == "logstart":
|
elif eventname == "logstart":
|
||||||
# self.notify_inproc(eventname, node=self, **kwargs)
|
self.notify_inproc(eventname, node=self, **kwargs)
|
||||||
elif eventname in ("testreport", "collectreport", "teardownreport"):
|
elif eventname in ("testreport", "collectreport", "teardownreport"):
|
||||||
|
item_index = kwargs.pop("item_index", None)
|
||||||
rep = unserialize_report(eventname, kwargs['data'])
|
rep = unserialize_report(eventname, kwargs['data'])
|
||||||
|
if item_index is not None:
|
||||||
|
rep.item_index = item_index
|
||||||
self.notify_inproc(eventname, node=self, rep=rep)
|
self.notify_inproc(eventname, node=self, rep=rep)
|
||||||
elif eventname == "collectionfinish":
|
elif eventname == "collectionfinish":
|
||||||
self.notify_inproc(eventname, node=self, ids=kwargs['ids'])
|
self.notify_inproc(eventname, node=self, ids=kwargs['ids'])
|
||||||
@@ -296,8 +305,7 @@ class SlaveController(object):
|
|||||||
self.config.pluginmanager.notify_exception(excinfo)
|
self.config.pluginmanager.notify_exception(excinfo)
|
||||||
|
|
||||||
def unserialize_report(name, reportdict):
|
def unserialize_report(name, reportdict):
|
||||||
d = reportdict
|
|
||||||
if name == "testreport":
|
if name == "testreport":
|
||||||
return runner.TestReport(**d)
|
return runner.TestReport(**reportdict)
|
||||||
elif name == "collectreport":
|
elif name == "collectreport":
|
||||||
return runner.CollectReport(**d)
|
return runner.CollectReport(**reportdict)
|
||||||
|
|||||||
Reference in New Issue
Block a user