mirror of
https://github.com/1j01/textual-paint.git
synced 2024-12-22 14:21:33 +03:00
198 lines
8.6 KiB
Python
198 lines
8.6 KiB
Python
from pathlib import Path, PurePath
|
|
from typing import TYPE_CHECKING, Awaitable, Callable, Iterable, Protocol
|
|
|
|
import pytest
|
|
from textual.geometry import Offset
|
|
from textual.pilot import Pilot
|
|
from textual.widget import Widget
|
|
from textual.widgets import Input
|
|
|
|
if TYPE_CHECKING:
|
|
# When tests are run, paint.py is re-evaluated,
|
|
# leading to a different class of the same name at runtime.
|
|
from textual_paint.paint import PaintApp
|
|
|
|
|
|
class SnapCompareType(Protocol):
|
|
"""Type of the function returned by the snap_compare fixture."""
|
|
def __call__(
|
|
self,
|
|
app_path: str | PurePath,
|
|
press: Iterable[str] = (),
|
|
terminal_size: tuple[int, int] = (80, 24),
|
|
run_before: Callable[[Pilot], Awaitable[None] | None] | None = None, # type: ignore
|
|
) -> bool:
|
|
...
|
|
|
|
# These paths are treated as relative to this file.
|
|
APPS_DIR = Path("../src/textual_paint")
|
|
PAINT = APPS_DIR / "paint.py"
|
|
GALLERY = APPS_DIR / "gallery.py"
|
|
|
|
LARGER = (81, 38)
|
|
"""Large enough to show the Textual Paint app's main UI and most dialogs comfortably."""
|
|
LARGEST = (107, 42)
|
|
"""Large enough to show the Edit Colors dialog, which is a bit oversized."""
|
|
|
|
# Prevent flaky tests due to timing issues.
|
|
Input.cursor_blink = False # type: ignore
|
|
# paint.DOUBLE_CLICK_TIME = 20.0 # seconds; ridiculously high; probably ineffective since paint.py is re-evaluated for each test
|
|
|
|
@pytest.fixture(params=[
|
|
{"theme": "light", "ascii_only": False},
|
|
{"theme": "dark", "ascii_only": False},
|
|
{"theme": "light", "ascii_only": True},
|
|
{"theme": "dark", "ascii_only": True},
|
|
], ids=lambda param: f"{param['theme']}_{'ascii' if param['ascii_only'] else 'unicode'}")
|
|
def each_theme(request: pytest.FixtureRequest):
|
|
"""Fixture to test each combination of UI styles."""
|
|
theme = request.param.get("theme")
|
|
ascii_only = request.param.get("ascii_only")
|
|
# os.environ["PYTEST_TEXTUAL_PAINT_ARGS"] = f"--theme {theme}" + (" --ascii-only" if ascii_only else "")
|
|
from textual_paint.args import args
|
|
args.theme = theme
|
|
args.ascii_only = ascii_only
|
|
yield
|
|
# del os.environ["PYTEST_TEXTUAL_PAINT_ARGS"]
|
|
args.theme = "light"
|
|
args.ascii_only = False
|
|
|
|
|
|
def test_paint_app(snap_compare: SnapCompareType, each_theme: None):
|
|
assert snap_compare(PAINT, terminal_size=LARGER)
|
|
|
|
def test_paint_stretch_skew_dialog(snap_compare: SnapCompareType, each_theme: None):
|
|
assert snap_compare(PAINT, press=["ctrl+w"])
|
|
|
|
def test_paint_flip_rotate_dialog(snap_compare: SnapCompareType, each_theme: None):
|
|
assert snap_compare(PAINT, press=["ctrl+r"])
|
|
|
|
def test_paint_image_attributes_dialog(snap_compare: SnapCompareType, each_theme: None):
|
|
assert snap_compare(PAINT, press=["ctrl+e"])
|
|
|
|
def test_paint_open_dialog(snap_compare: SnapCompareType, each_theme: None):
|
|
assert snap_compare(PAINT, press=["ctrl+o"], terminal_size=LARGER)
|
|
|
|
def test_paint_save_dialog(snap_compare: SnapCompareType, each_theme: None):
|
|
assert snap_compare(PAINT, press=["ctrl+s"], terminal_size=LARGER)
|
|
|
|
def test_paint_help_dialog(snap_compare: SnapCompareType, each_theme: None):
|
|
assert snap_compare(PAINT, press=["f1"], terminal_size=LARGER)
|
|
|
|
def test_paint_view_bitmap(snap_compare: SnapCompareType):
|
|
assert snap_compare(PAINT, press=["ctrl+f"])
|
|
|
|
def test_paint_invert_and_exit(snap_compare: SnapCompareType, each_theme: None):
|
|
assert snap_compare(PAINT, press=["ctrl+i", "ctrl+q"])
|
|
|
|
def test_swap_selected_colors(snap_compare: SnapCompareType):
|
|
async def swap_selected_colors(pilot: Pilot[None]):
|
|
await pilot.click("CharInput", control=True)
|
|
|
|
assert snap_compare(PAINT, run_before=swap_selected_colors)
|
|
|
|
def test_paint_character_picker_dialog(snap_compare: SnapCompareType, each_theme: None):
|
|
async def open_character_picker(pilot: Pilot[None]):
|
|
# app.dark = True caused it to fail to open the dialog in the dark theme,
|
|
# due to `self.call_later(self.refresh_css)` in `watch_dark` in `App`
|
|
# (verified by replacing `app.dark = args.theme == "dark"` with `app.call_later(app.refresh_css)`)
|
|
# Adding a delay works around this.
|
|
await pilot.pause(1.0)
|
|
await pilot.click("CharInput")
|
|
await pilot.click("CharInput")
|
|
assert pilot.app.query_one("CharacterSelectorDialogWindow")
|
|
|
|
assert snap_compare(PAINT, run_before=open_character_picker, terminal_size=LARGER)
|
|
|
|
def test_paint_edit_colors_dialog(snap_compare: SnapCompareType, each_theme: None):
|
|
async def open_edit_colors(pilot: Pilot[None]):
|
|
await pilot.pause(1.0) # see comment in test_paint_character_picker_dialog
|
|
pilot.app.query("ColorsBox Button")[0].id = "a_color_button"
|
|
await pilot.click("#a_color_button")
|
|
await pilot.click("#a_color_button")
|
|
assert pilot.app.query_one("EditColorsDialogWindow")
|
|
|
|
assert snap_compare(PAINT, run_before=open_edit_colors, terminal_size=LARGEST)
|
|
|
|
def test_paint_expand_canvas_dialog(snap_compare: SnapCompareType, each_theme: None):
|
|
async def paste_large_content(pilot: Pilot[None]):
|
|
if TYPE_CHECKING:
|
|
# Will be a different class at runtime, per test, due to re-evaluating the module.
|
|
assert isinstance(pilot.app, PaintApp)
|
|
pilot.app.paste("a" * 1000)
|
|
|
|
assert snap_compare(PAINT, run_before=paste_large_content, terminal_size=LARGER)
|
|
|
|
def test_paint_error_dialog(snap_compare: SnapCompareType, each_theme: None):
|
|
async def show_error(pilot: Pilot[None]):
|
|
if TYPE_CHECKING:
|
|
# Will be a different class at runtime, per test, due to re-evaluating the module.
|
|
assert isinstance(pilot.app, PaintApp)
|
|
pilot.app.message_box("EMIT", "Error Message Itself Test", "ok", error=Exception("Error Message Itself Test"))
|
|
assert pilot.app.query_one("MessageBox")
|
|
await pilot.pause(1.0)
|
|
assert pilot.app.query_one("MessageBox .details_button")
|
|
# pilot.app.query_one("MessageBox .details_button", Button).press()
|
|
await pilot.click("MessageBox .details_button")
|
|
await pilot.pause(0.5) # avoid pressed state
|
|
|
|
assert snap_compare(PAINT, run_before=show_error)
|
|
|
|
def test_paint_custom_zoom_dialog(snap_compare: SnapCompareType, each_theme: None):
|
|
async def show_custom_zoom(pilot: Pilot[None]):
|
|
if TYPE_CHECKING:
|
|
# Will be a different class at runtime, per test, due to re-evaluating the module.
|
|
assert isinstance(pilot.app, PaintApp)
|
|
pilot.app.action_custom_zoom()
|
|
|
|
assert snap_compare(PAINT, run_before=show_custom_zoom)
|
|
|
|
def test_paint_about_paint_dialog(snap_compare: SnapCompareType, each_theme: None):
|
|
async def show_about_paint(pilot: Pilot[None]):
|
|
if TYPE_CHECKING:
|
|
# Will be a different class at runtime, per test, due to re-evaluating the module.
|
|
assert isinstance(pilot.app, PaintApp)
|
|
pilot.app.action_about_paint()
|
|
|
|
assert snap_compare(PAINT, run_before=show_about_paint)
|
|
|
|
def test_paint_polygon_tool(snap_compare: SnapCompareType):
|
|
async def draw_polygon(pilot: Pilot[None]):
|
|
tool_buttons = pilot.app.query("ToolsBox Button")
|
|
color_buttons = pilot.app.query("ColorsBox Button")
|
|
for button in tool_buttons:
|
|
if button.tooltip == "Polygon":
|
|
polygon_tool_button = button
|
|
break
|
|
else:
|
|
raise Exception("Couldn't find Polygon tool button")
|
|
|
|
async def clickity(button: Widget) -> None:
|
|
button.add_class("to_click")
|
|
await pilot.click(".to_click")
|
|
button.remove_class("to_click")
|
|
await pilot.pause(1.0) # for good luck
|
|
|
|
await clickity(polygon_tool_button)
|
|
await pilot.click("Canvas", offset=Offset(2, 2))
|
|
await pilot.click("Canvas", offset=Offset(2, 20))
|
|
await pilot.click("Canvas", offset=Offset(30, 20))
|
|
await pilot.click("Canvas", offset=Offset(30, 2))
|
|
await pilot.click("Canvas", offset=Offset(2, 2)) # end by clicking on the start point
|
|
# await clickity(color_buttons[16]) # red
|
|
# await pilot.click("Canvas", offset=Offset(10, 5))
|
|
# await pilot.click("Canvas", offset=Offset(10, 9))
|
|
# await pilot.click("Canvas", offset=Offset(10, 9))
|
|
# await pilot.click("Canvas", offset=Offset(1, 5))
|
|
# await pilot.click("Canvas", offset=Offset(1, 5)) # end by double clicking
|
|
# await clickity(color_buttons[17]) # yellow
|
|
# await pilot.click("Canvas", offset=Offset(10, 13))
|
|
# await pilot.click("Canvas", offset=Offset(15, 13))
|
|
# await pilot.click("Canvas", offset=Offset(12, 16)) # don't end, leave as polyline
|
|
|
|
assert snap_compare(PAINT, run_before=draw_polygon, terminal_size=LARGER)
|
|
|
|
def test_gallery_app(snap_compare: SnapCompareType):
|
|
assert snap_compare(GALLERY)
|
|
|