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

Saner tempdir removal

Browse Source
This commit is contained in:
Kovid Goyal 2022-02-21 18:05:36 +05:30
parent 081d6a3f16
commit a565443d4a
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
kitty_tests/shell_integration.py
View File

@ -4,9 +4,9 @@
import os import os
import shutil import shutil
import tempfile
import unittest import unittest
from contextlib import contextmanager from contextlib import contextmanager
from tempfile import TemporaryDirectory
from kitty.constants import terminfo_dir from kitty.constants import terminfo_dir
from kitty.fast_data_types import CURSOR_BEAM from kitty.fast_data_types import CURSOR_BEAM
@ -37,13 +37,17 @@ class ShellIntegration(BaseTest):
@contextmanager @contextmanager
def run_shell(self, shell='zsh', rc=''): def run_shell(self, shell='zsh', rc=''):
with TemporaryDirectory() as home_dir: home_dir = os.path.realpath(tempfile.mkdtemp())
try:
pty = self.create_pty(f'{shell} -il', cwd=home_dir, env=safe_env_for_running_shell(home_dir, rc)) pty = self.create_pty(f'{shell} -il', cwd=home_dir, env=safe_env_for_running_shell(home_dir, rc))
i = 10 i = 10
while i > 0 and not pty.screen_contents().strip(): while i > 0 and not pty.screen_contents().strip():
pty.process_input_from_child() pty.process_input_from_child()
i -= 1 i -= 1
yield pty yield pty
finally:
if os.path.exists(home_dir):
shutil.rmtree(home_dir)
@unittest.skipUnless(shutil.which('zsh'), 'zsh not installed') @unittest.skipUnless(shutil.which('zsh'), 'zsh not installed')
def test_zsh_integration(self): def test_zsh_integration(self):