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
|
import os
import pytest
import libqtile
from libqtile.widget import generic_poll_text
# Use local httpbin in CI, fallback to httpbin.org for local development
HTTPBIN_BASE = (
"http://localhost:8080"
if os.environ.get("GITHUB_ACTIONS") == "true"
else "https://httpbin.org"
)
def test_gen_poll_text():
gpt_no_func = generic_poll_text.GenPollText()
assert gpt_no_func.poll() == "You need a poll function"
gpt_with_func = generic_poll_text.GenPollText(func=lambda: "Has function")
assert gpt_with_func.poll() == "Has function"
@pytest.mark.asyncio
async def test_gen_poll_url_not_configured():
gpurl = generic_poll_text.GenPollUrl()
assert await gpurl.apoll() == "Invalid config"
def test_gen_poll_url_no_json():
gpurl = generic_poll_text.GenPollUrl(json=False)
assert "Content-Type" not in gpurl.headers
def test_gen_poll_url_headers_and_json():
gpurl = generic_poll_text.GenPollUrl(
headers={"fake-header": "fake-value"},
data={"argument": "data value"},
user_agent="qtile test",
)
assert gpurl.headers["User-agent"] == "qtile test"
assert gpurl.headers["fake-header"] == "fake-value"
assert gpurl.headers["Content-Type"] == "application/json"
assert gpurl.data.decode() == '{"argument": "data value"}'
@pytest.mark.asyncio
async def test_gen_poll_url_text():
gpurl = generic_poll_text.GenPollUrl(
json=False, parse=lambda x: x, url=f"{HTTPBIN_BASE}/anything"
)
result = await gpurl.apoll()
assert isinstance(result, str)
assert "anything" in result
@pytest.mark.asyncio
async def test_gen_poll_url_json_with_data():
gpurl = generic_poll_text.GenPollUrl(
parse=lambda x: x["data"], data={"test": "value"}, url=f"{HTTPBIN_BASE}/anything"
)
result = await gpurl.apoll()
assert result == '{"test": "value"}'
@pytest.mark.asyncio
async def test_gen_poll_url_xml_no_xmltodict():
gpurl = generic_poll_text.GenPollUrl(
json=False, xml=True, parse=lambda x: x, url=f"{HTTPBIN_BASE}/anything"
)
result = await gpurl.apoll()
assert result == "Can't parse"
@pytest.mark.asyncio
async def test_gen_poll_url_broken_parse():
gpurl = generic_poll_text.GenPollUrl(
json=False, parse=lambda x: x.foo, url=f"{HTTPBIN_BASE}/anything"
)
result = await gpurl.apoll()
assert result == "Can't parse"
@pytest.mark.asyncio
async def test_gen_poll_url_custom_headers():
gpurl = generic_poll_text.GenPollUrl(
headers={"X-Custom-Header": "test-value", "X-Another-Header": "another-value"},
parse=lambda x: x["headers"],
url=f"{HTTPBIN_BASE}/headers",
)
result = await gpurl.apoll()
assert "X-Custom-Header" in result
assert result["X-Custom-Header"] == "test-value"
assert "X-Another-Header" in result
assert result["X-Another-Header"] == "another-value"
def test_gen_poll_command(manager_nospawn, minimal_conf_noscreen):
gpcommand = generic_poll_text.GenPollCommand(cmd=["echo", "hello"])
config = minimal_conf_noscreen
config.screens = [libqtile.config.Screen(top=libqtile.bar.Bar([gpcommand], 10))]
manager_nospawn.start(config)
command = manager_nospawn.c.widget["genpollcommand"]
assert command.info()["text"] == "hello"
|
