Unix fixes
[shibboleth/cpp-xmltooling.git] / xmltoolingtest / XMLObjectBaseTestCase.h
1 /*
2  *  Copyright 2001-2005 Internet2
3  * 
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 #include <cxxtest/TestSuite.h>
18 #include <xmltooling/AbstractAttributeExtensibleXMLObject.h>
19 #include <xmltooling/AbstractDOMCachingXMLObject.h>
20 #include <xmltooling/AbstractExtensibleXMLObject.h>
21 #include <xmltooling/exceptions.h>
22 #include <xmltooling/XMLObjectBuilder.h>
23 #include <xmltooling/XMLToolingConfig.h>
24 #include <xmltooling/io/AbstractXMLObjectMarshaller.h>
25 #include <xmltooling/io/AbstractXMLObjectUnmarshaller.h>
26 #include <xmltooling/util/ParserPool.h>
27 #include <xmltooling/util/XMLObjectChildrenList.h>
28 #include <xmltooling/util/XMLHelper.h>
29
30 using namespace xmltooling;
31 using namespace std;
32
33 extern ParserPool* validatingPool;
34 extern ParserPool* nonvalidatingPool;
35 extern string data_path;
36
37 #if defined (_MSC_VER)
38     #pragma warning( push )
39     #pragma warning( disable : 4250 4251 )
40 #endif
41
42 class SimpleXMLObject : public AbstractDOMCachingXMLObject
43 {
44 public:
45     static const XMLCh NAMESPACE[];
46     static const XMLCh NAMESPACE_PREFIX[];
47     static const XMLCh LOCAL_NAME[];
48     static const XMLCh ID_ATTRIB_NAME[];
49
50     SimpleXMLObject() : AbstractDOMCachingXMLObject(NAMESPACE, LOCAL_NAME, NAMESPACE_PREFIX), m_id(NULL), m_value(NULL) {}
51     virtual ~SimpleXMLObject() {
52         XMLString::release(&m_id);
53         XMLString::release(&m_value);
54     }
55     
56     const XMLCh* getId() const { return m_id; }
57     void setId(const XMLCh* id) { m_id=prepareForAssignment(m_id,id); }
58
59     const XMLCh* getValue() const { return m_value; }
60     void setValue(const XMLCh* value) { m_value=prepareForAssignment(m_value,value); }
61     
62     VectorOf(SimpleXMLObject) getSimpleXMLObjects() {
63         return VectorOf(SimpleXMLObject)(this, m_simples, &m_children, m_children.end());
64     }
65     
66     SimpleXMLObject* clone() const {
67         auto_ptr<XMLObject> domClone(AbstractDOMCachingXMLObject::clone());
68         SimpleXMLObject* ret=dynamic_cast<SimpleXMLObject*>(domClone.get());
69         if (ret) {
70             domClone.release();
71             return ret;
72         }
73
74         ret=new SimpleXMLObject();
75         ret->setId(m_id);
76         ret->setValue(m_value);
77         xmltooling::clone(m_children, ret->m_children);
78         return ret;
79     }
80
81 private:
82     XMLCh* m_id;
83     XMLCh* m_value;
84     vector<SimpleXMLObject*> m_simples;
85 };
86
87 class SimpleXMLObjectBuilder : public XMLObjectBuilder
88 {
89 public:
90     SimpleXMLObject* buildObject() const {
91         return new SimpleXMLObject();
92     }
93 };
94
95 class SimpleXMLObjectMarshaller : public AbstractXMLObjectMarshaller
96 {
97 public:
98     SimpleXMLObjectMarshaller() {}
99
100 private:
101     void marshallAttributes(const XMLObject& xmlObject, DOMElement* domElement) const {
102         const SimpleXMLObject& simpleXMLObject = dynamic_cast<const SimpleXMLObject&>(xmlObject);
103         
104         if(simpleXMLObject.getId()) {
105             domElement->setAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME, simpleXMLObject.getId());
106             domElement->setIdAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME);
107         }
108     }
109
110     void marshallElementContent(const XMLObject& xmlObject, DOMElement* domElement) const {
111         const SimpleXMLObject& simpleXMLObject = dynamic_cast<const SimpleXMLObject&>(xmlObject);
112
113         if(simpleXMLObject.getValue()) {
114             domElement->setTextContent(simpleXMLObject.getValue());
115         }
116     }
117 };
118
119 class SimpleXMLObjectUnmarshaller : public AbstractXMLObjectUnmarshaller
120 {
121 public:
122     SimpleXMLObjectUnmarshaller() {}
123
124 private:
125     void processChildElement(XMLObject& parentXMLObject, XMLObject* childXMLObject) const {
126         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(parentXMLObject);
127
128         SimpleXMLObject* child = dynamic_cast<SimpleXMLObject*>(childXMLObject);
129         if (child) {
130             simpleXMLObject.getSimpleXMLObjects().push_back(child);
131         }
132         else {
133             throw UnmarshallingException("Unknown child element cannot be added to parent object.");
134         }
135     }
136
137     void processAttribute(XMLObject& xmlObject, const DOMAttr* attribute) const {
138         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(xmlObject);
139
140         if (XMLString::equals(attribute->getLocalName(),SimpleXMLObject::ID_ATTRIB_NAME)) {
141             simpleXMLObject.setId(attribute->getValue());
142         }
143         else {
144             throw UnmarshallingException("Unknown attribute cannot be processed by parent object.");
145         }
146     }
147
148     void processElementContent(XMLObject& xmlObject, const XMLCh* elementContent) const {
149         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(xmlObject);
150         
151         simpleXMLObject.setValue(elementContent);
152     }
153
154 };
155
156 class WildcardXMLObjectMarshaller;
157
158 class WildcardXMLObject : public AbstractExtensibleXMLObject, public AbstractAttributeExtensibleXMLObject
159 {
160     friend class WildcardXMLObjectMarshaller;
161 public:
162     WildcardXMLObject(const XMLCh* nsURI, const XMLCh* localName, const XMLCh* prefix)
163         : AbstractDOMCachingXMLObject(nsURI, localName, prefix),
164         AbstractExtensibleXMLObject(nsURI, localName, prefix),
165         AbstractAttributeExtensibleXMLObject(nsURI, localName, prefix) {}
166     virtual ~WildcardXMLObject() {}
167     
168     WildcardXMLObject* clone() const {
169         auto_ptr<XMLObject> domClone(AbstractDOMCachingXMLObject::clone());
170         WildcardXMLObject* ret=dynamic_cast<WildcardXMLObject*>(domClone.get());
171         if (ret) {
172             domClone.release();
173             return ret;
174         }
175
176         ret=new WildcardXMLObject(
177             getElementQName().getNamespaceURI(),getElementQName().getLocalPart(),getElementQName().getPrefix()
178             );
179         for (map<QName,XMLCh*>::const_iterator i=m_attributeMap.begin(); i!=m_attributeMap.end(); i++) {
180             ret->m_attributeMap[i->first]=XMLString::replicate(i->second);
181         }
182         xmltooling::clone(m_children, ret->m_children);
183         return ret;
184     }
185 };
186
187 class WildcardXMLObjectBuilder : public XMLObjectBuilder
188 {
189 public:
190     XMLObject* buildObject() const {
191         throw XMLObjectException("No default builder available.");
192     }
193
194     WildcardXMLObject* buildObject(const XMLCh* nsURI, const XMLCh* localName, const XMLCh* prefix) const {
195         return new WildcardXMLObject(nsURI,localName,prefix);
196     }
197 };
198
199 class WildcardXMLObjectMarshaller : public AbstractXMLObjectMarshaller
200 {
201 public:
202     WildcardXMLObjectMarshaller() : AbstractXMLObjectMarshaller() {}
203
204 private:
205     void marshallAttributes(const XMLObject& xmlObject, DOMElement* domElement) const {
206         const WildcardXMLObject& wcXMLObject = dynamic_cast<const WildcardXMLObject&>(xmlObject);
207
208         for (map<QName,XMLCh*>::const_iterator i=wcXMLObject.m_attributeMap.begin(); i!=wcXMLObject.m_attributeMap.end(); i++) {
209             DOMAttr* attr=domElement->getOwnerDocument()->createAttributeNS(i->first.getNamespaceURI(),i->first.getLocalPart());
210             if (i->first.hasPrefix())
211                 attr->setPrefix(i->first.getPrefix());
212             attr->setNodeValue(i->second);
213             domElement->setAttributeNode(attr);
214         }
215     }
216
217     void marshallElementContent(const XMLObject& xmlObject, DOMElement* domElement) const {
218         const WildcardXMLObject& wcXMLObject = dynamic_cast<const WildcardXMLObject&>(xmlObject);
219
220         if(wcXMLObject.getTextContent()) {
221             domElement->appendChild(domElement->getOwnerDocument()->createTextNode(wcXMLObject.getTextContent()));
222         }
223     }
224 };
225
226 class WildcardXMLObjectUnmarshaller : public AbstractXMLObjectUnmarshaller
227 {
228 public:
229     WildcardXMLObjectUnmarshaller() {}
230
231 private:
232     XMLObject* buildXMLObject(const DOMElement* domElement) const {
233         const WildcardXMLObjectBuilder* builder =
234             dynamic_cast<const WildcardXMLObjectBuilder*>(XMLObjectBuilder::getBuilder(domElement));
235         if (builder)
236             return builder->buildObject(domElement->getNamespaceURI(),domElement->getLocalName(),domElement->getPrefix());
237         throw UnmarshallingException("Failed to locate WildcardObjectBuilder for element.");
238     }
239
240     void processChildElement(XMLObject& parentXMLObject, XMLObject* childXMLObject) const {
241         WildcardXMLObject& wcXMLObject = dynamic_cast<WildcardXMLObject&>(parentXMLObject);
242
243         wcXMLObject.getXMLObjects().push_back(childXMLObject);
244     }
245
246     void processAttribute(XMLObject& xmlObject, const DOMAttr* attribute) const {
247         WildcardXMLObject& wcXMLObject = dynamic_cast<WildcardXMLObject&>(xmlObject);
248        
249         QName q(attribute->getNamespaceURI(),attribute->getLocalName(),attribute->getPrefix()); 
250         wcXMLObject.setAttribute(q,attribute->getNodeValue());
251     }
252
253     void processElementContent(XMLObject& xmlObject, const XMLCh* elementContent) const {
254         WildcardXMLObject& wcXMLObject = dynamic_cast<WildcardXMLObject&>(xmlObject);
255         
256         wcXMLObject.setTextContent(elementContent);
257     }
258
259 };
260
261 #if defined (_MSC_VER)
262     #pragma warning( pop )
263 #endif