def test_execute_shell_handles_errors(run_mock):
run_mock.side_effect = run_error_func
try:
utils.execute_shell(["expecting error"])
assert False, "expected CalledProcessError"
except CalledProcessError:
pass
def create_watchdog_file():
if not os.path.exists('workdir'):
os.mkdir('workdir')
command = "mkdir -p ~/.ssh && \
ssh-keyscan -H 192.168.10.1 >> ~/.ssh/known_hosts && \
sshpass -e ssh -q [email protected] /opt/vyatta/bin/vyatta-op-cmd-wrapper \
show load-balance watchdog > workdir/watchdog.txt"
execute_shell(command)
def test_execute_shell(run_mock):
run_mock.side_effect = run_func
utils.execute_shell(["asdf"])
def test_execute_shell_handles_pwd():
pwd = utils.execute_shell(["pwd"])
assert pwd
def should_reboot_via_telnet(name):
command = f"scripts/uptime.sh {NAME_IP_MAP[name]}"
uptime = float(execute_shell(command))
LOG.info("checking to see if uptime (%s) is greater than %s seconds",
uptime, get_max_time_in_seconds())
return uptime > get_max_time_in_seconds()
def reboot_via_telnet(name):
if should_reboot_via_telnet(name):
command = f"scripts/reboot_via_telnet.sh {NAME_IP_MAP[name]}"
execute_shell(command)