def test_schedule_vcenter_task_on_vm(connection):
mock_CreateScheduledTask = mock.MagicMock()
connection.return_value.content.scheduledTaskManager.CreateScheduledTask = mock_CreateScheduledTask
vm = VirtualMachine()
vm._schedule_vcenter_task_on_vm(vim.VirtualMachine.PowerOff,
'Power off virtual machine "dummy"',
datetime.timedelta(hours=1))
assert mock_CreateScheduledTask.call_count == 1
def test_schedule_vcenter_task_on_vm_fail_on_bad_type(connection):
vm = VirtualMachine()
with pytest.raises(TypeError):
vm._schedule_vcenter_task_on_vm(vim.VirtualMachine.PowerOff,
'Power off virtual machine "dummy"',
"Wrong type")
