mirrors/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty synced 2026-06-08 22:28:24 +02:00
Code Issues Packages Projects Releases Wiki Activity

Allow mocking usage of timers in the test suite

Browse Source
This commit is contained in:
Kovid Goyal
2021-09-01 13:09:42 +05:30
parent 40786427b0
commit 87f6793552
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
kitty/file_transmission.py
View File

@@ -351,9 +351,12 @@ class FileTransmission:
self.pending_receive_responses: Deque[Tuple[str, str]] = deque()
self.pending_timer: Optional[int] = None
def callback_after(self, callback: Callable[[Optional[int]], None], timeout: float) -> Optional[int]:
return add_timer(callback, timeout, False)
def start_pending_timer(self) -> None:
if self.pending_timer is None:
self.pending_timer = add_timer(self.try_pending, 0.2, False)
self.pending_timer = self.callback_after(self.try_pending, 0.2)
def try_pending(self, timer_id: Optional[int]) -> None:
self.pending_timer = None
@@ -536,3 +539,6 @@ class TestFileTransmission(FileTransmission):
def start_receive(self, aid: str) -> None:
self.handle_send_confirmation(aid, {'response': 'y' if self.allow else 'n'})
def callback_after(self, callback: Callable[[Optional[int]], None], timeout: float) -> Optional[int]:
callback(None)