rostests/winetests/msxml3/xmlview.c

   1 /*
   2  * Copyright 2012 Piotr 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 WIN32_NO_STATUS
  20 #define _INC_WINDOWS
  21 #define COM_NO_WINDOWS_H
  22
  23 #define COBJMACROS
  24 #define CONST_VTABLE
  25
  26 //#include <stdio.h>
  27 //#include <assert.h>
  28
  29 #include <wine/test.h>
  30
  31 //#include "windows.h"
  32 #include <winnls.h>
  33 #include <wingdi.h>
  34 #include <ole2.h>
  35 #include <mshtml.h>
  36 #include <mshtmdid.h>
  37 #include <initguid.h>
  38 #include <perhist.h>
  39 #include <docobj.h>
  40 //#include "urlmon.h"
  41 #include <xmlparser.h>
  42
  43 HRESULT (WINAPI *pCreateURLMoniker)(IMoniker*, LPCWSTR, IMoniker**);
  44
  45 static const char xmlview_html[] =
  46 "\r\n"
  47 "<BODY><H2>Generated HTML</H2>\r\n"
  48 "<TABLE>\r\n"
  49 "<TBODY>\r\n"
  50 "<TR bgColor=green>\r\n"
  51 "<TH>Title</TH>\r\n"
  52 "<TH>Value</TH></TR>\r\n"
  53 "<TR>\r\n"
  54 "<TD>title1</TD>\r\n"
  55 "<TD>value1</TD></TR>\r\n"
  56 "<TR>\r\n"
  57 "<TD>title2</TD>\r\n"
  58 "<TD>value2</TD></TR></TBODY></TABLE></BODY>";
  59
  60 IHTMLDocument2 *html_doc;
  61 BOOL loaded;
  62
  63 static int html_src_compare(LPCWSTR strw, const char *stra)
  64 {
  65     char buf[2048], *p1;
  66     const char *p2;
  67
  68     WideCharToMultiByte(CP_ACP, 0, strw, -1, buf, sizeof(buf), NULL, NULL);
  69
  70     p1 = buf;
  71     p2 = stra;
  72     while(1) {
  73         while(*p1=='\r' || *p1=='\n' || *p1=='\"') p1++;
  74         while(*p2=='\r' || *p2=='\n') p2++;
  75
  76         if(!*p1 || !*p2 || tolower(*p1)!=tolower(*p2))
  77             break;
  78
  79         p1++;
  80         p2++;
  81     }
  82
  83     return *p1 != *p2;
  84 }
  85
  86 static HRESULT WINAPI HTMLEvents_QueryInterface(IDispatch *iface, REFIID riid, void **ppv)
  87 {
  88     if(IsEqualGUID(&IID_IUnknown, riid) || IsEqualGUID(&IID_IDispatch, riid)) {
  89         *ppv = iface;
  90         return S_OK;
  91     }
  92
  93     ok(0, "Unexpected call\n");
  94     return E_NOINTERFACE;
  95 }
  96
  97 static ULONG WINAPI HTMLEvents_AddRef(IDispatch *iface)
  98 {
  99     return 2;
 100 }
 101
 102 static ULONG WINAPI HTMLEvents_Release(IDispatch *iface)
 103 {
 104     return 1;
 105 }
 106
 107 static HRESULT WINAPI HTMLEvents_GetTypeInfoCount(IDispatch *iface, UINT *pctinfo)
 108 {
 109     ok(0, "unexpected call\n");
 110     return E_NOTIMPL;
 111 }
 112
 113 static HRESULT WINAPI HTMLEvents_GetTypeInfo(IDispatch *iface, UINT iTInfo, LCID lcid,
 114         ITypeInfo **ppTInfo)
 115 {
 116     ok(0, "unexpected call\n");
 117     return E_NOTIMPL;
 118 }
 119
 120 static HRESULT WINAPI HTMLEvents_GetIDsOfNames(IDispatch *iface, REFIID riid, LPOLESTR *rgszNames,
 121         UINT cNames, LCID lcid, DISPID *rgDispId)
 122 {
 123     ok(0, "unexpected call\n");
 124     return E_NOTIMPL;
 125 }
 126
 127 static HRESULT WINAPI HTMLEvents_Invoke(IDispatch *iface, DISPID dispIdMember, REFIID riid,
 128         LCID lcid, WORD wFlags, DISPPARAMS *pDispParams, VARIANT *pVarResult,
 129         EXCEPINFO *pExcepInfo, UINT *puArgErr)
 130 {
 131     if(dispIdMember == DISPID_HTMLDOCUMENTEVENTS2_ONREADYSTATECHANGE) {
 132         static const WCHAR completeW[] = {'c','o','m','p','l','e','t','e',0};
 133         HRESULT hr;
 134         BSTR state;
 135
 136         hr = IHTMLDocument2_get_readyState(html_doc, &state);
 137         ok(hr == S_OK, "got 0x%08x\n", hr);
 138         if(!memcmp(state, completeW, sizeof(completeW)))
 139             loaded = TRUE;
 140     }
 141
 142     return S_OK;
 143 }
 144
 145 static const IDispatchVtbl HTMLEventsVtbl = {
 146     HTMLEvents_QueryInterface,
 147     HTMLEvents_AddRef,
 148     HTMLEvents_Release,
 149     HTMLEvents_GetTypeInfoCount,
 150     HTMLEvents_GetTypeInfo,
 151     HTMLEvents_GetIDsOfNames,
 152     HTMLEvents_Invoke
 153 };
 154
 155 static IDispatch HTMLEvents = { &HTMLEventsVtbl };
 156
 157 static void test_QueryInterface(void)
 158 {
 159     IUnknown *xmlview, *unk;
 160     IHTMLDocument *htmldoc;
 161     HRESULT hres;
 162
 163     hres = CoCreateInstance(&CLSID_XMLView, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 164             &IID_IUnknown, (void**)&xmlview);
 165     if(FAILED(hres)) {
 166         win_skip("Failed to create XMLView instance\n");
 167         return;
 168     }
 169     ok(hres == S_OK, "CoCreateInstance returned %x, expected S_OK\n", hres);
 170
 171     hres = IUnknown_QueryInterface(xmlview, &IID_IPersistMoniker, (void**)&unk);
 172     ok(hres == S_OK, "QueryInterface(IID_IPersistMoniker) returned %x, expected S_OK\n", hres);
 173     IUnknown_Release(unk);
 174
 175     hres = IUnknown_QueryInterface(xmlview, &IID_IPersistHistory, (void**)&unk);
 176     ok(hres == S_OK, "QueryInterface(IID_IPersistHistory) returned %x, expected S_OK\n", hres);
 177     IUnknown_Release(unk);
 178
 179     hres = IUnknown_QueryInterface(xmlview, &IID_IOleCommandTarget, (void**)&unk);
 180     ok(hres == S_OK, "QueryInterface(IID_IOleCommandTarget) returned %x, expected S_OK\n", hres);
 181     IUnknown_Release(unk);
 182
 183     hres = IUnknown_QueryInterface(xmlview, &IID_IOleObject, (void**)&unk);
 184     ok(hres == S_OK, "QueryInterface(IID_IOleObject) returned %x, expected S_OK\n", hres);
 185     IUnknown_Release(unk);
 186
 187     hres = IUnknown_QueryInterface(xmlview, &IID_IHTMLDocument, (void**)&htmldoc);
 188     ok(hres == S_OK, "QueryInterface(IID_IHTMLDocument) returned %x, expected S_OK\n", hres);
 189     hres = IHTMLDocument_QueryInterface(htmldoc, &IID_IUnknown, (void**)&unk);
 190     ok(hres == S_OK, "QueryInterface(IID_IUnknown) returned %x, expected S_OK\n", hres);
 191     ok(unk == xmlview, "Aggregation is not working as expected\n");
 192     IUnknown_Release(unk);
 193     IHTMLDocument_Release(htmldoc);
 194
 195     IUnknown_Release(xmlview);
 196 }
 197
 198 static void test_Load(void)
 199 {
 200     static const WCHAR xmlview_xmlW[] = {'/','x','m','l','/','x','m','l','v','i','e','w','.','x','m','l',0};
 201     static const WCHAR res[] = {'r','e','s',':','/','/',0};
 202
 203     WCHAR buf[1024];
 204     IPersistMoniker *pers_mon;
 205     IConnectionPointContainer *cpc;
 206     IConnectionPoint *cp;
 207     IMoniker *mon;
 208     IBindCtx *bctx;
 209     IHTMLElement *elem;
 210     HRESULT hres;
 211     MSG msg;
 212     BSTR source;
 213
 214     lstrcpyW(buf, res);
 215     GetModuleFileNameW(NULL, buf+lstrlenW(buf), (sizeof(buf)-sizeof(res))/sizeof(WCHAR));
 216     lstrcatW(buf, xmlview_xmlW);
 217
 218     if(!pCreateURLMoniker) {
 219         win_skip("CreateURLMoniker not available\n");
 220         return;
 221     }
 222
 223     hres = CoCreateInstance(&CLSID_XMLView, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 224             &IID_IPersistMoniker, (void**)&pers_mon);
 225     if(FAILED(hres)) {
 226         win_skip("Failed to create XMLView instance\n");
 227         return;
 228     }
 229     ok(hres == S_OK, "CoCreateInstance returned %x, expected S_OK\n", hres);
 230
 231     hres = IPersistMoniker_QueryInterface(pers_mon, &IID_IHTMLDocument2, (void**)&html_doc);
 232     ok(hres == S_OK, "QueryInterface(HTMLDocument2) returned %x, expected S_OK\n", hres);
 233     hres = IPersistMoniker_QueryInterface(pers_mon, &IID_IConnectionPointContainer, (void**)&cpc);
 234     ok(hres == S_OK, "QueryInterface(IConnectionPointContainer) returned %x, expected S_OK\n", hres);
 235     hres = IConnectionPointContainer_FindConnectionPoint(cpc, &IID_IDispatch, &cp);
 236     ok(hres == S_OK, "FindConnectionPoint returned %x, expected S_OK\n", hres);
 237     hres = IConnectionPoint_Advise(cp, (IUnknown*)&HTMLEvents, NULL);
 238     ok(hres == S_OK, "Advise returned %x, expected S_OK\n", hres);
 239     IConnectionPoint_Release(cp);
 240     IConnectionPointContainer_Release(cpc);
 241
 242     hres = CreateBindCtx(0, &bctx);
 243     ok(hres == S_OK, "CreateBindCtx returned %x, expected S_OK\n", hres);
 244     hres = pCreateURLMoniker(NULL, buf, &mon);
 245     ok(hres == S_OK, "CreateUrlMoniker returned %x, expected S_OK\n", hres);
 246     loaded = FALSE;
 247     hres = IPersistMoniker_Load(pers_mon, TRUE, mon, bctx, 0);
 248     ok(hres == S_OK, "Load returned %x, expected S_OK\n", hres);
 249     IBindCtx_Release(bctx);
 250     IMoniker_Release(mon);
 251
 252     while(!loaded && GetMessageA(&msg, NULL, 0, 0)) {
 253         TranslateMessage(&msg);
 254         DispatchMessageA(&msg);
 255     }
 256
 257     hres = IHTMLDocument2_get_body(html_doc, &elem);
 258     ok(hres == S_OK, "get_body returned %x, expected S_OK\n", hres);
 259     hres = IHTMLElement_get_outerHTML(elem, &source);
 260     ok(hres == S_OK, "get_outerHTML returned %x, expected S_OK\n", hres);
 261     ok(!html_src_compare(source, xmlview_html), "Incorrect HTML source: %s\n", wine_dbgstr_w(source));
 262     IHTMLElement_Release(elem);
 263     SysFreeString(source);
 264
 265     IHTMLDocument2_Release(html_doc);
 266     html_doc = NULL;
 267     IPersistMoniker_Release(pers_mon);
 268 }
 269
 270 START_TEST(xmlview)
 271 {
 272     HMODULE urlmon = LoadLibraryA("urlmon.dll");
 273     pCreateURLMoniker = (void*)GetProcAddress(urlmon, "CreateURLMoniker");
 274
 275     CoInitialize(NULL);
 276     test_QueryInterface();
 277     test_Load();
 278     CoUninitialize();
 279 }