def test_virtual_machine_winrm_upload_fail(run_ps, connection, open, os_stat): st_size_mock = mock.Mock() st_size_mock.st_size = 3 os_stat.return_value = st_size_mock code_mock = mock.Mock() code_mock.status_code = 1 code_mock.std_err = 'Whatever'.encode('ascii') run_ps.return_value = code_mock read_mock = mock.Mock read_mock.read = lambda x, y: b'\0\0\0' open.__enter__ = read_mock open.__exit__ = mock.Mock() 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) with pytest.raises(WinRmError): vm.winrm_upload('whatever', 'whatever', step=2)
def test_virtual_machine_winrm_upload_success(run_ps, connection, open, os_stat): st_size_mock = mock.Mock() st_size_mock.st_size = 3 os_stat.return_value = st_size_mock code_mock = mock.Mock() code_mock.status_code = 0 run_ps.return_value = code_mock read_mock = mock.Mock read_mock.read = lambda x, y: b'\0\0\0' open.__enter__ = read_mock open.__exit__ = mock.Mock() os.environ['vcdriver_vm_winrm_username'] = '******' os.environ['vcdriver_vm_winrm_password'] = '******' load() vm = VirtualMachine() assert vm.winrm_upload('whatever', '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) assert vm.winrm_upload('whatever', 'whatever', step=2) is None assert vm.winrm_upload('whatever', 'whatever', step=2, quiet=True) is None