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

Allow Ctrl+Z also while replaying in test recorder

Browse Source
This commit is contained in:
Isaiah Odhner 2023-09-09 23:28:56 -04:00
parent c395f4e8f3
commit a10b9eac11
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/pilot_recorder.py
View File

@ -63,6 +63,8 @@ def get_selector(target: DOMNode) -> tuple[str, int|None]:
return selector, None return selector, None
original_on_event = PaintApp.on_event
class PilotRecorder(): class PilotRecorder():
def __init__(self) -> None: def __init__(self) -> None:
self.app: PaintApp | None = None self.app: PaintApp | None = None
@ -71,7 +73,6 @@ class PilotRecorder():
self.output_file = unique_file("tests/test_paint_something.py") self.output_file = unique_file("tests/test_paint_something.py")
self.next_after_exit: Callable[[], None] | None = None self.next_after_exit: Callable[[], None] | None = None
original_on_event = PaintApp.on_event
recorder = self recorder = self
async def on_event(self: PaintApp, event: Event) -> None: async def on_event(self: PaintApp, event: Event) -> None:
# Record before the event is handled, so a clicked button that closes a dialog, # Record before the event is handled, so a clicked button that closes a dialog,
@ -82,12 +83,12 @@ class PilotRecorder():
def record_event(self, event: Event) -> None: def record_event(self, event: Event) -> None:
assert self.app is not None, "app should be set if we're recording an event from it" assert self.app is not None, "app should be set if we're recording an event from it"
if self.replaying:
return
# Handling any event means including it in the undo stack right now. # Handling any event means including it in the undo stack right now.
# Don't want to undo a single mouse-move, especially when it doesn't do anything yet. # Don't want to undo a single mouse-move, especially when it doesn't do anything yet.
# if isinstance(event, (MouseDown, MouseMove, MouseUp)): # if isinstance(event, (MouseDown, MouseMove, MouseUp)):
if isinstance(event, MouseDown): if isinstance(event, MouseDown):
if self.replaying:
return
try: try:
widget, _ = self.app.get_widget_at(*event.screen_offset) widget, _ = self.app.get_widget_at(*event.screen_offset)
except NoWidget: except NoWidget:
@ -104,6 +105,8 @@ class PilotRecorder():
self.save_replay() self.save_replay()
self.app.exit(None, Text("Saved test recording to " + self.output_file)) self.app.exit(None, Text("Saved test recording to " + self.output_file))
else: else:
if self.replaying:
return
self.steps.append((event, Offset(), "", None)) self.steps.append((event, Offset(), "", None))
self.steps_changed() self.steps_changed()