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

Dont wait for cancel response when user denies permission

Browse Source
This commit is contained in:
Kovid Goyal
2023-07-25 11:47:41 +05:30
parent ca485cc0f7
commit 2cae0ab695
3 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
kitty_tests/file_transmission.py
View File

@@ -179,6 +179,7 @@ class TransferPTY(PTY):
class TestFileTransmission(BaseTest):
def setUp(self):
self.direction_receive = False
super().setUp()
self.tdir = os.path.realpath(tempfile.mkdtemp())
self.responses = []
@@ -327,7 +328,7 @@ class TestFileTransmission(BaseTest):
@contextmanager
def run_kitten(self, cmd, home_dir='', allow=True):
cwd = os.path.realpath(tempfile.mkdtemp(suffix='-cwd', dir=self.tdir))
cmd = [kitten_exe(), 'transfer'] + cmd
cmd = [kitten_exe(), 'transfer'] + (['--direction=receive'] if self.direction_receive else []) + cmd
env = {'PWD': cwd}
if home_dir:
env['HOME'] = home_dir
@@ -342,6 +343,17 @@ class TestFileTransmission(BaseTest):
if os.path.exists(cwd):
shutil.rmtree(cwd)
def test_transfer_receive(self):
self.direction_receive = True
src = os.path.join(self.tdir, 'src')
self.src_data = os.urandom(11113)
with open(src, 'wb') as s:
s.write(self.src_data)
dest = os.path.join(self.tdir, 'dest')
with self.run_kitten([src, dest], allow=False) as pty:
pty.wait_till_child_exits(require_exit_code=1)
self.assertFalse(os.path.exists(dest))
def test_transfer_send(self):
src = os.path.join(self.tdir, 'src')
self.src_data = os.urandom(9137)