dlls/msxml3/tests/xmldoc.c

   1 /*
   2  * IXMLDocument tests
   3  *
   4  * Copyright 2007 James Hawkins
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #define COBJMACROS
  22
  23 #include <stdio.h>
  24 #include "windows.h"
  25 #include "ole2.h"
  26 #include "msxml2.h"
  27 #include "ocidl.h"
  28
  29 #include "wine/test.h"
  30
  31 /* Deprecated Error Code */
  32 #define XML_E_INVALIDATROOTLEVEL    0xc00ce556
  33
  34 static void create_xml_file(LPCSTR filename)
  35 {
  36     DWORD dwNumberOfBytesWritten;
  37     HANDLE hf = CreateFile(filename, GENERIC_WRITE, 0, NULL,
  38                            CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
  39
  40     static const char data[] =
  41         "<?xml version=\"1.0\" ?>\n"
  42         "<BankAccount>\n"
  43         "  <Number>1234</Number>\n"
  44         "  <Name>Captain Ahab</Name>\n"
  45         "</BankAccount>";
  46
  47     WriteFile(hf, data, sizeof(data) - 1, &dwNumberOfBytesWritten, NULL);
  48     CloseHandle(hf);
  49 }
  50
  51 static void test_xmldoc(void)
  52 {
  53     HRESULT hr;
  54     IXMLDocument *doc = NULL;
  55     IXMLElement *element = NULL, *child = NULL, *value = NULL;
  56     IXMLElementCollection *collection = NULL, *inner = NULL;
  57     IPersistStreamInit *psi = NULL;
  58     IStream *stream = NULL;
  59     HGLOBAL hglobal;
  60     HANDLE hfile;
  61     LPVOID ptr;
  62     DWORD file_size, read;
  63     CHAR path[MAX_PATH];
  64     long type, num_child;
  65     VARIANT vIndex, vName;
  66     BSTR name = NULL;
  67
  68     static const WCHAR szBankAccount[] = {'B','A','N','K','A','C','C','O','U','N','T',0};
  69     static const WCHAR szNumber[] = {'N','U','M','B','E','R',0};
  70     static const WCHAR szNumVal[] = {'1','2','3','4',0};
  71     static const WCHAR szName[] = {'N','A','M','E',0};
  72     static const WCHAR szNameVal[] = {'C','a','p','t','a','i','n',' ','A','h','a','b',0};
  73
  74     hr = CoCreateInstance(&CLSID_XMLDocument, NULL, CLSCTX_INPROC_SERVER,
  75                           &IID_IXMLDocument, (LPVOID*)&doc);
  76     if (FAILED(hr))
  77     {
  78         skip("Failed to create XMLDocument instance\n");
  79         return;
  80     }
  81
  82     create_xml_file("bank.xml");
  83     GetFullPathNameA("bank.xml", MAX_PATH, path, NULL);
  84
  85     hfile = CreateFile(path, GENERIC_READ, FILE_SHARE_READ, NULL,
  86                        OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
  87     ok(hfile != INVALID_HANDLE_VALUE, "Expected a valid file handle\n");
  88     file_size = GetFileSize(hfile, NULL);
  89
  90     hglobal = GlobalAlloc(GHND, file_size);
  91     ptr = GlobalLock(hglobal);
  92
  93     ReadFile(hfile, ptr, file_size, &read, NULL);
  94     ok(file_size == read, "Expected to read the whole file, read %d\n", read);
  95
  96     hr = CreateStreamOnHGlobal(hglobal, TRUE, &stream);
  97     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
  98     ok(stream != NULL, "Expected non-NULL stream\n");
  99
 100     CloseHandle(hfile);
 101     GlobalUnlock(hglobal);
 102
 103     hr = IXMLDocument_QueryInterface(doc, &IID_IPersistStreamInit, (LPVOID *)&psi);
 104     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 105     ok(psi != NULL, "Expected non-NULL psi\n");
 106
 107     hr = IXMLDocument_get_root(doc, &element);
 108     ok(hr == E_FAIL, "Expected E_FAIL, got %08x\n", hr);
 109     ok(element == NULL, "Expected NULL element\n");
 110
 111     hr = IPersistStreamInit_Load(psi, stream);
 112     ok(hr == S_OK || hr == XML_E_INVALIDATROOTLEVEL, "Expected S_OK, got %08x\n", hr);
 113     if(hr == XML_E_INVALIDATROOTLEVEL)
 114         goto cleanup;
 115
 116     ok(stream != NULL, "Expected non-NULL stream\n");
 117
 118     hr = IXMLDocument_get_root(doc, &element);
 119     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 120     ok(element != NULL, "Expected non-NULL element\n");
 121
 122     hr = IXMLElement_get_type(element, &type);
 123     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 124     ok(type == XMLELEMTYPE_ELEMENT, "Expected XMLELEMTYPE_ELEMENT, got %ld\n", type);
 125
 126     hr = IXMLElement_get_tagName(element, &name);
 127     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 128     ok(!lstrcmpW(name, szBankAccount), "Expected BANKACCOUNT\n");
 129     SysFreeString(name);
 130
 131     hr = IXMLElement_get_children(element, &collection);
 132     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 133     ok(collection != NULL, "Expected non-NULL collection\n");
 134
 135     hr = IXMLElementCollection_get_length(collection, &num_child);
 136     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 137     ok(num_child == 2, "Expected 2, got %ld\n", num_child);
 138
 139     V_VT(&vIndex) = VT_I4;
 140     V_I4(&vIndex) = 0;
 141     V_VT(&vName) = VT_ERROR;
 142     V_ERROR(&vName) = DISP_E_PARAMNOTFOUND;
 143     hr = IXMLElementCollection_item(collection, vIndex, vName, (IDispatch **)&child);
 144     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 145     ok(child != NULL, "Expected non-NULL child\n");
 146
 147     hr = IXMLElement_get_type(child, &type);
 148     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 149     ok(type == XMLELEMTYPE_ELEMENT, "Expected XMLELEMTYPE_ELEMENT, got %ld\n", type);
 150
 151     hr = IXMLElement_get_tagName(child, &name);
 152     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 153     ok(!lstrcmpW(name, szNumber), "Expected NUMBER\n");
 154     SysFreeString(name);
 155
 156     hr = IXMLElement_get_children(child, &inner);
 157     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 158     ok(inner != NULL, "Expected non-NULL inner\n");
 159
 160     hr = IXMLElementCollection_get_length(inner, &num_child);
 161     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 162     ok(num_child == 1, "Expected 1, got %ld\n", num_child);
 163
 164     hr = IXMLElementCollection_item(inner, vIndex, vName, (IDispatch **)&value);
 165     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 166     ok(value != NULL, "Expected non-NULL value\n");
 167
 168     hr = IXMLElement_get_type(value, &type);
 169     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 170     ok(type == XMLELEMTYPE_TEXT, "Expected XMLELEMTYPE_TEXT, got %ld\n", type);
 171
 172     hr = IXMLElement_get_text(value, &name);
 173     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 174     ok(!lstrcmpW(name, szNumVal), "Expected '1234'\n");
 175     SysFreeString(name);
 176
 177     IXMLElementCollection_Release(inner);
 178
 179     inner = (IXMLElementCollection *)0xdeadbeef;
 180     hr = IXMLElement_get_children(value, &inner);
 181     ok(hr == 1, "Expected 1, got %08x\n", hr);
 182     ok(inner == NULL, "Expected NULL inner, got %p\n", inner);
 183
 184     IXMLElement_Release(value);
 185     IXMLElement_Release(child);
 186     value = NULL;
 187     child = NULL;
 188
 189     V_I4(&vIndex) = 1;
 190     hr = IXMLElementCollection_item(collection, vIndex, vName, (IDispatch **)&child);
 191     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 192     ok(child != NULL, "Expected non-NULL child\n");
 193
 194     hr = IXMLElement_get_type(child, &type);
 195     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 196     ok(type == XMLELEMTYPE_ELEMENT, "Expected XMLELEMTYPE_ELEMENT, got %ld\n", type);
 197
 198     hr = IXMLElement_get_tagName(child, &name);
 199     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 200     ok(!lstrcmpW(name, szName), "Expected NAME\n");
 201     SysFreeString(name);
 202
 203     hr = IXMLElement_get_children(child, &inner);
 204     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 205     ok(inner != NULL, "Expected non-NULL inner\n");
 206
 207     hr = IXMLElementCollection_get_length(inner, &num_child);
 208     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 209     ok(num_child == 1, "Expected 1, got %ld\n", num_child);
 210
 211     V_I4(&vIndex) = 0;
 212     hr = IXMLElementCollection_item(inner, vIndex, vName, (IDispatch **)&value);
 213     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 214     ok(value != NULL, "Expected non-NULL value\n");
 215
 216     hr = IXMLElement_get_type(value, &type);
 217     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 218     ok(type == XMLELEMTYPE_TEXT, "Expected XMLELEMTYPE_TEXT, got %ld\n", type);
 219
 220     hr = IXMLElement_get_text(value, &name);
 221     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 222     ok(!lstrcmpW(name, szNameVal), "Expected 'Captain Ahab'\n");
 223     SysFreeString(name);
 224
 225     IXMLElementCollection_Release(inner);
 226
 227     inner = (IXMLElementCollection *)0xdeadbeef;
 228     hr = IXMLElement_get_children(value, &inner);
 229     ok(hr == 1, "Expected 1, got %08x\n", hr);
 230     ok(inner == NULL, "Expected NULL inner, got %p\n", inner);
 231
 232     IXMLElement_Release(value);
 233     IXMLElement_Release(child);
 234     IXMLElementCollection_Release(collection);
 235     IXMLElement_Release(element);
 236 cleanup:
 237     IStream_Release(stream);
 238     IPersistStreamInit_Release(psi);
 239     IXMLDocument_Release(doc);
 240
 241     DeleteFileA("bank.xml");
 242 }
 243
 244 static void test_createElement(void)
 245 {
 246     HRESULT hr;
 247     IXMLDocument *doc = NULL;
 248     IXMLElement *element = NULL, *root = NULL;
 249     VARIANT vType, vName;
 250     long type;
 251
 252     hr = CoCreateInstance(&CLSID_XMLDocument, NULL, CLSCTX_INPROC_SERVER,
 253                           &IID_IXMLDocument, (LPVOID*)&doc);
 254     if (FAILED(hr))
 255     {
 256         skip("Failed to create XMLDocument instance\n");
 257         return;
 258     }
 259
 260     /* invalid vType type */
 261     V_VT(&vType) = VT_NULL;
 262     V_VT(&vName) = VT_NULL;
 263     element = (IXMLElement *)0xdeadbeef;
 264     hr = IXMLDocument_createElement(doc, vType, vName, &element);
 265     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
 266     ok(element == NULL, "Expected NULL element\n");
 267
 268     /* invalid vType value */
 269     V_VT(&vType) = VT_I4;
 270     V_I4(&vType) = -1;
 271     V_VT(&vName) = VT_NULL;
 272     hr = IXMLDocument_createElement(doc, vType, vName, &element);
 273     /* Up to and including SP7, createElement returns an element. */
 274     if(hr == S_OK)
 275     {
 276         ok(element != NULL, "Expected element\n");
 277         if (element != NULL)
 278         {
 279             hr = IXMLElement_get_type(element, &type);
 280             ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 281             /* SP7 returns an XMLELEMTYPE_ELEMENT */
 282             ok(type == XMLELEMTYPE_OTHER || type == XMLELEMTYPE_ELEMENT,
 283                          "Expected XMLELEMTYPE_OTHER || XMLELEMTYPE_ELEMENT, got %ld\n", type);
 284
 285             IXMLElement_Release(element);
 286         }
 287     }
 288     else
 289     {
 290         ok(hr == E_NOTIMPL, "Expected E_NOTIMPL, got %08x\n", hr);
 291         ok(element == NULL, "Expected NULL element\n");
 292     }
 293
 294     /* invalid vName type */
 295     V_VT(&vType) = VT_I4;
 296     V_I4(&vType) = XMLELEMTYPE_ELEMENT;
 297     V_VT(&vName) = VT_I4;
 298     hr = IXMLDocument_createElement(doc, vType, vName, &element);
 299     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 300     ok(element != NULL, "Expected non-NULL element\n");
 301
 302     hr = IXMLElement_get_type(element, &type);
 303     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 304     ok(type == XMLELEMTYPE_ELEMENT, "Expected XMLELEMTYPE_ELEMENT, got %ld\n", type);
 305
 306     IXMLElement_Release(element);
 307
 308     /* NULL element */
 309     V_VT(&vType) = VT_I4;
 310     V_I4(&vType) = XMLELEMTYPE_ELEMENT;
 311     V_VT(&vName) = VT_I4;
 312     hr = IXMLDocument_createElement(doc, vType, vName, NULL);
 313     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
 314
 315     root = (IXMLElement *)0xdeadbeef;
 316     hr = IXMLDocument_get_root(doc, &root);
 317     ok(hr == E_FAIL, "Expected E_FAIL, got %08x\n", hr);
 318     ok(root == NULL, "Expected NULL root\n");
 319
 320     V_VT(&vType) = VT_I4;
 321     V_I4(&vType) = XMLELEMTYPE_ELEMENT;
 322     V_VT(&vName) = VT_NULL;
 323     hr = IXMLDocument_createElement(doc, vType, vName, &element);
 324     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 325     ok(element != NULL, "Expected non-NULL element\n");
 326
 327     /* createElement does not set the new element as root */
 328     root = (IXMLElement *)0xdeadbeef;
 329     hr = IXMLDocument_get_root(doc, &root);
 330     ok(hr == E_FAIL, "Expected E_FAIL, got %08x\n", hr);
 331     ok(root == NULL, "Expected NULL root\n");
 332
 333     IXMLElement_Release(element);
 334     IXMLDocument_Release(doc);
 335 }
 336
 337 START_TEST(xmldoc)
 338 {
 339     HRESULT hr;
 340
 341     hr = CoInitialize(NULL);
 342     ok(hr == S_OK, "failed to init com\n");
 343
 344     test_xmldoc();
 345     test_createElement();
 346
 347     CoUninitialize();
 348 }