[PATCH 1/7] msvcr120/tests: Add test for canceling running chores.

[Torge Matthies]

Signed-off-by: Torge Matthies <tmatthies at codeweavers.com>
---
 dlls/msvcr120/tests/msvcr120.c | 53 ++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)

diff --git a/dlls/msvcr120/tests/msvcr120.c b/dlls/msvcr120/tests/msvcr120.c
index 26b363de6c59..88a32b382ef7 100644
--- a/dlls/msvcr120/tests/msvcr120.c
+++ b/dlls/msvcr120/tests/msvcr120.c
@@ -250,6 +250,8 @@ static _StructuredTaskCollection* (__thiscall *p__StructuredTaskCollection_ctor)
 static void (__thiscall *p__StructuredTaskCollection_dtor)(_StructuredTaskCollection*);
 static void (__thiscall *p__StructuredTaskCollection__Schedule)(_StructuredTaskCollection*, _UnrealizedChore*);
 static int (__stdcall *p__StructuredTaskCollection__RunAndWait)(_StructuredTaskCollection*, _UnrealizedChore*);
+static void (__thiscall *p__StructuredTaskCollection__Cancel)(_StructuredTaskCollection*);
+static MSVCRT_bool (__thiscall *p__StructuredTaskCollection__IsCanceling)(_StructuredTaskCollection*);
 
 #define SETNOFAIL(x,y) x = (void*)GetProcAddress(module,y)
 #define SET(x,y) do { SETNOFAIL(x,y); ok(x != NULL, "Export '%s' not found\n", y); } while(0)
@@ -313,6 +315,10 @@ static BOOL init(void)
                 "?_Schedule at _StructuredTaskCollection@details at Concurrency@@QEAAXPEAV_UnrealizedChore at 23@@Z");
         SET(p__StructuredTaskCollection__RunAndWait,
                 "?_RunAndWait at _StructuredTaskCollection@details at Concurrency@@QEAA?AW4_TaskCollectionStatus at 23@PEAV_UnrealizedChore at 23@@Z");
+        SET(p__StructuredTaskCollection__Cancel,
+                "?_Cancel at _StructuredTaskCollection@details at Concurrency@@QEAAXXZ");
+        SET(p__StructuredTaskCollection__IsCanceling,
+                "?_IsCanceling at _StructuredTaskCollection@details at Concurrency@@QEAA_NXZ");
         SET(p_critical_section_ctor,
                 "??0critical_section at Concurrency@@QEAA at XZ");
         SET(p_critical_section_dtor,
@@ -355,6 +361,10 @@ static BOOL init(void)
                 "?_Schedule at _StructuredTaskCollection@details at Concurrency@@QAAXPAV_UnrealizedChore at 23@@Z");
         SET(p__StructuredTaskCollection__RunAndWait,
                 "?_RunAndWait at _StructuredTaskCollection@details at Concurrency@@QAA?AW4_TaskCollectionStatus at 23@PAV_UnrealizedChore at 23@@Z");
+        SET(p__StructuredTaskCollection__Cancel,
+                "?_Cancel at _StructuredTaskCollection@details at Concurrency@@QAAXXZ");
+        SET(p__StructuredTaskCollection__IsCanceling,
+                "?_IsCanceling at _StructuredTaskCollection@details at Concurrency@@QAA_NXZ");
         SET(p_critical_section_ctor,
                 "??0critical_section at Concurrency@@QAA at XZ");
         SET(p_critical_section_dtor,
@@ -394,6 +404,10 @@ static BOOL init(void)
                 "?_Schedule at _StructuredTaskCollection@details at Concurrency@@QAEXPAV_UnrealizedChore at 23@@Z");
         SET(p__StructuredTaskCollection__RunAndWait,
                 "?_RunAndWait at _StructuredTaskCollection@details at Concurrency@@QAG?AW4_TaskCollectionStatus at 23@PAV_UnrealizedChore at 23@@Z");
+        SET(p__StructuredTaskCollection__Cancel,
+                "?_Cancel at _StructuredTaskCollection@details at Concurrency@@QAEXXZ");
+        SET(p__StructuredTaskCollection__IsCanceling,
+                "?_IsCanceling at _StructuredTaskCollection@details at Concurrency@@QAE_NXZ");
         SET(p_critical_section_ctor,
                 "??0critical_section at Concurrency@@QAE at XZ");
         SET(p_critical_section_dtor,
@@ -1334,6 +1348,7 @@ struct chore
     HANDLE set_event;
     HANDLE wait_event;
     DWORD main_tid;
+    BOOL check_canceling;
 };
 
 static void __cdecl chore_proc(_UnrealizedChore *_this)
@@ -1355,6 +1370,14 @@ static void __cdecl chore_proc(_UnrealizedChore *_this)
                 GetCurrentThreadId(), chore->main_tid);
     }
 
+    if (chore->check_canceling)
+    {
+        MSVCRT_bool canceling = call_func1(
+                p__StructuredTaskCollection__IsCanceling,
+                chore->chore.task_collection);
+        ok(!canceling, "Task is already canceling");
+    }
+
     if (!chore->wait_event)
         chore->executed = TRUE;
 
@@ -1370,6 +1393,14 @@ static void __cdecl chore_proc(_UnrealizedChore *_this)
         ok(ret == WAIT_OBJECT_0, "WaitForSingleObject returned %ld\n", ret);
         chore->executed = TRUE;
     }
+
+    if (chore->check_canceling)
+    {
+        MSVCRT_bool canceling = call_func1(
+                p__StructuredTaskCollection__IsCanceling,
+                chore->chore.task_collection);
+        ok(canceling, "Task is not canceling");
+    }
 }
 
 static void chore_ctor(struct chore *chore)
@@ -1533,6 +1564,28 @@ static void test_StructuredTaskCollection(void)
     ok(chore1.executed, "Chore was not executed\n");
     call_func1(p__StructuredTaskCollection_dtor, &task_coll);
 
+    /* test that running chores can be canceled */
+    call_func2(p__StructuredTaskCollection_ctor, &task_coll, NULL);
+    ResetEvent(chore_start_evt);
+    chore_ctor(&chore1);
+    chore1.check_canceling = TRUE;
+    chore1.wait_event = chore_evt2;
+    chore1.set_event = chore_evt1;
+
+    call_func2(p__StructuredTaskCollection__Schedule, &task_coll, &chore1.chore);
+
+    ret = WaitForSingleObject(chore_evt1, 5000);
+    ok(ret == WAIT_OBJECT_0, "WaitForSingleObject returned %ld\n", ret);
+
+    call_func1(p__StructuredTaskCollection__Cancel, &task_coll);
+
+    b = SetEvent(chore_evt2);
+    ok(b, "SetEvent failed\n");
+
+    status = p__StructuredTaskCollection__RunAndWait(&task_coll, NULL);
+    ok(status == 2, "_StructuredTaskCollection::_RunAndWait failed: %d\n", status);
+    call_func1(p__StructuredTaskCollection_dtor, &task_coll);
+
     CloseHandle(chore_start_evt);
     CloseHandle(chore_evt1);
     CloseHandle(chore_evt2);
-- 
2.37.0