Something went wrong. Try again.
Reactos
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125/* * 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()};
#ifndef __RATL__ // Avoid conflict (causing assertion at CAtlModule::CAtlModule)class CDumExe: public CAtlExeModuleT<CDumExe>{
};CDumExe dum;#endif
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); }}