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
|
import sys
from importlib import reload
from types import ModuleType
import pytest
from test.widgets.conftest import FakeBar
# Net widget only needs bytes_recv/sent attributes
# Widget displays increase since last poll therefore
# we need to increment value each time this is called.
class MockPsutil(ModuleType):
up = 0
down = 0
@classmethod
def net_io_counters(cls, pernic=False, _nowrap=True):
class IOCounters:
def __init__(self, up, down):
self.bytes_sent = up
self.bytes_recv = down
cls.up += 40000
cls.down += 1200000
if pernic:
return {"wlp58s0": IOCounters(cls.up, cls.down), "lo": IOCounters(cls.up, cls.down)}
return IOCounters(cls.up, cls.down)
# Patch the widget with our mock psutil module.
# Wrap widget so tests can pass keyword arguments.
@pytest.fixture
def patch_net(fake_qtile, monkeypatch, fake_window):
def build_widget(**kwargs):
monkeypatch.setitem(sys.modules, "psutil", MockPsutil("psutil"))
from libqtile.widget import net
# Reload fixes cases where psutil may have been imported previously
reload(net)
widget = net.Net(
format="{interface}: U {up}{up_suffix} {up_cumulative}{up_cumulative_suffix} D "
"{down}{down_suffix} {down_cumulative}{down_cumulative_suffix} T {total}"
"{total_suffix} {total_cumulative}{total_cumulative_suffix}",
**kwargs,
)
fakebar = FakeBar([widget], window=fake_window)
widget._configure(fake_qtile, fakebar)
return widget
return build_widget
def test_net_defaults(patch_net):
"""Default: widget shows `all` interfaces"""
net1 = patch_net()
assert net1.poll() == "all: U 40.0kB 80.0kB D 1.2MB 2.4MB T 1.24MB 2.48MB"
def test_net_single_interface(patch_net):
"""Display single named interface"""
net2 = patch_net(interface="wlp58s0")
assert net2.poll() == "wlp58s0: U 40.0kB 160.0kB D 1.2MB 4.8MB T 1.24MB 4.96MB"
def test_net_list_interface(patch_net):
"""Display multiple named interfaces"""
net2 = patch_net(interface=["wlp58s0", "lo"])
assert net2.poll() == (
"wlp58s0: U 40.0kB 240.0kB D 1.2MB 7.2MB T 1.24MB 7.44MB lo: U 40.0kB "
"240.0kB D 1.2MB 7.2MB T 1.24MB 7.44MB"
)
def test_net_invalid_interface(patch_net):
"""Pass an invalid interface value"""
with pytest.raises(AttributeError):
_ = patch_net(interface=12)
def test_net_use_bits(patch_net):
"""Display all interfaces in bits rather than bytes"""
net4 = patch_net(use_bits=True)
assert net4.poll() == "all: U 320.0kb 2.56Mb D 9.6Mb 76.8Mb T 9.92Mb 79.36Mb"
def test_net_convert_zero_b(patch_net):
"""Zero bytes is a special case in `convert_b`"""
net5 = patch_net()
assert net5.convert_b(0.0) == (0.0, "B")
def test_net_use_prefix(patch_net):
"""Tests `prefix` configurable option"""
net6 = patch_net(prefix="M")
assert net6.poll() == "all: U 0.04MB 440.0kB D 1.2MB 13.2MB T 1.24MB 13.64MB"
# Untested: 128-129 - generic exception catching
|
