dlls/msxml3/tests/saxreader.c

   1 /*
   2  * XML test
   3  *
   4  * Copyright 2008 Piotr Caban
   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 #define CONST_VTABLE
  23
  24 #include <stdio.h>
  25 #include "windows.h"
  26 #include "ole2.h"
  27 #include "msxml2.h"
  28 #include "ocidl.h"
  29
  30 #include "wine/test.h"
  31
  32 typedef enum _CH {
  33     CH_ENDTEST,
  34     CH_PUTDOCUMENTLOCATOR,
  35     CH_STARTDOCUMENT,
  36     CH_ENDDOCUMENT,
  37     CH_STARTPREFIXMAPPING,
  38     CH_ENDPREFIXMAPPING,
  39     CH_STARTELEMENT,
  40     CH_ENDELEMENT,
  41     CH_CHARACTERS,
  42     CH_IGNORABLEWHITESPACE,
  43     CH_PROCESSINGINSTRUCTION,
  44     CH_SKIPPEDENTITY
  45 } CH;
  46
  47 static const WCHAR szSimpleXML[] = {
  48 '<','?','x','m','l',' ','v','e','r','s','i','o','n','=','\"','1','.','0','\"',' ','?','>','\n',
  49 '<','B','a','n','k','A','c','c','o','u','n','t','>','\n',
  50 ' ',' ',' ','<','N','u','m','b','e','r','>','1','2','3','4','<','/','N','u','m','b','e','r','>','\n',
  51 ' ',' ',' ','<','N','a','m','e','>','C','a','p','t','a','i','n',' ','A','h','a','b','<','/','N','a','m','e','>','\n',
  52 '<','/','B','a','n','k','A','c','c','o','u','n','t','>','\n','\0'
  53 };
  54
  55 static const WCHAR szCarriageRetTest[] = {
  56 '<','?','x','m','l',' ','v','e','r','s','i','o','n','=','"','1','.','0','"','?','>','\r','\n',
  57 '<','B','a','n','k','A','c','c','o','u','n','t','>','\r','\n',
  58 '\t','<','N','u','m','b','e','r','>','1','2','3','4','<','/','N','u','m','b','e','r','>','\r','\n',
  59 '\t','<','N','a','m','e','>','C','a','p','t','a','i','n',' ','A','h','a','b','<','/','N','a','m','e','>','\r','\n',
  60 '<','/','B','a','n','k','A','c','c','o','u','n','t','>','\0'
  61 };
  62
  63 static const CHAR szTestXML[] =
  64 "<?xml version=\"1.0\" ?>\n"
  65 "<BankAccount>\n"
  66 "   <Number>1234</Number>\n"
  67 "   <Name>Captain Ahab</Name>\n"
  68 "</BankAccount>\n";
  69
  70 typedef struct _contenthandlercheck {
  71     CH id;
  72     int line;
  73     int column;
  74     const char *arg1;
  75     const char *arg2;
  76     const char *arg3;
  77 } content_handler_test;
  78
  79 static content_handler_test contentHandlerTest1[] = {
  80     { CH_PUTDOCUMENTLOCATOR, 0, 0 },
  81     { CH_STARTDOCUMENT, 0, 0 },
  82     { CH_STARTELEMENT, 2, 14, "", "BankAccount", "BankAccount" },
  83     { CH_CHARACTERS, 2, 14, "\n   " },
  84     { CH_STARTELEMENT, 3, 12, "", "Number", "Number" },
  85     { CH_CHARACTERS, 3, 12, "1234" },
  86     { CH_ENDELEMENT, 3, 18, "", "Number", "Number" },
  87     { CH_CHARACTERS, 3, 25, "\n   " },
  88     { CH_STARTELEMENT, 4, 10, "", "Name", "Name" },
  89     { CH_CHARACTERS, 4, 10, "Captain Ahab" },
  90     { CH_ENDELEMENT, 4, 24, "", "Name", "Name" },
  91     { CH_CHARACTERS, 4, 29, "\n" },
  92     { CH_ENDELEMENT, 5, 3, "", "BankAccount", "BankAccount" },
  93     { CH_ENDDOCUMENT, 0, 0 },
  94     { CH_ENDTEST }
  95 };
  96
  97 static content_handler_test contentHandlerTest2[] = {
  98     { CH_PUTDOCUMENTLOCATOR, 0, 0 },
  99     { CH_STARTDOCUMENT, 0, 0 },
 100     { CH_STARTELEMENT, 2, 14, "", "BankAccount", "BankAccount" },
 101     { CH_CHARACTERS, 2, 14, "\n" },
 102     { CH_CHARACTERS, 2, 16, "\t" },
 103     { CH_STARTELEMENT, 3, 10, "", "Number", "Number" },
 104     { CH_CHARACTERS, 3, 10, "1234" },
 105     { CH_ENDELEMENT, 3, 16, "", "Number", "Number" },
 106     { CH_CHARACTERS, 3, 23, "\n" },
 107     { CH_CHARACTERS, 3, 25, "\t" },
 108     { CH_STARTELEMENT, 4, 8, "", "Name", "Name" },
 109     { CH_CHARACTERS, 4, 8, "Captain Ahab" },
 110     { CH_ENDELEMENT, 4, 22, "", "Name", "Name" },
 111     { CH_CHARACTERS, 4, 27, "\n" },
 112     { CH_ENDELEMENT, 5, 3, "", "BankAccount", "BankAccount" },
 113     { CH_ENDDOCUMENT, 0, 0 },
 114     { CH_ENDTEST }
 115 };
 116
 117 static content_handler_test *expectCall;
 118 static ISAXLocator *locator;
 119
 120 static const char *debugstr_wn(const WCHAR *szStr, int len)
 121 {
 122     static char buf[1024];
 123     WideCharToMultiByte(CP_ACP, 0, szStr, len, buf, sizeof(buf), NULL, NULL);
 124     return buf;
 125 }
 126
 127 static void test_saxstr(unsigned line, const WCHAR *szStr, int nStr, const char *szTest)
 128 {
 129     WCHAR buf[1024];
 130     int len;
 131
 132     if(!szTest) {
 133         ok_(__FILE__,line) (szStr == NULL, "szStr != NULL\n");
 134         ok_(__FILE__,line) (nStr == 0, "nStr = %d, expected 0\n", nStr);
 135         return;
 136     }
 137
 138     len = strlen(szTest);
 139     ok_(__FILE__,line) (len == nStr, "nStr = %d, expected %d (%s)\n", nStr, len, szTest);
 140     if(len != nStr)
 141         return;
 142
 143     MultiByteToWideChar(CP_ACP, 0, szTest, -1, buf, sizeof(buf));
 144     ok_(__FILE__,line) (!memcmp(szStr, buf, len*sizeof(WCHAR)), "unexpected szStr %s, expected %s\n",
 145                         debugstr_wn(szStr, nStr), szTest);
 146 }
 147
 148 static BOOL test_expect_call(CH id)
 149 {
 150     ok(expectCall->id == id, "unexpected call %d, expected %d\n", id, expectCall->id);
 151     return expectCall->id == id;
 152 }
 153
 154 static void test_locator(unsigned line, int loc_line, int loc_column)
 155 {
 156     int rcolumn, rline;
 157     ISAXLocator_getLineNumber(locator, &rline);
 158     ISAXLocator_getColumnNumber(locator, &rcolumn);
 159
 160     ok_(__FILE__,line) (rline == loc_line,
 161             "unexpected line %d, expected %d\n", rline, loc_line);
 162     ok_(__FILE__,line) (rcolumn == loc_column,
 163             "unexpected column %d, expected %d\n", rcolumn, loc_column);
 164 }
 165
 166 static HRESULT WINAPI contentHandler_QueryInterface(
 167         ISAXContentHandler* iface,
 168         REFIID riid,
 169         void **ppvObject)
 170 {
 171     *ppvObject = NULL;
 172
 173     if(IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_ISAXContentHandler))
 174     {
 175         *ppvObject = iface;
 176     }
 177     else
 178     {
 179         return E_NOINTERFACE;
 180     }
 181
 182     return S_OK;
 183 }
 184
 185 static ULONG WINAPI contentHandler_AddRef(
 186         ISAXContentHandler* iface)
 187 {
 188     return 2;
 189 }
 190
 191 static ULONG WINAPI contentHandler_Release(
 192         ISAXContentHandler* iface)
 193 {
 194     return 1;
 195 }
 196
 197 static HRESULT WINAPI contentHandler_putDocumentLocator(
 198         ISAXContentHandler* iface,
 199         ISAXLocator *pLocator)
 200 {
 201     if(!test_expect_call(CH_PUTDOCUMENTLOCATOR))
 202         return E_FAIL;
 203
 204     locator = pLocator;
 205     test_locator(__LINE__, expectCall->line, expectCall->column);
 206
 207     expectCall++;
 208     return S_OK;
 209 }
 210
 211 static HRESULT WINAPI contentHandler_startDocument(
 212         ISAXContentHandler* iface)
 213 {
 214     if(!test_expect_call(CH_STARTDOCUMENT))
 215         return E_FAIL;
 216
 217     test_locator(__LINE__, expectCall->line, expectCall->column);
 218
 219     expectCall++;
 220     return S_OK;
 221 }
 222
 223 static HRESULT WINAPI contentHandler_endDocument(
 224         ISAXContentHandler* iface)
 225 {
 226     if(!test_expect_call(CH_ENDDOCUMENT))
 227         return E_FAIL;
 228
 229     test_locator(__LINE__, expectCall->line, expectCall->column);
 230
 231     expectCall++;
 232     return S_OK;
 233 }
 234
 235 static HRESULT WINAPI contentHandler_startPrefixMapping(
 236         ISAXContentHandler* iface,
 237         const WCHAR *pPrefix,
 238         int nPrefix,
 239         const WCHAR *pUri,
 240         int nUri)
 241 {
 242     if(!test_expect_call(CH_ENDDOCUMENT))
 243         return E_FAIL;
 244
 245     test_saxstr(__LINE__, pPrefix, nPrefix, expectCall->arg1);
 246     test_saxstr(__LINE__, pUri, nUri, expectCall->arg2);
 247     test_locator(__LINE__, expectCall->line, expectCall->column);
 248
 249     expectCall++;
 250     return S_OK;
 251 }
 252
 253 static HRESULT WINAPI contentHandler_endPrefixMapping(
 254         ISAXContentHandler* iface,
 255         const WCHAR *pPrefix,
 256         int nPrefix)
 257 {
 258     if(!test_expect_call(CH_ENDPREFIXMAPPING))
 259         return E_FAIL;
 260
 261     test_saxstr(__LINE__, pPrefix, nPrefix, expectCall->arg1);
 262     test_locator(__LINE__, expectCall->line, expectCall->column);
 263
 264     expectCall++;
 265     return S_OK;
 266 }
 267
 268 static HRESULT WINAPI contentHandler_startElement(
 269         ISAXContentHandler* iface,
 270         const WCHAR *pNamespaceUri,
 271         int nNamespaceUri,
 272         const WCHAR *pLocalName,
 273         int nLocalName,
 274         const WCHAR *pQName,
 275         int nQName,
 276         ISAXAttributes *pAttr)
 277 {
 278     if(!test_expect_call(CH_STARTELEMENT))
 279         return E_FAIL;
 280
 281     test_saxstr(__LINE__, pNamespaceUri, nNamespaceUri, expectCall->arg1);
 282     test_saxstr(__LINE__, pLocalName, nLocalName, expectCall->arg2);
 283     test_saxstr(__LINE__, pQName, nQName, expectCall->arg3);
 284     test_locator(__LINE__, expectCall->line, expectCall->column);
 285
 286     expectCall++;
 287     return S_OK;
 288 }
 289
 290 static HRESULT WINAPI contentHandler_endElement(
 291         ISAXContentHandler* iface,
 292         const WCHAR *pNamespaceUri,
 293         int nNamespaceUri,
 294         const WCHAR *pLocalName,
 295         int nLocalName,
 296         const WCHAR *pQName,
 297         int nQName)
 298 {
 299     if(!test_expect_call(CH_ENDELEMENT))
 300         return E_FAIL;
 301
 302     test_saxstr(__LINE__, pNamespaceUri, nNamespaceUri, expectCall->arg1);
 303     test_saxstr(__LINE__, pLocalName, nLocalName, expectCall->arg2);
 304     test_saxstr(__LINE__, pQName, nQName, expectCall->arg3);
 305     test_locator(__LINE__, expectCall->line, expectCall->column);
 306
 307     expectCall++;
 308     return S_OK;
 309 }
 310
 311 static HRESULT WINAPI contentHandler_characters(
 312         ISAXContentHandler* iface,
 313         const WCHAR *pChars,
 314         int nChars)
 315 {
 316     if(!test_expect_call(CH_CHARACTERS))
 317         return E_FAIL;
 318
 319     test_saxstr(__LINE__, pChars, nChars, expectCall->arg1);
 320     test_locator(__LINE__, expectCall->line, expectCall->column);
 321
 322     expectCall++;
 323     return S_OK;
 324 }
 325
 326 static HRESULT WINAPI contentHandler_ignorableWhitespace(
 327         ISAXContentHandler* iface,
 328         const WCHAR *pChars,
 329         int nChars)
 330 {
 331     if(!test_expect_call(CH_IGNORABLEWHITESPACE))
 332         return E_FAIL;
 333
 334     test_saxstr(__LINE__, pChars, nChars, expectCall->arg1);
 335     test_locator(__LINE__, expectCall->line, expectCall->column);
 336
 337     expectCall++;
 338     return S_OK;
 339 }
 340
 341 static HRESULT WINAPI contentHandler_processingInstruction(
 342         ISAXContentHandler* iface,
 343         const WCHAR *pTarget,
 344         int nTarget,
 345         const WCHAR *pData,
 346         int nData)
 347 {
 348     if(!test_expect_call(CH_PROCESSINGINSTRUCTION))
 349         return E_FAIL;
 350
 351     test_saxstr(__LINE__, pTarget, nTarget, expectCall->arg1);
 352     test_saxstr(__LINE__, pData, nData, expectCall->arg2);
 353     test_locator(__LINE__, expectCall->line, expectCall->column);
 354
 355     expectCall++;
 356     return S_OK;
 357 }
 358
 359 static HRESULT WINAPI contentHandler_skippedEntity(
 360         ISAXContentHandler* iface,
 361         const WCHAR *pName,
 362         int nName)
 363 {
 364     if(!test_expect_call(CH_SKIPPEDENTITY))
 365         return E_FAIL;
 366
 367     test_saxstr(__LINE__, pName, nName, expectCall->arg1);
 368     test_locator(__LINE__, expectCall->line, expectCall->column);
 369
 370     expectCall++;
 371     return S_OK;
 372 }
 373
 374
 375 static const ISAXContentHandlerVtbl contentHandlerVtbl =
 376 {
 377     contentHandler_QueryInterface,
 378     contentHandler_AddRef,
 379     contentHandler_Release,
 380     contentHandler_putDocumentLocator,
 381     contentHandler_startDocument,
 382     contentHandler_endDocument,
 383     contentHandler_startPrefixMapping,
 384     contentHandler_endPrefixMapping,
 385     contentHandler_startElement,
 386     contentHandler_endElement,
 387     contentHandler_characters,
 388     contentHandler_ignorableWhitespace,
 389     contentHandler_processingInstruction,
 390     contentHandler_skippedEntity
 391 };
 392
 393 static ISAXContentHandler contentHandler = { &contentHandlerVtbl };
 394
 395 static HRESULT WINAPI isaxerrorHandler_QueryInterface(
 396         ISAXErrorHandler* iface,
 397         REFIID riid,
 398         void **ppvObject)
 399 {
 400     *ppvObject = NULL;
 401
 402     if(IsEqualGUID(riid, &IID_IUnknown) || IsEqualGUID(riid, &IID_ISAXErrorHandler))
 403     {
 404         *ppvObject = iface;
 405     }
 406     else
 407     {
 408         return E_NOINTERFACE;
 409     }
 410
 411     return S_OK;
 412 }
 413
 414 static ULONG WINAPI isaxerrorHandler_AddRef(
 415         ISAXErrorHandler* iface)
 416 {
 417     return 2;
 418 }
 419
 420 static ULONG WINAPI isaxerrorHandler_Release(
 421         ISAXErrorHandler* iface)
 422 {
 423     return 1;
 424 }
 425
 426 static HRESULT WINAPI isaxerrorHandler_error(
 427         ISAXErrorHandler* iface,
 428         ISAXLocator *pLocator,
 429         const WCHAR *pErrorMessage,
 430         HRESULT hrErrorCode)
 431 {
 432     return S_OK;
 433 }
 434
 435 static HRESULT WINAPI isaxerrorHandler_fatalError(
 436         ISAXErrorHandler* iface,
 437         ISAXLocator *pLocator,
 438         const WCHAR *pErrorMessage,
 439         HRESULT hrErrorCode)
 440 {
 441     return S_OK;
 442 }
 443
 444 static HRESULT WINAPI isaxerrorHanddler_ignorableWarning(
 445         ISAXErrorHandler* iface,
 446         ISAXLocator *pLocator,
 447         const WCHAR *pErrorMessage,
 448         HRESULT hrErrorCode)
 449 {
 450     return S_OK;
 451 }
 452
 453 static const ISAXErrorHandlerVtbl errorHandlerVtbl =
 454 {
 455     isaxerrorHandler_QueryInterface,
 456     isaxerrorHandler_AddRef,
 457     isaxerrorHandler_Release,
 458     isaxerrorHandler_error,
 459     isaxerrorHandler_fatalError,
 460     isaxerrorHanddler_ignorableWarning
 461 };
 462
 463 static ISAXErrorHandler errorHandler = { &errorHandlerVtbl };
 464
 465 static void test_saxreader(void)
 466 {
 467     HRESULT hr;
 468     ISAXXMLReader *reader = NULL;
 469     VARIANT var;
 470     ISAXContentHandler *lpContentHandler;
 471     ISAXErrorHandler *lpErrorHandler;
 472     SAFEARRAY *pSA;
 473     SAFEARRAYBOUND SADim[1];
 474     char *pSAData = NULL;
 475     IStream *iStream;
 476     ULARGE_INTEGER liSize;
 477     LARGE_INTEGER liPos;
 478     ULONG bytesWritten;
 479     HANDLE file;
 480
 481     static const WCHAR testXmlW[] = {'t','e','s','t','.','x','m','l',0};
 482
 483     hr = CoCreateInstance(&CLSID_SAXXMLReader, NULL, CLSCTX_INPROC_SERVER,
 484             &IID_ISAXXMLReader, (LPVOID*)&reader);
 485
 486     if(FAILED(hr))
 487     {
 488         skip("Failed to create SAXXMLReader instance\n");
 489         return;
 490     }
 491
 492     hr = ISAXXMLReader_getContentHandler(reader, NULL);
 493     ok(hr == E_POINTER, "Expected E_POINTER, got %08x\n", hr);
 494
 495     hr = ISAXXMLReader_getErrorHandler(reader, NULL);
 496     ok(hr == E_POINTER, "Expected E_POINTER, got %08x\n", hr);
 497
 498     hr = ISAXXMLReader_getContentHandler(reader, &lpContentHandler);
 499     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 500     ok(lpContentHandler == NULL, "Expected %p, got %p\n", NULL, lpContentHandler);
 501
 502     hr = ISAXXMLReader_getErrorHandler(reader, &lpErrorHandler);
 503     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 504     ok(lpErrorHandler == NULL, "Expected %p, got %p\n", NULL, lpErrorHandler);
 505
 506     hr = ISAXXMLReader_putContentHandler(reader, NULL);
 507     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 508
 509     hr = ISAXXMLReader_putContentHandler(reader, &contentHandler);
 510     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 511
 512     hr = ISAXXMLReader_putErrorHandler(reader, &errorHandler);
 513     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 514
 515     hr = ISAXXMLReader_getContentHandler(reader, &lpContentHandler);
 516     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 517     ok(lpContentHandler == &contentHandler, "Expected %p, got %p\n", &contentHandler, lpContentHandler);
 518
 519     V_VT(&var) = VT_BSTR;
 520     V_BSTR(&var) = SysAllocString(szSimpleXML);
 521
 522     expectCall = contentHandlerTest1;
 523     hr = ISAXXMLReader_parse(reader, var);
 524     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 525     test_expect_call(CH_ENDTEST);
 526
 527     SADim[0].lLbound= 0;
 528     SADim[0].cElements= sizeof(szTestXML)-1;
 529     pSA = SafeArrayCreate(VT_UI1, 1, SADim);
 530     SafeArrayAccessData(pSA, (void**)&pSAData);
 531     memcpy(pSAData, szTestXML, sizeof(szTestXML)-1);
 532     SafeArrayUnaccessData(pSA);
 533     V_VT(&var) = VT_ARRAY|VT_UI1;
 534     V_ARRAY(&var) = pSA;
 535
 536     expectCall = contentHandlerTest1;
 537     hr = ISAXXMLReader_parse(reader, var);
 538     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 539     test_expect_call(CH_ENDTEST);
 540
 541     SafeArrayDestroy(pSA);
 542
 543     CreateStreamOnHGlobal(NULL, TRUE, &iStream);
 544     liSize.QuadPart = strlen(szTestXML);
 545     IStream_SetSize(iStream, liSize);
 546     IStream_Write(iStream, (void const*)szTestXML, strlen(szTestXML), &bytesWritten);
 547     liPos.QuadPart = 0;
 548     IStream_Seek(iStream, liPos, STREAM_SEEK_SET, NULL);
 549     V_VT(&var) = VT_UNKNOWN|VT_DISPATCH;
 550     V_UNKNOWN(&var) = (IUnknown*)iStream;
 551
 552     expectCall = contentHandlerTest1;
 553     hr = ISAXXMLReader_parse(reader, var);
 554     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 555     test_expect_call(CH_ENDTEST);
 556
 557     IStream_Release(iStream);
 558
 559     V_VT(&var) = VT_BSTR;
 560     V_BSTR(&var) = SysAllocString(szCarriageRetTest);
 561
 562     expectCall = contentHandlerTest2;
 563     hr = ISAXXMLReader_parse(reader, var);
 564     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 565     test_expect_call(CH_ENDTEST);
 566
 567     file = CreateFileW(testXmlW, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
 568     ok(file != INVALID_HANDLE_VALUE, "Could not create file: %u\n", GetLastError());
 569     WriteFile(file, szTestXML, sizeof(szTestXML)-1, &bytesWritten, NULL);
 570     CloseHandle(file);
 571
 572     expectCall = contentHandlerTest1;
 573     hr = ISAXXMLReader_parseURL(reader, testXmlW);
 574     ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
 575     test_expect_call(CH_ENDTEST);
 576
 577     DeleteFileW(testXmlW);
 578
 579     ISAXXMLReader_Release(reader);
 580 }
 581
 582 START_TEST(saxreader)
 583 {
 584     HRESULT hr;
 585
 586     hr = CoInitialize(NULL);
 587     ok(hr == S_OK, "failed to init com\n");
 588
 589     test_saxreader();
 590
 591     CoUninitialize();
 592 }