l.behm/bumblebee-status
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

[tests/cpu] adapt tests and add per-cpu tests

Browse source 
fixes #785
This commit is contained in:
tobi-wan-kenobi 2021-04-27 17:17:55 +02:00
parent fb6be007e5
commit 028932a560
1 changed files with 32 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

37
tests/modules/core/test_cpu.py
View file

@ -7,8 +7,9 @@ import modules.core.cpu
pytest.importorskip("psutil") pytest.importorskip("psutil")
def build_module(): def build_module(percpu=False):
config = core.config.Config([]) config = core.config.Config(["-p", "percpu={}".format(percpu)])
config.set("cpu.percpu", percpu)
return modules.core.cpu.Module(config=config, theme=None) return modules.core.cpu.Module(config=config, theme=None)
def cpu_widget(module): def cpu_widget(module):
@ -42,21 +43,47 @@ class TestCPU(TestCase):
cpu_percent_mock.return_value = 50 cpu_percent_mock.return_value = 50
module = build_module() module = build_module()
assert module.state(None) == None assert module.state(module.widget()) == None
@mock.patch('psutil.cpu_percent') @mock.patch('psutil.cpu_percent')
def test_warning_state(self, cpu_percent_mock): def test_warning_state(self, cpu_percent_mock):
cpu_percent_mock.return_value = 75 cpu_percent_mock.return_value = 75
module = build_module() module = build_module()
assert module.state(None) == 'warning' assert module.state(module.widget()) == 'warning'
@mock.patch('psutil.cpu_percent') @mock.patch('psutil.cpu_percent')
def test_critical_state(self, cpu_percent_mock): def test_critical_state(self, cpu_percent_mock):
cpu_percent_mock.return_value = 82 cpu_percent_mock.return_value = 82
module = build_module() module = build_module()
assert module.state(None) == 'critical' assert module.state(module.widget()) == 'critical'
@mock.patch('psutil.cpu_percent')
def test_healthy_state_percpu(self, cpu_percent_mock):
cpu_percent_mock.return_value = [50,42,47]
module = build_module(percpu=True)
for widget in module.widgets():
assert module.state(widget) == None
@mock.patch('psutil.cpu_percent')
def test_warning_state_percpu(self, cpu_percent_mock):
cpu_percent_mock.return_value = [50,72,47]
module = build_module(percpu=True)
assert module.state(module.widgets()[0]) == None
assert module.state(module.widgets()[1]) == "warning"
assert module.state(module.widgets()[2]) == None
@mock.patch('psutil.cpu_percent')
def test_warning_state_percpu(self, cpu_percent_mock):
cpu_percent_mock.return_value = [50,72,99]
module = build_module(percpu=True)
assert module.state(module.widgets()[0]) == None
assert module.state(module.widgets()[1]) == "warning"
assert module.state(module.widgets()[2]) == "critical"
@mock.patch('core.input.register') @mock.patch('core.input.register')
def test_register_left_mouse_action(self, input_register_mock): def test_register_left_mouse_action(self, input_register_mock):