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         SysFreeString(state);
 141     }
 142
 143     return S_OK;
 144 }
 145
 146 static const IDispatchVtbl HTMLEventsVtbl = {
 147     HTMLEvents_QueryInterface,
 148     HTMLEvents_AddRef,
 149     HTMLEvents_Release,
 150     HTMLEvents_GetTypeInfoCount,
 151     HTMLEvents_GetTypeInfo,
 152     HTMLEvents_GetIDsOfNames,
 153     HTMLEvents_Invoke
 154 };
 155
 156 static IDispatch HTMLEvents = { &HTMLEventsVtbl };
 157
 158 static void test_QueryInterface(void)
 159 {
 160     IUnknown *xmlview, *unk;
 161     IHTMLDocument *htmldoc;
 162     HRESULT hres;
 163
 164     hres = CoCreateInstance(&CLSID_XMLView, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 165             &IID_IUnknown, (void**)&xmlview);
 166     if(FAILED(hres)) {
 167         win_skip("Failed to create XMLView instance\n");
 168         return;
 169     }
 170     ok(hres == S_OK, "CoCreateInstance returned %x, expected S_OK\n", hres);
 171
 172     hres = IUnknown_QueryInterface(xmlview, &IID_IPersistMoniker, (void**)&unk);
 173     ok(hres == S_OK, "QueryInterface(IID_IPersistMoniker) returned %x, expected S_OK\n", hres);
 174     IUnknown_Release(unk);
 175
 176     hres = IUnknown_QueryInterface(xmlview, &IID_IPersistHistory, (void**)&unk);
 177     ok(hres == S_OK, "QueryInterface(IID_IPersistHistory) returned %x, expected S_OK\n", hres);
 178     IUnknown_Release(unk);
 179
 180     hres = IUnknown_QueryInterface(xmlview, &IID_IOleCommandTarget, (void**)&unk);
 181     ok(hres == S_OK, "QueryInterface(IID_IOleCommandTarget) returned %x, expected S_OK\n", hres);
 182     IUnknown_Release(unk);
 183
 184     hres = IUnknown_QueryInterface(xmlview, &IID_IOleObject, (void**)&unk);
 185     ok(hres == S_OK, "QueryInterface(IID_IOleObject) returned %x, expected S_OK\n", hres);
 186     IUnknown_Release(unk);
 187
 188     hres = IUnknown_QueryInterface(xmlview, &IID_IHTMLDocument, (void**)&htmldoc);
 189     ok(hres == S_OK, "QueryInterface(IID_IHTMLDocument) returned %x, expected S_OK\n", hres);
 190     hres = IHTMLDocument_QueryInterface(htmldoc, &IID_IUnknown, (void**)&unk);
 191     ok(hres == S_OK, "QueryInterface(IID_IUnknown) returned %x, expected S_OK\n", hres);
 192     ok(unk == xmlview, "Aggregation is not working as expected\n");
 193     IUnknown_Release(unk);
 194     IHTMLDocument_Release(htmldoc);
 195
 196     IUnknown_Release(xmlview);
 197 }
 198
 199 static void test_Load(void)
 200 {
 201     static const WCHAR xmlview_xmlW[] = {'/','x','m','l','/','x','m','l','v','i','e','w','.','x','m','l',0};
 202     static const WCHAR res[] = {'r','e','s',':','/','/',0};
 203
 204     WCHAR buf[1024];
 205     IPersistMoniker *pers_mon;
 206     IConnectionPointContainer *cpc;
 207     IConnectionPoint *cp;
 208     IMoniker *mon;
 209     IBindCtx *bctx;
 210     IHTMLElement *elem;
 211     HRESULT hres;
 212     MSG msg;
 213     BSTR source;
 214
 215     lstrcpyW(buf, res);
 216     GetModuleFileNameW(NULL, buf+lstrlenW(buf), (sizeof(buf)-sizeof(res))/sizeof(WCHAR));
 217     lstrcatW(buf, xmlview_xmlW);
 218
 219     if(!pCreateURLMoniker) {
 220         win_skip("CreateURLMoniker not available\n");
 221         return;
 222     }
 223
 224     hres = CoCreateInstance(&CLSID_XMLView, NULL, CLSCTX_INPROC_SERVER|CLSCTX_INPROC_HANDLER,
 225             &IID_IPersistMoniker, (void**)&pers_mon);
 226     if(FAILED(hres)) {
 227         win_skip("Failed to create XMLView instance\n");
 228         return;
 229     }
 230     ok(hres == S_OK, "CoCreateInstance returned %x, expected S_OK\n", hres);
 231
 232     hres = IPersistMoniker_QueryInterface(pers_mon, &IID_IHTMLDocument2, (void**)&html_doc);
 233     ok(hres == S_OK, "QueryInterface(HTMLDocument2) returned %x, expected S_OK\n", hres);
 234     hres = IPersistMoniker_QueryInterface(pers_mon, &IID_IConnectionPointContainer, (void**)&cpc);
 235     ok(hres == S_OK, "QueryInterface(IConnectionPointContainer) returned %x, expected S_OK\n", hres);
 236     hres = IConnectionPointContainer_FindConnectionPoint(cpc, &IID_IDispatch, &cp);
 237     ok(hres == S_OK, "FindConnectionPoint returned %x, expected S_OK\n", hres);
 238     hres = IConnectionPoint_Advise(cp, (IUnknown*)&HTMLEvents, NULL);
 239     ok(hres == S_OK, "Advise returned %x, expected S_OK\n", hres);
 240     IConnectionPoint_Release(cp);
 241     IConnectionPointContainer_Release(cpc);
 242
 243     hres = CreateBindCtx(0, &bctx);
 244     ok(hres == S_OK, "CreateBindCtx returned %x, expected S_OK\n", hres);
 245     hres = pCreateURLMoniker(NULL, buf, &mon);
 246     ok(hres == S_OK, "CreateUrlMoniker returned %x, expected S_OK\n", hres);
 247     loaded = FALSE;
 248     hres = IPersistMoniker_Load(pers_mon, TRUE, mon, bctx, 0);
 249     ok(hres == S_OK, "Load returned %x, expected S_OK\n", hres);
 250     IBindCtx_Release(bctx);
 251     IMoniker_Release(mon);
 252
 253     while(!loaded && GetMessageA(&msg, NULL, 0, 0)) {
 254         TranslateMessage(&msg);
 255         DispatchMessageA(&msg);
 256     }
 257
 258     hres = IHTMLDocument2_get_body(html_doc, &elem);
 259     ok(hres == S_OK, "get_body returned %x, expected S_OK\n", hres);
 260     hres = IHTMLElement_get_outerHTML(elem, &source);
 261     ok(hres == S_OK, "get_outerHTML returned %x, expected S_OK\n", hres);
 262     ok(!html_src_compare(source, xmlview_html), "Incorrect HTML source: %s\n", wine_dbgstr_w(source));
 263     IHTMLElement_Release(elem);
 264     SysFreeString(source);
 265
 266     IHTMLDocument2_Release(html_doc);
 267     html_doc = NULL;
 268     IPersistMoniker_Release(pers_mon);
 269 }
 270
 271 START_TEST(xmlview)
 272 {
 273     HMODULE urlmon = LoadLibraryA("urlmon.dll");
 274     pCreateURLMoniker = (void*)GetProcAddress(urlmon, "CreateURLMoniker");
 275
 276     CoInitialize(NULL);
 277     test_QueryInterface();
 278     test_Load();
 279     CoUninitialize();
 280 }