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