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