KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Dont rely on having source of test modules available

Browse Source
This commit is contained in:
Kovid Goyal 2022-07-05 20:42:55 +05:30
parent d4bc906c5c
commit af1c48e0f7
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 22 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
kitty_tests/prewarm.py
View File

@ -4,7 +4,6 @@
import json import json
import os import os
import re
import select import select
import signal import signal
import subprocess import subprocess
@ -12,7 +11,7 @@ import tempfile
import time import time
from contextlib import suppress from contextlib import suppress
from kitty.constants import kitty_exe, read_kitty_resource from kitty.constants import kitty_exe
from kitty.fast_data_types import ( from kitty.fast_data_types import (
CLD_EXITED, CLD_KILLED, get_options, has_sigqueue, install_signal_handlers, CLD_EXITED, CLD_KILLED, get_options, has_sigqueue, install_signal_handlers,
read_signals, remove_signal_handlers, sigqueue read_signals, remove_signal_handlers, sigqueue
@ -21,29 +20,6 @@ from kitty.fast_data_types import (
from . import BaseTest from . import BaseTest
def socket_child_main(exit_code=0):
import json
import os
import sys
from kitty.fast_data_types import get_options
from kitty.utils import read_screen_size
output = {
'test_env': os.environ.get('TEST_ENV_PASS', ''),
'cwd': os.getcwd(),
'font_family': get_options().font_family,
'cols': read_screen_size(fd=sys.stderr.fileno()).cols,
'stdin_data': sys.stdin.read(),
'done': 'hello',
}
print(json.dumps(output, indent=2), file=sys.stderr, flush=True)
print('testing stdout', end='')
raise SystemExit(exit_code)
# END_socket_child_main
class Prewarm(BaseTest): class Prewarm(BaseTest):
maxDiff = None maxDiff = None
@ -51,11 +27,28 @@ class Prewarm(BaseTest):
def test_socket_prewarming(self): def test_socket_prewarming(self):
from kitty.prewarm import fork_prewarm_process from kitty.prewarm import fork_prewarm_process
exit_code = 17 exit_code = 17
src = re.search( src = '''\
r'^(def socket_child_main.+?)^# END_socket_child_main', read_kitty_resource('prewarm.py', 'kitty_tests').decode(), def socket_child_main(exit_code=0):
flags=re.M | re.DOTALL).group(1) + '\n\n' import json
import os
import sys
cwd = tempfile.gettempdir() from kitty.fast_data_types import get_options
from kitty.utils import read_screen_size
output = {
'test_env': os.environ.get('TEST_ENV_PASS', ''),
'cwd': os.path.realpath(os.getcwd()),
'font_family': get_options().font_family,
'cols': read_screen_size(fd=sys.stderr.fileno()).cols,
'stdin_data': sys.stdin.read(),
'done': 'hello',
}
print(json.dumps(output, indent=2), file=sys.stderr, flush=True)
print('testing stdout', end='')
raise SystemExit(exit_code)
''' + '\n\n'
cwd = os.path.realpath(tempfile.gettempdir())
opts = self.set_options() opts = self.set_options()
opts.config_overrides = 'font_family prewarm', opts.config_overrides = 'font_family prewarm',
p = fork_prewarm_process(opts, use_exec=True) p = fork_prewarm_process(opts, use_exec=True)