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
|
diff --git a/src/pytestsysstats/plugin.py b/src/pytestsysstats/plugin.py
index 4009858..3067f73 100644
--- a/src/pytestsysstats/plugin.py
+++ b/src/pytestsysstats/plugin.py
@@ -231,20 +231,17 @@ def pytest_sessionstart(session: "Session") -> None:
):
stats_processes_instance = StatsProcesses()
stats_processes_instance.add("Test Suite Run", os.getpid())
- else:
- stats_processes_instance = None
+ session.config.pluginmanager.register(stats_processes_instance, "sysstats-processes")
- session.config.pluginmanager.register(stats_processes_instance, "sysstats-processes")
-
- terminalreporter = session.config.pluginmanager.getplugin(
- "terminalreporter"
- ) # type: "TerminalReporter"
- sys_stats_reporter = SystemStatsReporter(
- config=session.config,
- stats_processes=stats_processes_instance,
- terminalreporter=terminalreporter,
- )
- session.config.pluginmanager.register(sys_stats_reporter, "sysstats-reporter")
+ terminalreporter = session.config.pluginmanager.getplugin(
+ "terminalreporter"
+ ) # type: "TerminalReporter"
+ sys_stats_reporter = SystemStatsReporter(
+ config=session.config,
+ stats_processes=stats_processes_instance,
+ terminalreporter=terminalreporter,
+ )
+ session.config.pluginmanager.register(sys_stats_reporter, "sysstats-reporter")
@pytest.fixture(scope="session") # type: ignore[misc]
@@ -252,5 +249,5 @@ def stats_processes(request: "SubRequest") -> StatsProcesses:
"""
Session scoped process statistics tracker.
"""
- plugin = request.config.pluginmanager.get_plugin("sysstats-processes") # type: StatsProcesses
- return plugin
+ if request.config.pluginmanager.has_plugin("sysstats-processes"):
+ return request.config.pluginmanager.get_plugin("sysstats-processes") # type: StatsProcesses
|
