reactos/wine
1
0
Fork 0
mirror of https://github.com/reactos/wine.git synced 2024-11-29 22:50:43 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

rpcrt4: Tests for CStdStubBuffer_Disconnect and a tidy up of the implementation.

Browse Source
This commit is contained in:
Huw Davies 2006-08-23 16:36:43 +01:00 committed by Alexandre Julliard
parent f42f727f7c
commit 487c691dc5
2 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
dlls/rpcrt4/cstub.c
View File

@ -149,13 +149,14 @@ HRESULT WINAPI CStdStubBuffer_Connect(LPRPCSTUBBUFFER iface,
void WINAPI CStdStubBuffer_Disconnect(LPRPCSTUBBUFFER iface)
{
CStdStubBuffer *This = (CStdStubBuffer *)iface;
TRACE("(%p)->Disconnect()\n",This);
if (This->pvServerObject)
{
IUnknown_Release(This->pvServerObject);
This->pvServerObject = NULL;
}
CStdStubBuffer *This = (CStdStubBuffer *)iface;
IUnknown *old;
TRACE("(%p)->Disconnect()\n",This);
old = InterlockedExchangePointer((void**)&This->pvServerObject, NULL);
if(old)
IUnknown_Release(old);
}
HRESULT WINAPI CStdStubBuffer_Invoke(LPRPCSTUBBUFFER iface,

14
dlls/rpcrt4/tests/cstub.c
View File

@ -703,6 +703,19 @@ static void test_Connect(IPSFactoryBuffer *ppsf)
ok(connect_test_orig_release_called == 2, "release called %d\n", connect_test_orig_release_called);
}
static void test_Disconnect(IPSFactoryBuffer *ppsf)
{
IUnknownVtbl *orig_vtbl = &connect_test_orig_vtbl;
IUnknown *obj = (IUnknown*)&orig_vtbl;
IRpcStubBuffer *pstub = create_stub(ppsf, &IID_if1, obj, S_OK);
CStdStubBuffer *cstd_stub = (CStdStubBuffer*)pstub;
connect_test_orig_release_called = 0;
IRpcStubBuffer_Disconnect(pstub);
ok(connect_test_orig_release_called == 1, "release called %d\n", connect_test_orig_release_called);
ok(cstd_stub->pvServerObject == NULL, "pvServerObject %p\n", cstd_stub->pvServerObject);
}
START_TEST( cstub )
{
IPSFactoryBuffer *ppsf;
@ -713,6 +726,7 @@ START_TEST( cstub )
test_NdrStubForwardingFunction();
test_CreateStub(ppsf);
test_Connect(ppsf);
test_Disconnect(ppsf);
OleUninitialize();
}