dlls/atl/tests/atl_ax.c

   1 /*
   2  * Unit tests for Active Template Library ActiveX functions
   3  *
   4  * Copyright 2010 Andrew Nguyen
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22 #include <stdio.h>
  23
  24 #define COBJMACROS
  25
  26 #include <wine/test.h>
  27 #include <windef.h>
  28 #include <winbase.h>
  29 #include <winuser.h>
  30 #include <wingdi.h>
  31 #include <winnls.h>
  32 #include <winerror.h>
  33 #include <winnt.h>
  34 #include <wtypes.h>
  35 #include <olectl.h>
  36 #include <ocidl.h>
  37 #include <exdisp.h>
  38 #include <atlbase.h>
  39
  40 static HRESULT (WINAPI *pAtlAxAttachControl)(IUnknown *, HWND, IUnknown **);
  41
  42 static void init_function_pointers(void)
  43 {
  44     HMODULE hatl = GetModuleHandleA("atl.dll");
  45
  46     pAtlAxAttachControl = (void *)GetProcAddress(hatl, "AtlAxAttachControl");
  47 }
  48
  49 static ATOM register_class(void)
  50 {
  51     WNDCLASSA wndclassA;
  52
  53     wndclassA.style = 0;
  54     wndclassA.lpfnWndProc = DefWindowProcA;
  55     wndclassA.cbClsExtra = 0;
  56     wndclassA.cbWndExtra = 0;
  57     wndclassA.hInstance = GetModuleHandleA(NULL);
  58     wndclassA.hIcon = NULL;
  59     wndclassA.hCursor = LoadCursorA(NULL, (LPSTR)IDC_ARROW);
  60     wndclassA.hbrBackground = (HBRUSH)(COLOR_BTNFACE+1);
  61     wndclassA.lpszMenuName = NULL;
  62     wndclassA.lpszClassName = "WineAtlTestClass";
  63
  64     return RegisterClassA(&wndclassA);
  65 }
  66
  67 static void test_AtlAxAttachControl(void)
  68 {
  69     HWND hwnd = CreateWindowA("WineAtlTestClass", "Wine ATL Test Window", 0,
  70                               CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
  71                               CW_USEDEFAULT, NULL, NULL, NULL, NULL);
  72     HRESULT hr;
  73     IUnknown *pObj, *pContainer;
  74
  75     hr = pAtlAxAttachControl(NULL, NULL, NULL);
  76     ok(hr == E_INVALIDARG, "Expected AtlAxAttachControl to return E_INVALIDARG, got 0x%08x\n", hr);
  77
  78     pContainer = (IUnknown *)0xdeadbeef;
  79     hr = pAtlAxAttachControl(NULL, NULL, &pContainer);
  80     ok(hr == E_INVALIDARG, "Expected AtlAxAttachControl to return E_INVALIDARG, got 0x%08x\n", hr);
  81     ok(pContainer == (IUnknown *)0xdeadbeef,
  82        "Expected the output container pointer to be untouched, got %p\n", pContainer);
  83
  84     hr = pAtlAxAttachControl(NULL, hwnd, NULL);
  85     ok(hr == E_INVALIDARG, "Expected AtlAxAttachControl to return E_INVALIDARG, got 0x%08x\n", hr);
  86
  87     hr = CoCreateInstance(&CLSID_WebBrowser, NULL, CLSCTX_INPROC_SERVER | CLSCTX_INPROC_HANDLER,
  88                           &IID_IOleObject, (void **)&pObj);
  89     ok(hr == S_OK, "Expected CoCreateInstance to return S_OK, got 0x%08x\n", hr);
  90
  91     if (FAILED(hr))
  92     {
  93         skip("Couldn't obtain a test IOleObject instance\n");
  94         return;
  95     }
  96
  97     hr = pAtlAxAttachControl(pObj, NULL, NULL);
  98     ok(hr == S_FALSE, "Expected AtlAxAttachControl to return S_FALSE, got 0x%08x\n", hr);
  99
 100     pContainer = NULL;
 101     hr = pAtlAxAttachControl(pObj, NULL, &pContainer);
 102     ok(hr == S_FALSE, "Expected AtlAxAttachControl to return S_FALSE, got 0x%08x\n", hr);
 103     ok(pContainer != NULL, "got %p\n", pContainer);
 104     IUnknown_Release(pContainer);
 105
 106     hr = pAtlAxAttachControl(pObj, hwnd, NULL);
 107     ok(hr == S_OK, "Expected AtlAxAttachControl to return S_OK, got 0x%08x\n", hr);
 108
 109     IUnknown_Release(pObj);
 110
 111     DestroyWindow(hwnd);
 112 }
 113
 114 static void test_ax_win(void)
 115 {
 116     BOOL ret;
 117     WNDCLASSEXW wcex;
 118     static const WCHAR AtlAxWin[] = {'A','t','l','A','x','W','i','n',0};
 119     static HMODULE hinstance = 0;
 120
 121     ret = AtlAxWinInit();
 122     ok(ret, "AtlAxWinInit failed\n");
 123
 124     hinstance = GetModuleHandleA(NULL);
 125     memset(&wcex, 0, sizeof(wcex));
 126     wcex.cbSize = sizeof(wcex);
 127     ret = GetClassInfoExW(hinstance, AtlAxWin, &wcex);
 128     ok(ret, "AtlAxWin has not registered\n");
 129     ok(wcex.style == CS_GLOBALCLASS, "wcex.style %08x\n", wcex.style);
 130 }
 131
 132 START_TEST(atl_ax)
 133 {
 134     init_function_pointers();
 135
 136     if (!register_class())
 137         return;
 138
 139     CoInitialize(NULL);
 140
 141     if (pAtlAxAttachControl)
 142         test_AtlAxAttachControl();
 143     else
 144         win_skip("AtlAxAttachControl is not available\n");
 145
 146     test_ax_win();
 147
 148     CoUninitialize();
 149 }