summaryrefslogtreecommitdiff
path: root/src/_pytest/nose.py
blob: 16d5224e9fa82f81b7db23062bd2bd0f7a445c86 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
"""Run testsuites written for nose."""
from _pytest.config import hookimpl
from _pytest.fixtures import getfixturemarker
from _pytest.nodes import Item
from _pytest.python import Function
from _pytest.unittest import TestCaseFunction


@hookimpl(trylast=True)
def pytest_runtest_setup(item: Item) -> None:
    if not isinstance(item, Function):
        return
    # Don't do nose style setup/teardown on direct unittest style classes.
    if isinstance(item, TestCaseFunction):
        return

    # Capture the narrowed type of item for the teardown closure,
    # see https://github.com/python/mypy/issues/2608
    func = item

    if not call_optional(func.obj, "setup"):
        # Call module level setup if there is no object level one.
        assert func.parent is not None
        call_optional(func.parent.obj, "setup")  # type: ignore[attr-defined]

    def teardown_nose() -> None:
        if not call_optional(func.obj, "teardown"):
            assert func.parent is not None
            call_optional(func.parent.obj, "teardown")  # type: ignore[attr-defined]

    # XXX This implies we only call teardown when setup worked.
    func.addfinalizer(teardown_nose)


def call_optional(obj: object, name: str) -> bool:
    method = getattr(obj, name, None)
    if method is None:
        return False
    is_fixture = getfixturemarker(method) is not None
    if is_fixture:
        return False
    if not callable(method):
        return False
    # If there are any problems allow the exception to raise rather than
    # silently ignoring it.
    method()
    return True