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
