def test_virtual_machine_ssh_fail(sudo, helpers_run, vm_run, connection):
os.environ['vcdriver_vm_ssh_username'] = '******'
os.environ['vcdriver_vm_ssh_password'] = '******'
load()
vm = VirtualMachine()
vm_object_mock = mock.MagicMock()
vm_object_mock.summary.guest.ipAddress = 'fe80::250:56ff:febf:1a0a'
vm.__setattr__('_vm_object', vm_object_mock)
with pytest.raises(SshError):
vm.ssh('whatever', use_sudo=True)
def test_virtual_machine_ssh_timeout(helpers_run, vm_run, connection):
os.environ['vcdriver_vm_ssh_username'] = '******'
os.environ['vcdriver_vm_ssh_password'] = '******'
load()
vm = VirtualMachine(timeout=1)
vm_object_mock = mock.MagicMock()
vm_object_mock.summary.guest.ipAddress = '127.0.0.1'
vm.__setattr__('_vm_object', vm_object_mock)
helpers_run.side_effect = Exception
vm_run.side_effect = Exception
with pytest.raises(TimeoutError):
vm.ssh('whatever', use_sudo=True)
def test_virtual_machine_ssh_success(helpers_run, vm_run, sudo, connection):
os.environ['vcdriver_vm_ssh_username'] = '******'
os.environ['vcdriver_vm_ssh_password'] = '******'
load()
vm = VirtualMachine()
assert vm.ssh('whatever') is None
vm_object_mock = mock.MagicMock()
vm_object_mock.summary.guest.ipAddress = '127.0.0.1'
vm.__setattr__('_vm_object', vm_object_mock)
result_mock = mock.MagicMock()
result_mock.return_code = 3
result_mock.failed = False
helpers_run.return_value = result_mock
vm_run.return_value = result_mock
sudo.return_value = result_mock
assert vm.ssh('whatever', use_sudo=False).return_code == 3
assert vm.ssh('whatever', use_sudo=True).return_code == 3
