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)