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/AbstractElementProxy.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->m_namespaces=m_namespaces;
  76         ret->setId(m_id);
  77         ret->setValue(m_value);
  78         xmltooling::clone(m_children, ret->m_children);
  79         return ret;
  80     }
  81
  82 private:
  83     XMLCh* m_id;
  84     XMLCh* m_value;
  85     vector<SimpleXMLObject*> m_simples;
  86 };
  87
  88 class SimpleXMLObjectBuilder : public XMLObjectBuilder
  89 {
  90 public:
  91     SimpleXMLObject* buildObject() const {
  92         return new SimpleXMLObject();
  93     }
  94 };
  95
  96 class SimpleXMLObjectMarshaller : public AbstractXMLObjectMarshaller
  97 {
  98 public:
  99     SimpleXMLObjectMarshaller() {}
 100
 101 private:
 102     void marshallAttributes(const XMLObject& xmlObject, DOMElement* domElement) const {
 103         const SimpleXMLObject& simpleXMLObject = dynamic_cast<const SimpleXMLObject&>(xmlObject);
 104
 105         if(simpleXMLObject.getId()) {
 106             domElement->setAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME, simpleXMLObject.getId());
 107             domElement->setIdAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME);
 108         }
 109     }
 110
 111     void marshallElementContent(const XMLObject& xmlObject, DOMElement* domElement) const {
 112         const SimpleXMLObject& simpleXMLObject = dynamic_cast<const SimpleXMLObject&>(xmlObject);
 113
 114         if(simpleXMLObject.getValue()) {
 115             domElement->setTextContent(simpleXMLObject.getValue());
 116         }
 117     }
 118 };
 119
 120 class SimpleXMLObjectUnmarshaller : public AbstractXMLObjectUnmarshaller
 121 {
 122 public:
 123     SimpleXMLObjectUnmarshaller() {}
 124
 125 private:
 126     void processChildElement(XMLObject& parentXMLObject, XMLObject* childXMLObject) const {
 127         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(parentXMLObject);
 128
 129         SimpleXMLObject* child = dynamic_cast<SimpleXMLObject*>(childXMLObject);
 130         if (child) {
 131             simpleXMLObject.getSimpleXMLObjects().push_back(child);
 132         }
 133         else {
 134             throw UnmarshallingException("Unknown child element cannot be added to parent object.");
 135         }
 136     }
 137
 138     void processAttribute(XMLObject& xmlObject, const DOMAttr* attribute) const {
 139         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(xmlObject);
 140
 141         if (XMLString::equals(attribute->getLocalName(),SimpleXMLObject::ID_ATTRIB_NAME)) {
 142             simpleXMLObject.setId(attribute->getValue());
 143         }
 144         else {
 145             throw UnmarshallingException("Unknown attribute cannot be processed by parent object.");
 146         }
 147     }
 148
 149     void processElementContent(XMLObject& xmlObject, const XMLCh* elementContent) const {
 150         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(xmlObject);
 151
 152         simpleXMLObject.setValue(elementContent);
 153     }
 154
 155 };
 156
 157 class WildcardXMLObjectMarshaller;
 158
 159 class WildcardXMLObject : public AbstractElementProxy, public AbstractAttributeExtensibleXMLObject
 160 {
 161     friend class WildcardXMLObjectMarshaller;
 162 public:
 163     WildcardXMLObject(const XMLCh* nsURI, const XMLCh* localName, const XMLCh* prefix)
 164         : AbstractDOMCachingXMLObject(nsURI, localName, prefix),
 165         AbstractElementProxy(nsURI, localName, prefix),
 166         AbstractAttributeExtensibleXMLObject(nsURI, localName, prefix) {}
 167     virtual ~WildcardXMLObject() {}
 168
 169     WildcardXMLObject* clone() const {
 170         auto_ptr<XMLObject> domClone(AbstractDOMCachingXMLObject::clone());
 171         WildcardXMLObject* ret=dynamic_cast<WildcardXMLObject*>(domClone.get());
 172         if (ret) {
 173             domClone.release();
 174             return ret;
 175         }
 176
 177         ret=new WildcardXMLObject(
 178             getElementQName().getNamespaceURI(),getElementQName().getLocalPart(),getElementQName().getPrefix()
 179             );
 180         ret->m_namespaces=m_namespaces;
 181         for (map<QName,XMLCh*>::const_iterator i=m_attributeMap.begin(); i!=m_attributeMap.end(); i++) {
 182             ret->m_attributeMap[i->first]=XMLString::replicate(i->second);
 183         }
 184         ret->setTextContent(getTextContent());
 185         xmltooling::clone(m_children, ret->m_children);
 186         return ret;
 187     }
 188 };
 189
 190 class WildcardXMLObjectBuilder : public XMLObjectBuilder
 191 {
 192 public:
 193     XMLObject* buildObject() const {
 194         throw XMLObjectException("No default builder available.");
 195     }
 196
 197     WildcardXMLObject* buildObject(const XMLCh* nsURI, const XMLCh* localName, const XMLCh* prefix) const {
 198         return new WildcardXMLObject(nsURI,localName,prefix);
 199     }
 200 };
 201
 202 class WildcardXMLObjectMarshaller : public AbstractXMLObjectMarshaller
 203 {
 204 public:
 205     WildcardXMLObjectMarshaller() : AbstractXMLObjectMarshaller() {}
 206
 207 private:
 208     void marshallAttributes(const XMLObject& xmlObject, DOMElement* domElement) const {
 209         const WildcardXMLObject& wcXMLObject = dynamic_cast<const WildcardXMLObject&>(xmlObject);
 210
 211         for (map<QName,XMLCh*>::const_iterator i=wcXMLObject.m_attributeMap.begin(); i!=wcXMLObject.m_attributeMap.end(); i++) {
 212             DOMAttr* attr=domElement->getOwnerDocument()->createAttributeNS(i->first.getNamespaceURI(),i->first.getLocalPart());
 213             if (i->first.hasPrefix())
 214                 attr->setPrefix(i->first.getPrefix());
 215             attr->setNodeValue(i->second);
 216             domElement->setAttributeNode(attr);
 217         }
 218     }
 219
 220     void marshallElementContent(const XMLObject& xmlObject, DOMElement* domElement) const {
 221         const WildcardXMLObject& wcXMLObject = dynamic_cast<const WildcardXMLObject&>(xmlObject);
 222
 223         if(wcXMLObject.getTextContent()) {
 224             domElement->appendChild(domElement->getOwnerDocument()->createTextNode(wcXMLObject.getTextContent()));
 225         }
 226     }
 227 };
 228
 229 class WildcardXMLObjectUnmarshaller : public AbstractXMLObjectUnmarshaller
 230 {
 231 public:
 232     WildcardXMLObjectUnmarshaller() {}
 233
 234 private:
 235     XMLObject* buildXMLObject(const DOMElement* domElement) const {
 236         const WildcardXMLObjectBuilder* builder =
 237             dynamic_cast<const WildcardXMLObjectBuilder*>(XMLObjectBuilder::getBuilder(domElement));
 238         if (builder)
 239             return builder->buildObject(domElement->getNamespaceURI(),domElement->getLocalName(),domElement->getPrefix());
 240         throw UnmarshallingException("Failed to locate WildcardObjectBuilder for element.");
 241     }
 242
 243     void processChildElement(XMLObject& parentXMLObject, XMLObject* childXMLObject) const {
 244         WildcardXMLObject& wcXMLObject = dynamic_cast<WildcardXMLObject&>(parentXMLObject);
 245
 246         wcXMLObject.getXMLObjects().push_back(childXMLObject);
 247     }
 248
 249     void processAttribute(XMLObject& xmlObject, const DOMAttr* attribute) const {
 250         WildcardXMLObject& wcXMLObject = dynamic_cast<WildcardXMLObject&>(xmlObject);
 251
 252         QName q(attribute->getNamespaceURI(),attribute->getLocalName(),attribute->getPrefix());
 253         wcXMLObject.setAttribute(q,attribute->getNodeValue());
 254     }
 255
 256     void processElementContent(XMLObject& xmlObject, const XMLCh* elementContent) const {
 257         WildcardXMLObject& wcXMLObject = dynamic_cast<WildcardXMLObject&>(xmlObject);
 258
 259         wcXMLObject.setTextContent(elementContent);
 260     }
 261
 262 };
 263
 264 #if defined (_MSC_VER)
 265     #pragma warning( pop )
 266 #endif