def test_virtual_machine_winrm_fail(run_ps, connection): os.environ['vcdriver_vm_winrm_username'] = '******' os.environ['vcdriver_vm_winrm_password'] = '******' load() vm = VirtualMachine() vm_object_mock = mock.MagicMock() vm_object_mock.summary.guest.ipAddress = '127.0.0.1' vm.__setattr__('_vm_object', vm_object_mock) run_ps.return_value.status_code = 1 with pytest.raises(WinRmError): vm.winrm('script', dict())
def test_virtual_machine_winrm_timeout(run_ps, connection): os.environ['vcdriver_vm_winrm_username'] = '******' os.environ['vcdriver_vm_winrm_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) run_ps.side_effect = Exception with pytest.raises(TimeoutError): vm.winrm('script', dict())
def test_virtual_machine_winrm_success(run_ps, connection): os.environ['vcdriver_vm_winrm_username'] = '******' os.environ['vcdriver_vm_winrm_password'] = '******' load() vm = VirtualMachine() assert vm.winrm('whatever', dict()) is None vm_object_mock = mock.MagicMock() vm_object_mock.summary.guest.ipAddress = '127.0.0.1' vm.__setattr__('_vm_object', vm_object_mock) run_ps.return_value.status_code = 0 vm.winrm('script', dict()) run_ps.assert_called_with('script') assert run_ps.call_count == 2