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
