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