abereziny/frostfs-testlib
1
0
Fork 0
forked from TrueCloudLab/frostfs-testlib
Code Issues Pull requests Projects Releases Packages Wiki Activity

Added sudo parameter for getting shell with elevated rights or not

Browse source
This commit is contained in:
d.anurin 2023-07-12 09:41:17 +03:00 committed by anurindm
parent 917dc6f6d8
commit 59b4157991
2 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/frostfs_testlib/hosting/docker_host.py
View file

@ -61,10 +61,10 @@ class ServiceAttributes(ParsedAttributes):
class DockerHost(Host):
"""Manages services hosted in Docker containers running on a local or remote machine."""
def get_shell(self) -> Shell:
def get_shell(self, sudo: bool = True) -> Shell:
host_attributes = HostAttributes.parse(self._config.attributes)
command_inspectors = []
if host_attributes.sudo_shell:
if sudo:
command_inspectors.append(SudoInspector())
if not host_attributes.ssh_login:

5
src/frostfs_testlib/hosting/interfaces.py
View file

@ -65,9 +65,12 @@ class Host(ABC):
return cli_config
@abstractmethod
def get_shell(self) -> Shell:
def get_shell(self, sudo: bool = True) -> Shell:
"""Returns shell to this host.
Args:
sudo: if True, run all commands in shell with elevated rights
Returns:
Shell that executes commands on this host.
"""