Bug fixes and API changes from second unit test.
[shibboleth/cpp-xmltooling.git] / xmltoolingtest / XMLObjectBaseTestCase.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 <cxxtest/TestSuite.h>\r
18 #include <xmltooling/AbstractDOMCachingXMLObject.h>\r
19 #include <xmltooling/exceptions.h>\r
20 #include <xmltooling/XMLObjectBuilder.h>\r
21 #include <xmltooling/XMLToolingConfig.h>\r
22 #include <xmltooling/io/AbstractXMLObjectMarshaller.h>\r
23 #include <xmltooling/io/AbstractXMLObjectUnmarshaller.h>\r
24 #include <xmltooling/util/ParserPool.h>\r
25 #include <xmltooling/util/XMLHelper.h>\r
26 \r
27 using namespace xmltooling;\r
28 using namespace std;\r
29 \r
30 extern ParserPool* validatingPool;\r
31 extern ParserPool* nonvalidatingPool;\r
32 \r
33 #if defined (_MSC_VER)\r
34     #pragma warning( push )\r
35     #pragma warning( disable : 4250 4251 )\r
36 #endif\r
37 \r
38 class SimpleXMLObjectUnmarshaller;\r
39 \r
40 class SimpleXMLObject : public AbstractDOMCachingXMLObject\r
41 {\r
42 public:\r
43     static const XMLCh NAMESPACE[];\r
44     static const XMLCh NAMESPACE_PREFIX[];\r
45     static const XMLCh LOCAL_NAME[];\r
46     static const XMLCh ID_ATTRIB_NAME[];\r
47     \r
48     SimpleXMLObject() : AbstractDOMCachingXMLObject(NAMESPACE, LOCAL_NAME, NAMESPACE_PREFIX), m_id(NULL), m_value(NULL) {}\r
49     virtual ~SimpleXMLObject() {\r
50         XMLString::release(&m_id);\r
51         XMLString::release(&m_value);\r
52         for_each(m_children.begin(), m_children.end(), cleanup<SimpleXMLObject>());\r
53     }\r
54     \r
55     const XMLCh* getId() const { return m_id; }\r
56     void setId(const XMLCh* id) { m_id=prepareForAssignment(m_id,id); }\r
57 \r
58     const XMLCh* getValue() const { return m_value; }\r
59     void setValue(const XMLCh* value) { m_value=prepareForAssignment(m_value,value); }\r
60     \r
61     const list<SimpleXMLObject*>& getSimpleXMLObjects() const { return m_children; }\r
62     \r
63     bool hasChildren() const { return !m_children.empty(); }\r
64     size_t getOrderedChildren(vector<XMLObject*>& children) const {\r
65         children.assign(m_children.begin(),m_children.end());\r
66         return children.size();\r
67     }\r
68     SimpleXMLObject* clone() const {\r
69         SimpleXMLObject* ret=new SimpleXMLObject();\r
70         ret->setId(m_id);\r
71         ret->setValue(m_value);\r
72         xmltooling::clone(m_children, ret->m_children);\r
73         return ret;\r
74     }\r
75 \r
76 private:\r
77     XMLCh* m_id;\r
78     XMLCh* m_value;\r
79     list<SimpleXMLObject*> m_children;\r
80     \r
81     friend class SimpleXMLObjectUnmarshaller;\r
82 };\r
83 \r
84 class SimpleXMLObjectBuilder : public XMLObjectBuilder\r
85 {\r
86 public:\r
87     XMLObject* buildObject() const {\r
88         return new SimpleXMLObject();\r
89     }\r
90 };\r
91 \r
92 class SimpleXMLObjectMarshaller : public AbstractXMLObjectMarshaller\r
93 {\r
94 public:\r
95     SimpleXMLObjectMarshaller() : AbstractXMLObjectMarshaller(SimpleXMLObject::NAMESPACE, SimpleXMLObject::LOCAL_NAME) {}\r
96 \r
97 private:\r
98     void marshallAttributes(const XMLObject& xmlObject, DOMElement* domElement) const {\r
99         const SimpleXMLObject& simpleXMLObject = dynamic_cast<const SimpleXMLObject&>(xmlObject);\r
100         \r
101         if(simpleXMLObject.getId()) {\r
102             domElement->setAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME, simpleXMLObject.getId());\r
103             domElement->setIdAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME);\r
104         }\r
105     }\r
106 \r
107     void marshallElementContent(const XMLObject& xmlObject, DOMElement* domElement) const {\r
108         const SimpleXMLObject& simpleXMLObject = dynamic_cast<const SimpleXMLObject&>(xmlObject);\r
109 \r
110         if(simpleXMLObject.getValue()) {\r
111             domElement->setTextContent(simpleXMLObject.getValue());\r
112         }\r
113     }\r
114 };\r
115 \r
116 class SimpleXMLObjectUnmarshaller : public AbstractXMLObjectUnmarshaller\r
117 {\r
118 public:\r
119     SimpleXMLObjectUnmarshaller() : AbstractXMLObjectUnmarshaller(SimpleXMLObject::NAMESPACE, SimpleXMLObject::LOCAL_NAME) {}\r
120 \r
121 private:\r
122     void processChildElement(XMLObject& parentXMLObject, XMLObject* childXMLObject) const {\r
123         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(parentXMLObject);\r
124 \r
125         SimpleXMLObject* child = dynamic_cast<SimpleXMLObject*>(childXMLObject);\r
126         if (child) {\r
127             simpleXMLObject.m_children.push_back(child);\r
128         }\r
129         else {\r
130             throw UnmarshallingException("Unknown child element cannot be added to parent object.");\r
131         }\r
132     }\r
133 \r
134     void processAttribute(XMLObject& xmlObject, const DOMAttr* attribute) const {\r
135         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(xmlObject);\r
136 \r
137         if (XMLString::equals(attribute->getLocalName(),SimpleXMLObject::ID_ATTRIB_NAME)) {\r
138             simpleXMLObject.setId(attribute->getValue());\r
139         }\r
140     }\r
141 \r
142     void processElementContent(XMLObject& xmlObject, const XMLCh* elementContent) const {\r
143         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(xmlObject);\r
144         \r
145         simpleXMLObject.setValue(elementContent);\r
146     }\r
147 \r
148 };\r
149 \r
150 \r
151 #if defined (_MSC_VER)\r
152     #pragma warning( pop )\r
153 #endif\r