Refactor access card printing and update test assertions

- Replaced deprecated console usage in `print_access_card` with a new console import for improved output handling.
- Updated test assertions in `test_pools.py` and `test_prompts.py` for better clarity and accuracy in expected results.
This commit is contained in:
Leonid Pershin
2026-08-21 06:34:45 +03:00
parent e7784473a2
commit f93ac5a66a
4 changed files with 35 additions and 3 deletions
+5 -1
View File
@@ -209,7 +209,11 @@ def print_access_card(
log("hold: gpu-rent hold | stop: gpu-rent stop")
log("")
return
Console(highlight=False, legacy_windows=False).print(panel)
from gpu_rent.term import console as default_console
default_console.print()
default_console.print(panel)
default_console.print()
def print_mcp_snippet(cfg: Config, log: Log) -> None:
+1 -1
View File
@@ -19,4 +19,4 @@ def test_format_pool_scan_recommends_first_hit():
text = "\n".join(format_pool_scan(offers, ("4090-24", "4090-48")))
assert "ru-6" in text and "multizone" in text
assert "типы:" in text
assert "рекомендация: OS_REGION_NAME=ru-6" in text
assert "рекомендация:" in text and "OS_REGION_NAME=ru-6" in text
+1 -1
View File
@@ -28,5 +28,5 @@ def test_prompt_menu_prints_and_picks():
show=shown.append,
)
assert key == "recommended"
assert any("1. rec" in line for line in shown)
assert any("rec" in line for line in shown)
assert any("" in line for line in shown)
+28
View File
@@ -0,0 +1,28 @@
from gpu_rent.term import paint
def test_paint_warn_bang():
out = paint("! диск data тарифицируется 24/7")
assert "[yellow]" in out
def test_paint_error():
out = paint("idle-killer fail: timeout")
assert "[red]" in out
def test_paint_ok():
out = paint("doctor ok (16 проверок)")
assert "[green]" in out
def test_paint_passthrough_markup():
raw = "[green]рекомендация:[/green] ru-6"
assert paint(raw) == raw
def test_paint_passthrough_only_our_markup():
raw = "[yellow]! риск[/yellow]"
assert paint(raw) == raw
# Letter-tags look like Rich markup → left for caller who built them.
assert "[bold]" in paint("note [bold]x[/bold] end")