Flush stdin and exit stats screen on Enter

This commit is contained in:
thePR0M3TH3AN
2025-08-05 13:13:31 -04:00
parent d21ad78a02
commit 89cbef1aa4
2 changed files with 44 additions and 0 deletions

View File

@@ -88,3 +88,21 @@ def test_stats_display_resets_after_exit(monkeypatch, capsys):
main._display_live_stats(pm)
out = capsys.readouterr().out
assert out.count("stats") == 2
def test_stats_screen_breaks_on_enter(monkeypatch):
calls = {"display": 0}
def display():
calls["display"] += 1
print("stats")
pm = _make_pm()
pm.display_stats = display
monkeypatch.setattr(main, "get_notification_text", lambda *_: "")
monkeypatch.setattr(main, "timed_input", lambda *_args, **_kwargs: "")
main._display_live_stats(pm, interval=0.01)
assert calls["display"] == 1