Rob Shearman : rpcrt4: Add tests for some async RPC functions.

[Alexandre Julliard]

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

Author: Rob Shearman <rob(a)codeweavers.com>
Date:   Mon Jan  7 15:21:02 2008 +0000

rpcrt4: Add tests for some async RPC functions.

---

 dlls/rpcrt4/tests/Makefile.in |    1 +
 dlls/rpcrt4/tests/rpc_async.c |  102 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 103 insertions(+), 0 deletions(-)

diff --git a/dlls/rpcrt4/tests/Makefile.in b/dlls/rpcrt4/tests/Makefile.in
index c497aa3..3191973 100644
--- a/dlls/rpcrt4/tests/Makefile.in
+++ b/dlls/rpcrt4/tests/Makefile.in
@@ -15,6 +15,7 @@ CTESTS = \
 	generated.c \
 	ndr_marshall.c \
 	rpc.c \
+	rpc_async.c \
 	server.c
 
 @MAKE_TEST_RULES@
diff --git a/dlls/rpcrt4/tests/rpc_async.c b/dlls/rpcrt4/tests/rpc_async.c
new file mode 100644
index 0000000..0d32fd1
--- /dev/null
+++ b/dlls/rpcrt4/tests/rpc_async.c
@@ -0,0 +1,102 @@
+/*
+ * Unit test suite for rpc functions
+ *
+ * Copyright 2008 Robert Shearman (for CodeWeavers)
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#include <stdarg.h>
+
+#include "wine/test.h"
+
+#include <rpc.h>
+#include <rpcasync.h>
+
+RPC_STATUS (RPC_ENTRY *pRpcAsyncInitializeHandle)(PRPC_ASYNC_STATE,unsigned int);
+RPC_STATUS (RPC_ENTRY *pRpcAsyncGetCallStatus)(PRPC_ASYNC_STATE);
+
+static void test_RpcAsyncInitializeHandle(void)
+{
+    char buffer[256];
+    RPC_ASYNC_STATE async;
+    RPC_STATUS status;
+    int i;
+
+    status = pRpcAsyncInitializeHandle((PRPC_ASYNC_STATE)buffer, sizeof(buffer));
+    todo_wine
+    ok(status == ERROR_INVALID_PARAMETER, "RpcAsyncInitializeHandle with large Size should have returned ERROR_INVALID_PARAMETER instead of %ld\n", status);
+
+    status = pRpcAsyncInitializeHandle(&async, sizeof(async) - 1);
+    todo_wine
+    ok(status == ERROR_INVALID_PARAMETER, "RpcAsyncInitializeHandle with small Size should have returned ERROR_INVALID_PARAMETER instead of %ld\n", status);
+
+    memset(&async, 0xcc, sizeof(async));
+    status = pRpcAsyncInitializeHandle(&async, sizeof(async));
+    todo_wine
+    ok(status == RPC_S_OK, "RpcAsyncInitializeHandle failed with error %ld\n", status);
+
+    todo_wine
+    ok(async.Size == sizeof(async), "async.Size wrong: %d\n", async.Size);
+    todo_wine
+    ok(async.Signature == 0x43595341, "async.Signature should be 0x43595341, but is 0x%x instead\n", async.Signature);
+    todo_wine
+    ok(async.Lock == 0, "async.Lock should be 0, but is %d instead\n", async.Lock);
+    todo_wine
+    ok(async.Flags == 0, "async.Flags should be 0, but is %d instead\n", async.Flags);
+    todo_wine
+    ok(async.StubInfo == NULL, "async.StubInfo should be NULL, not %p\n", async.StubInfo);
+    ok(async.UserInfo == (void *)0xcccccccc, "async.UserInfo should be unset, not %p\n", async.UserInfo);
+    todo_wine
+    ok(async.RuntimeInfo == NULL, "async.RuntimeInfo should be NULL, not %p\n", async.RuntimeInfo);
+    ok(async.Event == 0xcccccccc, "async.Event should be unset, not %d\n", async.Event);
+    ok(async.NotificationType == 0xcccccccc, "async.NotificationType should be unset, not %d\n", async.NotificationType);
+    for (i = 0; i < 4; i++)
+        todo_wine
+        ok(async.Reserved[i] == 0x0, "async.Reserved[%d] should be 0x0, not 0x%lx\n", i, async.Reserved[i]);
+}
+
+static void test_RpcAsyncGetCallStatus(void)
+{
+    RPC_ASYNC_STATE async;
+    RPC_STATUS status;
+
+    status = pRpcAsyncInitializeHandle(&async, sizeof(async));
+    todo_wine
+    ok(status == RPC_S_OK, "RpcAsyncInitializeHandle failed with error %ld\n", status);
+
+    status = pRpcAsyncGetCallStatus(&async);
+    todo_wine
+    ok(status == RPC_S_INVALID_BINDING, "RpcAsyncGetCallStatus should have returned RPC_S_INVALID_BINDING instead of %ld\n", status);
+
+    memset(&async, 0, sizeof(async));
+    status = pRpcAsyncGetCallStatus(&async);
+    todo_wine
+    ok(status == RPC_S_INVALID_BINDING, "RpcAsyncGetCallStatus should have returned RPC_S_INVALID_BINDING instead of %ld\n", status);
+}
+
+START_TEST( rpc_async )
+{
+    HMODULE hRpcRt4 = GetModuleHandle("rpcrt4.dll");
+    pRpcAsyncInitializeHandle = (void *)GetProcAddress(hRpcRt4, "RpcAsyncInitializeHandle");
+    pRpcAsyncGetCallStatus = (void *)GetProcAddress(hRpcRt4, "RpcAsyncGetCallStatus");
+    if (!pRpcAsyncInitializeHandle || !pRpcAsyncGetCallStatus)
+    {
+        skip("asynchronous functions not available\n");
+        return;
+    }
+    test_RpcAsyncInitializeHandle();
+    test_RpcAsyncGetCallStatus();
+}