dlls/ole32/tests/dragdrop.c

   1 /*
   2  * Drag and Drop Tests
   3  *
   4  * Copyright 2007 Robert Shearman
   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 #define _WIN32_DCOM
  22 #define COBJMACROS
  23 #define CONST_VTABLE
  24
  25 #include <stdarg.h>
  26 #include <stdio.h>
  27
  28 #include "windef.h"
  29 #include "winbase.h"
  30 #include "objbase.h"
  31
  32 #include "wine/test.h"
  33
  34 /* functions that are not present on all versions of Windows */
  35 HRESULT (WINAPI * pCoInitializeEx)(LPVOID lpReserved, DWORD dwCoInit);
  36
  37 static int droptarget_addref_called;
  38 static int droptarget_release_called;
  39
  40 /* helper macros to make tests a bit leaner */
  41 #define ok_ole_success(hr, func) ok(hr == S_OK, func " failed with error 0x%08x\n", hr)
  42
  43 static HRESULT WINAPI DropTarget_QueryInterface(IDropTarget* iface, REFIID riid,
  44                                                 void** ppvObject)
  45 {
  46     trace("DropTarget_QueryInterface\n");
  47     if (IsEqualIID(riid, &IID_IUnknown) ||
  48         IsEqualIID(riid, &IID_IDropTarget))
  49     {
  50         IUnknown_AddRef(iface);
  51         *ppvObject = iface;
  52         return S_OK;
  53     }
  54     *ppvObject = NULL;
  55     return E_NOINTERFACE;
  56 }
  57
  58 static ULONG WINAPI DropTarget_AddRef(IDropTarget* iface)
  59 {
  60     droptarget_addref_called++;
  61     return 2;
  62 }
  63
  64 static ULONG WINAPI DropTarget_Release(IDropTarget* iface)
  65 {
  66     droptarget_release_called++;
  67     return 1;
  68 }
  69
  70 static HRESULT WINAPI DropTarget_DragEnter(IDropTarget* iface,
  71                                            IDataObject* pDataObj,
  72                                            DWORD grfKeyState, POINTL pt,
  73                                            DWORD* pdwEffect)
  74 {
  75     return E_NOTIMPL;
  76 }
  77
  78 static HRESULT WINAPI DropTarget_DragOver(IDropTarget* iface,
  79                                           DWORD grfKeyState,
  80                                           POINTL pt,
  81                                           DWORD* pdwEffect)
  82 {
  83     return E_NOTIMPL;
  84 }
  85
  86 static HRESULT WINAPI DropTarget_DragLeave(IDropTarget* iface)
  87 {
  88     return E_NOTIMPL;
  89 }
  90
  91 static HRESULT WINAPI DropTarget_Drop(IDropTarget* iface,
  92                                       IDataObject* pDataObj, DWORD grfKeyState,
  93                                       POINTL pt, DWORD* pdwEffect)
  94 {
  95     return E_NOTIMPL;
  96 }
  97
  98 static const IDropTargetVtbl DropTarget_VTbl =
  99 {
 100     DropTarget_QueryInterface,
 101     DropTarget_AddRef,
 102     DropTarget_Release,
 103     DropTarget_DragEnter,
 104     DropTarget_DragOver,
 105     DropTarget_DragLeave,
 106     DropTarget_Drop
 107 };
 108
 109 static IDropTarget DropTarget = { &DropTarget_VTbl };
 110
 111 START_TEST(dragdrop)
 112 {
 113     HRESULT hr;
 114
 115     hr = RegisterDragDrop(GetDesktopWindow(), &DropTarget);
 116     ok(hr == CO_E_NOTINITIALIZED, "RegisterDragDrop without OLE initialized should have returned CO_E_NOTINITIALIZED instead of 0x%08x\n", hr);
 117
 118     OleInitialize(NULL);
 119
 120     hr = RegisterDragDrop(GetDesktopWindow(), NULL);
 121     ok(hr == E_INVALIDARG, "RegisterDragDrop with NULL IDropTarget * should return E_INVALIDARG instead of 0x%08x\n", hr);
 122
 123     hr = RegisterDragDrop(NULL, &DropTarget);
 124     ok(hr == DRAGDROP_E_INVALIDHWND, "RegisterDragDrop with NULL hwnd should return DRAGDROP_E_INVALIDHWND instead of 0x%08x\n", hr);
 125
 126     hr = RegisterDragDrop((HWND)0xdeadbeef, &DropTarget);
 127     ok(hr == DRAGDROP_E_INVALIDHWND, "RegisterDragDrop with garbage hwnd should return DRAGDROP_E_INVALIDHWND instead of 0x%08x\n", hr);
 128
 129     ok(droptarget_addref_called == 0, "DropTarget_AddRef shouldn't have been called\n");
 130     hr = RegisterDragDrop(GetDesktopWindow(), &DropTarget);
 131     ok_ole_success(hr, "RegisterDragDrop");
 132     ok(droptarget_addref_called == 1, "DropTarget_AddRef should have been called once, not %d times\n", droptarget_addref_called);
 133
 134     hr = RegisterDragDrop(GetDesktopWindow(), &DropTarget);
 135     ok(hr == DRAGDROP_E_ALREADYREGISTERED, "RegisterDragDrop with already registered hwnd should return DRAGDROP_E_ALREADYREGISTERED instead of 0x%08x\n", hr);
 136
 137     ok(droptarget_release_called == 0, "DropTarget_Release shouldn't have been called\n");
 138     OleUninitialize();
 139     ok(droptarget_release_called == 0, "DropTarget_Release shouldn't have been called\n");
 140
 141     hr = RevokeDragDrop(GetDesktopWindow());
 142     ok_ole_success(hr, "RevokeDragDrop");
 143     ok(droptarget_release_called == 1, "DropTarget_Release should have been called once, not %d times\n", droptarget_release_called);
 144
 145     hr = RevokeDragDrop(NULL);
 146     ok(hr == DRAGDROP_E_INVALIDHWND, "RevokeDragDrop with NULL hwnd should return DRAGDROP_E_INVALIDHWND instead of 0x%08x\n", hr);
 147 }