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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
# -*- coding: utf-8 -*-
'''
pytestsalt.fixtures.stats
~~~~~~~~~~~~~~~~~~~~~~~~~
Test session process stats fixtures
'''
# Import Python libs
from __future__ import absolute_import, print_function, unicode_literals
import os
import sys
from collections import OrderedDict
# Import 3rd-party libs
import psutil
# Import pytest libs
import pytest
from _pytest.terminal import TerminalReporter
IS_WINDOWS = sys.platform.startswith('win')
class SaltTerminalReporter(TerminalReporter):
def __init__(self, config):
TerminalReporter.__init__(self, config)
@pytest.hookimpl(trylast=True)
def pytest_sessionstart(self, session):
TerminalReporter.pytest_sessionstart(self, session)
self._session = session
def pytest_runtest_logreport(self, report):
TerminalReporter.pytest_runtest_logreport(self, report)
if self.verbosity <= 0:
return
if report.when != 'call':
return
if self.config.getoption('--sys-stats') is False:
return
if self.verbosity > 1:
self.ensure_newline()
self.section('Processes Statistics', sep='-', bold=True)
left_padding = len(max(['System'] + list(self._session.stats_processes), key=len))
template = ' ...{} {} - CPU: {:6.2f} % MEM: {:6.2f} %'
if not IS_WINDOWS:
template += ' SWAP: {:6.2f} %'
template += '\n'
self.write(
template.format(
'.' * (left_padding - len('System')),
'System',
psutil.cpu_percent(),
psutil.virtual_memory().percent,
psutil.swap_memory().percent
)
)
for name, psproc in self._session.stats_processes.items():
with psproc.oneshot():
cpu = psproc.cpu_percent()
mem = psproc.memory_percent('vms')
dots = '.' * (left_padding - len(name))
if not IS_WINDOWS:
swap = psproc.memory_percent('swap')
formatted = template.format(dots, name, cpu, mem, swap)
else:
formatted = template.format(dots, name, cpu, mem)
self.write(formatted)
def _get_progress_information_message(self):
msg = TerminalReporter._get_progress_information_message(self)
if self.verbosity <= 0:
return msg
if self.config.getoption('--sys-stats') is False:
return msg
if self.verbosity == 1:
msg = ' [CPU:{}%] [MEM:{}%]{}'.format(psutil.cpu_percent(),
psutil.virtual_memory().percent,
msg)
return msg
def pytest_sessionstart(session):
session.stats_processes = OrderedDict((
('Test Suite Run', psutil.Process(os.getpid())),
))
def pytest_addoption(parser):
'''
register argparse-style options and ini-style config values.
'''
output_options_group = parser.getgroup('Output Options')
output_options_group.addoption(
'--sys-stats',
default=False,
action='store_true',
help='Print System CPU and MEM statistics after each test execution.'
)
@pytest.mark.trylast
def pytest_configure(config):
'''
called after command line options have been parsed
and all plugins and initial conftest files been loaded.
'''
# Register our terminal reporter
if not getattr(config, 'slaveinput', None):
standard_reporter = config.pluginmanager.getplugin('terminalreporter')
salt_reporter = SaltTerminalReporter(standard_reporter.config)
config.pluginmanager.unregister(standard_reporter)
config.pluginmanager.register(salt_reporter, 'terminalreporter')
|
