dlls/oledlg/tests/main.c

   1 /*
   2  * oledlg tests
   3  *
   4  * Copyright 2015 Nikolay Sivov for CodeWeavers
   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 COBJMACROS
  22
  23 #include "wine/test.h"
  24 #include <stdio.h>
  25
  26 #include "initguid.h"
  27 #include "oledlg.h"
  28
  29 static HRESULT WINAPI enumverbs_QueryInterface(IEnumOLEVERB *iface, REFIID riid, void **ppv)
  30 {
  31     if (IsEqualIID(riid, &IID_IEnumOLEVERB) || IsEqualIID(riid, &IID_IUnknown)) {
  32         *ppv = iface;
  33         IEnumOLEVERB_AddRef(iface);
  34         return S_OK;
  35     }
  36
  37     *ppv = NULL;
  38     return E_NOINTERFACE;
  39 }
  40
  41 static ULONG WINAPI enumverbs_AddRef(IEnumOLEVERB *iface)
  42 {
  43     return 2;
  44 }
  45
  46 static ULONG WINAPI enumverbs_Release(IEnumOLEVERB *iface)
  47 {
  48     return 1;
  49 }
  50
  51 static int g_enumpos;
  52 static const WCHAR verbW[] = {'v','e','r','b',0};
  53 static HRESULT WINAPI enumverbs_Next(IEnumOLEVERB *iface, ULONG count, OLEVERB *verbs, ULONG *fetched)
  54 {
  55     ok(count == 1, "got %u\n", count);
  56     ok(fetched == NULL, "got %p\n", fetched);
  57     ok(g_enumpos == 0 || g_enumpos == 1, "got pos %d\n", g_enumpos);
  58
  59     if (g_enumpos++ == 0) {
  60         verbs->lVerb = 123;
  61         verbs->lpszVerbName = CoTaskMemAlloc(sizeof(verbW));
  62         lstrcpyW(verbs->lpszVerbName, verbW);
  63         verbs->fuFlags = MF_ENABLED;
  64         verbs->grfAttribs = OLEVERBATTRIB_ONCONTAINERMENU;
  65         if (fetched) *fetched = 1;
  66         return S_OK;
  67     }
  68
  69     return S_FALSE;
  70 }
  71
  72 static HRESULT WINAPI enumverbs_Skip(IEnumOLEVERB *iface, ULONG count)
  73 {
  74     ok(0, "unexpected call\n");
  75     return E_NOTIMPL;
  76 }
  77
  78 static HRESULT WINAPI enumverbs_Reset(IEnumOLEVERB *iface)
  79 {
  80     ok(0, "unexpected call\n");
  81     return E_NOTIMPL;
  82 }
  83
  84 static HRESULT WINAPI enumverbs_Clone(IEnumOLEVERB *iface, IEnumOLEVERB **ppenum)
  85 {
  86     ok(0, "unexpected call\n");
  87     return E_NOTIMPL;
  88 }
  89
  90 static IEnumOLEVERBVtbl enumverbsvtbl = {
  91     enumverbs_QueryInterface,
  92     enumverbs_AddRef,
  93     enumverbs_Release,
  94     enumverbs_Next,
  95     enumverbs_Skip,
  96     enumverbs_Reset,
  97     enumverbs_Clone
  98 };
  99
 100 static IEnumOLEVERB enumverbs = { &enumverbsvtbl };
 101
 102 static HRESULT WINAPI oleobject_QueryInterface(IOleObject *iface, REFIID riid, void **ppv)
 103 {
 104     if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IOleObject)) {
 105         *ppv = iface;
 106         IOleObject_AddRef(iface);
 107         return S_OK;
 108     }
 109
 110     *ppv = NULL;
 111     return E_NOINTERFACE;
 112 }
 113
 114 static ULONG WINAPI oleobject_AddRef(IOleObject *iface)
 115 {
 116     return 2;
 117 }
 118
 119 static ULONG WINAPI oleobject_Release(IOleObject *iface)
 120 {
 121     return 1;
 122 }
 123
 124 static HRESULT WINAPI oleobject_SetClientSite(IOleObject *iface, IOleClientSite *site)
 125 {
 126     ok(0, "unexpected call\n");
 127     return E_NOTIMPL;
 128 }
 129
 130 static HRESULT WINAPI oleobject_GetClientSite(IOleObject *iface, IOleClientSite **site)
 131 {
 132     ok(0, "unexpected call\n");
 133     return E_NOTIMPL;
 134 }
 135
 136 static HRESULT WINAPI oleobject_SetHostNames(IOleObject *iface, LPCOLESTR containerapp,
 137     LPCOLESTR containerObj)
 138 {
 139     ok(0, "unexpected call\n");
 140     return E_NOTIMPL;
 141 }
 142
 143 static HRESULT WINAPI oleobject_Close(IOleObject *iface, DWORD saveopt)
 144 {
 145     ok(0, "unexpected call\n");
 146     return E_NOTIMPL;
 147 }
 148
 149 static HRESULT WINAPI oleobject_SetMoniker(IOleObject *iface, DWORD whichmoniker, IMoniker *mk)
 150 {
 151     ok(0, "unexpected call\n");
 152     return E_NOTIMPL;
 153 }
 154
 155 static HRESULT WINAPI oleobject_GetMoniker(IOleObject *iface, DWORD assign, DWORD whichmoniker,
 156     IMoniker **mk)
 157 {
 158     ok(0, "unexpected call\n");
 159     return E_NOTIMPL;
 160 }
 161
 162 static HRESULT WINAPI oleobject_InitFromData(IOleObject *iface, IDataObject *dataobject,
 163     BOOL creation, DWORD reserved)
 164 {
 165     ok(0, "unexpected call\n");
 166     return E_NOTIMPL;
 167 }
 168
 169 static HRESULT WINAPI oleobject_GetClipboardData(IOleObject *iface, DWORD reserved, IDataObject **dataobject)
 170 {
 171     ok(0, "unexpected call\n");
 172     return E_NOTIMPL;
 173 }
 174
 175 static HRESULT WINAPI oleobject_DoVerb(IOleObject *iface, LONG verb, MSG *msg, IOleClientSite *activesite,
 176     LONG index, HWND hwndParent, LPCRECT rect)
 177 {
 178     ok(0, "unexpected call\n");
 179     return E_NOTIMPL;
 180 }
 181
 182 static BOOL g_enumverbsfail;
 183 static HRESULT WINAPI oleobject_EnumVerbs(IOleObject *iface, IEnumOLEVERB **enumverb)
 184 {
 185     if (g_enumverbsfail) {
 186         *enumverb = NULL;
 187         return E_FAIL;
 188     }
 189     *enumverb = &enumverbs;
 190     return S_OK;
 191 }
 192
 193 static HRESULT WINAPI oleobject_Update(IOleObject *iface)
 194 {
 195     ok(0, "unexpected call\n");
 196     return E_NOTIMPL;
 197 }
 198
 199 static HRESULT WINAPI oleobject_IsUpToDate(IOleObject *iface)
 200 {
 201     ok(0, "unexpected call\n");
 202     return E_NOTIMPL;
 203 }
 204
 205 static HRESULT WINAPI oleobject_GetUserClassID(IOleObject *iface, CLSID *clsid)
 206 {
 207     ok(0, "unexpected call\n");
 208     return E_NOTIMPL;
 209 }
 210
 211 static const WCHAR testW[] = {'t','e','s','t',0};
 212 static HRESULT WINAPI oleobject_GetUserType(IOleObject *iface, DWORD formoftype,
 213     LPOLESTR *usertype)
 214 {
 215     ok(formoftype == USERCLASSTYPE_SHORT, "got %d\n", formoftype);
 216     *usertype = CoTaskMemAlloc(sizeof(testW));
 217     lstrcpyW(*usertype, testW);
 218     return S_OK;
 219 }
 220
 221 static HRESULT WINAPI oleobject_SetExtent(IOleObject *iface, DWORD aspect, SIZEL *size)
 222 {
 223     ok(0, "unexpected call\n");
 224     return E_NOTIMPL;
 225 }
 226
 227 static HRESULT WINAPI oleobject_GetExtent(IOleObject *iface, DWORD aspect, SIZEL *size)
 228 {
 229     ok(0, "unexpected call\n");
 230     return E_NOTIMPL;
 231 }
 232
 233 static HRESULT WINAPI oleobject_Advise(IOleObject *iface, IAdviseSink *sink, DWORD *connection)
 234 {
 235     ok(0, "unexpected call\n");
 236     return E_NOTIMPL;
 237 }
 238
 239 static HRESULT WINAPI oleobject_Unadvise(IOleObject *iface, DWORD connection)
 240 {
 241     ok(0, "unexpected call\n");
 242     return E_NOTIMPL;
 243 }
 244
 245 static HRESULT WINAPI oleobject_EnumAdvise(IOleObject *iface, IEnumSTATDATA **enumadvise)
 246 {
 247     ok(0, "unexpected call\n");
 248     return E_NOTIMPL;
 249 }
 250
 251 static HRESULT WINAPI oleobject_GetMiscStatus(IOleObject *iface, DWORD aspect, DWORD *status)
 252 {
 253     ok(0, "unexpected call\n");
 254     return E_NOTIMPL;
 255 }
 256
 257 static HRESULT WINAPI oleobject_SetColorScheme(IOleObject *iface, LOGPALETTE *pal)
 258 {
 259     ok(0, "unexpected call\n");
 260     return E_NOTIMPL;
 261 }
 262
 263 static IOleObjectVtbl oleobjectvtbl = {
 264     oleobject_QueryInterface,
 265     oleobject_AddRef,
 266     oleobject_Release,
 267     oleobject_SetClientSite,
 268     oleobject_GetClientSite,
 269     oleobject_SetHostNames,
 270     oleobject_Close,
 271     oleobject_SetMoniker,
 272     oleobject_GetMoniker,
 273     oleobject_InitFromData,
 274     oleobject_GetClipboardData,
 275     oleobject_DoVerb,
 276     oleobject_EnumVerbs,
 277     oleobject_Update,
 278     oleobject_IsUpToDate,
 279     oleobject_GetUserClassID,
 280     oleobject_GetUserType,
 281     oleobject_SetExtent,
 282     oleobject_GetExtent,
 283     oleobject_Advise,
 284     oleobject_Unadvise,
 285     oleobject_EnumAdvise,
 286     oleobject_GetMiscStatus,
 287     oleobject_SetColorScheme
 288 };
 289
 290 static IOleObject oleobject = { &oleobjectvtbl };
 291
 292 static void test_OleUIAddVerbMenu(void)
 293 {
 294     static const WCHAR cadabraW[] = {'c','a','d','a','b','r','a',0};
 295     HMENU hMenu, verbmenu;
 296     MENUITEMINFOW info;
 297     WCHAR buffW[50];
 298     int count;
 299     BOOL ret;
 300
 301     ret = OleUIAddVerbMenuW(NULL, NULL, NULL, 0, 0, 0, FALSE, 0, NULL);
 302     ok(!ret, "got %d\n", ret);
 303
 304     verbmenu = (HMENU)0xdeadbeef;
 305     ret = OleUIAddVerbMenuW(NULL, NULL, NULL, 0, 0, 0, FALSE, 0, &verbmenu);
 306     ok(!ret, "got %d\n", ret);
 307     ok(verbmenu == NULL, "got %p\n", verbmenu);
 308
 309     g_enumpos = 0;
 310     ret = OleUIAddVerbMenuW(&oleobject, NULL, NULL, 0, 0, 0, FALSE, 0, NULL);
 311     ok(!ret, "got %d\n", ret);
 312
 313     hMenu = CreatePopupMenu();
 314
 315     memset(&info, 0, sizeof(info));
 316     info.cbSize = sizeof(info);
 317     ret = InsertMenuItemW(hMenu, 0, TRUE, &info);
 318     ok(ret, "got %d\n", ret);
 319
 320     count = GetMenuItemCount(hMenu);
 321     ok(count == 1, "got %d\n", count);
 322
 323     g_enumpos = 0;
 324     ret = OleUIAddVerbMenuW(&oleobject, NULL, hMenu, 0, 0, 0, FALSE, 0, NULL);
 325     ok(!ret, "got %d\n", ret);
 326
 327     count = GetMenuItemCount(hMenu);
 328     ok(count == 1, "got %d\n", count);
 329
 330     ret = InsertMenuItemW(hMenu, 0, TRUE, &info);
 331     ok(ret, "got %d\n", ret);
 332
 333     count = GetMenuItemCount(hMenu);
 334     ok(count == 2, "got %d\n", count);
 335
 336     verbmenu = (HMENU)0xdeadbeef;
 337     g_enumpos = 0;
 338     ret = OleUIAddVerbMenuW(&oleobject, NULL, hMenu, 1, 0, 0, FALSE, 0, &verbmenu);
 339     ok(ret, "got %d\n", ret);
 340     ok(verbmenu == NULL, "got %p\n", verbmenu);
 341
 342     count = GetMenuItemCount(hMenu);
 343     ok(count == 2, "got %d\n", count);
 344
 345     /* object doesn't support EnumVerbs() */
 346     g_enumverbsfail = TRUE;
 347     g_enumpos = 0;
 348     verbmenu = (HMENU)0xdeadbeef;
 349     ret = OleUIAddVerbMenuW(&oleobject, NULL, hMenu, 2, 0, 0, FALSE, 0, &verbmenu);
 350     ok(!ret, "got %d\n", ret);
 351     ok(verbmenu == NULL, "got %p\n", verbmenu);
 352     g_enumverbsfail = FALSE;
 353
 354     /* added disabled item */
 355     memset(&info, 0, sizeof(info));
 356     info.cbSize = sizeof(info);
 357     info.fMask = MIIM_STATE|MIIM_SUBMENU;
 358     ret = GetMenuItemInfoW(hMenu, 2, TRUE, &info);
 359     ok(ret, "got %d\n", ret);
 360     ok(info.fState & MFS_DISABLED, "got state 0x%08x\n", info.fState);
 361     ok(info.hSubMenu == NULL, "got submenu %p\n", info.hSubMenu);
 362
 363     count = GetMenuItemCount(hMenu);
 364     ok(count == 3, "got %d\n", count);
 365
 366     /* now without object */
 367     verbmenu = (HMENU)0xdeadbeef;
 368     ret = OleUIAddVerbMenuW(NULL, testW, hMenu, 3, 42, 0, FALSE, 0, &verbmenu);
 369     ok(!ret, "got %d\n", ret);
 370     ok(verbmenu == NULL, "got %p\n", verbmenu);
 371
 372     memset(&info, 0, sizeof(info));
 373     info.cbSize = sizeof(info);
 374     info.fMask = MIIM_STATE|MIIM_ID|MIIM_STRING|MIIM_SUBMENU;
 375     info.dwTypeData = buffW;
 376     info.cch = ARRAY_SIZE(buffW);
 377     ret = GetMenuItemInfoW(hMenu, 3, TRUE, &info);
 378     ok(ret, "got %d\n", ret);
 379     ok(info.fState == MF_GRAYED, "got state 0x%08x\n", info.fState);
 380     ok(info.wID == 42, "got id %d\n", info.wID);
 381     ok(info.hSubMenu == NULL, "got submenu %p\n", info.hSubMenu);
 382
 383     count = GetMenuItemCount(hMenu);
 384     ok(count == 4, "got %d\n", count);
 385
 386     verbmenu = (HMENU)0xdeadbeef;
 387     g_enumpos = 0;
 388     ret = OleUIAddVerbMenuW(&oleobject, NULL, hMenu, 4, 0, 0, FALSE, 0, &verbmenu);
 389     ok(ret, "got %d\n", ret);
 390     ok(verbmenu == NULL, "got %p\n", verbmenu);
 391
 392     /* check newly added item */
 393     memset(&info, 0, sizeof(info));
 394     info.cbSize = sizeof(info);
 395     info.fMask = MIIM_STRING|MIIM_STATE|MIIM_SUBMENU;
 396     info.dwTypeData = buffW;
 397     info.cch = ARRAY_SIZE(buffW);
 398     ret = GetMenuItemInfoW(hMenu, 4, TRUE, &info);
 399     ok(ret, "got %d\n", ret);
 400     /* Item string contains verb, usertype and localized string for 'Object' word,
 401        exact format depends on localization. */
 402     ok(wcsstr(buffW, verbW) != NULL, "str %s\n", wine_dbgstr_w(buffW));
 403     ok(info.fState == 0, "got state 0x%08x\n", info.fState);
 404     ok(info.hSubMenu == NULL, "got submenu %p\n", info.hSubMenu);
 405
 406     count = GetMenuItemCount(hMenu);
 407     ok(count == 5, "got %d\n", count);
 408
 409     DestroyMenu(hMenu);
 410
 411     /* try to add verb menu repeatedly, with same id */
 412     hMenu = CreatePopupMenu();
 413
 414     count = GetMenuItemCount(hMenu);
 415     ok(count == 0, "got %d\n", count);
 416
 417     verbmenu = NULL;
 418     ret = OleUIAddVerbMenuW(NULL, NULL, hMenu, 0, 5, 10, TRUE, 3, &verbmenu);
 419     ok(!ret, "got %d\n", ret);
 420     ok(verbmenu == NULL, "got %p\n", verbmenu);
 421
 422     count = GetMenuItemCount(hMenu);
 423     ok(count == 1, "got %d\n", count);
 424
 425     verbmenu = NULL;
 426     ret = OleUIAddVerbMenuW(NULL, NULL, hMenu, 0, 5, 10, TRUE, 3, &verbmenu);
 427     ok(!ret, "got %d\n", ret);
 428     ok(verbmenu == NULL, "got %p\n", verbmenu);
 429
 430     count = GetMenuItemCount(hMenu);
 431     ok(count == 1, "got %d\n", count);
 432
 433     /* same position, different id */
 434     verbmenu = NULL;
 435     ret = OleUIAddVerbMenuW(NULL, NULL, hMenu, 0, 6, 10, TRUE, 3, &verbmenu);
 436     ok(!ret, "got %d\n", ret);
 437     ok(verbmenu == NULL, "got %p\n", verbmenu);
 438
 439     count = GetMenuItemCount(hMenu);
 440     ok(count == 1, "got %d\n", count);
 441
 442     /* change added item string and state */
 443     memset(&info, 0, sizeof(info));
 444     info.cbSize = sizeof(info);
 445     info.fMask = MIIM_STRING|MIIM_STATE;
 446     info.fState = MFS_ENABLED;
 447     info.dwTypeData = buffW;
 448     lstrcpyW(buffW, cadabraW);
 449     ret = SetMenuItemInfoW(hMenu, 0, TRUE, &info);
 450     ok(ret, "got %d\n", ret);
 451
 452     buffW[0] = 0;
 453     GetMenuStringW(hMenu, 0, buffW, ARRAY_SIZE(buffW), MF_BYPOSITION);
 454     ok(!lstrcmpW(buffW, cadabraW), "got %s\n", wine_dbgstr_w(buffW));
 455
 456     verbmenu = NULL;
 457     ret = OleUIAddVerbMenuW(NULL, NULL, hMenu, 0, 5, 10, TRUE, 3, &verbmenu);
 458     ok(!ret, "got %d\n", ret);
 459     ok(verbmenu == NULL, "got %p\n", verbmenu);
 460
 461     memset(&info, 0, sizeof(info));
 462     info.cbSize = sizeof(info);
 463     info.fMask = MIIM_STRING|MIIM_STATE;
 464     buffW[0] = 0;
 465     info.dwTypeData = buffW;
 466     info.cch = ARRAY_SIZE(buffW);
 467     ret = GetMenuItemInfoW(hMenu, 0, TRUE, &info);
 468     ok(ret, "got %d\n", ret);
 469     ok(lstrcmpW(buffW, cadabraW), "got %s\n", wine_dbgstr_w(buffW));
 470     ok(info.fState == MF_GRAYED, "got state 0x%08x\n", info.fState);
 471
 472     count = GetMenuItemCount(hMenu);
 473     ok(count == 1, "got %d\n", count);
 474
 475     DestroyMenu(hMenu);
 476 }
 477
 478 START_TEST(main)
 479 {
 480     test_OleUIAddVerbMenu();
 481 }