acoolname/neotest-python
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(pytest): wrap pytest types

Browse Source
This commit is contained in:
Rónán Carrigan
2022-11-10 14:57:14 +00:00
parent 16c3c90b55
commit 438744e274
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
neotest_python/pytest.py
View File

@@ -32,11 +32,11 @@ class NeotestResultCollector:
self.stream = stream self.stream = stream
self.adapter = adapter self.adapter = adapter
self.pytest_config: Optional[pytest.Config] = None # type: ignore self.pytest_config: Optional["pytest.Config"] = None # type: ignore
self.results: Dict[str, NeotestResult] = {} self.results: Dict[str, NeotestResult] = {}
def _get_short_output( def _get_short_output(
self, config: pytest.Config, report: pytest.TestReport self, config: "pytest.Config", report: "pytest.TestReport"
) -> Optional[str]: ) -> Optional[str]:
buffer = StringIO() buffer = StringIO()
# Hack to get pytest to write ANSI codes # Hack to get pytest to write ANSI codes
@@ -57,7 +57,7 @@ class NeotestResultCollector:
buffer.seek(0) buffer.seek(0)
return buffer.read() return buffer.read()
def pytest_deselected(self, items: List[pytest.Item]): def pytest_deselected(self, items: List["pytest.Item"]):
for report in items: for report in items:
file_path, *name_path = report.nodeid.split("::") file_path, *name_path = report.nodeid.split("::")
abs_path = str(Path(self.pytest_config.rootdir, file_path)) abs_path = str(Path(self.pytest_config.rootdir, file_path))
@@ -76,11 +76,11 @@ class NeotestResultCollector:
self.stream(pos_id, result) self.stream(pos_id, result)
self.results[pos_id] = result self.results[pos_id] = result
def pytest_cmdline_main(self, config: pytest.Config): def pytest_cmdline_main(self, config: "pytest.Config"):
self.pytest_config = config self.pytest_config = config
@pytest.hookimpl(hookwrapper=True) @pytest.hookimpl(hookwrapper=True)
def pytest_runtest_makereport(self, item: pytest.Item, call: pytest.CallInfo) -> None: def pytest_runtest_makereport(self, item: "pytest.Item", call: "pytest.CallInfo") -> None:
# pytest generates the report.outcome field in its internal # pytest generates the report.outcome field in its internal
# pytest_runtest_makereport implementation, so call it first. (We don't # pytest_runtest_makereport implementation, so call it first. (We don't
# implement pytest_runtest_logreport because it doesn't have access to # implement pytest_runtest_logreport because it doesn't have access to
@@ -143,9 +143,9 @@ class NeotestDebugpyPlugin:
def pytest_exception_interact( def pytest_exception_interact(
self, self,
node: Union[pytest.Item, pytest.Collector], node: Union["pytest.Item", "pytest.Collector"],
call: pytest.CallInfo, call: "pytest.CallInfo",
report: Union[pytest.CollectReport, pytest.TestReport], report: Union["pytest.CollectReport", "pytest.TestReport"],
): ):
# call.excinfo: _pytest._code.ExceptionInfo # call.excinfo: _pytest._code.ExceptionInfo
self.maybe_debugpy_postmortem(call.excinfo._excinfo) self.maybe_debugpy_postmortem(call.excinfo._excinfo)