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
|
import datetime
import sys
from importlib import reload
import pytest
from libqtile.config import Bar, Screen
from libqtile.widget import vertical_clock
from test.helpers import BareConfig
# Mock Datetime object that returns a set datetime and also
# has a simplified timezone method to check functionality of
# the widget.
class MockDatetime(datetime.datetime):
@classmethod
def now(cls, *args, **kwargs):
return cls(2024, 1, 1, 10, 20, 30)
def astimezone(self, tzone=None):
if tzone is None:
return self
return self + tzone.utcoffset(None)
@pytest.fixture
def patched_clock(monkeypatch):
# Stop system importing these modules in case they exist on environment
monkeypatch.setitem(sys.modules, "pytz", None)
monkeypatch.setitem(sys.modules, "dateutil", None)
monkeypatch.setitem(sys.modules, "dateutil.tz", None)
# Reload module to force ImportErrors
reload(vertical_clock)
# Override datetime.
# This is key for testing as we can fix time.
monkeypatch.setattr("libqtile.widget.vertical_clock.datetime", MockDatetime)
class TestVerticalClock(vertical_clock.VerticalClock):
def __init__(self, **config):
vertical_clock.VerticalClock.__init__(self, **config)
self.name = "verticalclock"
def info(self):
info = vertical_clock.VerticalClock.info(self)
info["text"] = "|".join(layout.text for layout in self.layouts)
return info
yield TestVerticalClock
@pytest.fixture(scope="function")
def vclock_manager(manager_nospawn, request, patched_clock):
class VClockConfig(BareConfig):
screens = [
Screen(
left=Bar(
[
patched_clock(
**getattr(request, "param", dict()),
)
],
30,
)
)
]
manager_nospawn.start(VClockConfig)
yield manager_nospawn
def config(**kwargs):
return pytest.mark.parametrize("vclock_manager", [kwargs], indirect=True)
def test_vclock_default(vclock_manager):
assert vclock_manager.c.widget["verticalclock"].info()["text"] == "10|20"
@config(format=["%H", "%M", "-", "%d", "%m", "%Y"])
def test_vclock_extra_lines(vclock_manager):
assert vclock_manager.c.widget["verticalclock"].info()["text"] == "10|20|-|01|01|2024"
@pytest.mark.parametrize(
"vclock_manager",
[
dict(fontsize=[10]), # too few
dict(fontsize=[10, 20, 30, 40]), # too many
dict(fontsize=[10, "fff"]), # mix values
dict(foreground=["fff"]), # too few
dict(foreground=["fff"] * 4), # too many
dict(foreground=["fff", 10]), # mix values
],
indirect=True,
)
def test_vclock_invalid_configs(vclock_manager):
assert vclock_manager.c.bar["left"].info()["widgets"][0]["name"] == "configerrorwidget"
|
