reactos/modules/rostests/apitests/atl/CComObject.cpp
Katayama Hirofumi MZ a800fa7bb2
[ATL_APITEST] Add 'atltest.h' and use it (#1822)
- Add "atltest.h" header file for common use in atl_apitest.
- Delete the duplicated and simplify.
2019-08-12 19:24:08 +09:00

125 lines
3.5 KiB
C++

/*
* PROJECT: ReactOS api tests
* LICENSE: GPL-2.0+ (https://spdx.org/licenses/GPL-2.0+)
* PURPOSE: Test for CComObject
* COPYRIGHT: Copyright 2017 Mark Jansen (mark.jansen@reactos.org)
*/
#include <atlbase.h>
#include <atlcom.h>
#ifdef HAVE_APITEST
#include <apitest.h>
#else
#include "atltest.h"
#endif
static LONG g_CTOR = 0;
static LONG g_DTOR = 0;
static LONG g_BLIND = 0;
class CTestObject :
public CComObjectRootEx<CComMultiThreadModelNoCS>,
public IPersist,
public IStdMarshalInfo
{
public:
CTestObject()
{
InterlockedIncrement(&g_CTOR);
}
~CTestObject()
{
InterlockedIncrement(&g_DTOR);
}
// *** IPersist methods ***
virtual HRESULT STDMETHODCALLTYPE GetClassID(CLSID *pClassID)
{
return E_NOTIMPL;
}
// *** IStdMarshalInfo methods ***
virtual HRESULT STDMETHODCALLTYPE GetClassForHandler(DWORD dwDestContext, void *pvDestContext, CLSID *pClsid)
{
return E_NOTIMPL;
}
static HRESULT WINAPI FuncBlind(void* pv, REFIID riid, LPVOID* ppv, DWORD_PTR dw)
{
InterlockedIncrement(&g_BLIND);
return E_FAIL;
}
DECLARE_NOT_AGGREGATABLE(CTestObject)
DECLARE_PROTECT_FINAL_CONSTRUCT()
BEGIN_COM_MAP(CTestObject)
COM_INTERFACE_ENTRY_IID(IID_IPersist, IPersist) /* First entry has to be a simple entry, otherwise ATL asserts */
COM_INTERFACE_ENTRY_FUNC_BLIND(0, FuncBlind) /* Showing that even after a Blind func, entryies can be found */
COM_INTERFACE_ENTRY_IID(IID_IStdMarshalInfo, IStdMarshalInfo)
END_COM_MAP()
};
class CDumExe: public CAtlExeModuleT<CDumExe>
{
};
CDumExe dum;
START_TEST(CComObject)
{
g_CTOR = g_DTOR = g_BLIND = 0;
CComObject<CTestObject>* pTest;
HRESULT hr = CComObject<CTestObject>::CreateInstance(&pTest);
ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr);
ok(g_CTOR == 1, "Expected 1, got %lu\n", g_CTOR);
ok(g_DTOR == 0, "Expected 0, got %lu\n", g_DTOR);
ok(g_BLIND == 0, "Expected 0, got %lu\n", g_BLIND);
if (hr == S_OK)
{
ULONG ref = pTest->AddRef();
ok(ref == 1, "Expected 1, got %lu\n", ref);
{
CComPtr<IUnknown> ppv;
hr = pTest->QueryInterface(IID_IUnknown, (void **) &ppv);
ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr);
ok(g_CTOR == 1, "Expected 1, got %lu\n", g_CTOR);
ok(g_DTOR == 0, "Expected 0, got %lu\n", g_DTOR);
ok(g_BLIND == 0, "Expected 0, got %lu\n", g_BLIND);
CComPtr<IPersist> ppersist;
hr = pTest->QueryInterface(IID_IPersist, (void **)&ppersist);
ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr);
ok(g_CTOR == 1, "Expected 1, got %lu\n", g_CTOR);
ok(g_DTOR == 0, "Expected 0, got %lu\n", g_DTOR);
ok(g_BLIND == 0, "Expected 0, got %lu\n", g_BLIND);
}
{
CComPtr<IStdMarshalInfo> pstd;
hr = pTest->QueryInterface(IID_IStdMarshalInfo, (void **)&pstd);
ok(hr == S_OK, "Expected S_OK, got 0x%lx\n", hr);
ok(g_CTOR == 1, "Expected 1, got %lu\n", g_CTOR);
ok(g_DTOR == 0, "Expected 0, got %lu\n", g_DTOR);
ok(g_BLIND == 1, "Expected 1, got %lu\n", g_BLIND);
}
ref = pTest->Release();
ok(ref == 0, "Expected 0, got %lu\n", ref);
ok(g_CTOR == 1, "Expected 1, got %lu\n", g_CTOR);
ok(g_DTOR == 1, "Expected 1, got %lu\n", g_DTOR);
ok(g_BLIND == 1, "Expected 1, got %lu\n", g_BLIND);
}
}