def test_wait_all_boot(join, run_command): # pylint: disable=unused-argument """Test wait for ssh nodes to be available.""" config_ssh = { 'user': '******', 'exp_id': 123, } test_command = 'test' groups = _nodes_grouped(_ROOT_NODES) # normal boot node_ssh = OpenLinuxSsh(config_ssh, groups, verbose=True) output = [HostOutput(host, 'test', 0) for host in _ROOT_NODES] run_command.return_value = output node_ssh.wait(120) assert run_command.call_count == 2 run_command.assert_called_with('uptime', stop_on_errors=False) run_command.reset_mock() node_ssh.run(test_command) assert run_command.call_count == 2 run_command.assert_called_with(test_command, stop_on_errors=False)
def wait_for_boot(config_ssh, nodes, max_wait=120, verbose=False): """Wait for the open Linux nodes boot """ # Configure ssh. groups = _nodes_grouped(nodes) ssh = OpenLinuxSsh(config_ssh, groups, verbose=verbose) # Wait for boot return {"wait-for-boot": ssh.wait(max_wait)}