initializespy.cpp File Reference

#include <stdio.h>
#include <wine/test.h>
#include <winuser.h>
#include <winreg.h>
#include <shlwapi.h>
#include <unknownbase.h>

Include dependency graph for initializespy.cpp:

Go to the source code of this file.

Classes
class	CTestSpy

Macros
#define	WIN32_NO_STATUS
#define	_INC_WINDOWS
#define	COM_NO_WINDOWS_H
#define	test_S_OK(hres, message)   ok((hres) == S_OK, "%s (0x%lx instead of S_OK)\n", (message), (hres))
#define	test_HRES(hres, hresExpected, message)   ok((hres) == (hresExpected), "%s (0x%lx instead of 0x%lx)\n", (message), (hres), (hresExpected))
#define	test_ref(spy, expectedRef)   ok((spy)->GetRef() == (expectedRef), "unexpected refcount, %ld instead of %d\n", (spy)->GetRef(), (expectedRef))

Typedefs
typedef HRESULT(WINAPI *	pCoRegisterInitializeSpy_t) (_In_ LPINITIALIZESPY pSpy, _Out_ ULARGE_INTEGER *puliCookie)
typedef HRESULT(WINAPI *	pCoRevokeInitializeSpy_t) (_In_ ULARGE_INTEGER uliCookie)

Functions
void	test_IInitializeSpy_register2 ()
void	test_IInitializeSpy_switch_apt ()
void	test_IInitializeSpy_fail ()
void	test_IInitializeSpy_twice ()
	START_TEST (initializespy)

Variables
pCoRegisterInitializeSpy_t	pCoRegisterInitializeSpy
pCoRevokeInitializeSpy_t	pCoRevokeInitializeSpy
const DWORD	INVALID_VALUE = 0xdeadbeef

Macro Definition Documentation

_INC_WINDOWS

#define _INC_WINDOWS

Definition at line 9 of file initializespy.cpp.

COM_NO_WINDOWS_H

#define COM_NO_WINDOWS_H

Definition at line 10 of file initializespy.cpp.

test_HRES

#define test_HRES	(		hres,
			hresExpected,
			message
	)		ok((hres) == (hresExpected), "%s (0x%lx instead of 0x%lx)\n", (message), (hres), (hresExpected))

Definition at line 22 of file initializespy.cpp.

test_ref

#define test_ref	(		spy,
			expectedRef
	)		ok((spy)->GetRef() == (expectedRef), "unexpected refcount, %ld instead of %d\n", (spy)->GetRef(), (expectedRef))

Definition at line 23 of file initializespy.cpp.

test_S_OK

#define test_S_OK	(		hres,
			message
	)		ok((hres) == S_OK, "%s (0x%lx instead of S_OK)\n", (message), (hres))

Definition at line 21 of file initializespy.cpp.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 8 of file initializespy.cpp.

Typedef Documentation

pCoRegisterInitializeSpy_t

typedef HRESULT(WINAPI * pCoRegisterInitializeSpy_t) (_In_ LPINITIALIZESPY pSpy, _Out_ ULARGE_INTEGER *puliCookie)

Definition at line 26 of file initializespy.cpp.

pCoRevokeInitializeSpy_t

typedef HRESULT(WINAPI * pCoRevokeInitializeSpy_t) (_In_ ULARGE_INTEGER uliCookie)

Definition at line 27 of file initializespy.cpp.

Function Documentation

START_TEST()

START_TEST

(

initializespy

)

Definition at line 311 of file initializespy.cpp.

{
    HMODULE ole32 = LoadLibraryA("ole32.dll");
    pCoRegisterInitializeSpy = (pCoRegisterInitializeSpy_t)GetProcAddress(ole32, "CoRegisterInitializeSpy");
    pCoRevokeInitializeSpy = (pCoRevokeInitializeSpy_t)GetProcAddress(ole32, "CoRevokeInitializeSpy");
 
    test_IInitializeSpy_register2();
    test_IInitializeSpy_switch_apt();
    test_IInitializeSpy_fail();
    test_IInitializeSpy_twice();
}

test_IInitializeSpy_fail()

void test_IInitializeSpy_fail

(

)

Definition at line 260 of file initializespy.cpp.

{
    CTestSpy spy;
 
    spy.m_FailQueryInterface = true;
 
    spy.hr = pCoRegisterInitializeSpy(&spy, &spy.Cookie);
    test_HRES(spy.hr, E_NOINTERFACE, "Unexpected hr while registering invalid interface");
    test_ref(&spy, 0);
    ok(spy.Cookie.HighPart == 0xffffffff, "Unexpected Cookie.HighPart, expected 0xffffffff got: 0x%08lx\n", spy.Cookie.HighPart);
    ok(spy.Cookie.LowPart == 0xffffffff, "Unexpected Cookie.HighPart, expected 0xffffffff got: 0x%08lx\n", spy.Cookie.LowPart);
 
    spy.Cookie.HighPart = spy.Cookie.LowPart = 0xffffffff;
    HRESULT hr = pCoRevokeInitializeSpy(spy.Cookie);
    test_HRES(hr, E_INVALIDARG, "Unexpected hr while unregistering invalid interface");
    test_ref(&spy, 0);
 
    spy.Cookie.HighPart = spy.Cookie.LowPart = 0;
    hr = pCoRevokeInitializeSpy(spy.Cookie);
    test_HRES(hr, E_INVALIDARG, "Unexpected hr while unregistering invalid interface");
    test_ref(&spy, 0);
 
    /* we should not crash here, just return E_NOINTERFACE
        do note the Cookie is not even being touched at all, compared to calling this with an interface
        that does not respond to IID_IInitializeSpy */
    spy.Cookie.HighPart = spy.Cookie.LowPart = INVALID_VALUE;
    hr = pCoRegisterInitializeSpy(NULL, &spy.Cookie);
    test_HRES(spy.hr, E_NOINTERFACE, "Unexpected hr while registering NULL interface");
    ok(spy.Cookie.HighPart == INVALID_VALUE, "Unexpected Cookie.HighPart, expected 0xdeadbeef got: %lx\n", spy.Cookie.HighPart);
    ok(spy.Cookie.LowPart == INVALID_VALUE, "Unexpected Cookie.HighPart, expected 0xdeadbeef got: %lx\n", spy.Cookie.LowPart);
}

Referenced by START_TEST().

test_IInitializeSpy_register2()

void test_IInitializeSpy_register2

(

)

Definition at line 159 of file initializespy.cpp.

{
    CTestSpy spy, spy2;
 
    // first we register 2 spies
    spy.hr = pCoRegisterInitializeSpy(&spy, &spy.Cookie);
    test_S_OK(spy.hr, "CoRegisterInitializeSpy");
    test_ref(&spy, 1);
 
    spy2.hr = pCoRegisterInitializeSpy(&spy2, &spy2.Cookie);
    test_S_OK(spy2.hr, "CoRegisterInitializeSpy");
    test_ref(&spy, 1);
 
    // tell them what we expect
    spy.Expect(S_OK, COINIT_APARTMENTTHREADED, 1);
    spy2.Expect(S_OK, COINIT_APARTMENTTHREADED, 1);
 
    // Call CoInitializeEx and validate the results
    HRESULT hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
    test_S_OK(hr, "CoInitializeEx");
    spy.Check(1, 1, 0, 0);
    spy2.Check(1, 1, 0, 0);
 
    // Calling CoInit twice with the same apartment makes it return S_FALSE but still increment count
    spy.Expect(S_FALSE, COINIT_APARTMENTTHREADED, 2);
    spy2.Expect(S_FALSE, COINIT_APARTMENTTHREADED, 2);
 
    hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
    test_HRES(hr, S_FALSE, "CoInitializeEx");
    spy.Check(2, 2, 0, 0);
    spy2.Check(2, 2, 0, 0);
 
    /* the order we registered the spies in is important here.
        we have the second one to forcibly return S_OK, which makes the first spy see
        S_OK instead of S_FALSE.. */
    spy.Expect(S_OK, COINIT_APARTMENTTHREADED, 3);
    spy2.m_AlwaysReturnOK = true;
    spy2.Expect(S_FALSE, COINIT_APARTMENTTHREADED, 3);
 
    // and the S_OK also influences the returned value from CoInit.
    hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
    test_S_OK(hr, "CoInitializeEx");
    spy.Check(3, 3, 0, 0);
    spy2.Check(3, 3, 0, 0);
 
    CoUninitialize();
    spy.Check(3, 3, 1, 1);
    spy2.Check(3, 3, 1, 1);
 
    spy.m_CurAptRefs = spy2.m_CurAptRefs = 2;
 
    CoUninitialize();
    spy.Check(3, 3, 2, 2);
    spy2.Check(3, 3, 2, 2);
 
    spy.m_CurAptRefs = spy2.m_CurAptRefs = 1;
 
    CoUninitialize();
    spy.Check(3, 3, 3, 3);
    spy2.Check(3, 3, 3, 3);
 
    spy.m_CurAptRefs = spy2.m_CurAptRefs = 0;
 
    CoUninitialize();
    spy.Check(3, 3, 4, 4);
    spy2.Check(3, 3, 4, 4);
}

Referenced by START_TEST().

test_IInitializeSpy_switch_apt()

void test_IInitializeSpy_switch_apt

(

)

Definition at line 227 of file initializespy.cpp.

{
    CTestSpy spy;
 
    spy.hr = pCoRegisterInitializeSpy(&spy, &spy.Cookie);
    test_S_OK(spy.hr, "CoRegisterInitializeSpy");
    test_ref(&spy, 1);
 
    spy.Expect(S_OK, COINIT_APARTMENTTHREADED, 1);
 
    HRESULT hr = CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
    test_S_OK(hr, "CoInitializeEx");
    spy.Check(1, 1, 0, 0);
 
    spy.Expect(RPC_E_CHANGED_MODE, COINIT_MULTITHREADED, 1);
 
    hr = CoInitializeEx(NULL, COINIT_MULTITHREADED);
    test_HRES(hr, RPC_E_CHANGED_MODE, "CoInitializeEx");
    spy.Check(2, 2, 0, 0);
 
 
    CoUninitialize();
    spy.Check(2, 2, 1, 1);
 
    spy.m_CurAptRefs = 0;
 
    CoUninitialize();
    spy.Check(2, 2, 2, 2);
 
    CoUninitialize();
    spy.Check(2, 2, 3, 3);
}

Referenced by START_TEST().

test_IInitializeSpy_twice()

void test_IInitializeSpy_twice

(

)

Definition at line 292 of file initializespy.cpp.

{
    CTestSpy spy;
 
    spy.hr = pCoRegisterInitializeSpy(&spy, &spy.Cookie);
    test_S_OK(spy.hr, "CoRegisterInitializeSpy");
    test_ref(&spy, 1);
 
    ULARGE_INTEGER Cookie = { { INVALID_VALUE, INVALID_VALUE } };
    HRESULT hr = pCoRegisterInitializeSpy(&spy, &Cookie);
    test_S_OK(hr, "CoRegisterInitializeSpy");
    test_ref(&spy, 2);
 
    hr = pCoRevokeInitializeSpy(Cookie);
    test_S_OK(hr, "CoRevokeInitializeSpy");
    test_ref(&spy, 1);
}

Referenced by START_TEST().

Variable Documentation

INVALID_VALUE

const DWORD INVALID_VALUE = 0xdeadbeef

Definition at line 32 of file initializespy.cpp.

Referenced by CTestSpy::CTestSpy(), test_IInitializeSpy_fail(), and test_IInitializeSpy_twice().

pCoRegisterInitializeSpy

pCoRegisterInitializeSpy_t pCoRegisterInitializeSpy

Definition at line 28 of file initializespy.cpp.

Referenced by START_TEST(), test_IInitializeSpy_fail(), test_IInitializeSpy_register2(), test_IInitializeSpy_switch_apt(), and test_IInitializeSpy_twice().

pCoRevokeInitializeSpy

pCoRevokeInitializeSpy_t pCoRevokeInitializeSpy

Definition at line 29 of file initializespy.cpp.

Referenced by START_TEST(), test_IInitializeSpy_fail(), test_IInitializeSpy_twice(), and CTestSpy::~CTestSpy().