1j01/textual-paint
1
1
Fork 0
mirror of https://github.com/1j01/textual-paint.git synced 2024-12-19 04:41:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

Refactor tests that click a tool button

Browse Source
This commit is contained in:
Isaiah Odhner 2023-09-11 20:07:36 -04:00
parent 6e7a3445e0
commit 85d601b262
2 changed files with 19 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/pilot_helpers.py
View File

@ -5,6 +5,7 @@ Ideally this functionality would be part of Pilot.
from typing import Any
from textual.errors import NoWidget
from textual.events import MouseDown, MouseMove, MouseUp
from textual.geometry import Offset
from textual.pilot import Pilot, _get_mouse_message_arguments
@ -25,6 +26,18 @@ async def click_by_index(pilot: Pilot[Any], selector: str, index: int, shift: bo
await click_widget(pilot, widget, shift=shift, meta=meta, control=control)
async def click_by_attr(pilot: Pilot[Any], selector: str, attr: str, value: Any, shift: bool = False, meta: bool = False, control: bool = False) -> None:
"""Click on widget, query disambiguated by an attribute"""
# await pilot.pause(0.5)
widgets = pilot.app.query(selector)
for widget in widgets:
if getattr(widget, attr) == value:
break
else:
raise NoWidget(f"Could not find widget with {attr}={value}")
await click_widget(pilot, widget)
async def drag(pilot: Pilot[Any], selector: str, offsets: list[Offset], shift: bool = False, meta: bool = False, control: bool = False) -> None:
"""Drag across the given points."""
# await pilot.pause(0.5)

21
tests/test_snapshots.py
View File

@ -6,7 +6,7 @@ from textual.geometry import Offset
from textual.pilot import Pilot
from textual.widgets import Input
from tests.pilot_helpers import click_by_index, click_widget, drag
from tests.pilot_helpers import click_by_attr, click_by_index, drag
if TYPE_CHECKING:
# When tests are run, paint.py is re-evaluated,
@ -160,30 +160,21 @@ def test_paint_about_paint_dialog(snap_compare: SnapCompareType, each_theme: Non
@pytest.mark.skip(reason="@FIXME: polygon gets closed prematurely")
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")
await click_widget(pilot, polygon_tool_button)
await click_by_attr(pilot, "ToolsBox Button", "tooltip", "Polygon")
await pilot.pause(1.0) # for good luck
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 click_widget(pilot, color_buttons[16]) # red
await click_by_index(pilot, "ColorsBox Button", 16) # red
await pilot.pause(1.0) # for good luck
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 click_widget(pilot, color_buttons[17]) # yellow
await click_by_index(pilot, "ColorsBox Button", 17) # yellow
await pilot.pause(1.0) # for good luck
await pilot.click("Canvas", offset=Offset(10, 13))
await pilot.click("Canvas", offset=Offset(15, 13))
@ -193,7 +184,7 @@ def test_paint_polygon_tool(snap_compare: SnapCompareType):
def test_text_tool_wrapping(snap_compare: SnapCompareType):
async def automate_app(pilot: Pilot[None]):
await click_by_index(pilot, '#tools_box Button', 9)
await click_by_attr(pilot, "ToolsBox Button", "tooltip", "Text")
await drag(pilot, '#canvas', [Offset(x=5, y=8), Offset(x=24, y=16)])
for key in ('T', 'e', 'x', 't', 'space', 'T', 'o', 'o', 'l', 'space', 'T', 'e', 's', 't', 'space', 'left_parenthesis', 'T', 'T', 'T', 'right_parenthesis', 'n', 'e', 'w', 'space', 'l', 'i', 'n', 'e', 'space', 's', 't', 'a', 'r', 't', 's', 'space', 'h', 'e', 'r', 'e', 'a', 'n', 'd', 'space', 'h', 'e', 'r', 'e', 'space', 'a', 'u', 't', 'o', 'm', 'a', 't', 'i', 'c', 'a', 'l', 'hyphen', 'l', 'y'):
await pilot.press(key)
@ -202,7 +193,7 @@ def test_text_tool_wrapping(snap_compare: SnapCompareType):
def test_text_tool_cursor_keys_and_color(snap_compare: SnapCompareType):
async def automate_app(pilot: Pilot[None]):
await click_by_index(pilot, '#tools_box Button', 9)
await click_by_attr(pilot, "ToolsBox Button", "tooltip", "Text")
await drag(pilot, '#canvas', [Offset(x=8, y=5), Offset(x=21, y=10)])
for key in ('s', 'end', 'pagedown', '1', 'home', '2', 'pageup', '3', 'end', '4', 'pageup', 'home', 'right', 'right', 'c', 'r', 'e', 't', 'backspace', 'backspace', 'backspace', 'backspace', 'v', '3', 'n'):
await pilot.press(key)