773 lines
23 KiB
C
773 lines
23 KiB
C
/*
|
|
* XMLLite IXmlWriter tests
|
|
*
|
|
* Copyright 2011 (C) Alistair Leslie-Hughes
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
#define COBJMACROS
|
|
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "ole2.h"
|
|
#include "xmllite.h"
|
|
#include "wine/test.h"
|
|
|
|
#include "initguid.h"
|
|
DEFINE_GUID(IID_IXmlWriterOutput, 0xc1131708, 0x0f59, 0x477f, 0x93, 0x59, 0x7d, 0x33, 0x24, 0x51, 0xbc, 0x1a);
|
|
|
|
static HRESULT (WINAPI *pCreateXmlWriter)(REFIID riid, void **ppvObject, IMalloc *pMalloc);
|
|
static HRESULT (WINAPI *pCreateXmlWriterOutputWithEncodingName)(IUnknown *stream,
|
|
IMalloc *imalloc,
|
|
LPCWSTR encoding_name,
|
|
IXmlWriterOutput **output);
|
|
static HRESULT (WINAPI *pCreateXmlWriterOutputWithEncodingCodePage)(IUnknown *stream,
|
|
IMalloc *imalloc,
|
|
UINT codepage,
|
|
IXmlWriterOutput **output);
|
|
|
|
static HRESULT WINAPI testoutput_QueryInterface(IUnknown *iface, REFIID riid, void **obj)
|
|
{
|
|
if (IsEqualGUID(riid, &IID_IUnknown)) {
|
|
*obj = iface;
|
|
return S_OK;
|
|
}
|
|
else {
|
|
ok(0, "unknown riid=%s\n", wine_dbgstr_guid(riid));
|
|
return E_NOINTERFACE;
|
|
}
|
|
}
|
|
|
|
static ULONG WINAPI testoutput_AddRef(IUnknown *iface)
|
|
{
|
|
return 2;
|
|
}
|
|
|
|
static ULONG WINAPI testoutput_Release(IUnknown *iface)
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
static const IUnknownVtbl testoutputvtbl = {
|
|
testoutput_QueryInterface,
|
|
testoutput_AddRef,
|
|
testoutput_Release
|
|
};
|
|
|
|
static IUnknown testoutput = { &testoutputvtbl };
|
|
|
|
static HRESULT WINAPI teststream_QueryInterface(ISequentialStream *iface, REFIID riid, void **obj)
|
|
{
|
|
if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_ISequentialStream))
|
|
{
|
|
*obj = iface;
|
|
return S_OK;
|
|
}
|
|
|
|
*obj = NULL;
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
static ULONG WINAPI teststream_AddRef(ISequentialStream *iface)
|
|
{
|
|
return 2;
|
|
}
|
|
|
|
static ULONG WINAPI teststream_Release(ISequentialStream *iface)
|
|
{
|
|
return 1;
|
|
}
|
|
|
|
static HRESULT WINAPI teststream_Read(ISequentialStream *iface, void *pv, ULONG cb, ULONG *pread)
|
|
{
|
|
ok(0, "unexpected call\n");
|
|
return E_NOTIMPL;
|
|
}
|
|
|
|
static ULONG g_write_len;
|
|
static HRESULT WINAPI teststream_Write(ISequentialStream *iface, const void *pv, ULONG cb, ULONG *written)
|
|
{
|
|
g_write_len = cb;
|
|
*written = cb;
|
|
return S_OK;
|
|
}
|
|
|
|
static const ISequentialStreamVtbl teststreamvtbl =
|
|
{
|
|
teststream_QueryInterface,
|
|
teststream_AddRef,
|
|
teststream_Release,
|
|
teststream_Read,
|
|
teststream_Write
|
|
};
|
|
|
|
static ISequentialStream teststream = { &teststreamvtbl };
|
|
|
|
static void test_writer_create(void)
|
|
{
|
|
HRESULT hr;
|
|
IXmlWriter *writer;
|
|
LONG_PTR value;
|
|
|
|
/* crashes native */
|
|
if (0)
|
|
{
|
|
pCreateXmlWriter(&IID_IXmlWriter, NULL, NULL);
|
|
pCreateXmlWriter(NULL, (void**)&writer, NULL);
|
|
}
|
|
|
|
hr = pCreateXmlWriter(&IID_IXmlWriter, (void**)&writer, NULL);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
|
|
/* check default properties values */
|
|
value = 0;
|
|
hr = IXmlWriter_GetProperty(writer, XmlWriterProperty_ByteOrderMark, &value);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
ok(value == TRUE, "got %ld\n", value);
|
|
|
|
value = TRUE;
|
|
hr = IXmlWriter_GetProperty(writer, XmlWriterProperty_Indent, &value);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
ok(value == FALSE, "got %ld\n", value);
|
|
|
|
value = TRUE;
|
|
hr = IXmlWriter_GetProperty(writer, XmlWriterProperty_OmitXmlDeclaration, &value);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
ok(value == FALSE, "got %ld\n", value);
|
|
|
|
value = XmlConformanceLevel_Auto;
|
|
hr = IXmlWriter_GetProperty(writer, XmlWriterProperty_ConformanceLevel, &value);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
ok(value == XmlConformanceLevel_Document, "got %ld\n", value);
|
|
|
|
IXmlWriter_Release(writer);
|
|
}
|
|
|
|
static BOOL init_pointers(void)
|
|
{
|
|
/* don't free module here, it's to be unloaded on exit */
|
|
HMODULE mod = LoadLibraryA("xmllite.dll");
|
|
|
|
if (!mod)
|
|
{
|
|
win_skip("xmllite library not available\n");
|
|
return FALSE;
|
|
}
|
|
|
|
#define MAKEFUNC(f) if (!(p##f = (void*)GetProcAddress(mod, #f))) return FALSE;
|
|
MAKEFUNC(CreateXmlWriter);
|
|
MAKEFUNC(CreateXmlWriterOutputWithEncodingName);
|
|
MAKEFUNC(CreateXmlWriterOutputWithEncodingCodePage);
|
|
#undef MAKEFUNC
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static void test_writeroutput(void)
|
|
{
|
|
static const WCHAR utf16W[] = {'u','t','f','-','1','6',0};
|
|
IXmlWriterOutput *output;
|
|
IUnknown *unk;
|
|
HRESULT hr;
|
|
|
|
output = NULL;
|
|
hr = pCreateXmlWriterOutputWithEncodingName(&testoutput, NULL, NULL, &output);
|
|
ok(hr == S_OK, "got %08x\n", hr);
|
|
IUnknown_Release(output);
|
|
|
|
hr = pCreateXmlWriterOutputWithEncodingName(&testoutput, NULL, utf16W, &output);
|
|
ok(hr == S_OK, "got %08x\n", hr);
|
|
unk = NULL;
|
|
hr = IUnknown_QueryInterface(output, &IID_IXmlWriterOutput, (void**)&unk);
|
|
ok(hr == S_OK, "got %08x\n", hr);
|
|
ok(unk != NULL, "got %p\n", unk);
|
|
/* releasing 'unk' crashes on native */
|
|
IUnknown_Release(output);
|
|
|
|
output = NULL;
|
|
hr = pCreateXmlWriterOutputWithEncodingCodePage(&testoutput, NULL, ~0u, &output);
|
|
ok(hr == S_OK, "got %08x\n", hr);
|
|
IUnknown_Release(output);
|
|
|
|
hr = pCreateXmlWriterOutputWithEncodingCodePage(&testoutput, NULL, CP_UTF8, &output);
|
|
ok(hr == S_OK, "got %08x\n", hr);
|
|
unk = NULL;
|
|
hr = IUnknown_QueryInterface(output, &IID_IXmlWriterOutput, (void**)&unk);
|
|
ok(hr == S_OK, "got %08x\n", hr);
|
|
ok(unk != NULL, "got %p\n", unk);
|
|
/* releasing 'unk' crashes on native */
|
|
IUnknown_Release(output);
|
|
}
|
|
|
|
static void test_writestartdocument(void)
|
|
{
|
|
static const char fullprolog[] = "<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>";
|
|
static const char prologversion[] = "<?xml version=\"1.0\"?>";
|
|
static const WCHAR versionW[] = {'v','e','r','s','i','o','n','=','"','1','.','0','"',0};
|
|
static const WCHAR xmlW[] = {'x','m','l',0};
|
|
IXmlWriter *writer;
|
|
HGLOBAL hglobal;
|
|
IStream *stream;
|
|
HRESULT hr;
|
|
char *ptr;
|
|
|
|
hr = pCreateXmlWriter(&IID_IXmlWriter, (void**)&writer, NULL);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
|
|
/* output not set */
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == E_UNEXPECTED, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteProcessingInstruction(writer, xmlW, versionW);
|
|
ok(hr == E_UNEXPECTED, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, (IUnknown*)stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
/* nothing written yet */
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = GetHGlobalFromStream(stream, &hglobal);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(!strncmp(ptr, fullprolog, strlen(fullprolog)), "got %s, expected %s\n", ptr, fullprolog);
|
|
GlobalUnlock(hglobal);
|
|
|
|
/* one more time */
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
IStream_Release(stream);
|
|
|
|
/* now add PI manually, and try to start a document */
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, (IUnknown*)stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteProcessingInstruction(writer, xmlW, versionW);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
|
|
/* another attempt to add 'xml' PI */
|
|
hr = IXmlWriter_WriteProcessingInstruction(writer, xmlW, versionW);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = GetHGlobalFromStream(stream, &hglobal);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(!strncmp(ptr, prologversion, strlen(prologversion)), "got %s\n", ptr);
|
|
GlobalUnlock(hglobal);
|
|
|
|
IStream_Release(stream);
|
|
IXmlWriter_Release(writer);
|
|
}
|
|
|
|
static void test_flush(void)
|
|
{
|
|
IXmlWriter *writer;
|
|
HRESULT hr;
|
|
|
|
hr = pCreateXmlWriter(&IID_IXmlWriter, (void**)&writer, NULL);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, (IUnknown*)&teststream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
g_write_len = 0;
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
ok(g_write_len > 0, "got %d\n", g_write_len);
|
|
|
|
g_write_len = 1;
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
ok(g_write_len == 0, "got %d\n", g_write_len);
|
|
|
|
/* Release() flushes too */
|
|
g_write_len = 1;
|
|
IXmlWriter_Release(writer);
|
|
ok(g_write_len == 0, "got %d\n", g_write_len);
|
|
}
|
|
|
|
static void test_omitxmldeclaration(void)
|
|
{
|
|
static const char prologversion[] = "<?xml version=\"1.0\"?>";
|
|
static const WCHAR versionW[] = {'v','e','r','s','i','o','n','=','"','1','.','0','"',0};
|
|
static const WCHAR xmlW[] = {'x','m','l',0};
|
|
IXmlWriter *writer;
|
|
HGLOBAL hglobal;
|
|
IStream *stream;
|
|
HRESULT hr;
|
|
char *ptr;
|
|
|
|
hr = pCreateXmlWriter(&IID_IXmlWriter, (void**)&writer, NULL);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, (IUnknown*)stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetProperty(writer, XmlWriterProperty_OmitXmlDeclaration, TRUE);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = GetHGlobalFromStream(stream, &hglobal);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(!ptr, "got %p\n", ptr);
|
|
GlobalUnlock(hglobal);
|
|
|
|
/* one more time */
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
|
|
IStream_Release(stream);
|
|
|
|
/* now add PI manually, and try to start a document */
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, (IUnknown*)stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteProcessingInstruction(writer, xmlW, versionW);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = GetHGlobalFromStream(stream, &hglobal);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(!strncmp(ptr, prologversion, strlen(prologversion)), "got %s\n", ptr);
|
|
GlobalUnlock(hglobal);
|
|
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(!strncmp(ptr, prologversion, strlen(prologversion)), "got %s\n", ptr);
|
|
GlobalUnlock(hglobal);
|
|
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(!strncmp(ptr, prologversion, strlen(prologversion)), "got %s\n", ptr);
|
|
GlobalUnlock(hglobal);
|
|
|
|
/* another attempt to add 'xml' PI */
|
|
hr = IXmlWriter_WriteProcessingInstruction(writer, xmlW, versionW);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
IStream_Release(stream);
|
|
IXmlWriter_Release(writer);
|
|
}
|
|
|
|
static void test_bom(void)
|
|
{
|
|
static const WCHAR versionW[] = {'v','e','r','s','i','o','n','=','"','1','.','0','"',0};
|
|
static const WCHAR utf16W[] = {'u','t','f','-','1','6',0};
|
|
static const WCHAR xmlW[] = {'x','m','l',0};
|
|
static const WCHAR aW[] = {'a',0};
|
|
IXmlWriterOutput *output;
|
|
unsigned char *ptr;
|
|
IXmlWriter *writer;
|
|
IStream *stream;
|
|
HGLOBAL hglobal;
|
|
HRESULT hr;
|
|
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = pCreateXmlWriterOutputWithEncodingName((IUnknown*)stream, NULL, utf16W, &output);
|
|
ok(hr == S_OK, "got %08x\n", hr);
|
|
|
|
hr = pCreateXmlWriter(&IID_IXmlWriter, (void**)&writer, NULL);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetProperty(writer, XmlWriterProperty_OmitXmlDeclaration, TRUE);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, output);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
/* BOM is on by default */
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = GetHGlobalFromStream(stream, &hglobal);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(ptr[0] == 0xff && ptr[1] == 0xfe, "got %x,%x\n", ptr[0], ptr[1]);
|
|
GlobalUnlock(hglobal);
|
|
|
|
IStream_Release(stream);
|
|
IUnknown_Release(output);
|
|
|
|
/* start with PI */
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = pCreateXmlWriterOutputWithEncodingName((IUnknown*)stream, NULL, utf16W, &output);
|
|
ok(hr == S_OK, "got %08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, output);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteProcessingInstruction(writer, xmlW, versionW);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = GetHGlobalFromStream(stream, &hglobal);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(ptr[0] == 0xff && ptr[1] == 0xfe, "got %x,%x\n", ptr[0], ptr[1]);
|
|
GlobalUnlock(hglobal);
|
|
|
|
IUnknown_Release(output);
|
|
IStream_Release(stream);
|
|
|
|
/* start with element */
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = pCreateXmlWriterOutputWithEncodingName((IUnknown*)stream, NULL, utf16W, &output);
|
|
ok(hr == S_OK, "got %08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, output);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, NULL, aW, NULL);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = GetHGlobalFromStream(stream, &hglobal);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(ptr[0] == 0xff && ptr[1] == 0xfe, "got %x,%x\n", ptr[0], ptr[1]);
|
|
GlobalUnlock(hglobal);
|
|
|
|
IUnknown_Release(output);
|
|
IStream_Release(stream);
|
|
|
|
/* WriteElementString */
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = pCreateXmlWriterOutputWithEncodingName((IUnknown*)stream, NULL, utf16W, &output);
|
|
ok(hr == S_OK, "got %08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, output);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteElementString(writer, NULL, aW, NULL, NULL);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = GetHGlobalFromStream(stream, &hglobal);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(ptr[0] == 0xff && ptr[1] == 0xfe, "got %x,%x\n", ptr[0], ptr[1]);
|
|
GlobalUnlock(hglobal);
|
|
|
|
IUnknown_Release(output);
|
|
IStream_Release(stream);
|
|
|
|
IXmlWriter_Release(writer);
|
|
}
|
|
|
|
static void test_writestartelement(void)
|
|
{
|
|
static const WCHAR valueW[] = {'v','a','l','u','e',0};
|
|
static const char *str = "<a><b>value</b>";
|
|
static const WCHAR aW[] = {'a',0};
|
|
static const WCHAR bW[] = {'b',0};
|
|
char *ptr;
|
|
IXmlWriter *writer;
|
|
IStream *stream;
|
|
HGLOBAL hglobal;
|
|
HRESULT hr;
|
|
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = pCreateXmlWriter(&IID_IXmlWriter, (void**)&writer, NULL);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, NULL, aW, NULL);
|
|
ok(hr == E_UNEXPECTED, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, (IUnknown*)stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, aW, NULL, NULL);
|
|
ok(hr == E_INVALIDARG, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, NULL, NULL, NULL);
|
|
ok(hr == E_INVALIDARG, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, NULL, NULL, aW);
|
|
ok(hr == E_INVALIDARG, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, NULL, aW, NULL);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = GetHGlobalFromStream(stream, &hglobal);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(!strncmp(ptr, "<a", 2), "got %s\n", ptr);
|
|
GlobalUnlock(hglobal);
|
|
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Yes);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, NULL, NULL, NULL);
|
|
ok(hr == E_INVALIDARG, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteProcessingInstruction(writer, aW, aW);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
|
|
IStream_Release(stream);
|
|
IXmlWriter_Release(writer);
|
|
|
|
/* WriteElementString */
|
|
hr = pCreateXmlWriter(&IID_IXmlWriter, (void**)&writer, NULL);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteElementString(writer, NULL, bW, NULL, valueW);
|
|
ok(hr == E_UNEXPECTED, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, (IUnknown*)stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, NULL, aW, NULL);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteElementString(writer, NULL, bW, NULL, valueW);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = GetHGlobalFromStream(stream, &hglobal);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(!strncmp(ptr, str, strlen(str)), "got %s\n", ptr);
|
|
GlobalUnlock(hglobal);
|
|
|
|
IStream_Release(stream);
|
|
IXmlWriter_Release(writer);
|
|
}
|
|
|
|
static void test_writeendelement(void)
|
|
{
|
|
static const WCHAR aW[] = {'a',0};
|
|
static const WCHAR bW[] = {'b',0};
|
|
char *ptr;
|
|
IXmlWriter *writer;
|
|
IStream *stream;
|
|
HGLOBAL hglobal;
|
|
HRESULT hr;
|
|
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = pCreateXmlWriter(&IID_IXmlWriter, (void**)&writer, NULL);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, (IUnknown*)stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, NULL, aW, NULL);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, NULL, bW, NULL);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteEndElement(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteEndElement(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = GetHGlobalFromStream(stream, &hglobal);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(!strncmp(ptr, "<a><b /></a>", 12), "got %s\n", ptr);
|
|
GlobalUnlock(hglobal);
|
|
|
|
IXmlWriter_Release(writer);
|
|
IStream_Release(stream);
|
|
}
|
|
|
|
static void test_writeenddocument(void)
|
|
{
|
|
static const WCHAR aW[] = {'a',0};
|
|
static const WCHAR bW[] = {'b',0};
|
|
IXmlWriter *writer;
|
|
IStream *stream;
|
|
HGLOBAL hglobal;
|
|
HRESULT hr;
|
|
char *ptr;
|
|
|
|
hr = CreateStreamOnHGlobal(NULL, TRUE, &stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = pCreateXmlWriter(&IID_IXmlWriter, (void**)&writer, NULL);
|
|
ok(hr == S_OK, "Expected S_OK, got %08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteEndDocument(writer);
|
|
ok(hr == E_UNEXPECTED, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, (IUnknown*)stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
/* WriteEndDocument resets it to initial state */
|
|
hr = IXmlWriter_WriteEndDocument(writer);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteEndDocument(writer);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartDocument(writer, XmlStandalone_Omit);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, NULL, aW, NULL);
|
|
ok(hr == WR_E_INVALIDACTION, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_SetOutput(writer, (IUnknown*)stream);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, NULL, aW, NULL);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteStartElement(writer, NULL, bW, NULL);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = IXmlWriter_WriteEndDocument(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
hr = GetHGlobalFromStream(stream, &hglobal);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(ptr == NULL, "got %p\n", ptr);
|
|
|
|
/* we still need to flush manually, WriteEndDocument doesn't do that */
|
|
hr = IXmlWriter_Flush(writer);
|
|
ok(hr == S_OK, "got 0x%08x\n", hr);
|
|
|
|
ptr = GlobalLock(hglobal);
|
|
ok(ptr != NULL, "got %p\n", ptr);
|
|
ok(!strncmp(ptr, "<a><b /></a>", 12), "got %s\n", ptr);
|
|
GlobalUnlock(hglobal);
|
|
|
|
IXmlWriter_Release(writer);
|
|
IStream_Release(stream);
|
|
}
|
|
|
|
START_TEST(writer)
|
|
{
|
|
if (!init_pointers())
|
|
return;
|
|
|
|
test_writer_create();
|
|
test_writeroutput();
|
|
test_writestartdocument();
|
|
test_writestartelement();
|
|
test_writeendelement();
|
|
test_flush();
|
|
test_omitxmldeclaration();
|
|
test_bom();
|
|
test_writeenddocument();
|
|
}
|