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     SysFreeString(name);
 131
 132     hr = IXMLElement_get_children(element, &collection);
 133     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 134     ok(collection != NULL, "Expected non-NULL collection\n");
 135
 136     hr = IXMLElementCollection_get_length(collection, &num_child);
 137     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 138     ok(num_child == 2, "Expected 2, got %ld\n", num_child);
 139
 140     V_VT(&vIndex) = VT_I4;
 141     V_I4(&vIndex) = 0;
 142     V_VT(&vName) = VT_ERROR;
 143     V_ERROR(&vName) = DISP_E_PARAMNOTFOUND;
 144     hr = IXMLElementCollection_item(collection, vIndex, vName, (IDispatch **)&child);
 145     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 146     ok(child != NULL, "Expected non-NULL child\n");
 147
 148     hr = IXMLElement_get_type(child, &type);
 149     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 150     ok(type == XMLELEMTYPE_ELEMENT, "Expected XMLELEMTYPE_ELEMENT, got %ld\n", type);
 151
 152     hr = IXMLElement_get_tagName(child, &name);
 153     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 154     ok(!lstrcmpW(name, szNumber), "Expected NUMBER\n");
 155     SysFreeString(name);
 156
 157     hr = IXMLElement_get_children(child, &inner);
 158     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 159     ok(inner != NULL, "Expected non-NULL inner\n");
 160
 161     hr = IXMLElementCollection_get_length(inner, &num_child);
 162     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 163     ok(num_child == 1, "Expected 1, got %ld\n", num_child);
 164
 165     hr = IXMLElementCollection_item(inner, vIndex, vName, (IDispatch **)&value);
 166     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 167     ok(value != NULL, "Expected non-NULL value\n");
 168
 169     hr = IXMLElement_get_type(value, &type);
 170     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 171     ok(type == XMLELEMTYPE_TEXT, "Expected XMLELEMTYPE_TEXT, got %ld\n", type);
 172
 173     hr = IXMLElement_get_text(value, &name);
 174     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 175     ok(!lstrcmpW(name, szNumVal), "Expected '1234'\n");
 176     SysFreeString(name);
 177
 178     IXMLElementCollection_Release(inner);
 179
 180     inner = (IXMLElementCollection *)0xdeadbeef;
 181     hr = IXMLElement_get_children(value, &inner);
 182     ok(hr == 1, "Expected 1, got %d\n", hr);
 183     ok(inner == NULL, "Expected NULL inner, got %p\n", inner);
 184
 185     IXMLElement_Release(value);
 186     IXMLElement_Release(child);
 187     value = NULL;
 188     child = NULL;
 189
 190     V_I4(&vIndex) = 1;
 191     hr = IXMLElementCollection_item(collection, vIndex, vName, (IDispatch **)&child);
 192     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 193     ok(child != NULL, "Expected non-NULL child\n");
 194
 195     hr = IXMLElement_get_type(child, &type);
 196     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 197     ok(type == XMLELEMTYPE_ELEMENT, "Expected XMLELEMTYPE_ELEMENT, got %ld\n", type);
 198
 199     hr = IXMLElement_get_tagName(child, &name);
 200     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 201     ok(!lstrcmpW(name, szName), "Expected NAME\n");
 202     SysFreeString(name);
 203
 204     hr = IXMLElement_get_children(child, &inner);
 205     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 206     ok(inner != NULL, "Expected non-NULL inner\n");
 207
 208     hr = IXMLElementCollection_get_length(inner, &num_child);
 209     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 210     ok(num_child == 1, "Expected 1, got %ld\n", num_child);
 211
 212     V_I4(&vIndex) = 0;
 213     hr = IXMLElementCollection_item(inner, vIndex, vName, (IDispatch **)&value);
 214     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 215     ok(value != NULL, "Expected non-NULL value\n");
 216
 217     hr = IXMLElement_get_type(value, &type);
 218     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 219     ok(type == XMLELEMTYPE_TEXT, "Expected XMLELEMTYPE_TEXT, got %ld\n", type);
 220
 221     hr = IXMLElement_get_text(value, &name);
 222     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 223     ok(!lstrcmpW(name, szNameVal), "Expected 'Captain Ahab'\n");
 224     SysFreeString(name);
 225
 226     IXMLElementCollection_Release(inner);
 227
 228     inner = (IXMLElementCollection *)0xdeadbeef;
 229     hr = IXMLElement_get_children(value, &inner);
 230     ok(hr == 1, "Expected 1, got %d\n", hr);
 231     ok(inner == NULL, "Expected NULL inner, got %p\n", inner);
 232
 233     IXMLElement_Release(value);
 234     IXMLElement_Release(child);
 235     IXMLElementCollection_Release(collection);
 236     IXMLElement_Release(element);
 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 %d\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     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 274     ok(element != NULL, "Expected non-NULL element\n");
 275
 276     if (element != NULL)
 277     {
 278         hr = IXMLElement_get_type(element, &type);
 279         ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 280         ok(type == XMLELEMTYPE_OTHER, "Expected XMLELEMTYPE_OTHER, got %ld\n", type);
 281
 282         IXMLElement_Release(element);
 283     }
 284
 285     /* invalid vName type */
 286     V_VT(&vType) = VT_I4;
 287     V_I4(&vType) = XMLELEMTYPE_ELEMENT;
 288     V_VT(&vName) = VT_I4;
 289     hr = IXMLDocument_createElement(doc, vType, vName, &element);
 290     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 291     ok(element != NULL, "Expected non-NULL element\n");
 292
 293     hr = IXMLElement_get_type(element, &type);
 294     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 295     ok(type == XMLELEMTYPE_ELEMENT, "Expected XMLELEMTYPE_ELEMENT, got %ld\n", type);
 296
 297     IXMLElement_Release(element);
 298
 299     /* NULL element */
 300     V_VT(&vType) = VT_I4;
 301     V_I4(&vType) = XMLELEMTYPE_ELEMENT;
 302     V_VT(&vName) = VT_I4;
 303     hr = IXMLDocument_createElement(doc, vType, vName, NULL);
 304     ok(hr == E_INVALIDARG, "Expected E_INVALIDARG, got %08x\n", hr);
 305
 306     root = (IXMLElement *)0xdeadbeef;
 307     hr = IXMLDocument_get_root(doc, &root);
 308     ok(hr == E_FAIL, "Expected E_FAIL, got %08x\n", hr);
 309     ok(root == NULL, "Expected NULL root\n");
 310
 311     V_VT(&vType) = VT_I4;
 312     V_I4(&vType) = XMLELEMTYPE_ELEMENT;
 313     V_VT(&vName) = VT_NULL;
 314     hr = IXMLDocument_createElement(doc, vType, vName, &element);
 315     ok(hr == S_OK, "Expected S_OK, got %d\n", hr);
 316     ok(element != NULL, "Expected non-NULL element\n");
 317
 318     /* createElement does not set the new element as root */
 319     root = (IXMLElement *)0xdeadbeef;
 320     hr = IXMLDocument_get_root(doc, &root);
 321     ok(hr == E_FAIL, "Expected E_FAIL, got %d\n", hr);
 322     ok(root == NULL, "Expected NULL root\n");
 323
 324     IXMLElement_Release(element);
 325     IXMLDocument_Release(doc);
 326 }
 327
 328 START_TEST(xmldoc)
 329 {
 330     HRESULT hr;
 331
 332     hr = CoInitialize(NULL);
 333     ok(hr == S_OK, "failed to init com\n");
 334
 335     test_xmldoc();
 336     test_createElement();
 337 }