forked from TrueCloudLab/frostfs-testlib
61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
from abc import ABC, abstractmethod
|
|
from dataclasses import dataclass
|
|
from typing import Optional
|
|
|
|
|
|
@dataclass
|
|
class InteractiveInput:
|
|
"""
|
|
Interactive input for a shell command.
|
|
|
|
:attr str prompt_pattern: regular expression that defines expected prompt from the command.
|
|
:attr str input: user input that should be supplied to the command in response to the prompt.
|
|
"""
|
|
|
|
prompt_pattern: str
|
|
input: str
|
|
|
|
|
|
@dataclass
|
|
class CommandOptions:
|
|
"""
|
|
Options that control command execution.
|
|
|
|
:attr list interactive_inputs: user inputs that should be interactively supplied to
|
|
the command during execution.
|
|
:attr int timeout: timeout for command execution (in seconds).
|
|
:attr bool check: controls whether to check return code of the command. Set to False to
|
|
ignore non-zero return codes.
|
|
"""
|
|
|
|
interactive_inputs: Optional[list[InteractiveInput]] = None
|
|
timeout: int = 30
|
|
check: bool = True
|
|
|
|
|
|
@dataclass
|
|
class CommandResult:
|
|
"""
|
|
Represents a result of a command executed via shell.
|
|
"""
|
|
|
|
stdout: str
|
|
stderr: str
|
|
return_code: int
|
|
|
|
|
|
class Shell(ABC):
|
|
"""
|
|
Interface of a command shell on some system (local or remote).
|
|
"""
|
|
|
|
@abstractmethod
|
|
def exec(self, command: str, options: Optional[CommandOptions] = None) -> CommandResult:
|
|
"""
|
|
Executes specified command on this shell. To execute interactive command, user inputs
|
|
should be specified in *options*.
|
|
|
|
:param str command: command to execute on the shell.
|
|
:param CommandOptions options: options that control command execution.
|
|
:return command result.
|
|
"""
|