Rob Shearman : kernel32: Add tests for RegisterWaitForSingleObject and UnregisterWait.

[Alexandre Julliard]

Module: wine
Branch: master
Commit: 910cf617230ca5ca9c1e04589a8c846dfeb4541e
URL:    http://source.winehq.org/git/wine.git/?a=commit;h=910cf617230ca5ca9c1e04589a8c846dfeb4541e

Author: Rob Shearman <rob at codeweavers.com>
Date:   Tue Jan  1 22:37:34 2008 +0000

kernel32: Add tests for RegisterWaitForSingleObject and UnregisterWait.

---

 dlls/kernel32/tests/thread.c |   74 ++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 74 insertions(+), 0 deletions(-)

diff --git a/dlls/kernel32/tests/thread.c b/dlls/kernel32/tests/thread.c
index 6b90ad1..5989ccc 100644
--- a/dlls/kernel32/tests/thread.c
+++ b/dlls/kernel32/tests/thread.c
@@ -68,6 +68,12 @@ static SetThreadIdealProcessor_t pSetThreadIdealProcessor=NULL;
 typedef BOOL (WINAPI *SetThreadPriorityBoost_t)(HANDLE,BOOL);
 static SetThreadPriorityBoost_t pSetThreadPriorityBoost=NULL;
 
+typedef BOOL (WINAPI *RegisterWaitForSingleObject_t)(PHANDLE,HANDLE,WAITORTIMERCALLBACK,PVOID,ULONG,ULONG);
+static RegisterWaitForSingleObject_t pRegisterWaitForSingleObject=NULL;
+
+typedef BOOL (WINAPI *UnregisterWait_t)(HANDLE);
+static UnregisterWait_t pUnregisterWait=NULL;
+
 static HANDLE create_target_process(const char *arg)
 {
     char **argv;
@@ -878,6 +884,71 @@ static void test_QueueUserWorkItem(void)
     ok(times_executed == 100, "didn't execute all of the work items\n");
 }
 
+static void CALLBACK signaled_function(PVOID p, BOOLEAN TimerOrWaitFired)
+{
+    HANDLE event = p;
+    SetEvent(event);
+    ok(!TimerOrWaitFired, "wait shouldn't have timed out\n");
+}
+
+static void CALLBACK timeout_function(PVOID p, BOOLEAN TimerOrWaitFired)
+{
+    HANDLE event = p;
+    SetEvent(event);
+    ok(TimerOrWaitFired, "wait should have timed out\n");
+}
+
+static void test_RegisterWaitForSingleObject(void)
+{
+    BOOL ret;
+    HANDLE wait_handle;
+    HANDLE handle;
+    HANDLE complete_event;
+
+    if (!pRegisterWaitForSingleObject || !pUnregisterWait)
+    {
+        skip("RegisterWaitForSingleObject or UnregisterWait not implemented\n");
+        return;
+    }
+
+    /* test signaled case */
+
+    handle = CreateEvent(NULL, TRUE, TRUE, NULL);
+    complete_event = CreateEvent(NULL, FALSE, FALSE, NULL);
+
+    ret = pRegisterWaitForSingleObject(&wait_handle, handle, signaled_function, complete_event, INFINITE, WT_EXECUTEONLYONCE);
+    ok(ret, "RegisterWaitForSingleObject failed with error %d\n", GetLastError());
+
+    WaitForSingleObject(complete_event, INFINITE);
+    /* give worker thread chance to complete */
+    Sleep(100);
+
+    ret = pUnregisterWait(wait_handle);
+    ok(ret, "UnregisterWait failed with error %d\n", GetLastError());
+
+    /* test cancel case */
+
+    ResetEvent(handle);
+
+    ret = pRegisterWaitForSingleObject(&wait_handle, handle, signaled_function, complete_event, INFINITE, WT_EXECUTEONLYONCE);
+    ok(ret, "RegisterWaitForSingleObject failed with error %d\n", GetLastError());
+
+    ret = pUnregisterWait(wait_handle);
+    ok(ret, "UnregisterWait failed with error %d\n", GetLastError());
+
+    /* test timeout case */
+
+    ret = pRegisterWaitForSingleObject(&wait_handle, handle, timeout_function, complete_event, 0, WT_EXECUTEONLYONCE);
+    ok(ret, "RegisterWaitForSingleObject failed with error %d\n", GetLastError());
+
+    WaitForSingleObject(complete_event, INFINITE);
+    /* give worker thread chance to complete */
+    Sleep(100);
+
+    ret = pUnregisterWait(wait_handle);
+    ok(ret, "UnregisterWait failed with error %d\n", GetLastError());
+}
+
 START_TEST(thread)
 {
    HINSTANCE lib;
@@ -894,6 +965,8 @@ START_TEST(thread)
    pQueueUserWorkItem=(QueueUserWorkItem_t)GetProcAddress(lib,"QueueUserWorkItem");
    pSetThreadIdealProcessor=(SetThreadIdealProcessor_t)GetProcAddress(lib,"SetThreadIdealProcessor");
    pSetThreadPriorityBoost=(SetThreadPriorityBoost_t)GetProcAddress(lib,"SetThreadPriorityBoost");
+   pRegisterWaitForSingleObject=(RegisterWaitForSingleObject_t)GetProcAddress(lib,"RegisterWaitForSingleObject");
+   pUnregisterWait=(UnregisterWait_t)GetProcAddress(lib,"UnregisterWait");
 
    if (argc >= 3)
    {
@@ -934,4 +1007,5 @@ START_TEST(thread)
    test_SetThreadContext();
 #endif
    test_QueueUserWorkItem();
+   test_RegisterWaitForSingleObject();
 }