atl: Use atlbase.h in tests.

This commit is contained in:
Jacek Caban 2012-12-26 13:49:50 +01:00 committed by Alexandre Julliard
parent 6e485f13c0
commit 65493cf137
1 changed files with 22 additions and 65 deletions

View File

@ -24,84 +24,41 @@
#define COBJMACROS #define COBJMACROS
#include <atlbase.h>
#include <wine/test.h> #include <wine/test.h>
#include <windef.h>
#include <winbase.h>
#include <winuser.h>
#include <wingdi.h>
#include <winnls.h>
#include <winerror.h>
#include <winnt.h>
#include <wtypes.h>
#include <olectl.h>
#include <ocidl.h>
struct _ATL_OBJMAP_ENTRYW;
struct _AtlCreateWndData;
struct _ATL_TERMFUNC_ELEM;
struct _ATL_MODULEW
{
UINT cbSize;
HINSTANCE m_hInst;
HINSTANCE m_hInstResource;
HINSTANCE m_hInstTypeLib;
struct _ATL_OBJMAP_ENTRYW* m_pObjMap;
LONG m_nLockCnt;
HANDLE m_hHeap;
union
{
CRITICAL_SECTION m_csTypeInfoHolder;
CRITICAL_SECTION m_csStaticDataInit;
} u;
CRITICAL_SECTION m_csWindowCreate;
CRITICAL_SECTION m_csObjMap;
DWORD dwAtlBuildVer;
struct _AtlCreateWndData* m_pCreateWndList;
BOOL m_bDestroyHeap;
GUID* pguidVer;
DWORD m_dwHeaps;
HANDLE* m_phHeaps;
int m_nHeap;
struct _ATL_TERMFUNC_ELEM* m_pTermFuncs;
};
HRESULT WINAPI AtlModuleInit(struct _ATL_MODULEW* pM, struct _ATL_OBJMAP_ENTRYW* p, HINSTANCE h);
#define MAXSIZE 512 #define MAXSIZE 512
static void test_StructSize(void) static void test_StructSize(void)
{ {
struct _ATL_MODULEW *tst; _ATL_MODULEW *tst;
HRESULT hres; HRESULT hres;
int i; int i;
tst = HeapAlloc (GetProcessHeap(),HEAP_ZERO_MEMORY,MAXSIZE); tst = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, MAXSIZE);
for (i=0;i<MAXSIZE;i++) { for (i=0;i<MAXSIZE;i++) {
tst->cbSize = i; tst->cbSize = i;
hres = AtlModuleInit(tst, NULL, NULL); hres = AtlModuleInit(tst, NULL, NULL);
switch (i) { switch (i) {
case FIELD_OFFSET( struct _ATL_MODULEW, dwAtlBuildVer ): case FIELD_OFFSET(_ATL_MODULEW, dwAtlBuildVer):
case sizeof(struct _ATL_MODULEW): case sizeof(_ATL_MODULEW):
#ifdef _WIN64 #ifdef _WIN64
case sizeof(struct _ATL_MODULEW) + sizeof(void *): case sizeof(_ATL_MODULEW) + sizeof(void *):
#endif #endif
ok (hres == S_OK, "AtlModuleInit with %d failed (0x%x).\n", i, (int)hres); ok (hres == S_OK, "AtlModuleInit with %d failed (0x%x).\n", i, (int)hres);
break; break;
default: default:
ok (FAILED(hres) || ok (FAILED(hres), "AtlModuleInit with %d succeeded? (0x%x).\n", i, (int)hres);
broken((i > FIELD_OFFSET( struct _ATL_MODULEW, dwAtlBuildVer )) && (hres == S_OK)), /* Win95 */ break;
"AtlModuleInit with %d succeeded? (0x%x).\n", i, (int)hres); }
break; }
}
}
HeapFree (GetProcessHeap(), 0, tst); HeapFree (GetProcessHeap(), 0, tst);
} }
START_TEST(module) START_TEST(module)
{ {
test_StructSize(); test_StructSize();
} }