[PATCH 3/5] taskschd: Implement ITaskSettings_put_RestartCount.

[Hans Leidekker]

Signed-off-by: Hans Leidekker 
---
 dlls/taskschd/task.c            | 9 +++++++--
 dlls/taskschd/tests/scheduler.c | 6 +++---
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/dlls/taskschd/task.c b/dlls/taskschd/task.c
index e184ca813a..683a20c114 100644
--- a/dlls/taskschd/task.c
+++ b/dlls/taskschd/task.c
@@ -951,8 +951,13 @@ static HRESULT WINAPI TaskSettings_get_RestartCount(ITaskSettings *iface, INT *c

static HRESULT WINAPI TaskSettings_put_RestartCount(ITaskSettings *iface, INT count)
 {
-    FIXME("%p,%d: stub\n", iface, count);
-    return E_NOTIMPL;
+    TaskSettings *taskset = impl_from_ITaskSettings(iface);
+
+    TRACE("%p,%d\n", iface, count);
+
+    taskset->restart_count = count;
+
+    return S_OK;
 }

static HRESULT WINAPI TaskSettings_get_MultipleInstances(ITaskSettings *iface, TASK_INSTANCES_POLICY *policy)
diff --git a/dlls/taskschd/tests/scheduler.c b/dlls/taskschd/tests/scheduler.c
index 0e1c2ebc3d..eac17340cd 100644
--- a/dlls/taskschd/tests/scheduler.c
+++ b/dlls/taskschd/tests/scheduler.c
@@ -1105,6 +1105,9 @@ static void change_settings(ITaskDefinition *taskdef, struct settings *test)
     ok(hr == S_OK, "expected S_OK, got %#x\n", hr);

hr = ITaskSettings_put_RestartCount(set, test->restart_count);
+    ok(hr == S_OK, "expected S_OK, got %#x\n", hr);
+
+    hr = ITaskSettings_put_MultipleInstances(set, test->policy);
 todo_wine
     ok(hr == S_OK, "expected S_OK, got %#x\n", hr);
     /* FIXME: Remove once implemented */
@@ -1114,9 +1117,6 @@ todo_wine
         return;
     }

-    hr = ITaskSettings_put_MultipleInstances(set, test->policy);
-    ok(hr == S_OK, "expected S_OK, got %#x\n", hr);
-
     hr = ITaskSettings_put_StopIfGoingOnBatteries(set, test->stop_if_going_on_batteries);
     ok(hr == S_OK, "expected S_OK, got %#x\n", hr);