summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
authorRan Benita <ran@unusedvar.com>2020-12-30 16:02:16 +0200
committerRan Benita <ran@unusedvar.com>2021-01-24 14:08:39 +0200
commitda70f61f67cc37209a5d97bca18303bdd7bccbc8 (patch)
tree830871b387c4515303fc4aa2a5fbfc4e0656d2cd /src
parentd4f8e4b40ce481df0f34b2eac5d61aefae4ed2e1 (diff)
downloadpytest-da70f61f67cc37209a5d97bca18303bdd7bccbc8.tar.gz
runner: complete type annotations of SetupState
Diffstat (limited to 'src')
-rw-r--r--src/_pytest/runner.py16
1 files changed, 8 insertions, 8 deletions
diff --git a/src/_pytest/runner.py b/src/_pytest/runner.py
index 844e41f80..a49879432 100644
--- a/src/_pytest/runner.py
+++ b/src/_pytest/runner.py
@@ -403,22 +403,22 @@ def pytest_make_collect_report(collector: Collector) -> CollectReport:
class SetupState:
"""Shared state for setting up/tearing down test items or collectors."""
- def __init__(self):
+ def __init__(self) -> None:
self.stack: List[Node] = []
self._finalizers: Dict[Node, List[Callable[[], object]]] = {}
- def addfinalizer(self, finalizer: Callable[[], object], colitem) -> None:
+ def addfinalizer(self, finalizer: Callable[[], object], colitem: Node) -> None:
"""Attach a finalizer to the given colitem."""
assert colitem and not isinstance(colitem, tuple)
assert callable(finalizer)
# assert colitem in self.stack # some unit tests don't setup stack :/
self._finalizers.setdefault(colitem, []).append(finalizer)
- def _pop_and_teardown(self):
+ def _pop_and_teardown(self) -> None:
colitem = self.stack.pop()
self._teardown_with_finalization(colitem)
- def _callfinalizers(self, colitem) -> None:
+ def _callfinalizers(self, colitem: Node) -> None:
finalizers = self._finalizers.pop(colitem, None)
exc = None
while finalizers:
@@ -433,7 +433,7 @@ class SetupState:
if exc:
raise exc
- def _teardown_with_finalization(self, colitem) -> None:
+ def _teardown_with_finalization(self, colitem: Node) -> None:
self._callfinalizers(colitem)
colitem.teardown()
for colitem in self._finalizers:
@@ -446,11 +446,11 @@ class SetupState:
self._teardown_with_finalization(key)
assert not self._finalizers
- def teardown_exact(self, item, nextitem) -> None:
+ def teardown_exact(self, item: Item, nextitem: Optional[Item]) -> None:
needed_collectors = nextitem and nextitem.listchain() or []
self._teardown_towards(needed_collectors)
- def _teardown_towards(self, needed_collectors) -> None:
+ def _teardown_towards(self, needed_collectors: List[Node]) -> None:
exc = None
while self.stack:
if self.stack == needed_collectors[: len(self.stack)]:
@@ -465,7 +465,7 @@ class SetupState:
if exc:
raise exc
- def prepare(self, colitem) -> None:
+ def prepare(self, colitem: Item) -> None:
"""Setup objects along the collector chain to the test-method."""
# Check if the last collection node has raised an error.