Began to implement collection handling.
[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 public:\r
24     MarshallingTest() : m_qname(SimpleXMLObject::NAMESPACE,SimpleXMLObject::LOCAL_NAME) {}\r
25 \r
26     void setUp() {\r
27         XMLObjectBuilder::registerBuilder(m_qname, new SimpleXMLObjectBuilder());\r
28         Marshaller::registerMarshaller(m_qname, new SimpleXMLObjectMarshaller());\r
29         Unmarshaller::registerUnmarshaller(m_qname, new SimpleXMLObjectUnmarshaller());\r
30     }\r
31 \r
32     void tearDown() {\r
33         XMLObjectBuilder::deregisterBuilder(m_qname);\r
34         Marshaller::deregisterMarshaller(m_qname);\r
35         Unmarshaller::deregisterUnmarshaller(m_qname);\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::getBuilder(m_qname)->buildObject()));\r
43         TS_ASSERT(sxObject.get()!=NULL);\r
44         sxObject->setId(expected.get());\r
45         \r
46         DOMElement* rootElement = Marshaller::getMarshaller(sxObject.get())->marshall(sxObject.get());\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::getBuilder(m_qname)->buildObject()));\r
62         TS_ASSERT(sxObject.get()!=NULL);\r
63         sxObject->setValue(expected.get());\r
64         \r
65         DOMElement* rootElement = Marshaller::getMarshaller(sxObject.get())->marshall(sxObject.get());\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 XMLObjectBuilder* b=XMLObjectBuilder::getBuilder(m_qname);\r
80         TS_ASSERT(b!=NULL);\r
81         \r
82         auto_ptr<SimpleXMLObject> sxObject(dynamic_cast<SimpleXMLObject*>(b->buildObject()));\r
83         TS_ASSERT(sxObject.get()!=NULL);\r
84         ListOf(SimpleXMLObject) kids=sxObject->getSimpleXMLObjects();\r
85         kids.push_back(dynamic_cast<SimpleXMLObject*>(b->buildObject()));\r
86         kids.push_back(dynamic_cast<SimpleXMLObject*>(b->buildObject()));\r
87         kids.push_back(dynamic_cast<SimpleXMLObject*>(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         kids[0]->setId(foo.get());\r
93         kids.at(2)->setValue(bar.get());\r
94         kids.erase(kids.begin()+1);\r
95         \r
96         DOMElement* rootElement = Marshaller::getMarshaller(sxObject.get())->marshall(sxObject.get());\r
97 \r
98         string path=data_path + "SimpleXMLObjectWithChildren.xml";\r
99         ifstream fs(path.c_str());\r
100         DOMDocument* doc=nonvalidatingPool->parse(fs);\r
101         TS_ASSERT(doc!=NULL);\r
102 \r
103         TS_ASSERT(rootElement->isEqualNode(doc->getDocumentElement()));\r
104         doc->release();\r
105     }\r
106 \r
107 };\r