From 621453b068823d506ea7b868e85edf832e805516 Mon Sep 17 00:00:00 2001 From: pagedown Date: Wed, 16 Mar 2022 01:33:23 +0800 Subject: [PATCH] ... --- kitty_tests/ssh.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/kitty_tests/ssh.py b/kitty_tests/ssh.py index c542b7f52..c6ca16135 100644 --- a/kitty_tests/ssh.py +++ b/kitty_tests/ssh.py @@ -10,8 +10,7 @@ from functools import lru_cache from kittens.ssh.config import load_config from kittens.ssh.main import ( - bootstrap_script, get_connection_data, get_remote_command, - wrap_bootstrap_script + bootstrap_script, get_connection_data, wrap_bootstrap_script ) from kittens.ssh.options.types import Options as SSHOptions from kittens.ssh.options.utils import DELETE_ENV_VAR @@ -80,7 +79,8 @@ print(' '.join(map(str, buf)))'''), lines=13, cols=77) self.ae(parse(conf, '2').env, {'a': 'c', 'b': 'b'}) def test_ssh_bootstrap_sh_cmd_limit(self): - rcmd, _, _ = get_remote_command([], SSHOptions({'interpreter': 'sh'})) + sh_script, _, _ = bootstrap_script(SSHOptions({'interpreter': 'sh'}), script_type='sh', remote_args=[], request_id='123-123') + rcmd = wrap_bootstrap_script(sh_script, 'sh') # dropbear has a 9000 bytes maximum command length limit self.assertLessEqual(sum(len(x) for x in rcmd), 9000)