Dmitry Timoshkov : mstask: Implement ITaskScheduler::AddWorkItem().

[Alexandre Julliard]

Module: wine
Branch: master
Commit: f429c8188b321246af6e2a7ce8feee96f458d7a4
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=f429c8188b321246af6e2a7ce...

Author: Dmitry Timoshkov 
Date:   Wed Apr 25 11:34:52 2018 +0800

mstask: Implement ITaskScheduler::AddWorkItem().

Signed-off-by: Dmitry Timoshkov 
Signed-off-by: Alexandre Julliard

---

dlls/mstask/task_scheduler.c       | 29 +++++++++++++++++++++++------
 dlls/mstask/tests/task_scheduler.c |  8 --------
 2 files changed, 23 insertions(+), 14 deletions(-)

diff --git a/dlls/mstask/task_scheduler.c b/dlls/mstask/task_scheduler.c
index dd6eaf6..c890cb6 100644
--- a/dlls/mstask/task_scheduler.c
+++ b/dlls/mstask/task_scheduler.c
@@ -331,13 +331,30 @@ static HRESULT WINAPI MSTASK_ITaskScheduler_NewWorkItem(
     return TaskConstructor(This->service, task_name, (ITask **)task);
 }

-static HRESULT WINAPI MSTASK_ITaskScheduler_AddWorkItem(
-        ITaskScheduler* iface,
-        LPCWSTR pwszTaskName,
-        IScheduledWorkItem *pWorkItem)
+static HRESULT WINAPI MSTASK_ITaskScheduler_AddWorkItem(ITaskScheduler *iface, LPCWSTR name, IScheduledWorkItem *item)
 {
-    FIXME("%p, %s, %p: stub\n", iface, debugstr_w(pwszTaskName), pWorkItem);
-    return E_NOTIMPL;
+    static const WCHAR tasksW[] = { '\','T','a','s','k','s','\',0 };
+    static const WCHAR jobW[] = { '.','j','o','b',0 };
+    WCHAR task_name[MAX_PATH];
+    IPersistFile *pfile;
+    HRESULT hr;
+
+    TRACE("%p, %s, %p\n", iface, debugstr_w(name), item);
+
+    if (strchrW(name, '.')) return E_INVALIDARG;
+
+    GetWindowsDirectoryW(task_name, MAX_PATH);
+    lstrcatW(task_name, tasksW);
+    lstrcatW(task_name, name);
+    lstrcatW(task_name, jobW);
+
+    hr = IScheduledWorkItem_QueryInterface(item, &IID_IPersistFile, (void **)&pfile);
+    if (hr == S_OK)
+    {
+        hr = IPersistFile_Save(pfile, task_name, TRUE);
+        IPersistFile_Release(pfile);
+    }
+    return hr;
 }

static HRESULT WINAPI MSTASK_ITaskScheduler_IsOfType(
diff --git a/dlls/mstask/tests/task_scheduler.c b/dlls/mstask/tests/task_scheduler.c
index fbee5ec..8482972 100644
--- a/dlls/mstask/tests/task_scheduler.c
+++ b/dlls/mstask/tests/task_scheduler.c
@@ -343,30 +343,23 @@ static void test_task_storage(void)
     ok(task2 == (ITask *)0xdeadbeef, "task should not be set to NULL\n");

hr = ITaskScheduler_AddWorkItem(scheduler, Task1, (IScheduledWorkItem *)task);
-todo_wine
     ok(hr == S_OK, "AddWorkItem error %#x\n", hr);

curfile = get_task_curfile(task);
-todo_wine
     ok(file_exists(curfile), "curfile should exist\n");

hr = ITaskScheduler_AddWorkItem(scheduler, Task1, (IScheduledWorkItem *)task);
-todo_wine
     ok(hr == HRESULT_FROM_WIN32(ERROR_FILE_EXISTS), "wrong error %#x\n", hr);

curfile = get_task_curfile(task);
-todo_wine
     ok(file_exists(curfile), "curfile should exist\n");

ITask_Release(task);

task = NULL;
     hr = ITaskScheduler_Activate(scheduler, Task1, &IID_ITask, (IUnknown **)&task);
-todo_wine
     ok(hr == S_OK, "Activate error %#x\n", hr);
-todo_wine
     ok(task != NULL, "task should not be set to NULL\n");
-    if (task == NULL) goto fail;

curfile2 = get_task_curfile(task);
     ok(file_exists(curfile2), "curfile2 should exist\n");
@@ -380,7 +373,6 @@ todo_wine
     CoTaskMemFree(curfile);

ITask_Release(task);
-fail:
     ITaskScheduler_Release(scheduler);
 }