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