winecrt0: DriverEntry is a WINAPI function.
[wine] / 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 }