wehub-resource-sync
e52ddb7dd4
CodeQL / Analyze (python) (push) Failing after 1s
Linting and testing / build (3.11) (push) Failing after 0s
Package exe with PyInstaller - Windows / build (push) Failing after 2s
Linting and testing / build (3.10) (push) Failing after 1s
Linting and testing / minimal-install (push) Failing after 1s
Update sites rating and statistics / build (push) Failing after 1s
Build docker image and push to DockerHub / docker (push) Failing after 1s
Linting and testing / build (3.12) (push) Failing after 1s
Linting and testing / build (3.14) (push) Failing after 1s
Linting and testing / build (3.13) (push) Failing after 2s
122 lines
4.2 KiB
Python
122 lines
4.2 KiB
Python
from colorama import Fore, Style
|
|
|
|
from maigret.errors import CheckError
|
|
from maigret.notify import QueryNotifyPrint
|
|
from maigret.result import MaigretCheckStatus, MaigretCheckResult
|
|
|
|
|
|
def test_notify_illegal():
|
|
n = QueryNotifyPrint(color=False)
|
|
|
|
assert (
|
|
n.update(
|
|
MaigretCheckResult(
|
|
username="test",
|
|
status=MaigretCheckStatus.ILLEGAL,
|
|
site_name="TEST_SITE",
|
|
site_url_user="http://example.com/test",
|
|
)
|
|
)
|
|
== "[-] TEST_SITE: Illegal Username Format For This Site!"
|
|
)
|
|
|
|
|
|
def test_notify_claimed():
|
|
n = QueryNotifyPrint(color=False)
|
|
|
|
assert (
|
|
n.update(
|
|
MaigretCheckResult(
|
|
username="test",
|
|
status=MaigretCheckStatus.CLAIMED,
|
|
site_name="TEST_SITE",
|
|
site_url_user="http://example.com/test",
|
|
)
|
|
)
|
|
== "[+] TEST_SITE: http://example.com/test"
|
|
)
|
|
|
|
|
|
def test_notify_available():
|
|
n = QueryNotifyPrint(color=False)
|
|
|
|
assert (
|
|
n.update(
|
|
MaigretCheckResult(
|
|
username="test",
|
|
status=MaigretCheckStatus.AVAILABLE,
|
|
site_name="TEST_SITE",
|
|
site_url_user="http://example.com/test",
|
|
)
|
|
)
|
|
== "[-] TEST_SITE: Not found!"
|
|
)
|
|
|
|
|
|
def test_notify_unknown():
|
|
n = QueryNotifyPrint(color=False)
|
|
result = MaigretCheckResult(
|
|
username="test",
|
|
status=MaigretCheckStatus.UNKNOWN,
|
|
site_name="TEST_SITE",
|
|
site_url_user="http://example.com/test",
|
|
)
|
|
result.error = CheckError('Type', 'Reason')
|
|
|
|
assert n.update(result) == "[?] TEST_SITE: Type error: Reason"
|
|
|
|
|
|
# `warning(message, symbol, advice=None)` was added so that the "Too many
|
|
# errors..." summary can render the count line bold and the advice in
|
|
# normal weight. The pieces must stay visually distinct because the advice
|
|
# is multi-line guidance, not part of the alarm.
|
|
|
|
|
|
def _capture_warning_string(monkeypatch, **warning_kwargs):
|
|
"""Patch builtins.print so we capture the *exact* string warning passed —
|
|
independent of colorama's terminal/TTY heuristics, which strip ANSI in
|
|
capsys mode. We care about what notify generated, not what colorama
|
|
decided to do with it on a captured pipe."""
|
|
captured = []
|
|
import builtins
|
|
monkeypatch.setattr(builtins, 'print', lambda *a, **kw: captured.append(a[0] if a else ''))
|
|
n = QueryNotifyPrint(color=warning_kwargs.pop('color', True))
|
|
n.warning(**warning_kwargs)
|
|
assert captured, "warning() did not call print()"
|
|
return captured[0]
|
|
|
|
|
|
def test_warning_no_advice_renders_as_single_yellow_bold_line(monkeypatch):
|
|
out = _capture_warning_string(
|
|
monkeypatch, color=True, message='something happened', symbol='!',
|
|
)
|
|
# Existing behaviour preserved for the no-advice path
|
|
assert out == Style.BRIGHT + Fore.YELLOW + '[!] something happened'
|
|
|
|
|
|
def test_warning_with_advice_keeps_header_bold_and_advice_normal(monkeypatch):
|
|
"""Advice must come after Style.NORMAL so terminals stop boldfacing it.
|
|
The Fore.YELLOW colour stays — the visual cue is weight, not colour."""
|
|
out = _capture_warning_string(
|
|
monkeypatch, color=True, message='count line', symbol='!', advice='do the thing',
|
|
)
|
|
# Header is bold...
|
|
assert out.startswith(Style.BRIGHT + Fore.YELLOW + '[!] count line')
|
|
# ...advice is preceded by Style.NORMAL so the boldface ends before it
|
|
assert Style.NORMAL + '. do the thing' in out
|
|
# ...and the whole line resets all SGR state at the end so no styling
|
|
# leaks into the next print
|
|
assert out.endswith(Style.RESET_ALL)
|
|
|
|
|
|
def test_warning_with_advice_no_color_uses_plain_dot_separator(monkeypatch):
|
|
"""In no-colour mode the visual distinction is impossible to render, so
|
|
a plain ". " separator between the count line and the advice is enough.
|
|
No ANSI codes must leak into the output."""
|
|
out = _capture_warning_string(
|
|
monkeypatch, color=False, message='count line', symbol='!', advice='do the thing',
|
|
)
|
|
assert out == '[!] count line. do the thing'
|
|
# Defence in depth: no escape codes survived the no-colour branch
|
|
assert '\x1b[' not in out
|