winealsa: Unify the checks for wBitsPerSample.
[wine] / dlls / ieframe / tests / webbrowser.c
1 /*
2  * Copyright 2006 Jacek Caban for CodeWeavers
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #define COBJMACROS
20 #define CONST_VTABLE
21
22 #include <wine/test.h>
23 #include <stdarg.h>
24 #include <stdio.h>
25
26 #include "windef.h"
27 #include "winbase.h"
28 #include "initguid.h"
29 #include "ole2.h"
30 #include "exdisp.h"
31 #include "htiframe.h"
32 #include "mshtmhst.h"
33 #include "mshtmcid.h"
34 #include "mshtml.h"
35 #include "idispids.h"
36 #include "olectl.h"
37 #include "mshtmdid.h"
38 #include "shobjidl.h"
39 #include "shdeprecated.h"
40 #include "shlguid.h"
41 #include "exdispid.h"
42 #include "mimeinfo.h"
43 #include "hlink.h"
44 #include "docobjectservice.h"
45
46 DEFINE_GUID(GUID_NULL,0,0,0,0,0,0,0,0,0,0,0);
47 DEFINE_OLEGUID(CGID_DocHostCmdPriv, 0x000214D4L, 0, 0);
48
49 #define DEFINE_EXPECT(func) \
50     static BOOL expect_ ## func = FALSE, called_ ## func = FALSE
51
52 #define SET_EXPECT(func) \
53     expect_ ## func = TRUE
54
55 #define CHECK_EXPECT2(func) \
56     do { \
57         ok(expect_ ##func, "unexpected call " #func "\n"); \
58         called_ ## func = TRUE; \
59     }while(0)
60
61 #define CHECK_EXPECT(func) \
62     do { \
63         CHECK_EXPECT2(func); \
64         expect_ ## func = FALSE; \
65     }while(0)
66
67 #define CHECK_CALLED(func) \
68     do { \
69         ok(called_ ## func, "expected " #func "\n"); \
70         expect_ ## func = called_ ## func = FALSE; \
71     }while(0)
72
73 #define CLEAR_CALLED(func) \
74     expect_ ## func = called_ ## func = FALSE
75
76 DEFINE_EXPECT(GetContainer);
77 DEFINE_EXPECT(Site_GetWindow);
78 DEFINE_EXPECT(ShowObject);
79 DEFINE_EXPECT(CanInPlaceActivate);
80 DEFINE_EXPECT(OnInPlaceActivate);
81 DEFINE_EXPECT(OnUIActivate);
82 DEFINE_EXPECT(GetWindowContext);
83 DEFINE_EXPECT(Frame_GetWindow);
84 DEFINE_EXPECT(Frame_SetActiveObject);
85 DEFINE_EXPECT(UIWindow_SetActiveObject);
86 DEFINE_EXPECT(SetMenu);
87 DEFINE_EXPECT(Invoke_AMBIENT_USERMODE);
88 DEFINE_EXPECT(Invoke_AMBIENT_DLCONTROL);
89 DEFINE_EXPECT(Invoke_AMBIENT_USERAGENT);
90 DEFINE_EXPECT(Invoke_AMBIENT_PALETTE);
91 DEFINE_EXPECT(Invoke_AMBIENT_SILENT);
92 DEFINE_EXPECT(Invoke_AMBIENT_OFFLINEIFNOTCONNECTED);
93 DEFINE_EXPECT(Invoke_STATUSTEXTCHANGE);
94 DEFINE_EXPECT(Invoke_PROPERTYCHANGE);
95 DEFINE_EXPECT(Invoke_DOWNLOADBEGIN);
96 DEFINE_EXPECT(Invoke_BEFORENAVIGATE2);
97 DEFINE_EXPECT(Invoke_SETSECURELOCKICON);
98 DEFINE_EXPECT(Invoke_FILEDOWNLOAD);
99 DEFINE_EXPECT(Invoke_COMMANDSTATECHANGE);
100 DEFINE_EXPECT(Invoke_DOWNLOADCOMPLETE);
101 DEFINE_EXPECT(Invoke_ONMENUBAR);
102 DEFINE_EXPECT(Invoke_ONADDRESSBAR);
103 DEFINE_EXPECT(Invoke_ONSTATUSBAR);
104 DEFINE_EXPECT(Invoke_ONTOOLBAR);
105 DEFINE_EXPECT(Invoke_ONFULLSCREEN);
106 DEFINE_EXPECT(Invoke_ONTHEATERMODE);
107 DEFINE_EXPECT(Invoke_WINDOWSETRESIZABLE);
108 DEFINE_EXPECT(Invoke_TITLECHANGE);
109 DEFINE_EXPECT(Invoke_NAVIGATECOMPLETE2);
110 DEFINE_EXPECT(Invoke_PROGRESSCHANGE);
111 DEFINE_EXPECT(Invoke_DOCUMENTCOMPLETE);
112 DEFINE_EXPECT(Invoke_282);
113 DEFINE_EXPECT(EnableModeless_TRUE);
114 DEFINE_EXPECT(EnableModeless_FALSE);
115 DEFINE_EXPECT(GetHostInfo);
116 DEFINE_EXPECT(GetOptionKeyPath);
117 DEFINE_EXPECT(GetOverridesKeyPath);
118 DEFINE_EXPECT(SetStatusText);
119 DEFINE_EXPECT(UpdateUI);
120 DEFINE_EXPECT(Exec_SETDOWNLOADSTATE_0);
121 DEFINE_EXPECT(Exec_SETDOWNLOADSTATE_1);
122 DEFINE_EXPECT(Exec_SETPROGRESSMAX);
123 DEFINE_EXPECT(Exec_SETPROGRESSPOS);
124 DEFINE_EXPECT(Exec_UPDATECOMMANDS);
125 DEFINE_EXPECT(Exec_SETTITLE);
126 DEFINE_EXPECT(QueryStatus_SETPROGRESSTEXT);
127 DEFINE_EXPECT(Exec_STOP);
128 DEFINE_EXPECT(Exec_IDM_STOP);
129 DEFINE_EXPECT(QueryStatus_STOP);
130 DEFINE_EXPECT(QueryStatus_IDM_STOP);
131 DEFINE_EXPECT(DocHost_EnableModeless_TRUE);
132 DEFINE_EXPECT(DocHost_EnableModeless_FALSE);
133 DEFINE_EXPECT(DocHost_TranslateAccelerator);
134 DEFINE_EXPECT(GetDropTarget);
135 DEFINE_EXPECT(TranslateUrl);
136 DEFINE_EXPECT(ShowUI);
137 DEFINE_EXPECT(HideUI);
138 DEFINE_EXPECT(OnUIDeactivate);
139 DEFINE_EXPECT(OnInPlaceDeactivate);
140 DEFINE_EXPECT(RequestUIActivate);
141 DEFINE_EXPECT(ControlSite_TranslateAccelerator);
142 DEFINE_EXPECT(OnFocus);
143
144 static const WCHAR wszItem[] = {'i','t','e','m',0};
145 static const WCHAR emptyW[] = {0};
146
147 static VARIANT_BOOL exvb;
148
149 static IWebBrowser2 *wb;
150
151 static HWND container_hwnd, shell_embedding_hwnd;
152 static BOOL is_downloading, is_first_load, use_container_olecmd, test_close;
153 static HRESULT hr_dochost_TranslateAccelerator = E_NOTIMPL;
154 static HRESULT hr_site_TranslateAccelerator = E_NOTIMPL;
155 static const char *current_url;
156
157 #define DWL_EXPECT_BEFORE_NAVIGATE  0x01
158 #define DWL_FROM_PUT_HREF           0x02
159
160 static DWORD dwl_flags;
161
162
163 /* Returns true if the user interface is in English. Note that this does not
164  * presume of the formatting of dates, numbers, etc.
165  */
166 static BOOL is_lang_english(void)
167 {
168     static HMODULE hkernel32 = NULL;
169     static LANGID (WINAPI *pGetThreadUILanguage)(void) = NULL;
170     static LANGID (WINAPI *pGetUserDefaultUILanguage)(void) = NULL;
171
172     if (!hkernel32)
173     {
174         hkernel32 = GetModuleHandleA("kernel32.dll");
175         pGetThreadUILanguage = (void*)GetProcAddress(hkernel32, "GetThreadUILanguage");
176         pGetUserDefaultUILanguage = (void*)GetProcAddress(hkernel32, "GetUserDefaultUILanguage");
177     }
178     if (pGetThreadUILanguage)
179         return PRIMARYLANGID(pGetThreadUILanguage()) == LANG_ENGLISH;
180     if (pGetUserDefaultUILanguage)
181         return PRIMARYLANGID(pGetUserDefaultUILanguage()) == LANG_ENGLISH;
182
183     return PRIMARYLANGID(GetUserDefaultLangID()) == LANG_ENGLISH;
184 }
185
186 static int strcmp_wa(LPCWSTR strw, const char *stra)
187 {
188     CHAR buf[512];
189     WideCharToMultiByte(CP_ACP, 0, strw, -1, buf, sizeof(buf), NULL, NULL);
190     return lstrcmpA(stra, buf);
191 }
192
193 static BOOL iface_cmp(IUnknown *iface1, IUnknown *iface2)
194 {
195     IUnknown *unk1, *unk2;
196
197     if(iface1 == iface2)
198         return TRUE;
199
200     IUnknown_QueryInterface(iface1, &IID_IUnknown, (void**)&unk1);
201     IUnknown_Release(unk1);
202     IUnknown_QueryInterface(iface2, &IID_IUnknown, (void**)&unk2);
203     IUnknown_Release(unk2);
204
205     return unk1 == unk2;
206 }
207
208 static const char *debugstr_guid(REFIID riid)
209 {
210     static char buf[50];
211
212     sprintf(buf, "{%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X}",
213             riid->Data1, riid->Data2, riid->Data3, riid->Data4[0],
214             riid->Data4[1], riid->Data4[2], riid->Data4[3], riid->Data4[4],
215             riid->Data4[5], riid->Data4[6], riid->Data4[7]);
216
217     return buf;
218 }
219
220 static BSTR a2bstr(const char *str)
221 {
222     BSTR ret;
223     int len;
224
225     len = MultiByteToWideChar(CP_ACP, 0, str, -1, NULL, 0);
226     ret = SysAllocStringLen(NULL, len);
227     MultiByteToWideChar(CP_ACP, 0, str, -1, ret, len);
228
229     return ret;
230 }
231
232 #define create_WebBrowser(a) _create_WebBrowser(__LINE__,a)
233 static HRESULT _create_WebBrowser(unsigned line, IUnknown **unk)
234 {
235     HRESULT hres;
236
237     hres = CoCreateInstance(&CLSID_WebBrowser, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
238             &IID_IUnknown, (void**)unk);
239     ok_(__FILE__,line)(hres == S_OK, "Creating WebBrowser object failed: %08x\n", hres);
240     return hres;
241 }
242
243 #define test_LocationURL(a,b) _test_LocationURL(__LINE__,a,b)
244 static void _test_LocationURL(unsigned line, IUnknown *unk, const char *exurl)
245 {
246     IWebBrowser2 *wb;
247     BSTR url = (void*)0xdeadbeef;
248     HRESULT hres;
249
250     hres = IUnknown_QueryInterface(unk, &IID_IWebBrowser2, (void**)&wb);
251     ok(hres == S_OK, "Could not get IWebBrowser2 interface: %08x\n", hres);
252     if(FAILED(hres))
253         return;
254
255     hres = IWebBrowser2_get_LocationURL(wb, &url);
256     ok_(__FILE__,line) (hres == (*exurl ? S_OK : S_FALSE), "get_LocationURL failed: %08x\n", hres);
257     if (hres == S_OK)
258     {
259         ok_(__FILE__,line) (!strcmp_wa(url, exurl), "unexpected URL: %s\n", wine_dbgstr_w(url));
260         SysFreeString(url);
261     }
262
263     IWebBrowser2_Release(wb);
264 }
265
266 #define test_ready_state(ex) _test_ready_state(__LINE__,ex);
267 static void _test_ready_state(unsigned line, READYSTATE exstate)
268 {
269     READYSTATE state;
270     HRESULT hres;
271
272     hres = IWebBrowser2_get_ReadyState(wb, &state);
273     ok_(__FILE__,line)(hres == S_OK, "get_ReadyState failed: %08x\n", hres);
274     ok_(__FILE__,line)(state == exstate, "ReadyState = %d, expected %d\n", state, exstate);
275 }
276
277 #define get_document(u) _get_document(__LINE__,u)
278 static IDispatch *_get_document(unsigned line, IUnknown *unk)
279 {
280     IHTMLDocument2 *html_doc;
281     IWebBrowser2 *wb;
282     IDispatch *disp;
283     HRESULT hres;
284
285     hres = IUnknown_QueryInterface(unk, &IID_IWebBrowser2, (void**)&wb);
286     ok_(__FILE__,line)(hres == S_OK, "QueryInterface(IID_IWebBrowser2) failed: %08x\n", hres);
287
288     disp = NULL;
289     hres = IWebBrowser2_get_Document(wb, &disp);
290     IWebBrowser2_Release(wb);
291     ok_(__FILE__,line)(hres == S_OK, "get_Document failed: %08x\n", hres);
292     ok_(__FILE__,line)(disp != NULL, "doc_disp == NULL\n");
293
294     hres = IDispatch_QueryInterface(disp, &IID_IHTMLDocument2, (void**)&html_doc);
295     ok_(__FILE__,line)(hres == S_OK, "Could not get IHTMLDocument iface: %08x\n", hres);
296     ok(disp == (IDispatch*)html_doc, "disp != html_doc\n");
297     IHTMLDocument_Release(html_doc);
298
299     return disp;
300 }
301
302 #define get_dochost(u) _get_dochost(__LINE__,u)
303 static IOleClientSite *_get_dochost(unsigned line, IUnknown *unk)
304 {
305     IOleClientSite *client_site;
306     IOleObject *oleobj;
307     IDispatch *doc;
308     HRESULT hres;
309
310     doc = _get_document(line, unk);
311     hres = IDispatch_QueryInterface(doc, &IID_IOleObject, (void**)&oleobj);
312     IDispatch_Release(doc);
313     ok_(__FILE__,line)(hres == S_OK, "Got 0x%08x\n", hres);
314
315     hres = IOleObject_GetClientSite(oleobj, &client_site);
316     IOleObject_Release(oleobj);
317     ok_(__FILE__,line)(hres == S_OK, "Got 0x%08x\n", hres);
318
319     return client_site;
320 }
321
322 static HRESULT QueryInterface(REFIID,void**);
323
324 static HRESULT WINAPI OleCommandTarget_QueryInterface(IOleCommandTarget *iface,
325         REFIID riid, void **ppv)
326 {
327     ok(0, "unexpected call\n");
328     return E_NOINTERFACE;
329 }
330
331 static ULONG WINAPI OleCommandTarget_AddRef(IOleCommandTarget *iface)
332 {
333     return 2;
334 }
335
336 static ULONG WINAPI OleCommandTarget_Release(IOleCommandTarget *iface)
337 {
338     return 1;
339 }
340
341 static HRESULT WINAPI OleCommandTarget_QueryStatus(IOleCommandTarget *iface, const GUID *pguidCmdGroup,
342         ULONG cCmds, OLECMD prgCmds[], OLECMDTEXT *pCmdText)
343 {
344     ok(!pguidCmdGroup, "pguidCmdGroup != MULL\n");
345     ok(cCmds == 1, "cCmds=%d, expected 1\n", cCmds);
346     ok(!pCmdText, "pCmdText != NULL\n");
347
348     switch(prgCmds[0].cmdID) {
349     case OLECMDID_STOP:
350         CHECK_EXPECT2(QueryStatus_STOP);
351         prgCmds[0].cmdf = OLECMDF_SUPPORTED;
352         return S_OK;
353     case OLECMDID_SETPROGRESSTEXT:
354         CHECK_EXPECT(QueryStatus_SETPROGRESSTEXT);
355         prgCmds[0].cmdf = OLECMDF_ENABLED;
356         return S_OK;
357     case IDM_STOP:
358         /* Note:
359          *  IDM_STOP is a command specific to CGID_MSHTML (not an OLECMDID command).
360          *  This command appears here for the ExecWB and QueryStatusWB tests in order
361          *  to help demonstrate that these routines use a NULL pguidCmdGroup.
362          */
363         CHECK_EXPECT(QueryStatus_IDM_STOP);
364         prgCmds[0].cmdf = 0;
365         return S_OK;
366     default:
367         ok(0, "unexpected command %d\n", prgCmds[0].cmdID);
368     }
369
370     return E_FAIL;
371 }
372
373 static HRESULT WINAPI OleCommandTarget_Exec(IOleCommandTarget *iface, const GUID *pguidCmdGroup,
374         DWORD nCmdID, DWORD nCmdexecopt, VARIANT *pvaIn, VARIANT *pvaOut)
375 {
376     if(!pguidCmdGroup) {
377         switch(nCmdID) {
378         case OLECMDID_SETPROGRESSMAX:
379             CHECK_EXPECT2(Exec_SETPROGRESSMAX);
380             ok(nCmdexecopt == OLECMDEXECOPT_DONTPROMPTUSER, "nCmdexecopts=%08x\n", nCmdexecopt);
381             ok(pvaIn != NULL, "pvaIn == NULL\n");
382             if(pvaIn)
383                 ok(V_VT(pvaIn) == VT_I4, "V_VT(pvaIn)=%d, expected VT_I4\n", V_VT(pvaIn));
384             ok(pvaOut == NULL, "pvaOut=%p, expected NULL\n", pvaOut);
385             return S_OK;
386         case OLECMDID_SETPROGRESSPOS:
387             CHECK_EXPECT2(Exec_SETPROGRESSPOS);
388             ok(nCmdexecopt == OLECMDEXECOPT_DONTPROMPTUSER, "nCmdexecopts=%08x\n", nCmdexecopt);
389             ok(pvaIn != NULL, "pvaIn == NULL\n");
390             if(pvaIn)
391                 ok(V_VT(pvaIn) == VT_I4, "V_VT(pvaIn)=%d, expected VT_I4\n", V_VT(pvaIn));
392             ok(pvaOut == NULL, "pvaOut=%p, expected NULL\n", pvaOut);
393             return S_OK;
394         case OLECMDID_SETDOWNLOADSTATE:
395             if(is_downloading)
396                 ok(nCmdexecopt == OLECMDEXECOPT_DONTPROMPTUSER || !nCmdexecopt,
397                    "nCmdexecopts=%08x\n", nCmdexecopt);
398             else
399                 ok(!nCmdexecopt, "nCmdexecopts=%08x\n", nCmdexecopt);
400             ok(pvaOut == NULL, "pvaOut=%p\n", pvaOut);
401             ok(pvaIn != NULL, "pvaIn == NULL\n");
402             ok(V_VT(pvaIn) == VT_I4, "V_VT(pvaIn)=%d\n", V_VT(pvaIn));
403             switch(V_I4(pvaIn)) {
404             case 0:
405                 CHECK_EXPECT2(Exec_SETDOWNLOADSTATE_0);
406                 break;
407             case 1:
408                 CHECK_EXPECT2(Exec_SETDOWNLOADSTATE_1);
409                 break;
410             default:
411                 ok(0, "unexpevted V_I4(pvaIn)=%d\n", V_I4(pvaIn));
412             }
413             return S_OK;
414         case OLECMDID_UPDATECOMMANDS:
415             CHECK_EXPECT(Exec_UPDATECOMMANDS);
416             ok(nCmdexecopt == OLECMDEXECOPT_DONTPROMPTUSER, "nCmdexecopts=%08x\n", nCmdexecopt);
417             ok(!pvaIn, "pvaIn != NULL\n");
418             ok(!pvaOut, "pvaOut=%p, expected NULL\n", pvaOut);
419             return S_OK;
420         case OLECMDID_SETTITLE:
421             CHECK_EXPECT(Exec_SETTITLE);
422             /* TODO: test args */
423             return S_OK;
424         case OLECMDID_STOP:
425             CHECK_EXPECT(Exec_STOP);
426             return S_OK;
427         case IDM_STOP:
428             /* Note:
429              *  IDM_STOP is a command specific to CGID_MSHTML (not an OLECMDID command).
430              *  This command appears here for the ExecWB and QueryStatusWB tests in order
431              *  to help demonstrate that these routines use a NULL pguidCmdGroup.
432              */
433             CHECK_EXPECT(Exec_IDM_STOP);
434             return OLECMDERR_E_NOTSUPPORTED;
435         case 6058: /* TODO */
436             return E_NOTIMPL;
437         default:
438             ok(0, "unexpected nsCmdID %d\n", nCmdID);
439         }
440     }else if(IsEqualGUID(&CGID_Explorer, pguidCmdGroup)) {
441         switch(nCmdID) {
442         case 24: /* TODO */
443         case 25: /* IE5 */
444         case 37: /* TODO */
445         case 39: /* TODO */
446         case 66: /* TODO */
447         case 67: /* TODO */
448         case 69: /* TODO */
449         case 101: /* TODO (IE8) */
450         case 109: /* TODO (IE9) */
451             return E_FAIL;
452         default:
453             ok(0, "unexpected nCmdID %d\n", nCmdID);
454         }
455     }else if(IsEqualGUID(&CGID_ShellDocView, pguidCmdGroup)) {
456         switch(nCmdID) {
457         case 105: /* TODO */
458         case 140: /* TODO (Win7) */
459             return E_FAIL;
460         default:
461             ok(0, "unexpected nCmdID %d\n", nCmdID);
462         }
463     }else if(IsEqualGUID(&CGID_DocHostCmdPriv, pguidCmdGroup)) {
464         switch(nCmdID) {
465         case 11: /* TODO */
466             break;
467         default:
468             ok(0, "unexpected nCmdID %d of CGID_DocHostCmdPriv\n", nCmdID);
469         }
470     }else if(IsEqualGUID(&CGID_DocHostCommandHandler, pguidCmdGroup)) {
471         switch(nCmdID) {
472         case 6041: /* TODO */
473             break;
474         default:
475             ok(0, "unexpected nCmdID %d of CGID_DocHostCommandHandler\n", nCmdID);
476         }
477     }else {
478         ok(0, "unexpected pguidCmdGroup %s\n", debugstr_guid(pguidCmdGroup));
479     }
480
481     return E_FAIL;
482 }
483
484 static IOleCommandTargetVtbl OleCommandTargetVtbl = {
485     OleCommandTarget_QueryInterface,
486     OleCommandTarget_AddRef,
487     OleCommandTarget_Release,
488     OleCommandTarget_QueryStatus,
489     OleCommandTarget_Exec
490 };
491
492 static IOleCommandTarget OleCommandTarget = { &OleCommandTargetVtbl };
493
494 static HRESULT WINAPI OleContainer_QueryInterface(IOleContainer *iface, REFIID riid, void **ppv)
495 {
496     if(IsEqualGUID(&IID_ITargetContainer, riid))
497         return E_NOINTERFACE; /* TODO */
498
499     if(IsEqualGUID(&IID_IOleCommandTarget, riid)) {
500         if(use_container_olecmd)
501         {
502             *ppv = &OleCommandTarget;
503             return S_OK;
504         }
505         else
506             return E_NOINTERFACE;
507     }
508
509     ok(0, "unexpected call\n");
510     return E_NOINTERFACE;
511 }
512
513 static ULONG WINAPI OleContainer_AddRef(IOleContainer *iface)
514 {
515     return 2;
516 }
517
518 static ULONG WINAPI OleContainer_Release(IOleContainer *iface)
519 {
520     return 1;
521 }
522
523 static HRESULT WINAPI OleContainer_ParseDisplayName(IOleContainer *iface, IBindCtx *pbc,
524         LPOLESTR pszDiaplayName, ULONG *pchEaten, IMoniker **ppmkOut)
525 {
526     ok(0, "unexpected call\n");
527     return E_NOTIMPL;
528 }
529
530 static HRESULT WINAPI OleContainer_EnumObjects(IOleContainer *iface, DWORD grfFlags,
531         IEnumUnknown **ppenum)
532 {
533     ok(0, "unexpected call\n");
534     return E_NOTIMPL;
535 }
536
537 static HRESULT WINAPI OleContainer_LockContainer(IOleContainer *iface, BOOL fLock)
538 {
539     ok(0, "unexpected call\n");
540     return E_NOTIMPL;
541 }
542
543 static const IOleContainerVtbl OleContainerVtbl = {
544     OleContainer_QueryInterface,
545     OleContainer_AddRef,
546     OleContainer_Release,
547     OleContainer_ParseDisplayName,
548     OleContainer_EnumObjects,
549     OleContainer_LockContainer
550 };
551
552 static IOleContainer OleContainer = { &OleContainerVtbl };
553
554 static HRESULT WINAPI Dispatch_QueryInterface(IDispatch *iface, REFIID riid, void **ppv)
555 {
556     return QueryInterface(riid, ppv);
557 }
558
559 static ULONG WINAPI Dispatch_AddRef(IDispatch *iface)
560 {
561     return 2;
562 }
563
564 static ULONG WINAPI Dispatch_Release(IDispatch *iface)
565 {
566     return 1;
567 }
568
569 static HRESULT WINAPI Dispatch_GetTypeInfoCount(IDispatch *iface, UINT *pctinfo)
570 {
571     ok(0, "unexpected call\n");
572     return E_NOTIMPL;
573 }
574
575 static HRESULT WINAPI Dispatch_GetTypeInfo(IDispatch *iface, UINT iTInfo, LCID lcid,
576         ITypeInfo **ppTInfo)
577 {
578     ok(0, "unexpected call\n");
579     return E_NOTIMPL;
580 }
581
582 static HRESULT WINAPI Dispatch_GetIDsOfNames(IDispatch *iface, REFIID riid, LPOLESTR *rgszNames,
583         UINT cNames, LCID lcid, DISPID *rgDispId)
584 {
585     ok(0, "unexpected call\n");
586     return E_NOTIMPL;
587 }
588
589 static HRESULT WINAPI Dispatch_Invoke(IDispatch *iface, DISPID dispIdMember, REFIID riid,
590         LCID lcid, WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult,
591         EXCEPINFO *pExcepInfo, UINT *puArgErr)
592 {
593     ok(IsEqualGUID(&IID_NULL, riid), "riid != IID_NULL\n");
594     ok(pDispParams != NULL, "pDispParams == NULL\n");
595     ok(pExcepInfo == NULL, "pExcepInfo=%p, expected NULL\n", pExcepInfo);
596     ok(V_VT(pVarResult) == VT_EMPTY, "V_VT(pVarResult)=%d\n", V_VT(pVarResult));
597     ok(wFlags == DISPATCH_PROPERTYGET, "wFlags=%08x, expected DISPATCH_PROPERTYGET\n", wFlags);
598     ok(pDispParams->rgvarg == NULL, "pDispParams->rgvarg = %p\n", pDispParams->rgvarg);
599     ok(pDispParams->rgdispidNamedArgs == NULL,
600        "pDispParams->rgdispidNamedArgs = %p\n", pDispParams->rgdispidNamedArgs);
601     ok(pDispParams->cArgs == 0, "pDispParams->cArgs = %d\n", pDispParams->cArgs);
602     ok(pDispParams->cNamedArgs == 0, "pDispParams->cNamedArgs = %d\n", pDispParams->cNamedArgs);
603
604     switch(dispIdMember) {
605     case DISPID_AMBIENT_USERMODE:
606         CHECK_EXPECT2(Invoke_AMBIENT_USERMODE);
607         return E_FAIL;
608     case DISPID_AMBIENT_DLCONTROL:
609         CHECK_EXPECT(Invoke_AMBIENT_DLCONTROL);
610         ok(puArgErr != NULL, "puArgErr=%p\n", puArgErr);
611         return E_FAIL;
612     case DISPID_AMBIENT_USERAGENT:
613        CHECK_EXPECT(Invoke_AMBIENT_USERAGENT);
614         ok(puArgErr != NULL, "puArgErr=%p\n", puArgErr);
615         return E_FAIL;
616     case DISPID_AMBIENT_PALETTE:
617         CHECK_EXPECT(Invoke_AMBIENT_PALETTE);
618         ok(puArgErr != NULL, "puArgErr=%p\n", puArgErr);
619         return E_FAIL;
620     case DISPID_AMBIENT_OFFLINEIFNOTCONNECTED:
621         CHECK_EXPECT(Invoke_AMBIENT_OFFLINEIFNOTCONNECTED);
622         ok(puArgErr == NULL, "puArgErr=%p\n", puArgErr);
623         V_VT(pVarResult) = VT_BOOL;
624         V_BOOL(pVarResult) = VARIANT_FALSE;
625         return S_OK;
626     case DISPID_AMBIENT_SILENT:
627         CHECK_EXPECT(Invoke_AMBIENT_SILENT);
628         ok(puArgErr == NULL, "puArgErr=%p\n", puArgErr);
629         V_VT(pVarResult) = VT_BOOL;
630         V_BOOL(pVarResult) = VARIANT_FALSE;
631         return S_OK;
632     }
633
634     ok(0, "unexpected dispIdMember %d\n", dispIdMember);
635     return E_NOTIMPL;
636 }
637
638 static IDispatchVtbl DispatchVtbl = {
639     Dispatch_QueryInterface,
640     Dispatch_AddRef,
641     Dispatch_Release,
642     Dispatch_GetTypeInfoCount,
643     Dispatch_GetTypeInfo,
644     Dispatch_GetIDsOfNames,
645     Dispatch_Invoke
646 };
647
648 static IDispatch Dispatch = { &DispatchVtbl };
649
650 static HRESULT WINAPI WebBrowserEvents2_QueryInterface(IDispatch *iface, REFIID riid, void **ppv)
651 {
652     *ppv = NULL;
653
654     if(IsEqualGUID(&DIID_DWebBrowserEvents2, riid)) {
655         *ppv = iface;
656         return S_OK;
657     }
658
659     ok(0, "unexpected riid %s\n", debugstr_guid(riid));
660     return E_NOINTERFACE;
661 }
662
663 #define test_invoke_bool(p,s) _test_invoke_bool(__LINE__,p,s)
664 static void _test_invoke_bool(unsigned line, const DISPPARAMS *params, BOOL strict)
665 {
666     ok_(__FILE__,line) (params->rgvarg != NULL, "rgvarg == NULL\n");
667     ok_(__FILE__,line) (params->cArgs == 1, "cArgs=%d, expected 1\n", params->cArgs);
668     ok_(__FILE__,line) (V_VT(params->rgvarg) == VT_BOOL, "V_VT(arg)=%d\n", V_VT(params->rgvarg));
669     if(strict)
670         ok_(__FILE__,line) (V_BOOL(params->rgvarg) == exvb, "V_VT(arg)=%x, expected %x\n",
671                             V_BOOL(params->rgvarg), exvb);
672     else
673         ok_(__FILE__,line) (!V_BOOL(params->rgvarg) == !exvb, "V_VT(arg)=%x, expected %x\n",
674                             V_BOOL(params->rgvarg), exvb);
675 }
676
677 static void test_OnBeforeNavigate(const VARIANT *disp, const VARIANT *url, const VARIANT *flags,
678         const VARIANT *frame, const VARIANT *post_data, const VARIANT *headers, const VARIANT *cancel)
679 {
680     ok(V_VT(disp) == VT_DISPATCH, "V_VT(disp)=%d, expected VT_DISPATCH\n", V_VT(disp));
681     ok(V_DISPATCH(disp) != NULL, "V_DISPATCH(disp) == NULL\n");
682     ok(V_DISPATCH(disp) == (IDispatch*)wb, "V_DISPATCH(disp)=%p, wb=%p\n", V_DISPATCH(disp), wb);
683
684     ok(V_VT(url) == (VT_BYREF|VT_VARIANT), "V_VT(url)=%x, expected VT_BYREF|VT_VARIANT\n", V_VT(url));
685     ok(V_VARIANTREF(url) != NULL, "V_VARIANTREF(url) == NULL)\n");
686     if(V_VARIANTREF(url)) {
687         ok(V_VT(V_VARIANTREF(url)) == VT_BSTR, "V_VT(V_VARIANTREF(url))=%d, expected VT_BSTR\n",
688            V_VT(V_VARIANTREF(url)));
689         ok(V_BSTR(V_VARIANTREF(url)) != NULL, "V_BSTR(V_VARIANTREF(url)) == NULL\n");
690         ok(!strcmp_wa(V_BSTR(V_VARIANTREF(url)), current_url), "unexpected url %s, expected %s\n",
691            wine_dbgstr_w(V_BSTR(V_VARIANTREF(url))), current_url);
692     }
693
694     ok(V_VT(flags) == (VT_BYREF|VT_VARIANT), "V_VT(flags)=%x, expected VT_BYREF|VT_VARIANT\n",
695        V_VT(flags));
696     ok(V_VT(flags) == (VT_BYREF|VT_VARIANT), "V_VT(flags)=%x, expected VT_BYREF|VT_VARIANT\n",
697        V_VT(flags));
698     ok(V_VARIANTREF(flags) != NULL, "V_VARIANTREF(flags) == NULL)\n");
699     if(V_VARIANTREF(flags)) {
700         ok(V_VT(V_VARIANTREF(flags)) == VT_I4, "V_VT(V_VARIANTREF(flags))=%d, expected VT_I4\n",
701            V_VT(V_VARIANTREF(flags)));
702         if(is_first_load) {
703             ok(V_I4(V_VARIANTREF(flags)) == 0, "V_I4(V_VARIANTREF(flags)) = %x, expected 0\n",
704                V_I4(V_VARIANTREF(flags)));
705         }else {
706             ok((V_I4(V_VARIANTREF(flags)) & ~0x40) == 0, "V_I4(V_VARIANTREF(flags)) = %x, expected 0x40 or 0\n",
707                V_I4(V_VARIANTREF(flags)));
708         }
709     }
710
711     ok(V_VT(frame) == (VT_BYREF|VT_VARIANT), "V_VT(frame)=%x, expected VT_BYREF|VT_VARIANT\n",
712        V_VT(frame));
713     ok(V_VT(frame) == (VT_BYREF|VT_VARIANT), "V_VT(frame)=%x, expected VT_BYREF|VT_VARIANT\n",
714        V_VT(frame));
715     ok(V_VARIANTREF(frame) != NULL, "V_VARIANTREF(frame) == NULL)\n");
716     if(V_VARIANTREF(frame)) {
717         ok(V_VT(V_VARIANTREF(frame)) == VT_BSTR, "V_VT(V_VARIANTREF(frame))=%d, expected VT_BSTR\n",
718            V_VT(V_VARIANTREF(frame)));
719         ok(V_BSTR(V_VARIANTREF(frame)) == NULL, "V_BSTR(V_VARIANTREF(frame)) = %p, expected NULL\n",
720            V_BSTR(V_VARIANTREF(frame)));
721     }
722
723     ok(V_VT(post_data) == (VT_BYREF|VT_VARIANT), "V_VT(post_data)=%x, expected VT_BYREF|VT_VARIANT\n",
724        V_VT(post_data));
725     ok(V_VT(post_data) == (VT_BYREF|VT_VARIANT), "V_VT(post_data)=%x, expected VT_BYREF|VT_VARIANT\n",
726        V_VT(post_data));
727     ok(V_VARIANTREF(post_data) != NULL, "V_VARIANTREF(post_data) == NULL)\n");
728     if(V_VARIANTREF(post_data)) {
729         ok(V_VT(V_VARIANTREF(post_data)) == (VT_VARIANT|VT_BYREF),
730            "V_VT(V_VARIANTREF(post_data))=%d, expected VT_VARIANT|VT_BYREF\n",
731            V_VT(V_VARIANTREF(post_data)));
732         ok(V_VARIANTREF(V_VARIANTREF(post_data)) != NULL,
733            "V_VARIANTREF(V_VARIANTREF(post_data)) == NULL\n");
734         if(V_VARIANTREF(V_VARIANTREF(post_data))) {
735             ok(V_VT(V_VARIANTREF(V_VARIANTREF(post_data))) == VT_EMPTY,
736                "V_VT(V_VARIANTREF(V_VARIANTREF(post_data))) = %d, expected VT_EMPTY\n",
737                V_VT(V_VARIANTREF(V_VARIANTREF(post_data))));
738
739             if(V_VT(V_VARIANTREF(V_VARIANTREF(post_data))) == (VT_UI1|VT_ARRAY)) {
740                 const SAFEARRAY *sa = V_ARRAY(V_VARIANTREF(V_VARIANTREF(post_data)));
741
742                 ok(sa->cDims == 1, "sa->cDims = %d, expected 1\n", sa->cDims);
743                 ok(sa->fFeatures == 0, "sa->fFeatures = %d, expected 0\n", sa->fFeatures);
744             }
745         }
746     }
747
748     ok(V_VT(headers) == (VT_BYREF|VT_VARIANT), "V_VT(headers)=%x, expected VT_BYREF|VT_VARIANT\n",
749        V_VT(headers));
750     ok(V_VARIANTREF(headers) != NULL, "V_VARIANTREF(headers) == NULL)\n");
751     if(V_VARIANTREF(headers)) {
752         ok(V_VT(V_VARIANTREF(headers)) == VT_BSTR, "V_VT(V_VARIANTREF(headers))=%d, expected VT_BSTR\n",
753            V_VT(V_VARIANTREF(headers)));
754         ok(V_BSTR(V_VARIANTREF(headers)) == NULL, "V_BSTR(V_VARIANTREF(headers)) = %p, expected NULL\n",
755            V_BSTR(V_VARIANTREF(headers)));
756     }
757
758     ok(V_VT(cancel) == (VT_BYREF|VT_BOOL), "V_VT(cancel)=%x, expected VT_BYREF|VT_BOOL\n",
759        V_VT(cancel));
760     ok(V_BOOLREF(cancel) != NULL, "V_BOOLREF(pDispParams->rgvarg[0] == NULL)\n");
761     if(V_BOOLREF(cancel))
762         ok(*V_BOOLREF(cancel) == VARIANT_FALSE, "*V_BOOLREF(cancel) = %x, expected VARIANT_FALSE\n",
763            *V_BOOLREF(cancel));
764 }
765
766 static HRESULT WINAPI WebBrowserEvents2_Invoke(IDispatch *iface, DISPID dispIdMember, REFIID riid,
767         LCID lcid, WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult,
768         EXCEPINFO *pExcepInfo, UINT *puArgErr)
769 {
770     ok(IsEqualGUID(&IID_NULL, riid), "riid != IID_NULL\n");
771     ok(pDispParams != NULL, "pDispParams == NULL\n");
772     ok(pExcepInfo == NULL, "pExcepInfo=%p, expected NULL\n", pExcepInfo);
773     ok(pVarResult == NULL, "pVarResult=%p\n", pVarResult);
774     ok(wFlags == DISPATCH_METHOD, "wFlags=%08x, expected DISPATCH_METHOD\n", wFlags);
775     ok(pDispParams->rgdispidNamedArgs == NULL,
776        "pDispParams->rgdispidNamedArgs = %p\n", pDispParams->rgdispidNamedArgs);
777     ok(pDispParams->cNamedArgs == 0, "pDispParams->cNamedArgs = %d\n", pDispParams->cNamedArgs);
778
779     switch(dispIdMember) {
780     case DISPID_STATUSTEXTCHANGE:
781         CHECK_EXPECT2(Invoke_STATUSTEXTCHANGE);
782
783         ok(pDispParams->rgvarg != NULL, "rgvarg == NULL\n");
784         ok(pDispParams->cArgs == 1, "cArgs=%d, expected 1\n", pDispParams->cArgs);
785         ok(V_VT(pDispParams->rgvarg) == VT_BSTR, "V_VT(pDispParams->rgvarg)=%d, expected VT_BSTR\n",
786            V_VT(pDispParams->rgvarg));
787         /* TODO: Check text */
788         break;
789
790     case DISPID_PROPERTYCHANGE:
791         CHECK_EXPECT2(Invoke_PROPERTYCHANGE);
792
793         ok(pDispParams->rgvarg != NULL, "rgvarg == NULL\n");
794         ok(pDispParams->cArgs == 1, "cArgs=%d, expected 1\n", pDispParams->cArgs);
795         /* TODO: Check args */
796         break;
797
798     case DISPID_DOWNLOADBEGIN:
799         CHECK_EXPECT(Invoke_DOWNLOADBEGIN);
800
801         ok(pDispParams->rgvarg == NULL, "rgvarg=%p, expected NULL\n", pDispParams->rgvarg);
802         ok(pDispParams->cArgs == 0, "cArgs=%d, expected 0\n", pDispParams->cArgs);
803         test_ready_state(READYSTATE_LOADING);
804         break;
805
806     case DISPID_BEFORENAVIGATE2:
807         CHECK_EXPECT(Invoke_BEFORENAVIGATE2);
808
809         ok(pDispParams->rgvarg != NULL, "rgvarg == NULL\n");
810         ok(pDispParams->cArgs == 7, "cArgs=%d, expected 7\n", pDispParams->cArgs);
811         test_OnBeforeNavigate(pDispParams->rgvarg+6, pDispParams->rgvarg+5, pDispParams->rgvarg+4,
812                               pDispParams->rgvarg+3, pDispParams->rgvarg+2, pDispParams->rgvarg+1,
813                               pDispParams->rgvarg);
814         test_ready_state((dwl_flags & DWL_FROM_PUT_HREF) ? READYSTATE_COMPLETE : READYSTATE_LOADING);
815         break;
816
817     case DISPID_SETSECURELOCKICON:
818         CHECK_EXPECT2(Invoke_SETSECURELOCKICON);
819
820         ok(pDispParams->rgvarg != NULL, "rgvarg == NULL\n");
821         ok(pDispParams->cArgs == 1, "cArgs=%d, expected 1\n", pDispParams->cArgs);
822         /* TODO: Check args */
823         break;
824
825     case DISPID_FILEDOWNLOAD:
826         CHECK_EXPECT(Invoke_FILEDOWNLOAD);
827
828         ok(pDispParams->rgvarg != NULL, "rgvarg == NULL\n");
829         ok(pDispParams->cArgs == 2, "cArgs=%d, expected 2\n", pDispParams->cArgs);
830         /* TODO: Check args */
831         break;
832
833     case DISPID_COMMANDSTATECHANGE:
834         CHECK_EXPECT2(Invoke_COMMANDSTATECHANGE);
835
836         ok(pDispParams->rgvarg != NULL, "rgvarg == NULL\n");
837         ok(pDispParams->cArgs == 2, "cArgs=%d, expected 2\n", pDispParams->cArgs);
838         /* TODO: Check args */
839         break;
840
841     case DISPID_DOWNLOADCOMPLETE:
842         CHECK_EXPECT(Invoke_DOWNLOADCOMPLETE);
843
844         ok(pDispParams->rgvarg == NULL, "rgvarg=%p, expected NULL\n", pDispParams->rgvarg);
845         ok(pDispParams->cArgs == 0, "cArgs=%d, expected 0\n", pDispParams->cArgs);
846         test_ready_state(READYSTATE_LOADING);
847         break;
848
849     case DISPID_ONMENUBAR:
850         CHECK_EXPECT(Invoke_ONMENUBAR);
851         test_invoke_bool(pDispParams, TRUE);
852         break;
853
854     case DISPID_ONADDRESSBAR:
855         CHECK_EXPECT(Invoke_ONADDRESSBAR);
856         test_invoke_bool(pDispParams, TRUE);
857         break;
858
859     case DISPID_ONSTATUSBAR:
860         CHECK_EXPECT(Invoke_ONSTATUSBAR);
861         test_invoke_bool(pDispParams, TRUE);
862         break;
863
864     case DISPID_ONTOOLBAR:
865         CHECK_EXPECT(Invoke_ONTOOLBAR);
866         test_invoke_bool(pDispParams, FALSE);
867         break;
868
869     case DISPID_ONFULLSCREEN:
870         CHECK_EXPECT(Invoke_ONFULLSCREEN);
871         test_invoke_bool(pDispParams, TRUE);
872         break;
873
874     case DISPID_ONTHEATERMODE:
875         CHECK_EXPECT(Invoke_ONTHEATERMODE);
876         test_invoke_bool(pDispParams, TRUE);
877         break;
878
879     case DISPID_WINDOWSETRESIZABLE:
880         CHECK_EXPECT(Invoke_WINDOWSETRESIZABLE);
881         test_invoke_bool(pDispParams, TRUE);
882         break;
883
884     case DISPID_TITLECHANGE:
885         CHECK_EXPECT2(Invoke_TITLECHANGE);
886         /* FIXME */
887         break;
888
889     case DISPID_NAVIGATECOMPLETE2:
890         CHECK_EXPECT(Invoke_NAVIGATECOMPLETE2);
891         /* FIXME */
892         test_ready_state((dwl_flags & DWL_FROM_PUT_HREF) ? READYSTATE_COMPLETE : READYSTATE_LOADING);
893         break;
894
895     case DISPID_PROGRESSCHANGE:
896         CHECK_EXPECT(Invoke_PROGRESSCHANGE);
897         /* FIXME */
898         break;
899
900     case DISPID_DOCUMENTCOMPLETE:
901         CHECK_EXPECT(Invoke_DOCUMENTCOMPLETE);
902         /* FIXME */
903         test_ready_state(READYSTATE_COMPLETE);
904         break;
905
906     case 282: /* FIXME */
907         CHECK_EXPECT2(Invoke_282);
908         break;
909
910     default:
911         ok(0, "unexpected dispIdMember %d\n", dispIdMember);
912     }
913
914     return S_OK;
915 }
916
917 static IDispatchVtbl WebBrowserEvents2Vtbl = {
918     WebBrowserEvents2_QueryInterface,
919     Dispatch_AddRef,
920     Dispatch_Release,
921     Dispatch_GetTypeInfoCount,
922     Dispatch_GetTypeInfo,
923     Dispatch_GetIDsOfNames,
924     WebBrowserEvents2_Invoke
925 };
926
927 static IDispatch WebBrowserEvents2 = { &WebBrowserEvents2Vtbl };
928
929 static HRESULT WINAPI ClientSite_QueryInterface(IOleClientSite *iface, REFIID riid, void **ppv)
930 {
931     return QueryInterface(riid, ppv);
932 }
933
934 static ULONG WINAPI ClientSite_AddRef(IOleClientSite *iface)
935 {
936     return 2;
937 }
938
939 static ULONG WINAPI ClientSite_Release(IOleClientSite *iface)
940 {
941     return 1;
942 }
943
944 static HRESULT WINAPI ClientSite_SaveObject(IOleClientSite *iface)
945 {
946     ok(0, "unexpected call\n");
947     return E_NOTIMPL;
948 }
949
950 static HRESULT WINAPI ClientSite_GetMoniker(IOleClientSite *iface, DWORD dwAssign, DWORD dwWhichMoniker,
951         IMoniker **ppmon)
952 {
953     ok(0, "unexpected call\n");
954     return E_NOTIMPL;
955 }
956
957 static HRESULT WINAPI ClientSite_GetContainer(IOleClientSite *iface, IOleContainer **ppContainer)
958 {
959     CHECK_EXPECT(GetContainer);
960
961     ok(ppContainer != NULL, "ppContainer == NULL\n");
962     if(ppContainer)
963         *ppContainer = &OleContainer;
964
965     return S_OK;
966 }
967
968 static HRESULT WINAPI ClientSite_ShowObject(IOleClientSite *iface)
969 {
970     CHECK_EXPECT(ShowObject);
971     return S_OK;
972 }
973
974 static HRESULT WINAPI ClientSite_OnShowWindow(IOleClientSite *iface, BOOL fShow)
975 {
976     ok(0, "unexpected call\n");
977     return E_NOTIMPL;
978 }
979
980 static HRESULT WINAPI ClientSite_RequestNewObjectLayout(IOleClientSite *iface)
981 {
982     ok(0, "unexpected call\n");
983     return E_NOTIMPL;
984 }
985
986 static const IOleClientSiteVtbl ClientSiteVtbl = {
987     ClientSite_QueryInterface,
988     ClientSite_AddRef,
989     ClientSite_Release,
990     ClientSite_SaveObject,
991     ClientSite_GetMoniker,
992     ClientSite_GetContainer,
993     ClientSite_ShowObject,
994     ClientSite_OnShowWindow,
995     ClientSite_RequestNewObjectLayout
996 };
997
998 static IOleClientSite ClientSite = { &ClientSiteVtbl };
999
1000 static HRESULT WINAPI IOleControlSite_fnQueryInterface(IOleControlSite *iface, REFIID riid, void **ppv)
1001 {
1002     *ppv = NULL;
1003     return E_NOINTERFACE;
1004 }
1005
1006 static ULONG WINAPI IOleControlSite_fnAddRef(IOleControlSite *iface)
1007 {
1008     return 2;
1009 }
1010
1011 static ULONG WINAPI IOleControlSite_fnRelease(IOleControlSite *iface)
1012 {
1013     return 1;
1014 }
1015
1016 static HRESULT WINAPI IOleControlSite_fnOnControlInfoChanged(IOleControlSite* This)
1017 {
1018     ok(0, "unexpected call\n");
1019     return E_NOTIMPL;
1020 }
1021
1022 static HRESULT WINAPI IOleControlSite_fnLockInPlaceActive(IOleControlSite* This,
1023                                                           BOOL fLock)
1024 {
1025     ok(0, "unexpected call\n");
1026     return E_NOTIMPL;
1027 }
1028
1029 static HRESULT WINAPI IOleControlSite_fnGetExtendedControl(IOleControlSite* This,
1030                                                            IDispatch **ppDisp)
1031 {
1032     ok(0, "unexpected call\n");
1033     return E_NOTIMPL;
1034 }
1035
1036 static HRESULT WINAPI IOleControlSite_fnTransformCoords(IOleControlSite* This,
1037                                                         POINTL *pPtlHimetric,
1038                                                         POINTF *pPtfContainer,
1039                                                         DWORD dwFlags)
1040 {
1041     ok(0, "unexpected call\n");
1042     return E_NOTIMPL;
1043 }
1044
1045 static HRESULT WINAPI IOleControlSite_fnTranslateAccelerator(IOleControlSite* This, MSG *pMsg,
1046                                                              DWORD grfModifiers)
1047 {
1048     CHECK_EXPECT(ControlSite_TranslateAccelerator);
1049     return hr_site_TranslateAccelerator;
1050 }
1051
1052 static HRESULT WINAPI IOleControlSite_fnOnFocus(IOleControlSite* This, BOOL fGotFocus)
1053 {
1054     CHECK_EXPECT2(OnFocus);
1055     return E_NOTIMPL;
1056 }
1057
1058 static HRESULT WINAPI IOleControlSite_fnShowPropertyFrame(IOleControlSite* This)
1059 {
1060     ok(0, "unexpected call\n");
1061     return E_NOTIMPL;
1062 }
1063
1064 static IOleControlSiteVtbl ControlSiteVtbl = {
1065     IOleControlSite_fnQueryInterface,
1066     IOleControlSite_fnAddRef,
1067     IOleControlSite_fnRelease,
1068     IOleControlSite_fnOnControlInfoChanged,
1069     IOleControlSite_fnLockInPlaceActive,
1070     IOleControlSite_fnGetExtendedControl,
1071     IOleControlSite_fnTransformCoords,
1072     IOleControlSite_fnTranslateAccelerator,
1073     IOleControlSite_fnOnFocus,
1074     IOleControlSite_fnShowPropertyFrame
1075 };
1076
1077 static IOleControlSite ControlSite = { &ControlSiteVtbl };
1078
1079 static HRESULT WINAPI InPlaceUIWindow_QueryInterface(IOleInPlaceFrame *iface,
1080                                                      REFIID riid, void **ppv)
1081 {
1082     ok(0, "unexpected call\n");
1083     return E_NOINTERFACE;
1084 }
1085
1086 static ULONG WINAPI InPlaceUIWindow_AddRef(IOleInPlaceFrame *iface)
1087 {
1088     return 2;
1089 }
1090
1091 static ULONG WINAPI InPlaceUIWindow_Release(IOleInPlaceFrame *iface)
1092 {
1093     return 1;
1094 }
1095
1096 static HRESULT WINAPI InPlaceUIWindow_GetWindow(IOleInPlaceFrame *iface, HWND *phwnd)
1097 {
1098     ok(0, "unexpected call\n");
1099     return E_NOTIMPL;
1100 }
1101
1102 static HRESULT WINAPI InPlaceFrame_GetWindow(IOleInPlaceFrame *iface, HWND *phwnd)
1103 {
1104     CHECK_EXPECT(Frame_GetWindow);
1105     ok(phwnd != NULL, "phwnd == NULL\n");
1106     if(phwnd)
1107         *phwnd = container_hwnd;
1108     return S_OK;
1109 }
1110
1111 static HRESULT WINAPI InPlaceUIWindow_ContextSensitiveHelp(IOleInPlaceFrame *iface, BOOL fEnterMode)
1112 {
1113     ok(0, "unexpected call\n");
1114     return E_NOTIMPL;
1115 }
1116
1117 static HRESULT WINAPI InPlaceUIWindow_GetBorder(IOleInPlaceFrame *iface, LPRECT lprectBorder)
1118 {
1119     ok(0, "unexpected call\n");
1120     return E_NOTIMPL;
1121 }
1122
1123 static HRESULT WINAPI InPlaceUIWindow_RequestBorderSpace(IOleInPlaceFrame *iface,
1124         LPCBORDERWIDTHS pborderwidths)
1125 {
1126     ok(0, "unexpected call\n");
1127     return E_NOTIMPL;
1128 }
1129
1130 static HRESULT WINAPI InPlaceUIWindow_SetBorderSpace(IOleInPlaceFrame *iface,
1131         LPCBORDERWIDTHS pborderwidths)
1132 {
1133     ok(0, "unexpected call\n");
1134     return E_NOTIMPL;
1135 }
1136
1137 static HRESULT WINAPI InPlaceUIWindow_SetActiveObject(IOleInPlaceFrame *iface,
1138         IOleInPlaceActiveObject *pActiveObject, LPCOLESTR pszObjName)
1139 {
1140     CHECK_EXPECT(UIWindow_SetActiveObject);
1141     if(!test_close) {
1142         ok(pActiveObject != NULL, "pActiveObject = NULL\n");
1143         ok(!lstrcmpW(pszObjName, wszItem), "unexpected pszObjName\n");
1144     } else {
1145         ok(!pActiveObject, "pActiveObject != NULL\n");
1146         ok(!pszObjName, "pszObjName != NULL\n");
1147     }
1148     return S_OK;
1149 }
1150
1151 static HRESULT WINAPI InPlaceFrame_SetActiveObject(IOleInPlaceFrame *iface,
1152         IOleInPlaceActiveObject *pActiveObject, LPCOLESTR pszObjName)
1153 {
1154     CHECK_EXPECT(Frame_SetActiveObject);
1155     if(!test_close) {
1156         ok(pActiveObject != NULL, "pActiveObject = NULL\n");
1157         ok(!lstrcmpW(pszObjName, wszItem), "unexpected pszObjName\n");
1158     } else {
1159         ok(!pActiveObject, "pActiveObject != NULL\n");
1160         ok(!pszObjName, "pszObjName != NULL\n");
1161     }
1162     return S_OK;
1163 }
1164
1165 static HRESULT WINAPI InPlaceFrame_InsertMenus(IOleInPlaceFrame *iface, HMENU hmenuShared,
1166         LPOLEMENUGROUPWIDTHS lpMenuWidths)
1167 {
1168     ok(0, "unexpected call\n");
1169     return E_NOTIMPL;
1170 }
1171
1172 static HRESULT WINAPI InPlaceFrame_SetMenu(IOleInPlaceFrame *iface, HMENU hmenuShared,
1173         HOLEMENU holemenu, HWND hwndActiveObject)
1174 {
1175     CHECK_EXPECT(SetMenu);
1176     ok(hmenuShared == NULL, "hmenuShared=%p\n", hmenuShared);
1177     ok(holemenu == NULL, "holemenu=%p\n", holemenu);
1178     ok(hwndActiveObject == shell_embedding_hwnd, "hwndActiveObject=%p, expected %p\n",
1179        hwndActiveObject, shell_embedding_hwnd);
1180     return S_OK;
1181 }
1182
1183 static HRESULT WINAPI InPlaceFrame_RemoveMenus(IOleInPlaceFrame *iface, HMENU hmenuShared)
1184 {
1185     ok(0, "unexpected call\n");
1186     return E_NOTIMPL;
1187 }
1188
1189 static HRESULT WINAPI InPlaceFrame_SetStatusText(IOleInPlaceFrame *iface, LPCOLESTR pszStatusText)
1190 {
1191     CHECK_EXPECT2(SetStatusText);
1192     /* FIXME: Check pszStatusText */
1193     return S_OK;
1194 }
1195
1196 static HRESULT WINAPI InPlaceFrame_EnableModeless(IOleInPlaceFrame *iface, BOOL fEnable)
1197 {
1198     if(fEnable)
1199         CHECK_EXPECT2(EnableModeless_TRUE);
1200     else
1201         CHECK_EXPECT2(EnableModeless_FALSE);
1202     return S_OK;
1203 }
1204
1205 static HRESULT WINAPI InPlaceFrame_TranslateAccelerator(IOleInPlaceFrame *iface, LPMSG lpmsg, WORD wID)
1206 {
1207     ok(0, "unexpected call\n");
1208     return E_NOTIMPL;
1209 }
1210
1211 static const IOleInPlaceFrameVtbl InPlaceUIWindowVtbl = {
1212     InPlaceUIWindow_QueryInterface,
1213     InPlaceUIWindow_AddRef,
1214     InPlaceUIWindow_Release,
1215     InPlaceUIWindow_GetWindow,
1216     InPlaceUIWindow_ContextSensitiveHelp,
1217     InPlaceUIWindow_GetBorder,
1218     InPlaceUIWindow_RequestBorderSpace,
1219     InPlaceUIWindow_SetBorderSpace,
1220     InPlaceUIWindow_SetActiveObject,
1221 };
1222
1223 static IOleInPlaceUIWindow InPlaceUIWindow = { (IOleInPlaceUIWindowVtbl*)&InPlaceUIWindowVtbl };
1224
1225 static const IOleInPlaceFrameVtbl InPlaceFrameVtbl = {
1226     InPlaceUIWindow_QueryInterface,
1227     InPlaceUIWindow_AddRef,
1228     InPlaceUIWindow_Release,
1229     InPlaceFrame_GetWindow,
1230     InPlaceUIWindow_ContextSensitiveHelp,
1231     InPlaceUIWindow_GetBorder,
1232     InPlaceUIWindow_RequestBorderSpace,
1233     InPlaceUIWindow_SetBorderSpace,
1234     InPlaceFrame_SetActiveObject,
1235     InPlaceFrame_InsertMenus,
1236     InPlaceFrame_SetMenu,
1237     InPlaceFrame_RemoveMenus,
1238     InPlaceFrame_SetStatusText,
1239     InPlaceFrame_EnableModeless,
1240     InPlaceFrame_TranslateAccelerator
1241 };
1242
1243 static IOleInPlaceFrame InPlaceFrame = { &InPlaceFrameVtbl };
1244
1245 static HRESULT WINAPI InPlaceSite_QueryInterface(IOleInPlaceSiteEx *iface, REFIID riid, void **ppv)
1246 {
1247     return QueryInterface(riid, ppv);
1248 }
1249
1250 static ULONG WINAPI InPlaceSite_AddRef(IOleInPlaceSiteEx *iface)
1251 {
1252     return 2;
1253 }
1254
1255 static ULONG WINAPI InPlaceSite_Release(IOleInPlaceSiteEx *iface)
1256 {
1257     return 1;
1258 }
1259
1260 static HRESULT WINAPI InPlaceSite_GetWindow(IOleInPlaceSiteEx *iface, HWND *phwnd)
1261 {
1262     CHECK_EXPECT(Site_GetWindow);
1263     ok(phwnd != NULL, "phwnd == NULL\n");
1264     if(phwnd)
1265         *phwnd = container_hwnd;
1266     return S_OK;
1267 }
1268
1269 static HRESULT WINAPI InPlaceSite_ContextSensitiveHelp(IOleInPlaceSiteEx *iface, BOOL fEnterMode)
1270 {
1271     ok(0, "unexpected call\n");
1272     return E_NOTIMPL;
1273 }
1274
1275 static HRESULT WINAPI InPlaceSite_CanInPlaceActivate(IOleInPlaceSiteEx *iface)
1276 {
1277     CHECK_EXPECT(CanInPlaceActivate);
1278     return S_OK;
1279 }
1280
1281 static HRESULT WINAPI InPlaceSite_OnInPlaceActivate(IOleInPlaceSiteEx *iface)
1282 {
1283     CHECK_EXPECT(OnInPlaceActivate);
1284     return S_OK;
1285 }
1286
1287 static HRESULT WINAPI InPlaceSite_OnUIActivate(IOleInPlaceSiteEx *iface)
1288 {
1289     CHECK_EXPECT(OnUIActivate);
1290     return S_OK;
1291 }
1292
1293 static HRESULT WINAPI InPlaceSite_GetWindowContext(IOleInPlaceSiteEx *iface,
1294         IOleInPlaceFrame **ppFrame, IOleInPlaceUIWindow **ppDoc, LPRECT lprcPosRect,
1295         LPRECT lprcClipRect, LPOLEINPLACEFRAMEINFO lpFrameInfo)
1296 {
1297     static const RECT pos_rect = {2,1,1002,901};
1298     static const RECT clip_rect = {10,10,990,890};
1299
1300     CHECK_EXPECT(GetWindowContext);
1301
1302     ok(ppFrame != NULL, "ppFrame = NULL\n");
1303     if(ppFrame)
1304         *ppFrame = &InPlaceFrame;
1305
1306     ok(ppDoc != NULL, "ppDoc = NULL\n");
1307     if(ppDoc)
1308         *ppDoc = &InPlaceUIWindow;
1309
1310     ok(lprcPosRect != NULL, "lprcPosRect = NULL\n");
1311     if(lprcPosRect)
1312         memcpy(lprcPosRect, &pos_rect, sizeof(RECT));
1313
1314     ok(lprcClipRect != NULL, "lprcClipRect = NULL\n");
1315     if(lprcClipRect)
1316         memcpy(lprcClipRect, &clip_rect, sizeof(RECT));
1317
1318     ok(lpFrameInfo != NULL, "lpFrameInfo = NULL\n");
1319     if(lpFrameInfo) {
1320         ok(lpFrameInfo->cb == sizeof(*lpFrameInfo), "lpFrameInfo->cb = %u, expected %u\n", lpFrameInfo->cb, (unsigned)sizeof(*lpFrameInfo));
1321         lpFrameInfo->fMDIApp = FALSE;
1322         lpFrameInfo->hwndFrame = container_hwnd;
1323         lpFrameInfo->haccel = NULL;
1324         lpFrameInfo->cAccelEntries = 0;
1325     }
1326
1327     return S_OK;
1328 }
1329
1330 static HRESULT WINAPI InPlaceSite_Scroll(IOleInPlaceSiteEx *iface, SIZE scrollExtant)
1331 {
1332     ok(0, "unexpected call\n");
1333     return E_NOTIMPL;
1334 }
1335
1336 static HRESULT WINAPI InPlaceSite_OnUIDeactivate(IOleInPlaceSiteEx *iface, BOOL fUndoable)
1337 {
1338     CHECK_EXPECT(OnUIDeactivate);
1339     ok(!fUndoable, "fUndoable should be FALSE\n");
1340     return S_OK;
1341 }
1342
1343 static HRESULT WINAPI InPlaceSite_OnInPlaceDeactivate(IOleInPlaceSiteEx *iface)
1344 {
1345     CHECK_EXPECT(OnInPlaceDeactivate);
1346     return S_OK;
1347 }
1348
1349 static HRESULT WINAPI InPlaceSite_DiscardUndoState(IOleInPlaceSiteEx *iface)
1350 {
1351     ok(0, "unexpected call\n");
1352     return E_NOTIMPL;
1353 }
1354
1355 static HRESULT WINAPI InPlaceSite_DeactivateAndUndo(IOleInPlaceSiteEx *iface)
1356 {
1357     ok(0, "unexpected call\n");
1358     return E_NOTIMPL;
1359 }
1360
1361 static HRESULT WINAPI InPlaceSite_OnPosRectChange(IOleInPlaceSiteEx *iface, LPCRECT lprcPosRect)
1362 {
1363     ok(0, "unexpected call\n");
1364     return E_NOTIMPL;
1365 }
1366
1367 static HRESULT WINAPI InPlaceSite_OnInPlaceActivateEx(IOleInPlaceSiteEx *iface,
1368                                                       BOOL *pNoRedraw, DWORD dwFlags)
1369 {
1370     ok(0, "unexpected call\n");
1371     return E_NOTIMPL;
1372 }
1373
1374 static HRESULT WINAPI InPlaceSite_OnInPlaceDeactivateEx(IOleInPlaceSiteEx *iface,
1375                                                         BOOL fNoRedraw)
1376 {
1377     ok(0, "unexpected call\n");
1378     return E_NOTIMPL;
1379 }
1380
1381 static HRESULT WINAPI InPlaceSite_RequestUIActivate(IOleInPlaceSiteEx *iface)
1382 {
1383     CHECK_EXPECT2(RequestUIActivate);
1384     return S_OK;
1385 }
1386
1387 static const IOleInPlaceSiteExVtbl InPlaceSiteExVtbl = {
1388     InPlaceSite_QueryInterface,
1389     InPlaceSite_AddRef,
1390     InPlaceSite_Release,
1391     InPlaceSite_GetWindow,
1392     InPlaceSite_ContextSensitiveHelp,
1393     InPlaceSite_CanInPlaceActivate,
1394     InPlaceSite_OnInPlaceActivate,
1395     InPlaceSite_OnUIActivate,
1396     InPlaceSite_GetWindowContext,
1397     InPlaceSite_Scroll,
1398     InPlaceSite_OnUIDeactivate,
1399     InPlaceSite_OnInPlaceDeactivate,
1400     InPlaceSite_DiscardUndoState,
1401     InPlaceSite_DeactivateAndUndo,
1402     InPlaceSite_OnPosRectChange,
1403     InPlaceSite_OnInPlaceActivateEx,
1404     InPlaceSite_OnInPlaceDeactivateEx,
1405     InPlaceSite_RequestUIActivate
1406 };
1407
1408 static IOleInPlaceSiteEx InPlaceSite = { &InPlaceSiteExVtbl };
1409
1410 static HRESULT WINAPI DocHostUIHandler_QueryInterface(IDocHostUIHandler2 *iface, REFIID riid, void **ppv)
1411 {
1412     return QueryInterface(riid, ppv);
1413 }
1414
1415 static ULONG WINAPI DocHostUIHandler_AddRef(IDocHostUIHandler2 *iface)
1416 {
1417     return 2;
1418 }
1419
1420 static ULONG WINAPI DocHostUIHandler_Release(IDocHostUIHandler2 *iface)
1421 {
1422     return 1;
1423 }
1424
1425 static HRESULT WINAPI DocHostUIHandler_ShowContextMenu(IDocHostUIHandler2 *iface, DWORD dwID, POINT *ppt,
1426         IUnknown *pcmdtReserved, IDispatch *pdicpReserved)
1427 {
1428     ok(0, "unexpected call %d %p %p %p\n", dwID, ppt, pcmdtReserved, pdicpReserved);
1429     return S_FALSE;
1430 }
1431
1432 static HRESULT WINAPI DocHostUIHandler_GetHostInfo(IDocHostUIHandler2 *iface, DOCHOSTUIINFO *pInfo)
1433 {
1434     CHECK_EXPECT2(GetHostInfo);
1435     ok(pInfo != NULL, "pInfo=NULL\n");
1436     if(pInfo) {
1437         ok(pInfo->cbSize == sizeof(DOCHOSTUIINFO) || broken(!pInfo->cbSize), "pInfo->cbSize=%u\n", pInfo->cbSize);
1438         ok(!pInfo->dwFlags, "pInfo->dwFlags=%08x, expected 0\n", pInfo->dwFlags);
1439         ok(!pInfo->dwDoubleClick, "pInfo->dwDoubleClick=%08x, expected 0\n", pInfo->dwDoubleClick);
1440         ok(!pInfo->pchHostCss, "pInfo->pchHostCss=%p, expected NULL\n", pInfo->pchHostCss);
1441         ok(!pInfo->pchHostNS, "pInfo->pchhostNS=%p, expected NULL\n", pInfo->pchHostNS);
1442     }
1443     return E_FAIL;
1444 }
1445
1446 static HRESULT WINAPI DocHostUIHandler_ShowUI(IDocHostUIHandler2 *iface, DWORD dwID,
1447         IOleInPlaceActiveObject *pActiveObject, IOleCommandTarget *pCommandTarget,
1448         IOleInPlaceFrame *pFrame, IOleInPlaceUIWindow *pDoc)
1449 {
1450     CHECK_EXPECT(ShowUI);
1451     return E_NOTIMPL;
1452 }
1453
1454 static HRESULT WINAPI DocHostUIHandler_HideUI(IDocHostUIHandler2 *iface)
1455 {
1456     CHECK_EXPECT(HideUI);
1457     return E_NOTIMPL;
1458 }
1459
1460 static HRESULT WINAPI DocHostUIHandler_UpdateUI(IDocHostUIHandler2 *iface)
1461 {
1462     CHECK_EXPECT(UpdateUI);
1463     return E_NOTIMPL;
1464 }
1465
1466 static HRESULT WINAPI DocHostUIHandler_EnableModeless(IDocHostUIHandler2 *iface, BOOL fEnable)
1467 {
1468     if(fEnable)
1469         CHECK_EXPECT(DocHost_EnableModeless_TRUE);
1470     else
1471         CHECK_EXPECT(DocHost_EnableModeless_FALSE);
1472
1473     return S_OK;
1474 }
1475
1476 static HRESULT WINAPI DocHostUIHandler_OnDocWindowActivate(IDocHostUIHandler2 *iface, BOOL fActivate)
1477 {
1478     ok(0, "unexpected call\n");
1479     return E_NOTIMPL;
1480 }
1481
1482 static HRESULT WINAPI DocHostUIHandler_OnFrameWindowActivate(IDocHostUIHandler2 *iface, BOOL fActivate)
1483 {
1484     ok(0, "unexpected call\n");
1485     return E_NOTIMPL;
1486 }
1487
1488 static HRESULT WINAPI DocHostUIHandler_ResizeBorder(IDocHostUIHandler2 *iface, LPCRECT prcBorder,
1489         IOleInPlaceUIWindow *pUIWindow, BOOL fRameWindow)
1490 {
1491     ok(0, "unexpected call\n");
1492     return E_NOTIMPL;
1493 }
1494
1495 static HRESULT WINAPI DocHostUIHandler_TranslateAccelerator(IDocHostUIHandler2 *iface, LPMSG lpMsg,
1496         const GUID *pguidCmdGroup, DWORD nCmdID)
1497 {
1498     CHECK_EXPECT(DocHost_TranslateAccelerator);
1499     ok(pguidCmdGroup != NULL, "Got NULL pguidCmdGroup.\n");
1500     if(pguidCmdGroup)
1501         ok(IsEqualGUID(pguidCmdGroup, &CGID_MSHTML), "Unexpected pguidCmdGroup\n");
1502     ok(lpMsg != NULL, "Got NULL lpMsg.\n");
1503     return hr_dochost_TranslateAccelerator;
1504 }
1505
1506 static HRESULT WINAPI DocHostUIHandler_GetOptionKeyPath(IDocHostUIHandler2 *iface,
1507         LPOLESTR *pchKey, DWORD dw)
1508 {
1509     CHECK_EXPECT(GetOptionKeyPath);
1510     ok(pchKey != NULL, "pchKey==NULL\n");
1511     if(pchKey)
1512         ok(*pchKey == NULL, "*pchKey=%p\n", *pchKey);
1513     ok(!dw, "dw=%x\n", dw);
1514     return E_NOTIMPL;
1515 }
1516
1517 static HRESULT WINAPI DocHostUIHandler_GetDropTarget(IDocHostUIHandler2 *iface,
1518         IDropTarget *pDropTarget, IDropTarget **ppDropTarget)
1519 {
1520     CHECK_EXPECT(GetDropTarget);
1521     return E_NOTIMPL;
1522 }
1523
1524 static HRESULT WINAPI DocHostUIHandler_GetExternal(IDocHostUIHandler2 *iface, IDispatch **ppDispatch)
1525 {
1526     ok(0, "unexpected call\n");
1527     return E_NOTIMPL;
1528 }
1529
1530 static HRESULT WINAPI DocHostUIHandler_TranslateUrl(IDocHostUIHandler2 *iface, DWORD dwTranslate,
1531         OLECHAR *pchURLIn, OLECHAR **ppchURLOut)
1532 {
1533     if(is_downloading && !(dwl_flags & DWL_EXPECT_BEFORE_NAVIGATE))
1534         todo_wine CHECK_EXPECT(TranslateUrl);
1535     else
1536         CHECK_EXPECT(TranslateUrl);
1537     return E_NOTIMPL;
1538 }
1539
1540 static HRESULT WINAPI DocHostUIHandler_FilterDataObject(IDocHostUIHandler2 *iface, IDataObject *pDO,
1541         IDataObject **ppPORet)
1542 {
1543     ok(0, "unexpected call\n");
1544     return E_NOTIMPL;
1545 }
1546
1547 static HRESULT WINAPI DocHostUIHandler_GetOverrideKeyPath(IDocHostUIHandler2 *iface,
1548         LPOLESTR *pchKey, DWORD dw)
1549 {
1550     CHECK_EXPECT(GetOverridesKeyPath);
1551     ok(pchKey != NULL, "pchKey==NULL\n");
1552     if(pchKey)
1553         ok(*pchKey == NULL, "*pchKey=%p\n", *pchKey);
1554     ok(!dw, "dw=%x\n", dw);
1555     return E_NOTIMPL;
1556 }
1557
1558 static const IDocHostUIHandler2Vtbl DocHostUIHandlerVtbl = {
1559     DocHostUIHandler_QueryInterface,
1560     DocHostUIHandler_AddRef,
1561     DocHostUIHandler_Release,
1562     DocHostUIHandler_ShowContextMenu,
1563     DocHostUIHandler_GetHostInfo,
1564     DocHostUIHandler_ShowUI,
1565     DocHostUIHandler_HideUI,
1566     DocHostUIHandler_UpdateUI,
1567     DocHostUIHandler_EnableModeless,
1568     DocHostUIHandler_OnDocWindowActivate,
1569     DocHostUIHandler_OnFrameWindowActivate,
1570     DocHostUIHandler_ResizeBorder,
1571     DocHostUIHandler_TranslateAccelerator,
1572     DocHostUIHandler_GetOptionKeyPath,
1573     DocHostUIHandler_GetDropTarget,
1574     DocHostUIHandler_GetExternal,
1575     DocHostUIHandler_TranslateUrl,
1576     DocHostUIHandler_FilterDataObject,
1577     DocHostUIHandler_GetOverrideKeyPath
1578 };
1579
1580 static IDocHostUIHandler2 DocHostUIHandler = { &DocHostUIHandlerVtbl };
1581
1582
1583 static HRESULT WINAPI ServiceProvider_QueryInterface(IServiceProvider *iface, REFIID riid, void **ppv)
1584 {
1585     return QueryInterface(riid, ppv);
1586 }
1587
1588 static ULONG WINAPI ServiceProvider_AddRef(IServiceProvider *iface)
1589 {
1590     return 2;
1591 }
1592
1593 static ULONG WINAPI ServiceProvider_Release(IServiceProvider *iface)
1594 {
1595     return 1;
1596 }
1597
1598 static HRESULT WINAPI ServiceProvider_QueryService(IServiceProvider *iface,
1599                                     REFGUID guidService, REFIID riid, void **ppv)
1600 {
1601     *ppv = NULL;
1602
1603     if(!winetest_interactive)
1604         return E_NOINTERFACE;
1605
1606     if (IsEqualGUID(&SID_STopLevelBrowser, guidService))
1607         trace("Service SID_STopLevelBrowser\n");
1608     else if (IsEqualGUID(&SID_SEditCommandTarget, guidService))
1609         trace("Service SID_SEditCommandTarget\n");
1610     else if (IsEqualGUID(&IID_ITargetFrame2, guidService))
1611         trace("Service IID_ITargetFrame2\n");
1612     else if (IsEqualGUID(&SID_SInternetSecurityManager, guidService))
1613         trace("Service SID_SInternetSecurityManager\n");
1614     else if (IsEqualGUID(&SID_SOleUndoManager, guidService))
1615         trace("Service SID_SOleUndoManager\n");
1616     else if (IsEqualGUID(&SID_IMimeInfo, guidService))
1617         trace("Service SID_IMimeInfo\n");
1618     else if (IsEqualGUID(&SID_STopWindow, guidService))
1619         trace("Service SID_STopWindow\n");
1620
1621     /* 30D02401-6A81-11D0-8274-00C04FD5AE38 Explorer Bar: Search */
1622     /* D1E7AFEC-6A2E-11D0-8C78-00C04FD918B4 no info */
1623     /* A9227C3C-7F8E-11D0-8CB0-00A0C92DBFE8 no info */
1624     /* 371EA634-DC5C-11D1-BA57-00C04FC2040E one reference to IVersionHost */
1625     /* 3050F429-98B5-11CF-BB82-00AA00BDCE0B IID_IElementBehaviorFactory */
1626     /* 6D12FE80-7911-11CF-9534-0000C05BAE0B SID_DefView */
1627     /* AD7F6C62-F6BD-11D2-959B-006097C553C8 no info */
1628     /* 53A2D5B1-D2FC-11D0-84E0-006097C9987D no info */
1629     /* 3050F312-98B5-11CF-BB82-00AA00BDCE0B HTMLFrameBaseClass */
1630     /* 639447BD-B2D3-44B9-9FB0-510F23CB45E4 no info */
1631     /* 20C46561-8491-11CF-960C-0080C7F4EE85 no info */
1632
1633     else
1634         trace("Service %s not supported\n", debugstr_guid(guidService));
1635
1636     return E_NOINTERFACE;
1637 }
1638
1639
1640 static const IServiceProviderVtbl ServiceProviderVtbl = {
1641     ServiceProvider_QueryInterface,
1642     ServiceProvider_AddRef,
1643     ServiceProvider_Release,
1644     ServiceProvider_QueryService
1645 };
1646
1647 static IServiceProvider ServiceProvider = { &ServiceProviderVtbl };
1648
1649
1650 static HRESULT QueryInterface(REFIID riid, void **ppv)
1651 {
1652     *ppv = NULL;
1653
1654     if(IsEqualGUID(&IID_IUnknown, riid)
1655             || IsEqualGUID(&IID_IOleClientSite, riid))
1656         *ppv = &ClientSite;
1657     else if(IsEqualGUID(&IID_IOleWindow, riid)
1658             || IsEqualGUID(&IID_IOleInPlaceSite, riid)
1659             || IsEqualGUID(&IID_IOleInPlaceSiteEx, riid))
1660         *ppv = &InPlaceSite;
1661     else if(IsEqualGUID(&IID_IDocHostUIHandler, riid)
1662             || IsEqualGUID(&IID_IDocHostUIHandler2, riid))
1663         *ppv = &DocHostUIHandler;
1664     else if(IsEqualGUID(&IID_IDispatch, riid))
1665         *ppv = &Dispatch;
1666     else if(IsEqualGUID(&IID_IServiceProvider, riid))
1667         *ppv = &ServiceProvider;
1668     else if(IsEqualGUID(&IID_IDocHostShowUI, riid))
1669         trace("interface IID_IDocHostShowUI\n");
1670     else if(IsEqualGUID(&IID_IOleControlSite, riid))
1671         *ppv = &ControlSite;
1672     else if(IsEqualGUID(&IID_IOleCommandTarget, riid))
1673         trace("interface IID_IOleCommandTarget\n");
1674
1675     /* B6EA2050-048A-11D1-82B9-00C04FB9942E IAxWinHostWindow */
1676
1677     else
1678     {
1679         /* are there more interfaces, that a host can support? */
1680         trace("%s: interface not supported\n", debugstr_guid(riid));
1681     }
1682
1683     return (*ppv) ? S_OK : E_NOINTERFACE;
1684 }
1685
1686 static LRESULT WINAPI wnd_proc(HWND hwnd, UINT msg, WPARAM wParam, LPARAM lParam)
1687 {
1688     return DefWindowProcW(hwnd, msg, wParam, lParam);
1689 }
1690
1691 static HWND create_container_window(void)
1692 {
1693     static const WCHAR wszWebBrowserContainer[] =
1694         {'W','e','b','B','r','o','w','s','e','r','C','o','n','t','a','i','n','e','r',0};
1695     static WNDCLASSEXW wndclass = {
1696         sizeof(WNDCLASSEXW),
1697         0,
1698         wnd_proc,
1699         0, 0, NULL, NULL, NULL, NULL, NULL,
1700         wszWebBrowserContainer,
1701         NULL
1702     };
1703
1704     RegisterClassExW(&wndclass);
1705     return CreateWindowW(wszWebBrowserContainer, wszWebBrowserContainer,
1706             WS_OVERLAPPEDWINDOW, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
1707             CW_USEDEFAULT, NULL, NULL, NULL, NULL);
1708 }
1709
1710 static void test_DoVerb(IUnknown *unk)
1711 {
1712     IOleObject *oleobj;
1713     RECT rect = {0,0,1000,1000};
1714     HRESULT hres;
1715
1716     hres = IUnknown_QueryInterface(unk, &IID_IOleObject, (void**)&oleobj);
1717     ok(hres == S_OK, "QueryInterface(IID_OleObject) failed: %08x\n", hres);
1718     if(FAILED(hres))
1719         return;
1720
1721     SET_EXPECT(CanInPlaceActivate);
1722     SET_EXPECT(Site_GetWindow);
1723     SET_EXPECT(OnInPlaceActivate);
1724     SET_EXPECT(GetWindowContext);
1725     SET_EXPECT(ShowObject);
1726     SET_EXPECT(GetContainer);
1727     SET_EXPECT(Frame_GetWindow);
1728     SET_EXPECT(OnUIActivate);
1729     SET_EXPECT(Frame_SetActiveObject);
1730     SET_EXPECT(UIWindow_SetActiveObject);
1731     SET_EXPECT(SetMenu);
1732     SET_EXPECT(OnFocus);
1733
1734     hres = IOleObject_DoVerb(oleobj, OLEIVERB_SHOW, NULL, &ClientSite,
1735                              0, (HWND)0xdeadbeef, &rect);
1736     ok(hres == S_OK, "DoVerb failed: %08x\n", hres);
1737
1738     CHECK_CALLED(CanInPlaceActivate);
1739     CHECK_CALLED(Site_GetWindow);
1740     CHECK_CALLED(OnInPlaceActivate);
1741     CHECK_CALLED(GetWindowContext);
1742     CHECK_CALLED(ShowObject);
1743     CHECK_CALLED(GetContainer);
1744     CHECK_CALLED(Frame_GetWindow);
1745     CHECK_CALLED(OnUIActivate);
1746     CHECK_CALLED(Frame_SetActiveObject);
1747     CHECK_CALLED(UIWindow_SetActiveObject);
1748     CHECK_CALLED(SetMenu);
1749     todo_wine CHECK_CALLED(OnFocus);
1750
1751     hres = IOleObject_DoVerb(oleobj, OLEIVERB_SHOW, NULL, &ClientSite,
1752                            0, (HWND)0xdeadbeef, &rect);
1753     ok(hres == S_OK, "DoVerb failed: %08x\n", hres);
1754
1755     IOleObject_Release(oleobj);
1756 }
1757
1758 static void test_GetMiscStatus(IOleObject *oleobj)
1759 {
1760     DWORD st, i;
1761     HRESULT hres;
1762
1763     for(i=0; i<10; i++) {
1764         st = 0xdeadbeef;
1765         hres = IOleObject_GetMiscStatus(oleobj, i, &st);
1766         ok(hres == S_OK, "GetMiscStatus failed: %08x\n", hres);
1767         ok(st == (OLEMISC_SETCLIENTSITEFIRST|OLEMISC_ACTIVATEWHENVISIBLE|OLEMISC_INSIDEOUT
1768                   |OLEMISC_CANTLINKINSIDE|OLEMISC_RECOMPOSEONRESIZE),
1769            "st=%08x, expected OLEMISC_SETCLIENTSITEFIRST|OLEMISC_ACTIVATEWHENVISIBLE|"
1770            "OLEMISC_INSIDEOUT|OLEMISC_CANTLINKINSIDE|OLEMISC_RECOMPOSEONRESIZE)\n", st);
1771     }
1772 }
1773
1774 static void test_SetHostNames(IOleObject *oleobj)
1775 {
1776     HRESULT hres;
1777
1778     static const WCHAR test_appW[] =  {'t','e','s','t',' ','a','p','p',0};
1779
1780     hres = IOleObject_SetHostNames(oleobj, test_appW, (void*)0xdeadbeef);
1781     ok(hres == S_OK, "SetHostNames failed: %08x\n", hres);
1782 }
1783
1784 static void test_ClientSite(IUnknown *unk, IOleClientSite *client, BOOL stop_download)
1785 {
1786     IOleObject *oleobj;
1787     IOleInPlaceObject *inplace;
1788     HWND hwnd;
1789     HRESULT hres;
1790
1791     hres = IUnknown_QueryInterface(unk, &IID_IOleObject, (void**)&oleobj);
1792     ok(hres == S_OK, "QueryInterface(IID_OleObject) failed: %08x\n", hres);
1793     if(FAILED(hres))
1794         return;
1795
1796     test_GetMiscStatus(oleobj);
1797     test_SetHostNames(oleobj);
1798
1799     hres = IUnknown_QueryInterface(unk, &IID_IOleInPlaceObject, (void**)&inplace);
1800     ok(hres == S_OK, "QueryInterface(IID_OleInPlaceObject) failed: %08x\n", hres);
1801     if(FAILED(hres)) {
1802         IOleObject_Release(oleobj);
1803         return;
1804     }
1805
1806     hres = IOleInPlaceObject_GetWindow(inplace, &hwnd);
1807     ok(hres == S_OK, "GetWindow failed: %08x\n", hres);
1808     ok((hwnd == NULL) ^ (client == NULL), "unexpected hwnd %p\n", hwnd);
1809
1810     if(client) {
1811         SET_EXPECT(GetContainer);
1812         SET_EXPECT(Site_GetWindow);
1813         SET_EXPECT(Invoke_AMBIENT_OFFLINEIFNOTCONNECTED);
1814         SET_EXPECT(Invoke_AMBIENT_SILENT);
1815     }else if(stop_download) {
1816         SET_EXPECT(Invoke_DOWNLOADCOMPLETE);
1817         if (use_container_olecmd) SET_EXPECT(Exec_SETDOWNLOADSTATE_0);
1818         SET_EXPECT(Invoke_COMMANDSTATECHANGE);
1819     }else {
1820         SET_EXPECT(Invoke_STATUSTEXTCHANGE);
1821     }
1822
1823     hres = IOleObject_SetClientSite(oleobj, client);
1824     ok(hres == S_OK, "SetClientSite failed: %08x\n", hres);
1825
1826     if(client) {
1827         CHECK_CALLED(GetContainer);
1828         CHECK_CALLED(Site_GetWindow);
1829         CHECK_CALLED(Invoke_AMBIENT_OFFLINEIFNOTCONNECTED);
1830         CHECK_CALLED(Invoke_AMBIENT_SILENT);
1831     }else if(stop_download) {
1832         todo_wine CHECK_CALLED(Invoke_DOWNLOADCOMPLETE);
1833         if (use_container_olecmd) todo_wine CHECK_CALLED(Exec_SETDOWNLOADSTATE_0);
1834         todo_wine CHECK_CALLED(Invoke_COMMANDSTATECHANGE);
1835     }else {
1836         CLEAR_CALLED(Invoke_STATUSTEXTCHANGE); /* Called by IE9 */
1837     }
1838
1839     hres = IOleInPlaceObject_GetWindow(inplace, &hwnd);
1840     ok(hres == S_OK, "GetWindow failed: %08x\n", hres);
1841     ok((hwnd == NULL) == (client == NULL), "unexpected hwnd %p\n", hwnd);
1842
1843     shell_embedding_hwnd = hwnd;
1844
1845     test_SetHostNames(oleobj);
1846
1847     IOleInPlaceObject_Release(inplace);
1848     IOleObject_Release(oleobj);
1849 }
1850
1851 static void test_ClassInfo(IUnknown *unk)
1852 {
1853     IProvideClassInfo2 *class_info;
1854     GUID guid;
1855     HRESULT hres;
1856
1857     hres = IUnknown_QueryInterface(unk, &IID_IProvideClassInfo2, (void**)&class_info);
1858     ok(hres == S_OK, "QueryInterface(IID_IProvideClassInfo)  failed: %08x\n", hres);
1859     if(FAILED(hres))
1860         return;
1861
1862     hres = IProvideClassInfo2_GetGUID(class_info, GUIDKIND_DEFAULT_SOURCE_DISP_IID, &guid);
1863     ok(hres == S_OK, "GetGUID failed: %08x\n", hres);
1864     ok(IsEqualGUID(&DIID_DWebBrowserEvents2, &guid), "wrong guid\n");
1865
1866     hres = IProvideClassInfo2_GetGUID(class_info, 0, &guid);
1867     ok(hres == E_FAIL, "GetGUID failed: %08x, expected E_FAIL\n", hres);
1868     ok(IsEqualGUID(&IID_NULL, &guid), "wrong guid\n");
1869
1870     hres = IProvideClassInfo2_GetGUID(class_info, 2, &guid);
1871     ok(hres == E_FAIL, "GetGUID failed: %08x, expected E_FAIL\n", hres);
1872     ok(IsEqualGUID(&IID_NULL, &guid), "wrong guid\n");
1873
1874     hres = IProvideClassInfo2_GetGUID(class_info, GUIDKIND_DEFAULT_SOURCE_DISP_IID, NULL);
1875     ok(hres == E_POINTER, "GetGUID failed: %08x, expected E_POINTER\n", hres);
1876
1877     hres = IProvideClassInfo2_GetGUID(class_info, 0, NULL);
1878     ok(hres == E_POINTER, "GetGUID failed: %08x, expected E_POINTER\n", hres);
1879
1880     IProvideClassInfo2_Release(class_info);
1881 }
1882
1883 static void test_ie_funcs(IUnknown *unk)
1884 {
1885     IWebBrowser2 *wb;
1886     IDispatch *disp;
1887     VARIANT_BOOL b;
1888     int i;
1889     LONG hwnd;
1890     HRESULT hres;
1891     BSTR sName;
1892
1893     hres = IUnknown_QueryInterface(unk, &IID_IWebBrowser2, (void**)&wb);
1894     ok(hres == S_OK, "Could not get IWebBrowser2 interface: %08x\n", hres);
1895     if(FAILED(hres))
1896         return;
1897
1898     /* HWND */
1899
1900     hwnd = 0xdeadbeef;
1901     hres = IWebBrowser2_get_HWND(wb, &hwnd);
1902     ok(hres == E_FAIL, "get_HWND failed: %08x, expected E_FAIL\n", hres);
1903     ok(hwnd == 0, "unexpected hwnd %x\n", hwnd);
1904
1905     /* MenuBar */
1906
1907     hres = IWebBrowser2_get_MenuBar(wb, &b);
1908     ok(hres == S_OK, "get_MenuBar failed: %08x\n", hres);
1909     ok(b == VARIANT_TRUE, "b=%x\n", b);
1910
1911     SET_EXPECT(Invoke_ONMENUBAR);
1912     hres = IWebBrowser2_put_MenuBar(wb, (exvb = VARIANT_FALSE));
1913     ok(hres == S_OK, "put_MenuBar failed: %08x\n", hres);
1914     CHECK_CALLED(Invoke_ONMENUBAR);
1915
1916     hres = IWebBrowser2_get_MenuBar(wb, &b);
1917     ok(hres == S_OK, "get_MenuBar failed: %08x\n", hres);
1918     ok(b == VARIANT_FALSE, "b=%x\n", b);
1919
1920     SET_EXPECT(Invoke_ONMENUBAR);
1921     hres = IWebBrowser2_put_MenuBar(wb, (exvb = 100));
1922     ok(hres == S_OK, "put_MenuBar failed: %08x\n", hres);
1923     CHECK_CALLED(Invoke_ONMENUBAR);
1924
1925     hres = IWebBrowser2_get_MenuBar(wb, &b);
1926     ok(hres == S_OK, "get_MenuBar failed: %08x\n", hres);
1927     ok(b == VARIANT_TRUE, "b=%x\n", b);
1928
1929     /* AddressBar */
1930
1931     hres = IWebBrowser2_get_AddressBar(wb, &b);
1932     ok(hres == S_OK, "get_AddressBar failed: %08x\n", hres);
1933     ok(b == VARIANT_TRUE, "b=%x\n", b);
1934
1935     SET_EXPECT(Invoke_ONADDRESSBAR);
1936     hres = IWebBrowser2_put_AddressBar(wb, (exvb = VARIANT_FALSE));
1937     ok(hres == S_OK, "put_AddressBar failed: %08x\n", hres);
1938     CHECK_CALLED(Invoke_ONADDRESSBAR);
1939
1940     hres = IWebBrowser2_get_AddressBar(wb, &b);
1941     ok(hres == S_OK, "get_MenuBar failed: %08x\n", hres);
1942     ok(b == VARIANT_FALSE, "b=%x\n", b);
1943
1944     SET_EXPECT(Invoke_ONADDRESSBAR);
1945     hres = IWebBrowser2_put_AddressBar(wb, (exvb = 100));
1946     ok(hres == S_OK, "put_AddressBar failed: %08x\n", hres);
1947     CHECK_CALLED(Invoke_ONADDRESSBAR);
1948
1949     hres = IWebBrowser2_get_AddressBar(wb, &b);
1950     ok(hres == S_OK, "get_AddressBar failed: %08x\n", hres);
1951     ok(b == VARIANT_TRUE, "b=%x\n", b);
1952
1953     SET_EXPECT(Invoke_ONADDRESSBAR);
1954     hres = IWebBrowser2_put_AddressBar(wb, (exvb = VARIANT_TRUE));
1955     ok(hres == S_OK, "put_MenuBar failed: %08x\n", hres);
1956     CHECK_CALLED(Invoke_ONADDRESSBAR);
1957
1958     /* StatusBar */
1959
1960     hres = IWebBrowser2_get_StatusBar(wb, &b);
1961     ok(hres == S_OK, "get_StatusBar failed: %08x\n", hres);
1962     ok(b == VARIANT_TRUE, "b=%x\n", b);
1963
1964     SET_EXPECT(Invoke_ONSTATUSBAR);
1965     hres = IWebBrowser2_put_StatusBar(wb, (exvb = VARIANT_TRUE));
1966     ok(hres == S_OK, "put_StatusBar failed: %08x\n", hres);
1967     CHECK_CALLED(Invoke_ONSTATUSBAR);
1968
1969     hres = IWebBrowser2_get_StatusBar(wb, &b);
1970     ok(hres == S_OK, "get_StatusBar failed: %08x\n", hres);
1971     ok(b == VARIANT_TRUE, "b=%x\n", b);
1972
1973     SET_EXPECT(Invoke_ONSTATUSBAR);
1974     hres = IWebBrowser2_put_StatusBar(wb, (exvb = VARIANT_FALSE));
1975     ok(hres == S_OK, "put_StatusBar failed: %08x\n", hres);
1976     CHECK_CALLED(Invoke_ONSTATUSBAR);
1977
1978     hres = IWebBrowser2_get_StatusBar(wb, &b);
1979     ok(hres == S_OK, "get_StatusBar failed: %08x\n", hres);
1980     ok(b == VARIANT_FALSE, "b=%x\n", b);
1981
1982     SET_EXPECT(Invoke_ONSTATUSBAR);
1983     hres = IWebBrowser2_put_StatusBar(wb, (exvb = 100));
1984     ok(hres == S_OK, "put_StatusBar failed: %08x\n", hres);
1985     CHECK_CALLED(Invoke_ONSTATUSBAR);
1986
1987     hres = IWebBrowser2_get_StatusBar(wb, &b);
1988     ok(hres == S_OK, "get_StatusBar failed: %08x\n", hres);
1989     ok(b == VARIANT_TRUE, "b=%x\n", b);
1990
1991     /* ToolBar */
1992
1993     hres = IWebBrowser2_get_ToolBar(wb, &i);
1994     ok(hres == S_OK, "get_ToolBar failed: %08x\n", hres);
1995     ok(i, "i=%x\n", i);
1996
1997     SET_EXPECT(Invoke_ONTOOLBAR);
1998     hres = IWebBrowser2_put_ToolBar(wb, (exvb = VARIANT_FALSE));
1999     ok(hres == S_OK, "put_ToolBar failed: %08x\n", hres);
2000     CHECK_CALLED(Invoke_ONTOOLBAR);
2001
2002     hres = IWebBrowser2_get_ToolBar(wb, &i);
2003     ok(hres == S_OK, "get_ToolBar failed: %08x\n", hres);
2004     ok(i == VARIANT_FALSE, "b=%x\n", i);
2005
2006     SET_EXPECT(Invoke_ONTOOLBAR);
2007     hres = IWebBrowser2_put_ToolBar(wb, (exvb = 100));
2008     ok(hres == S_OK, "put_ToolBar failed: %08x\n", hres);
2009     CHECK_CALLED(Invoke_ONTOOLBAR);
2010
2011     hres = IWebBrowser2_get_ToolBar(wb, &i);
2012     ok(hres == S_OK, "get_ToolBar failed: %08x\n", hres);
2013     ok(i, "i=%x\n", i);
2014
2015     SET_EXPECT(Invoke_ONTOOLBAR);
2016     hres = IWebBrowser2_put_ToolBar(wb, (exvb = VARIANT_TRUE));
2017     ok(hres == S_OK, "put_ToolBar failed: %08x\n", hres);
2018     CHECK_CALLED(Invoke_ONTOOLBAR);
2019
2020     /* FullScreen */
2021
2022     hres = IWebBrowser2_get_FullScreen(wb, &b);
2023     ok(hres == S_OK, "get_FullScreen failed: %08x\n", hres);
2024     ok(b == VARIANT_FALSE, "b=%x\n", b);
2025
2026     SET_EXPECT(Invoke_ONFULLSCREEN);
2027     hres = IWebBrowser2_put_FullScreen(wb, (exvb = VARIANT_TRUE));
2028     ok(hres == S_OK, "put_FullScreen failed: %08x\n", hres);
2029     CHECK_CALLED(Invoke_ONFULLSCREEN);
2030
2031     hres = IWebBrowser2_get_FullScreen(wb, &b);
2032     ok(hres == S_OK, "get_FullScreen failed: %08x\n", hres);
2033     ok(b == VARIANT_TRUE, "b=%x\n", b);
2034
2035     SET_EXPECT(Invoke_ONFULLSCREEN);
2036     hres = IWebBrowser2_put_FullScreen(wb, (exvb = 100));
2037     ok(hres == S_OK, "put_FullScreen failed: %08x\n", hres);
2038     CHECK_CALLED(Invoke_ONFULLSCREEN);
2039
2040     hres = IWebBrowser2_get_FullScreen(wb, &b);
2041     ok(hres == S_OK, "get_FullScreen failed: %08x\n", hres);
2042     ok(b == VARIANT_TRUE, "b=%x\n", b);
2043
2044     SET_EXPECT(Invoke_ONFULLSCREEN);
2045     hres = IWebBrowser2_put_FullScreen(wb, (exvb = VARIANT_FALSE));
2046     ok(hres == S_OK, "put_FullScreen failed: %08x\n", hres);
2047     CHECK_CALLED(Invoke_ONFULLSCREEN);
2048
2049     /* TheaterMode */
2050
2051     hres = IWebBrowser2_get_TheaterMode(wb, &b);
2052     ok(hres == S_OK, "get_TheaterMode failed: %08x\n", hres);
2053     ok(b == VARIANT_FALSE, "b=%x\n", b);
2054
2055     SET_EXPECT(Invoke_ONTHEATERMODE);
2056     hres = IWebBrowser2_put_TheaterMode(wb, (exvb = VARIANT_TRUE));
2057     ok(hres == S_OK, "put_TheaterMode failed: %08x\n", hres);
2058     CHECK_CALLED(Invoke_ONTHEATERMODE);
2059
2060     hres = IWebBrowser2_get_TheaterMode(wb, &b);
2061     ok(hres == S_OK, "get_TheaterMode failed: %08x\n", hres);
2062     ok(b == VARIANT_TRUE, "b=%x\n", b);
2063
2064     SET_EXPECT(Invoke_ONTHEATERMODE);
2065     hres = IWebBrowser2_put_TheaterMode(wb, (exvb = 100));
2066     ok(hres == S_OK, "put_TheaterMode failed: %08x\n", hres);
2067     CHECK_CALLED(Invoke_ONTHEATERMODE);
2068
2069     hres = IWebBrowser2_get_TheaterMode(wb, &b);
2070     ok(hres == S_OK, "get_TheaterMode failed: %08x\n", hres);
2071     ok(b == VARIANT_TRUE, "b=%x\n", b);
2072
2073     SET_EXPECT(Invoke_ONTHEATERMODE);
2074     hres = IWebBrowser2_put_TheaterMode(wb, (exvb = VARIANT_FALSE));
2075     ok(hres == S_OK, "put_TheaterMode failed: %08x\n", hres);
2076     CHECK_CALLED(Invoke_ONTHEATERMODE);
2077
2078     /* Resizable */
2079
2080     b = 0x100;
2081     hres = IWebBrowser2_get_Resizable(wb, &b);
2082     ok(hres == E_NOTIMPL, "get_Resizable failed: %08x\n", hres);
2083     ok(b == 0x100, "b=%x\n", b);
2084
2085     SET_EXPECT(Invoke_WINDOWSETRESIZABLE);
2086     hres = IWebBrowser2_put_Resizable(wb, (exvb = VARIANT_TRUE));
2087     ok(hres == S_OK, "put_Resizable failed: %08x\n", hres);
2088     CHECK_CALLED(Invoke_WINDOWSETRESIZABLE);
2089
2090     SET_EXPECT(Invoke_WINDOWSETRESIZABLE);
2091     hres = IWebBrowser2_put_Resizable(wb, (exvb = VARIANT_FALSE));
2092     ok(hres == S_OK, "put_Resizable failed: %08x\n", hres);
2093     CHECK_CALLED(Invoke_WINDOWSETRESIZABLE);
2094
2095     hres = IWebBrowser2_get_Resizable(wb, &b);
2096     ok(hres == E_NOTIMPL, "get_Resizable failed: %08x\n", hres);
2097     ok(b == 0x100, "b=%x\n", b);
2098
2099     /* Application */
2100
2101     disp = NULL;
2102     hres = IWebBrowser2_get_Application(wb, &disp);
2103     ok(hres == S_OK, "get_Application failed: %08x\n", hres);
2104     ok(disp == (void*)wb, "disp=%p, expected %p\n", disp, wb);
2105     if(disp)
2106         IDispatch_Release(disp);
2107
2108     hres = IWebBrowser2_get_Application(wb, NULL);
2109     ok(hres == E_POINTER, "get_Application failed: %08x, expected E_POINTER\n", hres);
2110
2111     /* Name */
2112     hres = IWebBrowser2_get_Name(wb, &sName);
2113     ok(hres == S_OK, "getName failed: %08x, expected S_OK\n", hres);
2114     if (is_lang_english())
2115         ok(!strcmp_wa(sName, "Microsoft Web Browser Control"), "got '%s', expected 'Microsoft Web Browser Control'\n", wine_dbgstr_w(sName));
2116     else /* Non-English cannot be blank. */
2117         ok(sName!=NULL, "get_Name return a NULL string.\n");
2118     SysFreeString(sName);
2119
2120     /* Quit */
2121
2122     hres = IWebBrowser2_Quit(wb);
2123     ok(hres == E_FAIL, "Quit failed: %08x, expected E_FAIL\n", hres);
2124
2125     IWebBrowser2_Release(wb);
2126 }
2127
2128 static void test_Silent(IWebBrowser2 *wb, IOleControl *control, BOOL is_clientsite)
2129 {
2130     VARIANT_BOOL b;
2131     HRESULT hres;
2132
2133     b = 100;
2134     hres = IWebBrowser2_get_Silent(wb, &b);
2135     ok(hres == S_OK, "get_Silent failed: %08x\n", hres);
2136     ok(b == VARIANT_FALSE, "b=%x\n", b);
2137
2138     hres = IWebBrowser2_put_Silent(wb, VARIANT_TRUE);
2139     ok(hres == S_OK, "set_Silent failed: %08x\n", hres);
2140
2141     b = 100;
2142     hres = IWebBrowser2_get_Silent(wb, &b);
2143     ok(hres == S_OK, "get_Silent failed: %08x\n", hres);
2144     ok(b == VARIANT_TRUE, "b=%x\n", b);
2145
2146     hres = IWebBrowser2_put_Silent(wb, 100);
2147     ok(hres == S_OK, "set_Silent failed: %08x\n", hres);
2148
2149     b = 100;
2150     hres = IWebBrowser2_get_Silent(wb, &b);
2151     ok(hres == S_OK, "get_Silent failed: %08x\n", hres);
2152     ok(b == VARIANT_TRUE, "b=%x\n", b);
2153
2154     hres = IWebBrowser2_put_Silent(wb, VARIANT_FALSE);
2155     ok(hres == S_OK, "set_Silent failed: %08x\n", hres);
2156
2157     b = 100;
2158     hres = IWebBrowser2_get_Silent(wb, &b);
2159     ok(hres == S_OK, "get_Silent failed: %08x\n", hres);
2160     ok(b == VARIANT_FALSE, "b=%x\n", b);
2161
2162     if(is_clientsite) {
2163         hres = IWebBrowser2_put_Silent(wb, VARIANT_TRUE);
2164         ok(hres == S_OK, "set_Silent failed: %08x\n", hres);
2165
2166         SET_EXPECT(Invoke_AMBIENT_SILENT);
2167     }
2168
2169     hres = IOleControl_OnAmbientPropertyChange(control, DISPID_AMBIENT_SILENT);
2170     ok(hres == S_OK, "OnAmbientPropertyChange failed %08x\n", hres);
2171
2172     if(is_clientsite)
2173         CHECK_CALLED(Invoke_AMBIENT_SILENT);
2174
2175     b = 100;
2176     hres = IWebBrowser2_get_Silent(wb, &b);
2177     ok(hres == S_OK, "get_Silent failed: %08x\n", hres);
2178     ok(b == VARIANT_FALSE, "b=%x\n", b);
2179 }
2180
2181 static void test_Offline(IWebBrowser2 *wb, IOleControl *control, BOOL is_clientsite)
2182 {
2183     VARIANT_BOOL b;
2184     HRESULT hres;
2185
2186     b = 100;
2187     hres = IWebBrowser2_get_Offline(wb, &b);
2188     ok(hres == S_OK, "get_Offline failed: %08x\n", hres);
2189     ok(b == VARIANT_FALSE, "b=%x\n", b);
2190
2191     hres = IWebBrowser2_put_Offline(wb, VARIANT_TRUE);
2192     ok(hres == S_OK, "set_Offline failed: %08x\n", hres);
2193
2194     b = 100;
2195     hres = IWebBrowser2_get_Offline(wb, &b);
2196     ok(hres == S_OK, "get_Offline failed: %08x\n", hres);
2197     ok(b == VARIANT_TRUE, "b=%x\n", b);
2198
2199     hres = IWebBrowser2_put_Offline(wb, 100);
2200     ok(hres == S_OK, "set_Offline failed: %08x\n", hres);
2201
2202     b = 100;
2203     hres = IWebBrowser2_get_Offline(wb, &b);
2204     ok(hres == S_OK, "get_Offline failed: %08x\n", hres);
2205     ok(b == VARIANT_TRUE, "b=%x\n", b);
2206
2207     hres = IWebBrowser2_put_Offline(wb, VARIANT_FALSE);
2208     ok(hres == S_OK, "set_Offline failed: %08x\n", hres);
2209
2210     b = 100;
2211     hres = IWebBrowser2_get_Offline(wb, &b);
2212     ok(hres == S_OK, "get_Offline failed: %08x\n", hres);
2213     ok(b == VARIANT_FALSE, "b=%x\n", b);
2214
2215     if(is_clientsite) {
2216         hres = IWebBrowser2_put_Offline(wb, VARIANT_TRUE);
2217         ok(hres == S_OK, "set_Offline failed: %08x\n", hres);
2218
2219         SET_EXPECT(Invoke_AMBIENT_OFFLINEIFNOTCONNECTED);
2220     }
2221
2222     hres = IOleControl_OnAmbientPropertyChange(control, DISPID_AMBIENT_OFFLINEIFNOTCONNECTED);
2223     ok(hres == S_OK, "OnAmbientPropertyChange failed %08x\n", hres);
2224
2225     if(is_clientsite)
2226         CHECK_CALLED(Invoke_AMBIENT_OFFLINEIFNOTCONNECTED);
2227
2228     b = 100;
2229     hres = IWebBrowser2_get_Offline(wb, &b);
2230     ok(hres == S_OK, "get_Offline failed: %08x\n", hres);
2231     ok(b == VARIANT_FALSE, "b=%x\n", b);
2232 }
2233
2234 static void test_ambient_unknown(IWebBrowser2 *wb, IOleControl *control, BOOL is_clientsite)
2235 {
2236     HRESULT hres;
2237
2238     SET_EXPECT(Invoke_AMBIENT_OFFLINEIFNOTCONNECTED);
2239     SET_EXPECT(Invoke_AMBIENT_SILENT);
2240     SET_EXPECT(Invoke_AMBIENT_USERMODE);
2241     SET_EXPECT(Invoke_AMBIENT_DLCONTROL);
2242     SET_EXPECT(Invoke_AMBIENT_USERAGENT);
2243     SET_EXPECT(Invoke_AMBIENT_PALETTE);
2244
2245     hres = IOleControl_OnAmbientPropertyChange(control, DISPID_UNKNOWN);
2246     ok(hres == S_OK, "OnAmbientPropertyChange failed %08x\n", hres);
2247
2248     CHECK_EXPECT(Invoke_AMBIENT_OFFLINEIFNOTCONNECTED);
2249     CHECK_EXPECT(Invoke_AMBIENT_SILENT);
2250     CHECK_EXPECT(Invoke_AMBIENT_USERMODE);
2251     CHECK_EXPECT(Invoke_AMBIENT_DLCONTROL);
2252     CHECK_EXPECT(Invoke_AMBIENT_USERAGENT);
2253     CHECK_EXPECT(Invoke_AMBIENT_PALETTE);
2254
2255     CLEAR_CALLED(Invoke_AMBIENT_OFFLINEIFNOTCONNECTED);
2256     CLEAR_CALLED(Invoke_AMBIENT_SILENT);
2257     CLEAR_CALLED(Invoke_AMBIENT_USERMODE);
2258     CLEAR_CALLED(Invoke_AMBIENT_DLCONTROL);
2259     CLEAR_CALLED(Invoke_AMBIENT_USERAGENT);
2260     CLEAR_CALLED(Invoke_AMBIENT_PALETTE);
2261 }
2262
2263 static void test_wb_funcs(IUnknown *unk, BOOL is_clientsite)
2264 {
2265     IWebBrowser2 *wb;
2266     IOleControl *control;
2267     HRESULT hres;
2268
2269     hres = IUnknown_QueryInterface(unk, &IID_IWebBrowser2, (void**)&wb);
2270     ok(hres == S_OK, "Could not get IWebBrowser2 interface: %08x\n", hres);
2271
2272     hres = IUnknown_QueryInterface(unk, &IID_IOleControl, (void**)&control);
2273     ok(hres == S_OK, "Could not get IOleControl interface: %08x\n", hres);
2274
2275     test_Silent(wb, control, is_clientsite);
2276     test_Offline(wb, control, is_clientsite);
2277     test_ambient_unknown(wb, control, is_clientsite);
2278
2279     IWebBrowser_Release(wb);
2280     IOleControl_Release(control);
2281 }
2282
2283 static void test_GetControlInfo(IUnknown *unk)
2284 {
2285     IOleControl *control;
2286     CONTROLINFO info;
2287     HRESULT hres;
2288
2289     hres = IUnknown_QueryInterface(unk, &IID_IOleControl, (void**)&control);
2290     ok(hres == S_OK, "Could not get IOleControl: %08x\n", hres);
2291     if(FAILED(hres))
2292         return;
2293
2294     hres = IOleControl_GetControlInfo(control, &info);
2295     ok(hres == E_NOTIMPL, "GetControlInfo failed: %08x, expected E_NOTIMPL\n", hres);
2296     hres = IOleControl_GetControlInfo(control, NULL);
2297     ok(hres == E_NOTIMPL, "GetControlInfo failed: %08x, expected E_NOTIMPL\n", hres);
2298
2299     IOleControl_Release(control);
2300 }
2301
2302 static void test_Extent(IUnknown *unk)
2303 {
2304     IOleObject *oleobj;
2305     SIZE size, expected;
2306     HRESULT hres;
2307     DWORD dpi_x;
2308     DWORD dpi_y;
2309     HDC hdc;
2310
2311     /* default aspect ratio is 96dpi / 96dpi */
2312     hdc = GetDC(0);
2313     dpi_x = GetDeviceCaps(hdc, LOGPIXELSX);
2314     dpi_y = GetDeviceCaps(hdc, LOGPIXELSY);
2315     ReleaseDC(0, hdc);
2316     if (dpi_x != 96 || dpi_y != 96)
2317         trace("dpi: %d / %d\n", dpi_y, dpi_y);
2318
2319     hres = IUnknown_QueryInterface(unk, &IID_IOleObject, (void**)&oleobj);
2320     ok(hres == S_OK, "Could not get IOleObkect: %08x\n", hres);
2321     if(FAILED(hres))
2322         return;
2323
2324     size.cx = size.cy = 0xdeadbeef;
2325     hres = IOleObject_GetExtent(oleobj, DVASPECT_CONTENT, &size);
2326     ok(hres == S_OK, "GetExtent failed: %08x\n", hres);
2327     /* Default size is 50x20 pixels, in himetric units */
2328     expected.cx = MulDiv( 50, 2540, dpi_x );
2329     expected.cy = MulDiv( 20, 2540, dpi_y );
2330     ok(size.cx == expected.cx && size.cy == expected.cy, "size = {%d %d} (expected %d %d)\n",
2331        size.cx, size.cy, expected.cx, expected.cy );
2332
2333     size.cx = 800;
2334     size.cy = 700;
2335     hres = IOleObject_SetExtent(oleobj, DVASPECT_CONTENT, &size);
2336     ok(hres == S_OK, "SetExtent failed: %08x\n", hres);
2337
2338     size.cx = size.cy = 0xdeadbeef;
2339     hres = IOleObject_GetExtent(oleobj, DVASPECT_CONTENT, &size);
2340     ok(hres == S_OK, "GetExtent failed: %08x\n", hres);
2341     ok(size.cx == 800 && size.cy == 700, "size = {%d %d}\n", size.cx, size.cy);
2342
2343     size.cx = size.cy = 0xdeadbeef;
2344     hres = IOleObject_GetExtent(oleobj, 0, &size);
2345     ok(hres == S_OK, "GetExtent failed: %08x\n", hres);
2346     ok(size.cx == 800 && size.cy == 700, "size = {%d %d}\n", size.cx, size.cy);
2347
2348     size.cx = 900;
2349     size.cy = 800;
2350     hres = IOleObject_SetExtent(oleobj, 0, &size);
2351     ok(hres == S_OK, "SetExtent failed: %08x\n", hres);
2352
2353     size.cx = size.cy = 0xdeadbeef;
2354     hres = IOleObject_GetExtent(oleobj, 0, &size);
2355     ok(hres == S_OK, "GetExtent failed: %08x\n", hres);
2356     ok(size.cx == 900 && size.cy == 800, "size = {%d %d}\n", size.cx, size.cy);
2357
2358     size.cx = size.cy = 0xdeadbeef;
2359     hres = IOleObject_GetExtent(oleobj, 0xdeadbeef, &size);
2360     ok(hres == S_OK, "GetExtent failed: %08x\n", hres);
2361     ok(size.cx == 900 && size.cy == 800, "size = {%d %d}\n", size.cx, size.cy);
2362
2363     size.cx = 1000;
2364     size.cy = 900;
2365     hres = IOleObject_SetExtent(oleobj, 0xdeadbeef, &size);
2366     ok(hres == S_OK, "SetExtent failed: %08x\n", hres);
2367
2368     size.cx = size.cy = 0xdeadbeef;
2369     hres = IOleObject_GetExtent(oleobj, 0xdeadbeef, &size);
2370     ok(hres == S_OK, "GetExtent failed: %08x\n", hres);
2371     ok(size.cx == 1000 && size.cy == 900, "size = {%d %d}\n", size.cx, size.cy);
2372
2373     size.cx = size.cy = 0xdeadbeef;
2374     hres = IOleObject_GetExtent(oleobj, DVASPECT_CONTENT, &size);
2375     ok(hres == S_OK, "GetExtent failed: %08x\n", hres);
2376     ok(size.cx == 1000 && size.cy == 900, "size = {%d %d}\n", size.cx, size.cy);
2377
2378     IOleObject_Release(oleobj);
2379 }
2380
2381 static void test_ConnectionPoint(IUnknown *unk, BOOL init)
2382 {
2383     IConnectionPointContainer *container;
2384     IConnectionPoint *point;
2385     HRESULT hres;
2386
2387     static DWORD dw = 100;
2388
2389     hres = IUnknown_QueryInterface(unk, &IID_IConnectionPointContainer, (void**)&container);
2390     ok(hres == S_OK, "QueryInterface(IID_IConnectionPointContainer) failed: %08x\n", hres);
2391     if(FAILED(hres))
2392         return;
2393
2394     hres = IConnectionPointContainer_FindConnectionPoint(container, &DIID_DWebBrowserEvents2, &point);
2395     IConnectionPointContainer_Release(container);
2396     ok(hres == S_OK, "FindConnectionPoint failed: %08x\n", hres);
2397     if(FAILED(hres))
2398         return;
2399
2400     if(init) {
2401         hres = IConnectionPoint_Advise(point, (IUnknown*)&WebBrowserEvents2, &dw);
2402         ok(hres == S_OK, "Advise failed: %08x\n", hres);
2403         ok(dw == 1, "dw=%d, expected 1\n", dw);
2404     }else {
2405         hres = IConnectionPoint_Unadvise(point, dw);
2406         ok(hres == S_OK, "Unadvise failed: %08x\n", hres);
2407     }
2408
2409     IConnectionPoint_Release(point);
2410 }
2411
2412 static void test_Navigate2(IUnknown *unk)
2413 {
2414     IWebBrowser2 *webbrowser;
2415     VARIANT url;
2416     HRESULT hres;
2417
2418     hres = IUnknown_QueryInterface(unk, &IID_IWebBrowser2, (void**)&webbrowser);
2419     ok(hres == S_OK, "QueryInterface(IID_IWebBrowser) failed: %08x\n", hres);
2420     if(FAILED(hres))
2421         return;
2422
2423     test_LocationURL(unk, is_first_load ? "" : current_url);
2424     test_ready_state(is_first_load ? READYSTATE_UNINITIALIZED : READYSTATE_COMPLETE);
2425
2426     V_VT(&url) = VT_BSTR;
2427     V_BSTR(&url) = a2bstr("about:blank");
2428     current_url = "about:blank";
2429
2430     if(is_first_load) {
2431         SET_EXPECT(Invoke_AMBIENT_USERMODE);
2432         SET_EXPECT(Invoke_PROPERTYCHANGE);
2433         SET_EXPECT(Invoke_BEFORENAVIGATE2);
2434         SET_EXPECT(Invoke_DOWNLOADBEGIN);
2435         if (use_container_olecmd) SET_EXPECT(Exec_SETDOWNLOADSTATE_1);
2436         SET_EXPECT(EnableModeless_FALSE);
2437         SET_EXPECT(Invoke_STATUSTEXTCHANGE);
2438         SET_EXPECT(SetStatusText);
2439         SET_EXPECT(GetHostInfo);
2440         SET_EXPECT(Invoke_AMBIENT_DLCONTROL);
2441         SET_EXPECT(Invoke_AMBIENT_USERAGENT);
2442         SET_EXPECT(Invoke_AMBIENT_PALETTE);
2443         SET_EXPECT(GetOptionKeyPath);
2444         SET_EXPECT(GetOverridesKeyPath);
2445         if (use_container_olecmd) SET_EXPECT(QueryStatus_SETPROGRESSTEXT);
2446         if (use_container_olecmd) SET_EXPECT(Exec_SETPROGRESSMAX);
2447         if (use_container_olecmd) SET_EXPECT(Exec_SETPROGRESSPOS);
2448         SET_EXPECT(Invoke_SETSECURELOCKICON);
2449         SET_EXPECT(Invoke_FILEDOWNLOAD);
2450         if (use_container_olecmd) SET_EXPECT(Exec_SETDOWNLOADSTATE_0);
2451         SET_EXPECT(Invoke_COMMANDSTATECHANGE);
2452         SET_EXPECT(EnableModeless_TRUE);
2453         if (!use_container_olecmd) SET_EXPECT(Invoke_DOWNLOADCOMPLETE);
2454     }
2455
2456     hres = IWebBrowser2_Navigate2(webbrowser, &url, NULL, NULL, NULL, NULL);
2457     ok(hres == S_OK, "Navigate2 failed: %08x\n", hres);
2458
2459     if(is_first_load) {
2460         CHECK_CALLED(Invoke_AMBIENT_USERMODE);
2461         todo_wine CHECK_CALLED(Invoke_PROPERTYCHANGE);
2462         CHECK_CALLED(Invoke_BEFORENAVIGATE2);
2463         todo_wine CHECK_CALLED(Invoke_DOWNLOADBEGIN);
2464         if (use_container_olecmd) todo_wine CHECK_CALLED(Exec_SETDOWNLOADSTATE_1);
2465         CHECK_CALLED(EnableModeless_FALSE);
2466         CHECK_CALLED(Invoke_STATUSTEXTCHANGE);
2467         CHECK_CALLED(SetStatusText);
2468         CHECK_CALLED(GetHostInfo);
2469         CHECK_CALLED(Invoke_AMBIENT_DLCONTROL);
2470         CHECK_CALLED(Invoke_AMBIENT_USERAGENT);
2471         CLEAR_CALLED(Invoke_AMBIENT_PALETTE); /* Not called by IE9 */
2472         CLEAR_CALLED(GetOptionKeyPath);
2473         CHECK_CALLED(GetOverridesKeyPath);
2474         if (use_container_olecmd) todo_wine CHECK_CALLED(QueryStatus_SETPROGRESSTEXT);
2475         if (use_container_olecmd) todo_wine CHECK_CALLED(Exec_SETPROGRESSMAX);
2476         if (use_container_olecmd) todo_wine CHECK_CALLED(Exec_SETPROGRESSPOS);
2477         todo_wine CHECK_CALLED(Invoke_SETSECURELOCKICON);
2478         todo_wine CHECK_CALLED(Invoke_FILEDOWNLOAD);
2479         todo_wine CHECK_CALLED(Invoke_COMMANDSTATECHANGE);
2480         if (use_container_olecmd) todo_wine CHECK_CALLED(Exec_SETDOWNLOADSTATE_0);
2481         CHECK_CALLED(EnableModeless_TRUE);
2482     }
2483
2484     VariantClear(&url);
2485     IWebBrowser2_Release(webbrowser);
2486
2487     test_ready_state(READYSTATE_LOADING);
2488 }
2489
2490 static void test_QueryStatusWB(IWebBrowser2 *webbrowser, BOOL use_custom_target, BOOL has_document)
2491 {
2492     HRESULT hres, success_state;
2493     OLECMDF success_flags;
2494     enum OLECMDF status;
2495
2496     /*
2497      * We can tell the difference between the custom container target and the built-in target
2498      * since the custom target returns OLECMDF_SUPPORTED instead of OLECMDF_ENABLED.
2499      */
2500     if (use_custom_target)
2501         success_flags = OLECMDF_SUPPORTED;
2502     else
2503         success_flags = OLECMDF_ENABLED;
2504
2505     /* When no target is available (no document or custom target) an error is returned */
2506     if (has_document)
2507         success_state = S_OK;
2508     else
2509         success_state = E_UNEXPECTED;
2510
2511     /*
2512      * Test a safe operation that exists as both a high-numbered MSHTML id and an OLECMDID.
2513      * These tests show that QueryStatusWB uses a NULL pguidCmdGroup, since OLECMDID_STOP
2514      * is enabled and IDM_STOP is not.
2515      */
2516     status = 0xdeadbeef;
2517     if (use_custom_target) SET_EXPECT(QueryStatus_STOP);
2518     hres = IWebBrowser2_QueryStatusWB(webbrowser, OLECMDID_STOP, &status);
2519     ok(hres == success_state, "QueryStatusWB failed: %08x %08x\n", hres, success_state);
2520     if (!use_custom_target && has_document)
2521         todo_wine ok((has_document && status == success_flags) || (!has_document && status == 0xdeadbeef),
2522                      "OLECMDID_STOP not enabled/supported: %08x %08x\n", status, success_flags);
2523     else
2524         ok((has_document && status == success_flags) || (!has_document && status == 0xdeadbeef),
2525            "OLECMDID_STOP not enabled/supported: %08x %08x\n", status, success_flags);
2526     status = 0xdeadbeef;
2527     if (use_custom_target) SET_EXPECT(QueryStatus_IDM_STOP);
2528     hres = IWebBrowser2_QueryStatusWB(webbrowser, IDM_STOP, &status);
2529     ok(hres == success_state, "QueryStatusWB failed: %08x %08x\n", hres, success_state);
2530     ok((has_document && status == 0) || (!has_document && status == 0xdeadbeef),
2531        "IDM_STOP is enabled/supported: %08x %d\n", status, has_document);
2532 }
2533
2534 static void test_ExecWB(IWebBrowser2 *webbrowser, BOOL use_custom_target, BOOL has_document)
2535 {
2536     HRESULT hres, olecmdid_state, idm_state;
2537
2538     /* When no target is available (no document or custom target) an error is returned */
2539     if (has_document)
2540     {
2541         olecmdid_state = S_OK;
2542         idm_state = OLECMDERR_E_NOTSUPPORTED;
2543     }
2544     else
2545     {
2546         olecmdid_state = E_UNEXPECTED;
2547         idm_state = E_UNEXPECTED;
2548     }
2549
2550     /*
2551      * Test a safe operation that exists as both a high-numbered MSHTML id and an OLECMDID.
2552      * These tests show that QueryStatusWB uses a NULL pguidCmdGroup, since OLECMDID_STOP
2553      * succeeds (S_OK) and IDM_STOP does not (OLECMDERR_E_NOTSUPPORTED).
2554      */
2555     if(use_custom_target) {
2556         SET_EXPECT(Exec_STOP);
2557     }else if(has_document) {
2558         SET_EXPECT(Invoke_STATUSTEXTCHANGE);
2559         SET_EXPECT(SetStatusText);
2560     }
2561     hres = IWebBrowser2_ExecWB(webbrowser, OLECMDID_STOP, OLECMDEXECOPT_DONTPROMPTUSER, 0, 0);
2562     if(!use_custom_target && has_document) {
2563         todo_wine ok(hres == olecmdid_state, "ExecWB failed: %08x %08x\n", hres, olecmdid_state);
2564         CLEAR_CALLED(Invoke_STATUSTEXTCHANGE); /* Called by IE9 */
2565         CLEAR_CALLED(SetStatusText); /* Called by IE9 */
2566     }else {
2567         ok(hres == olecmdid_state, "ExecWB failed: %08x %08x\n", hres, olecmdid_state);
2568     }
2569     if (use_custom_target)
2570         SET_EXPECT(Exec_IDM_STOP);
2571     hres = IWebBrowser2_ExecWB(webbrowser, IDM_STOP, OLECMDEXECOPT_DONTPROMPTUSER, 0, 0);
2572     ok(hres == idm_state, "ExecWB failed: %08x %08x\n", hres, idm_state);
2573 }
2574
2575 static void test_download(DWORD flags)
2576 {
2577     MSG msg;
2578
2579     is_downloading = TRUE;
2580     dwl_flags = flags;
2581
2582     test_ready_state((flags & DWL_FROM_PUT_HREF) ? READYSTATE_COMPLETE : READYSTATE_LOADING);
2583
2584     if(flags & DWL_EXPECT_BEFORE_NAVIGATE) {
2585         SET_EXPECT(Invoke_PROPERTYCHANGE);
2586         SET_EXPECT(Invoke_BEFORENAVIGATE2);
2587         SET_EXPECT(TranslateUrl);
2588     }
2589     SET_EXPECT(Exec_SETPROGRESSMAX);
2590     SET_EXPECT(Exec_SETPROGRESSPOS);
2591     SET_EXPECT(Exec_SETDOWNLOADSTATE_1);
2592     SET_EXPECT(DocHost_EnableModeless_FALSE);
2593     SET_EXPECT(DocHost_EnableModeless_TRUE);
2594     SET_EXPECT(Invoke_SETSECURELOCKICON);
2595     SET_EXPECT(Invoke_282);
2596     SET_EXPECT(EnableModeless_FALSE);
2597     SET_EXPECT(Invoke_COMMANDSTATECHANGE);
2598     SET_EXPECT(Invoke_STATUSTEXTCHANGE);
2599     SET_EXPECT(SetStatusText);
2600     SET_EXPECT(EnableModeless_TRUE);
2601     if(!is_first_load)
2602         SET_EXPECT(GetHostInfo);
2603     SET_EXPECT(Exec_SETDOWNLOADSTATE_0);
2604     SET_EXPECT(Invoke_TITLECHANGE);
2605     SET_EXPECT(Invoke_NAVIGATECOMPLETE2);
2606     if(is_first_load)
2607         SET_EXPECT(GetDropTarget);
2608     SET_EXPECT(Invoke_PROGRESSCHANGE);
2609     SET_EXPECT(Invoke_DOCUMENTCOMPLETE);
2610
2611     SET_EXPECT(UpdateUI);
2612     SET_EXPECT(Exec_UPDATECOMMANDS);
2613     SET_EXPECT(QueryStatus_STOP);
2614
2615     while(!called_Invoke_DOCUMENTCOMPLETE && GetMessage(&msg, NULL, 0, 0)) {
2616         TranslateMessage(&msg);
2617         DispatchMessage(&msg);
2618     }
2619
2620     if(flags & DWL_EXPECT_BEFORE_NAVIGATE) {
2621         todo_wine CHECK_CALLED(Invoke_PROPERTYCHANGE);
2622         CHECK_CALLED(Invoke_BEFORENAVIGATE2);
2623         CHECK_CALLED(TranslateUrl);
2624     }
2625     todo_wine CHECK_CALLED(Exec_SETPROGRESSMAX);
2626     todo_wine CHECK_CALLED(Exec_SETPROGRESSPOS);
2627     todo_wine CHECK_CALLED(Exec_SETDOWNLOADSTATE_1);
2628     CLEAR_CALLED(DocHost_EnableModeless_FALSE); /* IE 7 */
2629     CLEAR_CALLED(DocHost_EnableModeless_TRUE); /* IE 7 */
2630     todo_wine CHECK_CALLED(Invoke_SETSECURELOCKICON);
2631     CLEAR_CALLED(Invoke_282); /* IE 7 */
2632     if(is_first_load)
2633         todo_wine CHECK_CALLED(EnableModeless_FALSE);
2634     else
2635         CLEAR_CALLED(EnableModeless_FALSE); /* IE 8 */
2636     CLEAR_CALLED(Invoke_COMMANDSTATECHANGE);
2637     todo_wine CHECK_CALLED(Invoke_STATUSTEXTCHANGE);
2638     todo_wine CHECK_CALLED(SetStatusText);
2639     if(is_first_load)
2640         todo_wine CHECK_CALLED(EnableModeless_TRUE);
2641     else
2642         CLEAR_CALLED(EnableModeless_FALSE); /* IE 8 */
2643     if(!is_first_load)
2644         todo_wine CHECK_CALLED(GetHostInfo);
2645     todo_wine CHECK_CALLED(Exec_SETDOWNLOADSTATE_0);
2646     todo_wine CHECK_CALLED(Invoke_TITLECHANGE);
2647     CHECK_CALLED(Invoke_NAVIGATECOMPLETE2);
2648     if(is_first_load)
2649         todo_wine CHECK_CALLED(GetDropTarget);
2650     todo_wine CHECK_CALLED(Invoke_PROGRESSCHANGE);
2651     CHECK_CALLED(Invoke_DOCUMENTCOMPLETE);
2652
2653     is_downloading = FALSE;
2654
2655     test_ready_state(READYSTATE_COMPLETE);
2656
2657     while(!called_Exec_UPDATECOMMANDS && GetMessage(&msg, NULL, 0, 0)) {
2658         TranslateMessage(&msg);
2659         DispatchMessage(&msg);
2660     }
2661
2662     CHECK_CALLED(UpdateUI);
2663     CHECK_CALLED(Exec_UPDATECOMMANDS);
2664     CLEAR_CALLED(QueryStatus_STOP);
2665 }
2666
2667 static void test_olecmd(IUnknown *unk, BOOL loaded)
2668 {
2669     IOleCommandTarget *cmdtrg;
2670     OLECMD cmds[3];
2671     HRESULT hres;
2672
2673     hres = IUnknown_QueryInterface(unk, &IID_IOleCommandTarget, (void**)&cmdtrg);
2674     ok(hres == S_OK, "Could not get IOleCommandTarget iface: %08x\n", hres);
2675     if(FAILED(hres))
2676         return;
2677
2678     cmds[0].cmdID = OLECMDID_SPELL;
2679     cmds[0].cmdf = 0xdeadbeef;
2680     cmds[1].cmdID = OLECMDID_REFRESH;
2681     cmds[1].cmdf = 0xdeadbeef;
2682     hres = IOleCommandTarget_QueryStatus(cmdtrg, NULL, 2, cmds, NULL);
2683     if(loaded) {
2684         ok(hres == S_OK, "QueryStatus failed: %08x\n", hres);
2685         ok(cmds[0].cmdf == OLECMDF_SUPPORTED, "OLECMDID_SPELL cmdf = %x\n", cmds[0].cmdf);
2686         ok(cmds[1].cmdf == (OLECMDF_ENABLED|OLECMDF_SUPPORTED),
2687            "OLECMDID_REFRESH cmdf = %x\n", cmds[1].cmdf);
2688     }else {
2689         ok(hres == 0x80040104, "QueryStatus failed: %08x\n", hres);
2690         ok(cmds[0].cmdf == 0xdeadbeef, "OLECMDID_SPELL cmdf = %x\n", cmds[0].cmdf);
2691         ok(cmds[1].cmdf == 0xdeadbeef, "OLECMDID_REFRESH cmdf = %x\n", cmds[0].cmdf);
2692     }
2693
2694     IOleCommandTarget_Release(cmdtrg);
2695 }
2696
2697 static void test_IServiceProvider(IUnknown *unk)
2698 {
2699     IServiceProvider *servprov = (void*)0xdeadbeef;
2700     HRESULT hres;
2701     IUnknown *ret = NULL;
2702
2703     hres = IUnknown_QueryInterface(unk, &IID_IServiceProvider, (void**)&servprov);
2704     ok(hres == S_OK, "QueryInterface returned %08x, expected S_OK\n", hres);
2705     if(FAILED(hres))
2706         return;
2707
2708     hres = IServiceProvider_QueryService(servprov, &SID_STopLevelBrowser, &IID_IBrowserService2, (LPVOID*)&ret);
2709     ok(hres == E_FAIL, "QueryService returned %08x, expected E_FAIL\n", hres);
2710     ok(ret == NULL, "ret returned %p, expected NULL\n", ret);
2711     if(hres == S_OK)
2712     {
2713         IUnknown_Release(ret);
2714     }
2715
2716     IServiceProvider_Release(servprov);
2717 }
2718
2719 static void test_put_href(IUnknown *unk)
2720 {
2721     IHTMLLocation *location;
2722     IHTMLDocument2 *doc;
2723     IDispatch *doc_disp;
2724     BSTR str;
2725     HRESULT hres;
2726
2727     doc_disp = get_document(unk);
2728
2729     hres = IDispatch_QueryInterface(doc_disp, &IID_IHTMLDocument2, (void**)&doc);
2730     IDispatch_Release(doc_disp);
2731     ok(hres == S_OK, "QueryInterface(IID_IHTMLDocument2 failed: %08x\n", hres);
2732
2733     location = NULL;
2734     hres = IHTMLDocument2_get_location(doc, &location);
2735     IHTMLDocument2_Release(doc);
2736     ok(hres == S_OK, "get_location failed: %08x\n", hres);
2737     ok(location != NULL, "location == NULL\n");
2738
2739     SET_EXPECT(TranslateUrl);
2740     SET_EXPECT(Invoke_BEFORENAVIGATE2);
2741     SET_EXPECT(Invoke_PROPERTYCHANGE);
2742
2743     dwl_flags = DWL_FROM_PUT_HREF;
2744
2745     str = a2bstr("about:test");
2746     current_url = "about:test";
2747     is_first_load = FALSE;
2748     hres = IHTMLLocation_put_href(location, str);
2749
2750     CHECK_CALLED(TranslateUrl);
2751     CHECK_CALLED(Invoke_BEFORENAVIGATE2);
2752     todo_wine CHECK_CALLED(Invoke_PROPERTYCHANGE);
2753
2754     IHTMLLocation_Release(location);
2755     SysFreeString(str);
2756     ok(hres == S_OK, "put_href failed: %08x\n", hres);
2757
2758     test_ready_state(READYSTATE_COMPLETE);
2759 }
2760
2761 static void test_QueryInterface(IUnknown *unk)
2762 {
2763     IQuickActivate *qa = (IQuickActivate*)0xdeadbeef;
2764     IRunnableObject *runnable = (IRunnableObject*)0xdeadbeef;
2765     IPerPropertyBrowsing *propbrowse = (void*)0xdeadbeef;
2766     IOleInPlaceSite *inplace = (void*)0xdeadbeef;
2767     IOleCache *cache = (void*)0xdeadbeef;
2768     IObjectWithSite *site = (void*)0xdeadbeef;
2769     IViewObjectEx *viewex = (void*)0xdeadbeef;
2770     IOleLink *link = (void*)0xdeadbeef;
2771     IMarshal *marshal = (void*)0xdeadbeef;
2772     IStdMarshalInfo *marshalinfo = (void*)0xdeadbeef;
2773     HRESULT hres;
2774
2775     hres = IUnknown_QueryInterface(unk, &IID_IQuickActivate, (void**)&qa);
2776     ok(hres == E_NOINTERFACE, "QueryInterface returned %08x, expected E_NOINTERFACE\n", hres);
2777     ok(qa == NULL, "qa=%p, expected NULL\n", qa);
2778
2779     hres = IUnknown_QueryInterface(unk, &IID_IRunnableObject, (void**)&runnable);
2780     ok(hres == E_NOINTERFACE, "QueryInterface returned %08x, expected E_NOINTERFACE\n", hres);
2781     ok(runnable == NULL, "runnable=%p, expected NULL\n", runnable);
2782
2783     hres = IUnknown_QueryInterface(unk, &IID_IPerPropertyBrowsing, (void**)&propbrowse);
2784     ok(hres == E_NOINTERFACE, "QueryInterface returned %08x, expected E_NOINTERFACE\n", hres);
2785     ok(propbrowse == NULL, "propbrowse=%p, expected NULL\n", propbrowse);
2786
2787     hres = IUnknown_QueryInterface(unk, &IID_IOleCache, (void**)&cache);
2788     ok(hres == E_NOINTERFACE, "QueryInterface returned %08x, expected E_NOINTERFACE\n", hres);
2789     ok(cache == NULL, "cache=%p, expected NULL\n", cache);
2790
2791     hres = IUnknown_QueryInterface(unk, &IID_IOleInPlaceSite, (void**)&inplace);
2792     ok(hres == E_NOINTERFACE, "QueryInterface returned %08x, expected E_NOINTERFACE\n", hres);
2793     ok(inplace == NULL, "inplace=%p, expected NULL\n", inplace);
2794
2795     hres = IUnknown_QueryInterface(unk, &IID_IObjectWithSite, (void**)&site);
2796     ok(hres == E_NOINTERFACE, "QueryInterface returned %08x, expected E_NOINTERFACE\n", hres);
2797     ok(site == NULL, "site=%p, expected NULL\n", site);
2798
2799     hres = IUnknown_QueryInterface(unk, &IID_IViewObjectEx, (void**)&viewex);
2800     ok(hres == E_NOINTERFACE, "QueryInterface returned %08x, expected E_NOINTERFACE\n", hres);
2801     ok(viewex == NULL, "viewex=%p, expected NULL\n", viewex);
2802
2803     hres = IUnknown_QueryInterface(unk, &IID_IOleLink, (void**)&link);
2804     ok(hres == E_NOINTERFACE, "QueryInterface returned %08x, expected E_NOINTERFACE\n", hres);
2805     ok(link == NULL, "link=%p, expected NULL\n", link);
2806
2807     hres = IUnknown_QueryInterface(unk, &IID_IMarshal, (void**)&marshal);
2808     ok(hres == E_NOINTERFACE, "QueryInterface returned %08x, expected E_NOINTERFACE\n", hres);
2809     ok(marshal == NULL, "marshal=%p, expected NULL\n", marshal);
2810
2811     hres = IUnknown_QueryInterface(unk, &IID_IStdMarshalInfo, (void**)&marshalinfo);
2812     ok(hres == E_NOINTERFACE, "QueryInterface returned %08x, expected E_NOINTERFACE\n", hres);
2813     ok(marshalinfo == NULL, "marshalinfo=%p, expected NULL\n", marshalinfo);
2814
2815 }
2816
2817 static void test_UIActivate(IUnknown *unk, BOOL activate)
2818 {
2819     IOleDocumentView *docview;
2820     IDispatch *disp;
2821     HRESULT hres;
2822
2823     disp = get_document(unk);
2824
2825     hres = IDispatch_QueryInterface(disp, &IID_IOleDocumentView, (void**)&docview);
2826     ok(hres == S_OK, "Got 0x%08x\n", hres);
2827     if(SUCCEEDED(hres)) {
2828         if(activate) {
2829             SET_EXPECT(RequestUIActivate);
2830             SET_EXPECT(ShowUI);
2831             SET_EXPECT(HideUI);
2832             SET_EXPECT(OnFocus);
2833         }
2834
2835         hres = IOleDocumentView_UIActivate(docview, activate);
2836         if(activate)
2837             todo_wine ok(hres == S_OK, "Got 0x%08x\n", hres);
2838         else
2839             ok(hres == S_OK, "Got 0x%08x\n", hres);
2840
2841         if(activate) {
2842             todo_wine {
2843                 CHECK_CALLED(RequestUIActivate);
2844                 CHECK_CALLED(ShowUI);
2845                 CHECK_CALLED(HideUI);
2846                 CHECK_CALLED(OnFocus);
2847             }
2848         }
2849
2850         IOleDocumentView_Release(docview);
2851     }
2852
2853     IDispatch_Release(disp);
2854 }
2855
2856 static void test_TranslateAccelerator(IUnknown *unk)
2857 {
2858     IOleClientSite *doc_clientsite;
2859     IOleInPlaceActiveObject *pao;
2860     HRESULT hres;
2861     DWORD keycode;
2862     MSG msg_a = {
2863         container_hwnd,
2864         0, 0, 0,
2865         GetTickCount(),
2866         {5, 5}
2867     };
2868
2869     test_Navigate2(unk);
2870
2871     hres = IUnknown_QueryInterface(unk, &IID_IOleInPlaceActiveObject, (void**)&pao);
2872     ok(hres == S_OK, "Got 0x%08x\n", hres);
2873     if(SUCCEEDED(hres)) {
2874         /* One accelerator that should be handled by mshtml */
2875         msg_a.message = WM_KEYDOWN;
2876         msg_a.wParam = VK_F1;
2877         hres = IOleInPlaceActiveObject_TranslateAccelerator(pao, &msg_a);
2878         ok(hres == S_FALSE, "Got 0x%08x (%04x::%02lx)\n", hres, msg_a.message, msg_a.wParam);
2879
2880         /* And one that should not */
2881         msg_a.message = WM_KEYDOWN;
2882         msg_a.wParam = VK_F5;
2883         hres = IOleInPlaceActiveObject_TranslateAccelerator(pao, &msg_a);
2884         ok(hres == S_FALSE, "Got 0x%08x (%04x::%02lx)\n", hres, msg_a.message, msg_a.wParam);
2885
2886         IOleInPlaceActiveObject_Release(pao);
2887     }
2888
2889     test_UIActivate(unk, TRUE);
2890
2891     /* Test again after UIActivate */
2892     hres = IUnknown_QueryInterface(unk, &IID_IOleInPlaceActiveObject, (void**)&pao);
2893     ok(hres == S_OK, "Got 0x%08x\n", hres);
2894     if(SUCCEEDED(hres)) {
2895         /* One accelerator that should be handled by mshtml */
2896         msg_a.message = WM_KEYDOWN;
2897         msg_a.wParam = VK_F1;
2898         SET_EXPECT(DocHost_TranslateAccelerator);
2899         SET_EXPECT(ControlSite_TranslateAccelerator);
2900         hres = IOleInPlaceActiveObject_TranslateAccelerator(pao, &msg_a);
2901         ok(hres == S_FALSE, "Got 0x%08x (%04x::%02lx)\n", hres, msg_a.message, msg_a.wParam);
2902         todo_wine CHECK_CALLED(DocHost_TranslateAccelerator);
2903         todo_wine CHECK_CALLED(ControlSite_TranslateAccelerator);
2904
2905         /* And one that should not */
2906         msg_a.message = WM_KEYDOWN;
2907         msg_a.wParam = VK_F5;
2908         SET_EXPECT(DocHost_TranslateAccelerator);
2909         hres = IOleInPlaceActiveObject_TranslateAccelerator(pao, &msg_a);
2910         todo_wine ok(hres == S_OK, "Got 0x%08x (%04x::%02lx)\n", hres, msg_a.message, msg_a.wParam);
2911         todo_wine CHECK_CALLED(DocHost_TranslateAccelerator);
2912
2913         IOleInPlaceActiveObject_Release(pao);
2914     }
2915
2916     doc_clientsite = get_dochost(unk);
2917     if(doc_clientsite) {
2918         IDocHostUIHandler2 *dochost;
2919         IOleControlSite *doc_controlsite;
2920         IUnknown *unk_test;
2921
2922         hres = IOleClientSite_QueryInterface(doc_clientsite, &IID_IOleInPlaceFrame, (void**)&unk_test);
2923         ok(hres == E_NOINTERFACE, "Got 0x%08x\n", hres);
2924         if(SUCCEEDED(hres)) IUnknown_Release(unk_test);
2925
2926         hres = IOleClientSite_QueryInterface(doc_clientsite, &IID_IDocHostShowUI, (void**)&unk_test);
2927         todo_wine ok(hres == S_OK, "Got 0x%08x\n", hres);
2928         if(SUCCEEDED(hres)) IUnknown_Release(unk_test);
2929
2930         hres = IOleClientSite_QueryInterface(doc_clientsite, &IID_IDocHostUIHandler, (void**)&unk_test);
2931         ok(hres == S_OK, "Got 0x%08x\n", hres);
2932         if(SUCCEEDED(hres)) IUnknown_Release(unk_test);
2933
2934         hres = IOleClientSite_QueryInterface(doc_clientsite, &IID_IDocHostUIHandler2, (void**)&dochost);
2935         ok(hres == S_OK, "Got 0x%08x\n", hres);
2936         if(SUCCEEDED(hres)) {
2937             msg_a.message = WM_KEYDOWN;
2938             hr_dochost_TranslateAccelerator = 0xdeadbeef;
2939             for(keycode = 0; keycode <= 0x100; keycode++) {
2940                 msg_a.wParam = keycode;
2941                 SET_EXPECT(DocHost_TranslateAccelerator);
2942                 hres = IDocHostUIHandler_TranslateAccelerator(dochost, &msg_a, &CGID_MSHTML, 1234);
2943                 ok(hres == 0xdeadbeef, "Got 0x%08x\n", hres);
2944                 CHECK_CALLED(DocHost_TranslateAccelerator);
2945             }
2946             hr_dochost_TranslateAccelerator = E_NOTIMPL;
2947
2948             SET_EXPECT(DocHost_TranslateAccelerator);
2949             hres = IDocHostUIHandler_TranslateAccelerator(dochost, &msg_a, &CGID_MSHTML, 1234);
2950             ok(hres == E_NOTIMPL, "Got 0x%08x\n", hres);
2951             CHECK_CALLED(DocHost_TranslateAccelerator);
2952
2953             IDocHostUIHandler2_Release(dochost);
2954         }
2955
2956         hres = IOleClientSite_QueryInterface(doc_clientsite, &IID_IOleControlSite, (void**)&doc_controlsite);
2957         ok(hres == S_OK, "Got 0x%08x\n", hres);
2958         if(SUCCEEDED(hres)) {
2959             msg_a.message = WM_KEYDOWN;
2960             hr_site_TranslateAccelerator = 0xdeadbeef;
2961             for(keycode = 0; keycode < 0x100; keycode++) {
2962                 msg_a.wParam = keycode;
2963                 SET_EXPECT(ControlSite_TranslateAccelerator);
2964                 hres = IOleControlSite_TranslateAccelerator(doc_controlsite, &msg_a, 0);
2965                 if(keycode == 0x9 || keycode == 0x75)
2966                     todo_wine ok(hres == S_OK, "Got 0x%08x (keycode: %04x)\n", hres, keycode);
2967                 else
2968                     ok(hres == S_FALSE, "Got 0x%08x (keycode: %04x)\n", hres, keycode);
2969
2970                 CHECK_CALLED(ControlSite_TranslateAccelerator);
2971             }
2972             msg_a.wParam = VK_LEFT;
2973             SET_EXPECT(ControlSite_TranslateAccelerator);
2974             hres = IOleControlSite_TranslateAccelerator(doc_controlsite, &msg_a, 0);
2975             ok(hres == S_FALSE, "Got 0x%08x (keycode: %04x)\n", hres, keycode);
2976             CHECK_CALLED(ControlSite_TranslateAccelerator);
2977
2978             hr_site_TranslateAccelerator = S_OK;
2979             SET_EXPECT(ControlSite_TranslateAccelerator);
2980             hres = IOleControlSite_TranslateAccelerator(doc_controlsite, &msg_a, 0);
2981             ok(hres == S_OK, "Got 0x%08x (keycode: %04x)\n", hres, keycode);
2982             CHECK_CALLED(ControlSite_TranslateAccelerator);
2983
2984             hr_site_TranslateAccelerator = E_NOTIMPL;
2985             SET_EXPECT(ControlSite_TranslateAccelerator);
2986             hres = IOleControlSite_TranslateAccelerator(doc_controlsite, &msg_a, 0);
2987             ok(hres == S_FALSE, "Got 0x%08x (keycode: %04x)\n", hres, keycode);
2988             CHECK_CALLED(ControlSite_TranslateAccelerator);
2989
2990             IOleControlSite_Release(doc_controlsite);
2991         }
2992
2993         IOleClientSite_Release(doc_clientsite);
2994     }
2995
2996     test_UIActivate(unk, FALSE);
2997 }
2998
2999 static void test_dochost_qs(IUnknown *unk)
3000 {
3001     IOleClientSite *client_site;
3002     IServiceProvider *serv_prov;
3003     IUnknown *service;
3004     HRESULT hres;
3005
3006     client_site = get_dochost(unk);
3007     hres = IOleClientSite_QueryInterface(client_site, &IID_IServiceProvider, (void**)&serv_prov);
3008     IOleClientSite_Release(client_site);
3009     ok(hres == S_OK, "Could not get IServiceProvider iface: %08x\n", hres);
3010
3011     hres = IServiceProvider_QueryService(serv_prov, &IID_IHlinkFrame, &IID_IHlinkFrame, (void**)&service);
3012     ok(hres == S_OK, "QueryService failed: %08x\n", hres);
3013     ok(iface_cmp(service, unk), "service != unk\n");
3014     IUnknown_Release(service);
3015
3016     hres = IServiceProvider_QueryService(serv_prov, &IID_IWebBrowserApp, &IID_IHlinkFrame, (void**)&service);
3017     ok(hres == S_OK, "QueryService failed: %08x\n", hres);
3018     ok(iface_cmp(service, unk), "service != unk\n");
3019     IUnknown_Release(service);
3020
3021     hres = IServiceProvider_QueryService(serv_prov, &IID_IShellBrowser, &IID_IShellBrowser, (void**)&service);
3022     ok(hres == S_OK, "QueryService failed: %08x\n", hres);
3023     IUnknown_Release(service);
3024
3025     hres = IServiceProvider_QueryService(serv_prov, &IID_IShellBrowser, &IID_IBrowserService, (void**)&service);
3026     ok(hres == S_OK, "QueryService failed: %08x\n", hres);
3027     IUnknown_Release(service);
3028
3029     hres = IServiceProvider_QueryService(serv_prov, &IID_IShellBrowser, &IID_IDocObjectService, (void**)&service);
3030     ok(hres == S_OK, "QueryService failed: %08x\n", hres);
3031     IUnknown_Release(service);
3032
3033     IServiceProvider_Release(serv_prov);
3034 }
3035
3036 static void test_Close(IWebBrowser2 *wb, BOOL do_download)
3037 {
3038     IOleObject *oo;
3039     IOleClientSite *ocs;
3040     HRESULT hres;
3041
3042     hres = IWebBrowser2_QueryInterface(wb, &IID_IOleObject, (void**)&oo);
3043     ok(hres == S_OK, "QueryInterface failed: %08x\n", hres);
3044     if(hres != S_OK)
3045         return;
3046
3047     test_close = TRUE;
3048
3049     SET_EXPECT(Frame_SetActiveObject);
3050     SET_EXPECT(UIWindow_SetActiveObject);
3051     SET_EXPECT(OnUIDeactivate);
3052     SET_EXPECT(OnFocus);
3053     SET_EXPECT(OnInPlaceDeactivate);
3054     SET_EXPECT(Invoke_STATUSTEXTCHANGE);
3055     if(!do_download) {
3056         SET_EXPECT(Invoke_COMMANDSTATECHANGE);
3057         SET_EXPECT(Invoke_DOWNLOADCOMPLETE);
3058     }
3059     hres = IOleObject_Close(oo, OLECLOSE_NOSAVE);
3060     ok(hres == S_OK, "OleObject_Close failed: %x\n", hres);
3061     CHECK_CALLED(Frame_SetActiveObject);
3062     CHECK_CALLED(UIWindow_SetActiveObject);
3063     CHECK_CALLED(OnUIDeactivate);
3064     todo_wine CHECK_CALLED(OnFocus);
3065     CHECK_CALLED(OnInPlaceDeactivate);
3066     CLEAR_CALLED(Invoke_STATUSTEXTCHANGE); /* Called by IE9 */
3067     if(!do_download) {
3068         todo_wine CHECK_CALLED(Invoke_COMMANDSTATECHANGE);
3069         todo_wine CHECK_CALLED(Invoke_DOWNLOADCOMPLETE);
3070     }
3071
3072     hres = IOleObject_GetClientSite(oo, &ocs);
3073     ok(hres == S_OK, "hres = %x\n", hres);
3074     ok(!ocs, "ocs != NULL\n");
3075
3076     hres = IOleObject_Close(oo, OLECLOSE_NOSAVE);
3077     ok(hres == S_OK, "OleObject_Close failed: %x\n", hres);
3078
3079     test_close = FALSE;
3080     IOleObject_Release(oo);
3081 }
3082
3083 static void test_WebBrowser(BOOL do_download, BOOL do_close)
3084 {
3085     IUnknown *unk = NULL;
3086     ULONG ref;
3087     HRESULT hres;
3088
3089     if (FAILED(create_WebBrowser(&unk)))
3090         return;
3091
3092     is_downloading = FALSE;
3093     is_first_load = TRUE;
3094     use_container_olecmd = TRUE;
3095
3096     hres = IUnknown_QueryInterface(unk, &IID_IWebBrowser2, (void**)&wb);
3097     ok(hres == S_OK, "Could not get IWebBrowser2 iface: %08x\n", hres);
3098
3099     test_QueryStatusWB(wb, FALSE, FALSE);
3100     test_ExecWB(wb, FALSE, FALSE);
3101     test_QueryInterface(unk);
3102     test_ready_state(READYSTATE_UNINITIALIZED);
3103     test_ClassInfo(unk);
3104     test_LocationURL(unk, "");
3105     test_ConnectionPoint(unk, TRUE);
3106     test_ClientSite(unk, &ClientSite, !do_download);
3107     test_Extent(unk);
3108     test_wb_funcs(unk, TRUE);
3109     test_DoVerb(unk);
3110     test_olecmd(unk, FALSE);
3111     test_Navigate2(unk);
3112     test_QueryStatusWB(wb, TRUE, TRUE);
3113     test_ExecWB(wb, TRUE, TRUE);
3114
3115     if(do_download) {
3116         IDispatch *doc, *doc2;
3117
3118         test_download(0);
3119         test_olecmd(unk, TRUE);
3120         doc = get_document(unk);
3121
3122         test_put_href(unk);
3123         test_download(DWL_FROM_PUT_HREF);
3124         doc2 = get_document(unk);
3125         ok(doc == doc2, "doc != doc2\n");
3126         IDispatch_Release(doc2);
3127
3128         trace("Navigate2 repeated...\n");
3129         test_Navigate2(unk);
3130         test_download(DWL_EXPECT_BEFORE_NAVIGATE);
3131         test_TranslateAccelerator(unk);
3132         doc2 = get_document(unk);
3133         ok(doc == doc2, "doc != doc2\n");
3134         IDispatch_Release(doc2);
3135         IDispatch_Release(doc);
3136
3137         test_dochost_qs(unk);
3138     }
3139
3140     if(do_close)
3141         test_Close(wb, do_download);
3142     else
3143         test_ClientSite(unk, NULL, !do_download);
3144     test_ie_funcs(unk);
3145     test_GetControlInfo(unk);
3146     test_wb_funcs(unk, FALSE);
3147     test_ConnectionPoint(unk, FALSE);
3148     test_IServiceProvider(unk);
3149
3150     IWebBrowser2_Release(wb);
3151     ref = IUnknown_Release(unk);
3152     ok(ref == 0, "ref=%d, expected 0\n", ref);
3153 }
3154
3155 static void test_WebBrowser_NoContainerOlecmd(void)
3156 {
3157     IUnknown *unk = NULL;
3158     HRESULT hres;
3159     ULONG ref;
3160
3161     is_downloading = FALSE;
3162     is_first_load = TRUE;
3163     use_container_olecmd = FALSE;
3164
3165     /* Setup stage */
3166     if (FAILED(create_WebBrowser(&unk)))
3167         return;
3168     hres = IUnknown_QueryInterface(unk, &IID_IWebBrowser2, (void**)&wb);
3169     ok(hres == S_OK, "QueryInterface(IID_IWebBrowser) failed: %08x\n", hres);
3170     if(FAILED(hres))
3171         return;
3172     test_ConnectionPoint(unk, TRUE);
3173     test_ClientSite(unk, &ClientSite, TRUE);
3174     test_DoVerb(unk);
3175     test_Navigate2(unk);
3176
3177     /* Tests of interest */
3178     test_QueryStatusWB(wb, FALSE, TRUE);
3179     test_ExecWB(wb, FALSE, TRUE);
3180
3181     /* Cleanup stage */
3182     IWebBrowser2_Release(wb);
3183     test_ClientSite(unk, NULL, TRUE);
3184     test_ConnectionPoint(unk, FALSE);
3185     ref = IUnknown_Release(unk);
3186     ok(ref == 0, "ref=%d, expected 0\n", ref);
3187 }
3188
3189 START_TEST(webbrowser)
3190 {
3191     OleInitialize(NULL);
3192
3193     container_hwnd = create_container_window();
3194
3195     trace("Testing WebBrowser (no download)...\n");
3196     test_WebBrowser(FALSE, FALSE);
3197     test_WebBrowser(FALSE, TRUE);
3198     trace("Testing WebBrowser...\n");
3199     test_WebBrowser(TRUE, FALSE);
3200     trace("Testing WebBrowser (with close)...\n");
3201     test_WebBrowser(TRUE, TRUE);
3202     trace("Testing WebBrowser w/o container-based olecmd...\n");
3203     test_WebBrowser_NoContainerOlecmd();
3204
3205     OleUninitialize();
3206 }