From 029926dc75d2b7e38abc37ee6bb3f8bb3407ba28 Mon Sep 17 00:00:00 2001
From: Jefferson Carpenter <jeffersoncarpenter2@gmail.com>
Date: Wed, 20 Jan 2021 01:07:44 +0000
Subject: [PATCH] msxml3/tests: Add mxwriter DOMDocument test.

Signed-off-by: Jefferson Carpenter <jeffersoncarpenter2@gmail.com>
---
 dlls/msxml3/tests/saxreader.c | 122 ++++++++++++++++++++++++++++++++++
 1 file changed, 122 insertions(+)

diff --git a/dlls/msxml3/tests/saxreader.c b/dlls/msxml3/tests/saxreader.c
index ab814c5f764..adc16d5e9a8 100644
--- a/dlls/msxml3/tests/saxreader.c
+++ b/dlls/msxml3/tests/saxreader.c
@@ -4367,6 +4367,127 @@ static void test_mxwriter_stream(void)
     free_bstrs();
 }
 
+static void test_mxwriter_domdoc(void)
+{
+    IMXWriter *writer;
+    ISAXContentHandler *content;
+    IXMLDOMDocument3 *domdoc;
+    HRESULT hr;
+    VARIANT dest;
+
+    IXMLDOMNode *root = NULL;
+    IXMLDOMNodeList *node_list = NULL;
+    IXMLDOMNode *node = NULL;
+    LONG list_length = 0;
+    BSTR str;
+
+    // Create writer and attach DOMDocument output
+    hr = CoCreateInstance(&CLSID_MXXMLWriter60, NULL, CLSCTX_INPROC_SERVER,
+                          &IID_IMXWriter, (void**)&writer);
+    EXPECT_HR(hr, S_OK);
+
+    hr = IMXWriter_QueryInterface(writer, &IID_ISAXContentHandler, (void**)&content);
+    EXPECT_HR(hr, S_OK);
+
+    hr = CoCreateInstance(&CLSID_DOMDocument60, NULL, CLSCTX_INPROC_SERVER,
+                          &IID_IXMLDOMDocument3, (void**)&domdoc);
+    EXPECT_HR(hr, S_OK);
+
+    V_VT(&dest) = VT_DISPATCH;
+    V_DISPATCH(&dest) = (IDispatch*)domdoc;
+
+    hr = IMXWriter_put_output(writer, dest);
+    EXPECT_HR(hr, S_OK);
+
+
+    // Provide writer with events
+    hr = ISAXContentHandler_startDocument(content);
+    EXPECT_HR(hr, S_OK);
+
+    hr = ISAXContentHandler_startElement(content, L"", 0, L"", 0, L"BankAccount", 11, NULL);
+    EXPECT_HR(hr, S_OK);
+
+    hr = ISAXContentHandler_startElement(content, L"", 0, L"", 0, L"Number", 6, NULL);
+    EXPECT_HR(hr, S_OK);
+
+    hr = ISAXContentHandler_characters(content, L"12345", 5);
+    EXPECT_HR(hr, S_OK);
+
+    hr = ISAXContentHandler_endElement(content, L"", 0, L"", 0, L"Number", 6);
+    EXPECT_HR(hr, S_OK);
+
+    hr = ISAXContentHandler_startElement(content, L"", 0, L"", 0, L"Name", 4, NULL);
+    EXPECT_HR(hr, S_OK);
+
+    hr = ISAXContentHandler_characters(content, L"Captain Ahab", 12);
+    EXPECT_HR(hr, S_OK);
+
+    hr = ISAXContentHandler_endElement(content, L"", 0, L"", 0, L"Name", 4);
+    EXPECT_HR(hr, S_OK);
+
+    hr = ISAXContentHandler_endElement(content, L"", 0, L"", 0, L"BankAccount", 11);
+    EXPECT_HR(hr, S_OK);
+
+    hr = ISAXContentHandler_endDocument(content);
+    EXPECT_HR(hr, S_OK);
+
+
+    // Assert that it created document
+    hr = IXMLDOMDocument3_get_firstChild(domdoc, &root);
+    todo_wine EXPECT_HR(hr, S_OK);
+    if (!root) return;
+
+    hr = IXMLDOMNode_get_nodeName(root, &str);
+    todo_wine EXPECT_HR(hr, S_OK);
+    todo_wine ok(!lstrcmpW(L"BankAccount", str), "got %s\n", wine_dbgstr_w(str));
+    SysFreeString(str);
+
+    hr = IXMLDOMNode_get_childNodes(root, &node_list);
+    todo_wine EXPECT_HR(hr, S_OK);
+    if (!node_list) return;
+
+    hr = IXMLDOMNodeList_get_length(node_list, &list_length);
+    todo_wine EXPECT_HR(hr, S_OK);
+    todo_wine ok(2 == list_length, "list length %i, expected 1\n", list_length);
+
+    hr = IXMLDOMNodeList_get_item(node_list, 0, &node);
+    todo_wine EXPECT_HR(hr, S_OK);
+    if (!node) return;
+
+    hr = IXMLDOMNode_get_nodeName(node, &str);
+    todo_wine EXPECT_HR(hr, S_OK);
+    todo_wine ok(!lstrcmpW(L"Number", str), "got %s\n", wine_dbgstr_w(str));
+    SysFreeString(str);
+
+    hr = IXMLDOMNode_get_text(node, &str);
+    todo_wine EXPECT_HR(hr, S_OK);
+    todo_wine ok(!lstrcmpW(L"12345", str), "got %s\n", wine_dbgstr_w(str));
+    SysFreeString(str);
+    IXMLDOMNode_Release(node);
+
+    hr = IXMLDOMNodeList_get_item(node_list, 1, &node);
+    todo_wine EXPECT_HR(hr, S_OK);
+    if (!node) return;
+
+    hr = IXMLDOMNode_get_nodeName(node, &str);
+    todo_wine EXPECT_HR(hr, S_OK);
+    todo_wine ok(!lstrcmpW(L"Name", str), "got %s\n", wine_dbgstr_w(str));
+    SysFreeString(str);
+
+    hr = IXMLDOMNode_get_text(node, &str);
+    todo_wine EXPECT_HR(hr, S_OK);
+    todo_wine ok(!lstrcmpW(L"Captain Ahab", str), "got %s\n", wine_dbgstr_w(str));
+    SysFreeString(str);
+    IXMLDOMNode_Release(node);
+    IXMLDOMNodeList_Release(node_list);
+    IXMLDOMNode_Release(root);
+
+
+    IXMLDOMDocument3_Release(domdoc);
+    ISAXContentHandler_Release(content);
+    IMXWriter_Release(writer);
+}
+
 static const char *encoding_names[] = {
     "iso-8859-1",
     "iso-8859-2",
@@ -5768,6 +5889,7 @@ START_TEST(saxreader)
         test_mxwriter_properties();
         test_mxwriter_flush();
         test_mxwriter_stream();
+        test_mxwriter_domdoc();
         test_mxwriter_encoding();
         test_mxwriter_dispex();
         test_mxwriter_indent();
-- 
2.26.2