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 static void append_str(char **str, const char *data)
  32 {
  33     sprintf(*str, data);
  34     *str += strlen(*str);
  35 }
  36
  37 static void create_xml_file(LPCSTR filename)
  38 {
  39     char data[1024];
  40     char *ptr = data;
  41     DWORD dwNumberOfBytesWritten;
  42     HANDLE hf = CreateFile(filename, GENERIC_WRITE, 0, NULL,
  43                            CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
  44
  45     append_str(&ptr, "<?xml version=\"1.0\" ?>\n");
  46     append_str(&ptr, "<BankAccount>\n");
  47     append_str(&ptr, "  <Number>1234</Number>\n");
  48     append_str(&ptr, "  <Name>Captain Ahab</Name>\n");
  49     append_str(&ptr, "</BankAccount>");
  50
  51     WriteFile(hf, data, ptr - data, &dwNumberOfBytesWritten, NULL);
  52     CloseHandle(hf);
  53 }
  54
  55 static void test_xmldoc(void)
  56 {
  57     HRESULT hr;
  58     IXMLDocument *doc = NULL;
  59     IXMLElement *element = NULL, *child = NULL, *value = NULL;
  60     IXMLElementCollection *collection = NULL, *inner = NULL;
  61     IPersistStreamInit *psi = NULL;
  62     IStream *stream = NULL;
  63     HGLOBAL hglobal;
  64     HANDLE hfile;
  65     LPVOID ptr;
  66     DWORD file_size, read;
  67     CHAR path[MAX_PATH];
  68     long type, num_child;
  69     VARIANT vIndex, vName;
  70     BSTR name = NULL;
  71
  72     static const WCHAR szBankAccount[] = {'B','A','N','K','A','C','C','O','U','N','T',0};
  73     static const WCHAR szNumber[] = {'N','U','M','B','E','R',0};
  74     static const WCHAR szNumVal[] = {'1','2','3','4',0};
  75     static const WCHAR szName[] = {'N','A','M','E',0};
  76     static const WCHAR szNameVal[] = {'C','a','p','t','a','i','n',' ','A','h','a','b',0};
  77
  78     hr = CoCreateInstance(&CLSID_XMLDocument, NULL, CLSCTX_INPROC_SERVER,
  79                           &IID_IXMLDocument, (LPVOID*)&doc);
  80     if (FAILED(hr))
  81     {
  82         skip("Failed to create XMLDocument instance\n");
  83         return;
  84     }
  85
  86     create_xml_file("bank.xml");
  87     GetFullPathNameA("bank.xml", MAX_PATH, path, NULL);
  88
  89     hfile = CreateFile(path, GENERIC_READ, FILE_SHARE_READ, NULL,
  90                        OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
  91     ok(hfile != INVALID_HANDLE_VALUE, "Expected a valid file handle\n");
  92     file_size = GetFileSize(hfile, NULL);
  93
  94     hglobal = GlobalAlloc(GHND, file_size);
  95     ptr = GlobalLock(hglobal);
  96
  97     ReadFile(hfile, ptr, file_size, &read, NULL);
  98     ok(file_size == read, "Expected to read the whole file, read %d\n", read);
  99
 100     hr = CreateStreamOnHGlobal(hglobal, TRUE, &stream);
 101     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 102     ok(stream != NULL, "Expected non-NULL stream\n");
 103
 104     CloseHandle(hfile);
 105     GlobalUnlock(hglobal);
 106
 107     hr = IXMLDocument_QueryInterface(doc, &IID_IPersistStreamInit, (LPVOID *)&psi);
 108     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 109     ok(psi != NULL, "Expected non-NULL psi\n");
 110
 111     hr = IXMLDocument_get_root(doc, &element);
 112     ok(hr == E_FAIL, "Expected E_FAIL, got %d\n", hr);
 113     ok(element == NULL, "Expected NULL element\n");
 114
 115     hr = IPersistStreamInit_Load(psi, stream);
 116     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 117     ok(stream != NULL, "Expected non-NULL stream\n");
 118
 119     hr = IXMLDocument_get_root(doc, &element);
 120     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 121     ok(element != NULL, "Expected non-NULL element\n");
 122
 123     hr = IXMLElement_get_type(element, &type);
 124     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 125     ok(type == XMLELEMTYPE_ELEMENT, "Expected XMLELEMTYPE_ELEMENT, got %ld\n", type);
 126
 127     hr = IXMLElement_get_tagName(element, &name);
 128     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 129     ok(!lstrcmpW(name, szBankAccount), "Expected BANKACCOUNT\n");
 130
 131     hr = IXMLElement_get_children(element, &collection);
 132     ok(hr == S_OK, "Expected S_OK, got %d\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 %d\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 %d\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 %d\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 %d\n", hr);
 153     ok(!lstrcmpW(name, szNumber), "Expected NUMBER\n");
 154
 155     hr = IXMLElement_get_children(child, &inner);
 156     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 157     ok(inner != NULL, "Expected non-NULL inner\n");
 158
 159     hr = IXMLElementCollection_get_length(inner, &num_child);
 160     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 161     ok(num_child == 1, "Expected 1, got %ld\n", num_child);
 162
 163     hr = IXMLElementCollection_item(inner, vIndex, vName, (IDispatch **)&value);
 164     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 165     ok(value != NULL, "Expected non-NULL value\n");
 166
 167     hr = IXMLElement_get_type(value, &type);
 168     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 169     ok(type == XMLELEMTYPE_TEXT, "Expected XMLELEMTYPE_TEXT, got %ld\n", type);
 170
 171     hr = IXMLElement_get_text(value, &name);
 172     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 173     ok(!lstrcmpW(name, szNumVal), "Expected '1234'\n");
 174
 175     IXMLElementCollection_Release(inner);
 176
 177     inner = (IXMLElementCollection *)0xdeadbeef;
 178     hr = IXMLElement_get_children(value, &inner);
 179     ok(hr == 1, "Expected 1, got %d\n", hr);
 180     ok(inner == NULL, "Expected NULL inner, got %p\n", inner);
 181
 182     IXMLElement_Release(value);
 183     IXMLElement_Release(child);
 184     value = NULL;
 185     child = NULL;
 186
 187     V_I4(&vIndex) = 1;
 188     hr = IXMLElementCollection_item(collection, vIndex, vName, (IDispatch **)&child);
 189     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 190     ok(child != NULL, "Expected non-NULL child\n");
 191
 192     hr = IXMLElement_get_type(child, &type);
 193     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 194     ok(type == XMLELEMTYPE_ELEMENT, "Expected XMLELEMTYPE_ELEMENT, got %ld\n", type);
 195
 196     hr = IXMLElement_get_tagName(child, &name);
 197     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 198     ok(!lstrcmpW(name, szName), "Expected NAME\n");
 199
 200     hr = IXMLElement_get_children(child, &inner);
 201     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 202     ok(inner != NULL, "Expected non-NULL inner\n");
 203
 204     hr = IXMLElementCollection_get_length(inner, &num_child);
 205     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 206     ok(num_child == 1, "Expected 1, got %ld\n", num_child);
 207
 208     V_I4(&vIndex) = 0;
 209     hr = IXMLElementCollection_item(inner, vIndex, vName, (IDispatch **)&value);
 210     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 211     ok(value != NULL, "Expected non-NULL value\n");
 212
 213     hr = IXMLElement_get_type(value, &type);
 214     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 215     ok(type == XMLELEMTYPE_TEXT, "Expected XMLELEMTYPE_TEXT, got %ld\n", type);
 216
 217     hr = IXMLElement_get_text(value, &name);
 218     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 219     ok(!lstrcmpW(name, szNameVal), "Expected 'Captain Ahab'\n");
 220
 221     IXMLElementCollection_Release(inner);
 222
 223     inner = (IXMLElementCollection *)0xdeadbeef;
 224     hr = IXMLElement_get_children(value, &inner);
 225     ok(hr == 1, "Expected 1, got %d\n", hr);
 226     ok(inner == NULL, "Expected NULL inner, got %p\n", inner);
 227
 228     IXMLElement_Release(value);
 229     IXMLElement_Release(child);
 230     IXMLElementCollection_Release(collection);
 231     IXMLElement_Release(element);
 232     IStream_Release(stream);
 233     IPersistStreamInit_Release(psi);
 234     IXMLDocument_Release(doc);
 235
 236     DeleteFileA("bank.xml");
 237 }
 238
 239 static void test_createElement(void)
 240 {
 241     HRESULT hr;
 242     IXMLDocument *doc = NULL;
 243     IXMLElement *element = NULL, *root = NULL;
 244     VARIANT vType, vName;
 245     long type;
 246
 247     hr = CoCreateInstance(&CLSID_XMLDocument, NULL, CLSCTX_INPROC_SERVER,
 248                           &IID_IXMLDocument, (LPVOID*)&doc);
 249     if (FAILED(hr))
 250     {
 251         skip("Failed to create XMLDocument instance\n");
 252         return;
 253     }
 254
 255     /* invalid vType type */
 256     V_VT(&vType) = VT_NULL;
 257     V_VT(&vName) = VT_NULL;
 258     element = (IXMLElement *)0xdeadbeef;
 259     hr = IXMLDocument_createElement(doc, vType, vName, &element);
 260     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %d\n", hr);
 261     ok(element == NULL, "Expected NULL element\n");
 262
 263     /* invalid vType value */
 264     V_VT(&vType) = VT_I4;
 265     V_I4(&vType) = -1;
 266     V_VT(&vName) = VT_NULL;
 267     hr = IXMLDocument_createElement(doc, vType, vName, &element);
 268     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 269     ok(element != NULL, "Expected non-NULL element\n");
 270
 271     if (element != NULL)
 272     {
 273         hr = IXMLElement_get_type(element, &type);
 274         ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 275         ok(type == XMLELEMTYPE_OTHER, "Expected XMLELEMTYPE_OTHER, got %ld\n", type);
 276
 277         IXMLElement_Release(element);
 278     }
 279
 280     /* invalid vName type */
 281     V_VT(&vType) = VT_I4;
 282     V_I4(&vType) = XMLELEMTYPE_ELEMENT;
 283     V_VT(&vName) = VT_I4;
 284     hr = IXMLDocument_createElement(doc, vType, vName, &element);
 285     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 286     ok(element != NULL, "Expected non-NULL element\n");
 287
 288     hr = IXMLElement_get_type(element, &type);
 289     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 290     ok(type == XMLELEMTYPE_ELEMENT, "Expected XMLELEMTYPE_ELEMENT, got %ld\n", type);
 291
 292     IXMLElement_Release(element);
 293
 294     /* NULL element */
 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, NULL);
 299     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
 300
 301     root = (IXMLElement *)0xdeadbeef;
 302     hr = IXMLDocument_get_root(doc, &root);
 303     ok(hr == E_FAIL, "Expected E_FAIL, got %08x\n", hr);
 304     ok(root == NULL, "Expected NULL root\n");
 305
 306     V_VT(&vType) = VT_I4;
 307     V_I4(&vType) = XMLELEMTYPE_ELEMENT;
 308     V_VT(&vName) = VT_NULL;
 309     hr = IXMLDocument_createElement(doc, vType, vName, &element);
 310     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 311     ok(element != NULL, "Expected non-NULL element\n");
 312
 313     /* createElement does not set the new element as root */
 314     root = (IXMLElement *)0xdeadbeef;
 315     hr = IXMLDocument_get_root(doc, &root);
 316     ok(hr == E_FAIL, "Expected E_FAIL, got %d\n", hr);
 317     ok(root == NULL, "Expected NULL root\n");
 318
 319     IXMLElement_Release(element);
 320     IXMLDocument_Release(doc);
 321 }
 322
 323 START_TEST(xmldoc)
 324 {
 325     HRESULT hr;
 326
 327     hr = CoInitialize(NULL);
 328     ok(hr == S_OK, "failed to init com\n");
 329
 330     test_xmldoc();
 331     test_createElement();
 332 }