2010-11-06 06:37:25 +08:00
|
|
|
|
|
|
|
.. _`unittest.TestCase`:
|
|
|
|
|
2012-10-07 19:06:17 +08:00
|
|
|
Support for unittest.TestCase / Integration of fixtures
|
2010-10-12 16:59:04 +08:00
|
|
|
=====================================================================
|
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
.. _`unittest.py style`: http://docs.python.org/library/unittest.html
|
|
|
|
|
2012-10-09 20:35:17 +08:00
|
|
|
py.test has support for running Python `unittest.py style`_ tests.
|
2012-10-12 20:52:36 +08:00
|
|
|
It's meant for leveraging existing unittest-style projects
|
|
|
|
to use pytest features. Concretely, pytest will automatically
|
|
|
|
collect ``unittest.TestCase`` subclasses and their ``test`` methods in
|
|
|
|
test files. It will invoke typlical ``setUp/tearDown`` methods and
|
|
|
|
generally try to make test suites written to run on unittest, to also
|
|
|
|
run using pytest. We assume here that you are familiar with writing
|
|
|
|
``unittest.TestCase`` style tests and rather focus on
|
|
|
|
integration aspects.
|
|
|
|
|
|
|
|
Usage
|
|
|
|
-------------------------------------------------------------------
|
|
|
|
|
|
|
|
After :ref:`installation` type::
|
|
|
|
|
|
|
|
py.test
|
|
|
|
|
|
|
|
and you should be able to run your unittest-style tests if they
|
|
|
|
are contained in ``test_*`` modules. This way you can make
|
|
|
|
use of most :ref:`pytest features <features>`, for example
|
|
|
|
``--pdb`` debugging in failures, using :ref:`plain assert-statements <assert>`,
|
|
|
|
:ref:`more informative tracebacks <tbreportdemo>`, stdout-capturing or
|
|
|
|
distributing tests to multiple CPUs via the ``-nNUM`` option if you
|
|
|
|
installed the ``pytest-xdist`` plugin. Please refer to
|
|
|
|
the general pytest documentation for many more examples.
|
|
|
|
|
|
|
|
Mixing pytest fixtures into unittest.TestCase style tests
|
|
|
|
-----------------------------------------------------------
|
|
|
|
|
|
|
|
pytest supports using its :ref:`fixture mechanism <fixture>` with
|
|
|
|
``unittest.TestCase`` style tests. Assuming you have at least skimmed
|
|
|
|
the pytest fixture features, let's jump-start into an example that
|
|
|
|
integrates a pytest ``db_class`` fixture, setting up a
|
|
|
|
class-cached database object, and then reference it from
|
|
|
|
a unittest-style test::
|
|
|
|
|
|
|
|
# content of conftest.py
|
|
|
|
|
|
|
|
# hooks and fixtures in this file are available throughout all test
|
|
|
|
# modules living below the directory of this conftest.py file
|
|
|
|
|
|
|
|
import pytest
|
2010-10-12 16:59:04 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
@pytest.fixture(scope="class")
|
|
|
|
def db_class(request):
|
|
|
|
class DummyDB:
|
|
|
|
pass
|
|
|
|
request.cls.db = DummyDB()
|
|
|
|
|
|
|
|
This defines a fixture function ``db_class`` which - if used - is
|
|
|
|
called once for each test class and which sets the class-level
|
|
|
|
``db`` attribute to a ``DummyDB`` instance. The fixture function
|
|
|
|
achieves this by receiving a special ``request`` object which gives
|
|
|
|
access to :ref:`the requesting test context <request-context>` such
|
|
|
|
as the ``cls`` attribute, denoting the class from which the fixture
|
|
|
|
is used. This architecture de-couples fixture writing from actual test
|
|
|
|
code and allows re-use of the fixture by a minimal reference, the fixture
|
|
|
|
name. So let's write an actual ``unittest.TestCase`` class using our
|
|
|
|
fixture definition::
|
|
|
|
|
|
|
|
# content of test_unittest_db.py
|
2010-10-12 16:59:04 +08:00
|
|
|
|
|
|
|
import unittest
|
2012-10-12 20:52:36 +08:00
|
|
|
import pytest
|
|
|
|
|
|
|
|
@pytest.mark.usefixtures("db_class")
|
2010-10-12 16:59:04 +08:00
|
|
|
class MyTest(unittest.TestCase):
|
2012-10-12 20:52:36 +08:00
|
|
|
def test_method1(self):
|
|
|
|
assert hasattr(self, "db")
|
|
|
|
assert 0, self.db # fail for demo purposes
|
|
|
|
|
|
|
|
def test_method2(self):
|
|
|
|
assert 0, self.db # fail for demo purposes
|
|
|
|
|
|
|
|
The ``@pytest.mark.usefixtures("db_class")`` class-decorator makes sure that
|
|
|
|
the pytest fixture function ``db_class`` is called. Due to the deliberately
|
|
|
|
failing assert statements, we can take a look at the ``self.db`` values
|
|
|
|
in the traceback::
|
2010-10-12 16:59:04 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
$ py.test test_unittest_db.py
|
2010-10-12 16:59:04 +08:00
|
|
|
=========================== test session starts ============================
|
2012-10-12 20:52:36 +08:00
|
|
|
platform linux2 -- Python 2.7.3 -- pytest-2.3.0.dev22
|
|
|
|
plugins: xdist, bugzilla, cache, oejskit, cli, pep8, cov, timeout
|
|
|
|
collected 2 items
|
2010-10-12 16:59:04 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
test_unittest_db.py FF
|
2010-10-12 16:59:04 +08:00
|
|
|
|
|
|
|
================================= FAILURES =================================
|
2012-10-12 20:52:36 +08:00
|
|
|
___________________________ MyTest.test_method1 ____________________________
|
2010-10-12 16:59:04 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
self = <test_unittest_db.MyTest testMethod=test_method1>
|
2010-10-12 16:59:04 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
def test_method1(self):
|
|
|
|
assert hasattr(self, "db")
|
|
|
|
> assert 0, self.db # fail for demo purposes
|
|
|
|
E AssertionError: <conftest.DummyDB instance at 0x135dea8>
|
|
|
|
|
|
|
|
test_unittest_db.py:9: AssertionError
|
|
|
|
___________________________ MyTest.test_method2 ____________________________
|
|
|
|
|
|
|
|
self = <test_unittest_db.MyTest testMethod=test_method2>
|
|
|
|
|
|
|
|
def test_method2(self):
|
|
|
|
> assert 0, self.db # fail for demo purposes
|
|
|
|
E AssertionError: <conftest.DummyDB instance at 0x135dea8>
|
2010-10-12 16:59:04 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
test_unittest_db.py:12: AssertionError
|
|
|
|
========================= 2 failed in 0.04 seconds =========================
|
|
|
|
|
|
|
|
This default pytest traceback shows that, indeed, the two test methods
|
|
|
|
see the same ``self.db`` attribute instance which was our intention
|
|
|
|
when writing the class-scoped fixture function.
|
2010-10-12 16:59:04 +08:00
|
|
|
|
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
autouse fixtures and accessing other fixtures
|
|
|
|
-------------------------------------------------------------------
|
2012-09-18 16:54:12 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
Although it's usually better to explicitely declare use of fixtures you need
|
|
|
|
for a given test, you may sometimes want to have fixtures that are
|
|
|
|
automatically used in a given context. For this, you can flag
|
|
|
|
fixture functions with ``@pytest.fixture(autouse=True)`` and define
|
|
|
|
the fixture function in the context where you want it used. Let's look
|
|
|
|
at an example which makes all test methods of a ``TestCase`` class
|
|
|
|
execute in a clean temporary directory, using a ``initdir`` fixture
|
|
|
|
which itself uses the pytest builtin ``tmpdir`` fixture::
|
|
|
|
|
|
|
|
# content of test_unittest_cleandir.py
|
2012-09-18 16:54:12 +08:00
|
|
|
import pytest
|
|
|
|
import unittest
|
|
|
|
|
|
|
|
class MyTest(unittest.TestCase):
|
2012-10-12 20:52:36 +08:00
|
|
|
@pytest.fixture(autouse=True)
|
|
|
|
def initdir(self, tmpdir):
|
2012-09-18 16:54:12 +08:00
|
|
|
tmpdir.chdir() # change to pytest-provided temporary directory
|
|
|
|
tmpdir.join("samplefile.ini").write("# testdata")
|
|
|
|
|
|
|
|
def test_method(self):
|
|
|
|
s = open("samplefile.ini").read()
|
|
|
|
assert "testdata" in s
|
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
The ``initdir`` fixture function will be used for all methods of the
|
|
|
|
class where it is defined. This is basically just a shortcut for
|
|
|
|
using a ``@pytest.mark.usefixtures("initdir")`` on the class like in
|
|
|
|
the previous example. Note, that the ``initdir`` fixture function
|
|
|
|
accepts a :ref:`tmpdir <tmpdir>` argument, referencing a pytest
|
|
|
|
builtin fixture.
|
2012-09-18 16:54:12 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
Running this test module ...::
|
2012-09-18 16:54:12 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
$ py.test -q test_unittest_cleandir.py
|
|
|
|
.
|
2012-10-07 19:06:17 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
... gives us one passed test because the ``initdir`` fixture function
|
|
|
|
was executed ahead of the ``test_method``.
|
2012-10-07 19:06:17 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
.. note::
|
2012-10-07 19:06:17 +08:00
|
|
|
|
2012-10-12 20:52:36 +08:00
|
|
|
``unittest.TestCase`` methods cannot directly receive fixture or
|
|
|
|
function arguments as implementing that is likely to inflict
|
|
|
|
on the ability to run general unittest.TestCase test suites.
|
|
|
|
Given enough demand, attempts might be made, though. If
|
|
|
|
unittest finally grows a reasonable plugin system that should
|
|
|
|
help as well. In the meanwhile, the above ``usefixtures`` and
|
|
|
|
``autouse`` examples should help to mix in pytest fixtures into
|
|
|
|
unittest suites.
|