Handle variant element names, merge in wildcard class, add test cases.
[shibboleth/cpp-xmltooling.git] / xmltoolingtest / MarshallingTest.h
1 /*\r
2  *  Copyright 2001-2005 Internet2\r
3  * \r
4  * Licensed under the Apache License, Version 2.0 (the "License");\r
5  * you may not use this file except in compliance with the License.\r
6  * You may obtain a copy of the License at\r
7  *\r
8  *     http://www.apache.org/licenses/LICENSE-2.0\r
9  *\r
10  * Unless required by applicable law or agreed to in writing, software\r
11  * distributed under the License is distributed on an "AS IS" BASIS,\r
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
13  * See the License for the specific language governing permissions and\r
14  * limitations under the License.\r
15  */\r
16 \r
17 #include "XMLObjectBaseTestCase.h"\r
18 \r
19 #include <fstream>\r
20 \r
21 class MarshallingTest : public CxxTest::TestSuite {\r
22     QName m_qname;\r
23     QName m_qtype;\r
24 public:\r
25     MarshallingTest() : m_qname(SimpleXMLObject::NAMESPACE,SimpleXMLObject::LOCAL_NAME,SimpleXMLObject::NAMESPACE_PREFIX),\r
26         m_qtype(SimpleXMLObject::NAMESPACE,SimpleXMLObject::TYPE_NAME,SimpleXMLObject::NAMESPACE_PREFIX) {}\r
27 \r
28     void setUp() {\r
29         XMLObjectBuilder::registerBuilder(m_qname, new SimpleXMLObjectBuilder());\r
30         XMLObjectBuilder::registerBuilder(m_qtype, new SimpleXMLObjectBuilder());\r
31     }\r
32 \r
33     void tearDown() {\r
34         XMLObjectBuilder::deregisterBuilder(m_qname);\r
35         XMLObjectBuilder::deregisterBuilder(m_qtype);\r
36     }\r
37 \r
38     void testMarshallingWithAttributes() {\r
39         TS_TRACE("testMarshallingWithAttributes");\r
40 \r
41         auto_ptr_XMLCh expected("Firefly");\r
42         auto_ptr<SimpleXMLObject> sxObject(dynamic_cast<SimpleXMLObject*>(XMLObjectBuilder::buildOne(m_qname)));\r
43         TS_ASSERT(sxObject.get()!=NULL);\r
44         sxObject->setId(expected.get());\r
45         \r
46         DOMElement* rootElement = sxObject->marshall();\r
47 \r
48         string path=data_path + "SimpleXMLObjectWithAttribute.xml";\r
49         ifstream fs(path.c_str());\r
50         DOMDocument* doc=nonvalidatingPool->parse(fs);\r
51         TS_ASSERT(doc!=NULL);\r
52 \r
53         TS_ASSERT(rootElement->isEqualNode(doc->getDocumentElement()));\r
54         doc->release();\r
55     }\r
56 \r
57     void testMarshallingWithElementContent() {\r
58         TS_TRACE("testMarshallingWithElementContent");\r
59 \r
60         auto_ptr_XMLCh expected("Sample Content");\r
61         auto_ptr<SimpleXMLObject> sxObject(dynamic_cast<SimpleXMLObject*>(XMLObjectBuilder::buildOne(m_qname)));\r
62         TS_ASSERT(sxObject.get()!=NULL);\r
63         sxObject->setValue(expected.get());\r
64         \r
65         DOMElement* rootElement = sxObject->marshall();\r
66 \r
67         string path=data_path + "SimpleXMLObjectWithContent.xml";\r
68         ifstream fs(path.c_str());\r
69         DOMDocument* doc=nonvalidatingPool->parse(fs);\r
70         TS_ASSERT(doc!=NULL);\r
71 \r
72         TS_ASSERT(rootElement->isEqualNode(doc->getDocumentElement()));\r
73         doc->release();\r
74     }\r
75 \r
76     void testMarshallingWithChildElements() {\r
77         TS_TRACE("testMarshallingWithChildElements");\r
78 \r
79         const SimpleXMLObjectBuilder* b=dynamic_cast<const SimpleXMLObjectBuilder*>(XMLObjectBuilder::getBuilder(m_qname));\r
80         TS_ASSERT(b!=NULL);\r
81         \r
82         auto_ptr<SimpleXMLObject> sxObject(b->buildObject());\r
83         TS_ASSERT(sxObject.get()!=NULL);\r
84         VectorOf(SimpleXMLObject) kids=sxObject->getSimpleXMLObjects();\r
85         kids.push_back(b->buildObject());\r
86         kids.push_back(b->buildObject());\r
87         kids.push_back(b->buildObject());\r
88         \r
89         // Test some collection stuff\r
90         auto_ptr_XMLCh foo("Foo");\r
91         auto_ptr_XMLCh bar("Bar");\r
92         auto_ptr_XMLCh baz("Baz");\r
93         kids.begin()->setId(foo.get());\r
94         kids.at(2)->setValue(bar.get());\r
95         kids.erase(kids.begin()+1);\r
96         TS_ASSERT_SAME_DATA(kids.back()->getValue(), bar.get(), XMLString::stringLen(bar.get()));\r
97         \r
98         kids.push_back(b->buildObject(SimpleXMLObject::NAMESPACE,SimpleXMLObject::DERIVED_NAME,SimpleXMLObject::NAMESPACE_PREFIX));\r
99         kids.back()->setSchemaType(&m_qtype);\r
100         kids.back()->setValue(baz.get());\r
101         \r
102         DOMElement* rootElement = sxObject->marshall();\r
103 \r
104         string path=data_path + "SimpleXMLObjectWithChildren.xml";\r
105         ifstream fs(path.c_str());\r
106         DOMDocument* doc=nonvalidatingPool->parse(fs);\r
107         TS_ASSERT(doc!=NULL);\r
108 \r
109         TS_ASSERT(rootElement->isEqualNode(doc->getDocumentElement()));\r
110         doc->release();\r
111     }\r
112 \r
113 };\r