mirror of
https://github.com/kovidgoyal/kitty.git
synced 2024-09-11 14:45:31 +03:00
Follow PEP 0394 and use /usr/bin/env python so that the python in the users venv is respected. Not that the kitty python files are meant to be executed standalone anyway, but, whatever. Fixes #6810
280 lines
7.6 KiB
Python
280 lines
7.6 KiB
Python
#!/usr/bin/env python
|
|
# License: GPL v3 Copyright: 2016, Kovid Goyal <kovid at kovidgoyal.net>
|
|
|
|
from functools import partial
|
|
|
|
from kitty.fast_data_types import (
|
|
GLFW_MOD_ALT,
|
|
GLFW_MOD_CONTROL,
|
|
GLFW_MOUSE_BUTTON_LEFT,
|
|
GLFW_MOUSE_BUTTON_RIGHT,
|
|
create_mock_window,
|
|
mock_mouse_selection,
|
|
send_mock_mouse_event_to_window,
|
|
)
|
|
|
|
from . import BaseTest
|
|
|
|
|
|
def send_mouse_event(
|
|
window,
|
|
button=-1,
|
|
modifiers=0,
|
|
is_release=False,
|
|
x=0.0,
|
|
y=0,
|
|
clear_click_queue=False,
|
|
):
|
|
ix = int(x)
|
|
in_left_half_of_cell = x - ix < 0.5
|
|
send_mock_mouse_event_to_window(
|
|
window, button, modifiers, is_release, ix, y, clear_click_queue, in_left_half_of_cell
|
|
)
|
|
|
|
|
|
class TestMouse(BaseTest):
|
|
|
|
def test_mouse_selection(self):
|
|
s = self.create_screen(
|
|
options=dict(
|
|
rectangle_select_modifiers=GLFW_MOD_ALT | GLFW_MOD_CONTROL
|
|
)
|
|
)
|
|
w = create_mock_window(s)
|
|
ev = partial(send_mouse_event, w)
|
|
|
|
def mouse_selection(code: int) -> None:
|
|
mock_mouse_selection(w, s.callbacks.current_mouse_button, code)
|
|
|
|
s.callbacks.mouse_selection = mouse_selection
|
|
|
|
def sel():
|
|
return ''.join(s.text_for_selection())
|
|
|
|
def init():
|
|
s.reset()
|
|
s.draw('pqrst')
|
|
s.draw('uvwxy')
|
|
s.draw('ABCDE')
|
|
s.draw('FGHIJ')
|
|
s.draw('KLMNO')
|
|
s.draw('12345')
|
|
s.draw('67890')
|
|
s.draw('abcde')
|
|
s.draw('fghij')
|
|
s.draw('klmno')
|
|
|
|
def press(x=0, y=0, modifiers=0, button=GLFW_MOUSE_BUTTON_LEFT):
|
|
ev(button, x=x, y=y, modifiers=modifiers)
|
|
|
|
def release(x=0, y=0, button=GLFW_MOUSE_BUTTON_LEFT):
|
|
ev(
|
|
button,
|
|
x=x,
|
|
y=y,
|
|
is_release=True,
|
|
clear_click_queue=True
|
|
)
|
|
|
|
def move(x=0, y=0, button=-1, q=None):
|
|
ev(x=x, y=y, button=button)
|
|
if q is not None:
|
|
s = sel()
|
|
self.ae(s, q, f'{s!r} != {q!r} after movement to x={x} y={y}')
|
|
|
|
def multi_click(x=0, y=0, count=2):
|
|
clear_click_queue = True
|
|
while count > 0:
|
|
count -= 1
|
|
ev(GLFW_MOUSE_BUTTON_LEFT, x=x, y=y, clear_click_queue=clear_click_queue)
|
|
clear_click_queue = False
|
|
|
|
def scroll(x=0, y=0, up=True):
|
|
move(x=x, y=y, button=-2 if up else -3)
|
|
|
|
# Single line click, move, release test
|
|
init()
|
|
press()
|
|
move(x=3.6, q='1234')
|
|
release(x=3.6)
|
|
self.ae(sel(), '1234')
|
|
press(x=4), release(x=0.6)
|
|
self.ae(sel(), '234')
|
|
|
|
# multi line movement
|
|
init()
|
|
press(x=2, y=2)
|
|
move(x=2, y=1, q='890ab')
|
|
move(x=2.6, y=1, q='90ab')
|
|
move(y=1, q='67890ab')
|
|
move(x=4, y=1, q='0ab')
|
|
move(x=4.6, y=1, q='ab')
|
|
move(q='1234567890ab')
|
|
move(x=2, y=3, q='cdefg')
|
|
move(y=3, q='cde')
|
|
move(x=0.6, y=3, q='cdef')
|
|
move(x=2.6, y=3, q='cdefgh')
|
|
move(x=4.6, y=3, q='cdefghij')
|
|
|
|
# Single cell select
|
|
init()
|
|
press(), release(1)
|
|
self.ae(sel(), '1')
|
|
press(3), release(2)
|
|
self.ae(sel(), '3')
|
|
|
|
# Multi-line click release
|
|
init()
|
|
press(1, 1), release(3.6, 2)
|
|
self.ae(sel(), '7890abcd')
|
|
press(1.6, 1), release(3, 2)
|
|
self.ae(sel(), '890abc')
|
|
press(3.6, 4), release(2, 2)
|
|
self.ae(sel(), 'cdefghijklmn')
|
|
press(3, 4), release(2.6, 2)
|
|
self.ae(sel(), 'defghijklm')
|
|
|
|
# Word select with drag
|
|
s.reset()
|
|
s.draw('ab cd')
|
|
s.draw(' f gh')
|
|
s.draw(' stuv')
|
|
s.draw('X Y')
|
|
multi_click(x=1.4)
|
|
self.ae(sel(), 'ab')
|
|
move(2.6)
|
|
self.ae(sel(), 'ab ')
|
|
move(3.6)
|
|
self.ae(sel(), 'ab cd')
|
|
move(2.6)
|
|
self.ae(sel(), 'ab ')
|
|
release(3.6, 1)
|
|
self.ae(sel(), 'ab cd f gh')
|
|
multi_click(x=1, y=2)
|
|
self.ae(sel(), 'stuvX')
|
|
release()
|
|
multi_click(x=3.6)
|
|
self.ae(sel(), 'cd')
|
|
move(0.2)
|
|
release()
|
|
self.ae(sel(), 'ab cd')
|
|
multi_click(x=4.4)
|
|
self.ae(sel(), 'cd')
|
|
move(x=4.4, y=1)
|
|
self.ae(sel(), 'cd f gh')
|
|
move(x=4.4, y=0)
|
|
self.ae(sel(), 'cd')
|
|
release()
|
|
multi_click(x=4.4, y=1)
|
|
self.ae(sel(), 'gh')
|
|
move(x=4.4, y=0)
|
|
self.ae(sel(), 'cd f gh')
|
|
move(x=4.4, y=1)
|
|
self.ae(sel(), 'gh')
|
|
release()
|
|
multi_click(x=4.4)
|
|
self.ae(sel(), 'cd')
|
|
move()
|
|
self.ae(sel(), 'ab cd')
|
|
move(x=1, y=1)
|
|
self.ae(sel(), 'ab cd f')
|
|
move()
|
|
self.ae(sel(), 'ab cd')
|
|
release()
|
|
multi_click(x=1.4)
|
|
self.ae(sel(), 'ab')
|
|
move(x=4.4)
|
|
self.ae(sel(), 'ab cd')
|
|
move(x=4.4, y=1)
|
|
self.ae(sel(), 'ab cd f gh')
|
|
move(x=4.4)
|
|
self.ae(sel(), 'ab cd')
|
|
|
|
# Line select with drag
|
|
s.reset()
|
|
s.draw('1 2 3')
|
|
s.linefeed(), s.carriage_return()
|
|
s.draw('4 5 6')
|
|
s.linefeed(), s.carriage_return()
|
|
s.draw('7 8 9X')
|
|
multi_click(x=1, count=3)
|
|
self.ae(sel(), str(s.line(0)))
|
|
move(y=1)
|
|
self.ae(sel(), '1 2 3\n4 5 6')
|
|
move(y=2)
|
|
self.ae(sel(), '1 2 3\n4 5 6\n7 8 9X')
|
|
move(y=1)
|
|
self.ae(sel(), '1 2 3\n4 5 6')
|
|
move()
|
|
self.ae(sel(), str(s.line(0)))
|
|
release()
|
|
multi_click(y=1, count=3)
|
|
self.ae(sel(), '4 5 6')
|
|
move(y=0)
|
|
self.ae(sel(), '1 2 3\n4 5 6')
|
|
move(y=1)
|
|
self.ae(sel(), '4 5 6')
|
|
move(y=2)
|
|
self.ae(sel(), '4 5 6\n7 8 9X')
|
|
release()
|
|
s.reset()
|
|
s.draw(' 123')
|
|
s.linefeed(), s.carriage_return()
|
|
s.draw(' 456')
|
|
s.linefeed(), s.carriage_return()
|
|
multi_click(x=1, count=3)
|
|
self.ae(sel(), '123')
|
|
move(x=2, y=1)
|
|
self.ae(sel(), '123\n 456')
|
|
release()
|
|
press(x=2, y=1, button=GLFW_MOUSE_BUTTON_RIGHT)
|
|
release(x=2, y=1, button=GLFW_MOUSE_BUTTON_RIGHT)
|
|
self.ae(sel(), '123\n 456')
|
|
press(button=GLFW_MOUSE_BUTTON_RIGHT)
|
|
self.ae(sel(), ' 123\n 456')
|
|
release(button=GLFW_MOUSE_BUTTON_RIGHT)
|
|
|
|
# Rectangle select
|
|
init()
|
|
press(x=1, y=1, modifiers=GLFW_MOD_ALT | GLFW_MOD_CONTROL)
|
|
move(x=3.6, y=3)
|
|
self.ae(sel(), '789bcdghi')
|
|
release(x=3, y=3)
|
|
self.ae(sel(), '78bcgh')
|
|
press(x=3.6, y=1, modifiers=GLFW_MOD_ALT | GLFW_MOD_CONTROL)
|
|
self.ae(sel(), '')
|
|
move(x=1, y=3)
|
|
self.ae(sel(), '789bcdghi')
|
|
release(x=1.6)
|
|
self.ae(sel(), '3489')
|
|
|
|
# scrolling
|
|
init()
|
|
press(x=1.6)
|
|
scroll(x=1)
|
|
self.ae(sel(), 'LMNO12')
|
|
scroll(x=1)
|
|
self.ae(sel(), 'GHIJKLMNO12')
|
|
scroll(x=1, up=False)
|
|
self.ae(sel(), 'LMNO12')
|
|
scroll(x=2.6, up=False)
|
|
self.ae(sel(), '3')
|
|
release()
|
|
|
|
# extending selections
|
|
init()
|
|
press()
|
|
move(x=3.6, q='1234')
|
|
release(x=3.6)
|
|
self.ae(sel(), '1234')
|
|
press(x=1, y=1, button=GLFW_MOUSE_BUTTON_RIGHT)
|
|
self.ae(sel(), '123456')
|
|
move(x=2, y=1, q='1234567')
|
|
release(x=3, y=1, button=GLFW_MOUSE_BUTTON_RIGHT)
|
|
self.ae(sel(), '12345678')
|
|
init()
|
|
press(y=2)
|
|
move(x=3.6, y=2, q='abcd')
|
|
press(x=3, y=0, button=GLFW_MOUSE_BUTTON_RIGHT)
|
|
self.ae(sel(), '4567890abcd')
|