forked from TrueCloudLab/frostfs-testcases
38 lines
985 B
Python
38 lines
985 B
Python
#!/usr/bin/python3.8
|
|
|
|
"""
|
|
Helper functions to use with `neofs-cli`, `neo-go`
|
|
and other CLIs.
|
|
"""
|
|
|
|
import subprocess
|
|
import pexpect
|
|
|
|
from robot.api import logger
|
|
|
|
ROBOT_AUTO_KEYWORDS = False
|
|
|
|
|
|
def _cmd_run(cmd, timeout=30):
|
|
"""
|
|
Runs given shell command <cmd>, in case of success returns its stdout,
|
|
in case of failure returns error message.
|
|
"""
|
|
try:
|
|
compl_proc = subprocess.run(cmd, check=True, universal_newlines=True,
|
|
stdout=subprocess.PIPE, stderr=subprocess.STDOUT, timeout=timeout,
|
|
shell=True)
|
|
output = compl_proc.stdout
|
|
logger.info(f"Output: {output}")
|
|
return output
|
|
except subprocess.CalledProcessError as exc:
|
|
raise RuntimeError(f"Error:\nreturn code: {exc.returncode} "
|
|
f"\nOutput: {exc.output}") from exc
|
|
|
|
def _run_with_passwd(cmd):
|
|
p = pexpect.spawn(cmd)
|
|
p.expect(".*")
|
|
p.sendline('\r')
|
|
p.wait()
|
|
cmd = p.read()
|
|
return cmd.decode()
|