mirrors/kitty
1
0
Fork 0
mirror of https://github.com/kovidgoyal/kitty synced 2026-07-26 10:12:17 +02:00
Code Issues Packages Projects Releases Wiki Activity

Make dnd kitten testing infra a bit more robust

Browse Source
This commit is contained in:
Kovid Goyal
2026-04-19 21:00:09 +05:30
parent 0ddbffdf79
commit 36e8094afe
2 changed files with 16 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
kittens/dnd/main.go
View File

@@ -61,12 +61,13 @@ func run_loop(opts *Options, drop_dests map[string]drop_dest, drag_sources map[s
return err return err
} }
send_test_response := func(payload string) { send_test_response := func(payload string) {
in_test_mode = true
lp.DebugPrintln(payload) lp.DebugPrintln(payload)
} }
render_screen := func() error { render_screen := func() error {
lp.StartAtomicUpdate() if !in_test_mode {
defer lp.EndAtomicUpdate() lp.StartAtomicUpdate()
defer lp.EndAtomicUpdate()
}
lp.ClearScreen() lp.ClearScreen()
if allow_drags { if allow_drags {
if drag_started { if drag_started {
@@ -157,6 +158,9 @@ func run_loop(opts *Options, drop_dests map[string]drop_dest, drag_sources map[s
switch string(cmd.Payload) { switch string(cmd.Payload) {
case "PING": case "PING":
send_test_response("PONG") send_test_response("PONG")
case "SETUP":
in_test_mode = true
lp.NoRoundtripToTerminalOnExit()
default: default:
send_test_response("UNKNOWN TEST COMMAND: " + string(cmd.Payload)) send_test_response("UNKNOWN TEST COMMAND: " + string(cmd.Payload))
} }

10
kitty_tests/dnd_kitten.py
View File

@@ -5,7 +5,13 @@ import tempfile
from base64 import standard_b64encode from base64 import standard_b64encode
from kitty.constants import kitten_exe from kitty.constants import kitten_exe
from kitty.fast_data_types import DND_CODE, dnd_set_test_write_func, dnd_test_cleanup_fake_window, dnd_test_create_fake_window, dnd_test_probe_state from kitty.fast_data_types import (
DND_CODE,
dnd_set_test_write_func,
dnd_test_cleanup_fake_window,
dnd_test_create_fake_window,
dnd_test_probe_state,
)
from . import PTY, BaseTest from . import PTY, BaseTest
from .dnd import WriteCapture from .dnd import WriteCapture
@@ -28,6 +34,7 @@ class TestDnDKitten(BaseTest):
self.capture = capture self.capture = capture
self.test_dir = self.enterContext(tempfile.TemporaryDirectory()) self.test_dir = self.enterContext(tempfile.TemporaryDirectory())
self.messages_from_kitten = '' self.messages_from_kitten = ''
self.set_options({'tab_bar_style': 'hidden'})
def send_dnd_command_to_kitten(self, payload=b'', as_base64=False, flush=False, **metadata): def send_dnd_command_to_kitten(self, payload=b'', as_base64=False, flush=False, **metadata):
header = f'\x1b]{DND_CODE};' header = f'\x1b]{DND_CODE};'
@@ -63,6 +70,7 @@ class TestDnDKitten(BaseTest):
self.assertEqual(remote_client, self.probe_state('drop_is_remote_client')) self.assertEqual(remote_client, self.probe_state('drop_is_remote_client'))
if self.probe_state('drag_can_offer'): if self.probe_state('drag_can_offer'):
self.assertEqual(remote_client, self.probe_state('drag_is_remote_client')) self.assertEqual(remote_client, self.probe_state('drag_is_remote_client'))
self.send_dnd_command_to_kitten('SETUP', t='T')
def append(self, text): def append(self, text):
self.messages_from_kitten += text self.messages_from_kitten += text