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")