This commit is contained in:
pagedown 2022-03-16 01:33:23 +08:00
parent 6638546247
commit 621453b068
No known key found for this signature in database
GPG Key ID: E921CF18AC8FF6EB

View File

@ -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)