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 #ifndef XMLTOOLING_NO_XMLSEC
  27     #include <xmltooling/signature/Signature.h>
  28 #endif
  29 #include <xmltooling/util/ParserPool.h>
  30 #include <xmltooling/util/XMLConstants.h>
  31 #include <xmltooling/util/XMLHelper.h>
  32 #include <xmltooling/util/XMLObjectChildrenList.h>
  33
  34 using namespace xmltooling;
  35 using namespace std;
  36
  37 extern ParserPool* validatingPool;
  38 extern ParserPool* nonvalidatingPool;
  39 extern string data_path;
  40
  41 #if defined (_MSC_VER)
  42     #pragma warning( push )
  43     #pragma warning( disable : 4250 4251 )
  44 #endif
  45
  46 class SimpleXMLObject : public AbstractDOMCachingXMLObject
  47 {
  48 public:
  49     static const XMLCh NAMESPACE[];
  50     static const XMLCh NAMESPACE_PREFIX[];
  51     static const XMLCh LOCAL_NAME[];
  52     static const XMLCh ID_ATTRIB_NAME[];
  53
  54     SimpleXMLObject() : AbstractDOMCachingXMLObject(NAMESPACE, LOCAL_NAME, NAMESPACE_PREFIX), m_id(NULL), m_value(NULL) {
  55 #ifndef XMLTOOLING_NO_XMLSEC
  56         m_children.push_back(NULL);
  57         m_signature=m_children.begin();
  58 #endif
  59     }
  60
  61     virtual ~SimpleXMLObject() {
  62         XMLString::release(&m_id);
  63         XMLString::release(&m_value);
  64     }
  65
  66     const XMLCh* getId() const { return m_id; }
  67     void setId(const XMLCh* id) { m_id=prepareForAssignment(m_id,id); }
  68
  69     const XMLCh* getValue() const { return m_value; }
  70     void setValue(const XMLCh* value) { m_value=prepareForAssignment(m_value,value); }
  71
  72 #ifndef XMLTOOLING_NO_XMLSEC
  73     Signature* getSignature() const {
  74         return dynamic_cast<Signature*>(*m_signature);
  75     }
  76
  77     void setSignature(Signature* sig) {
  78         *m_signature=prepareForAssignment(*m_signature,sig);
  79     }
  80 #endif
  81
  82     VectorOf(SimpleXMLObject) getSimpleXMLObjects() {
  83         return VectorOf(SimpleXMLObject)(this, m_simples, &m_children, m_children.end());
  84     }
  85
  86     SimpleXMLObject* clone() const {
  87         auto_ptr<XMLObject> domClone(AbstractDOMCachingXMLObject::clone());
  88         SimpleXMLObject* ret=dynamic_cast<SimpleXMLObject*>(domClone.get());
  89         if (ret) {
  90             domClone.release();
  91             return ret;
  92         }
  93
  94         ret=new SimpleXMLObject();
  95         ret->m_namespaces=m_namespaces;
  96         ret->setId(m_id);
  97         ret->setValue(m_value);
  98         xmltooling::clone(m_children, ret->m_children);
  99         return ret;
 100     }
 101
 102 private:
 103     XMLCh* m_id;
 104     XMLCh* m_value;
 105     vector<SimpleXMLObject*> m_simples;
 106 #ifndef XMLTOOLING_NO_XMLSEC
 107     list<XMLObject*>::iterator m_signature;
 108 #endif
 109 };
 110
 111 class SimpleXMLObjectBuilder : public XMLObjectBuilder
 112 {
 113 public:
 114     SimpleXMLObject* buildObject() const {
 115         return new SimpleXMLObject();
 116     }
 117 };
 118
 119 class SimpleXMLObjectMarshaller : public AbstractXMLObjectMarshaller
 120 {
 121 public:
 122     SimpleXMLObjectMarshaller() {}
 123
 124 private:
 125     void marshallAttributes(const XMLObject& xmlObject, DOMElement* domElement) const {
 126         const SimpleXMLObject& simpleXMLObject = dynamic_cast<const SimpleXMLObject&>(xmlObject);
 127
 128         if(simpleXMLObject.getId()) {
 129             domElement->setAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME, simpleXMLObject.getId());
 130             domElement->setIdAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME);
 131         }
 132     }
 133
 134     void marshallElementContent(const XMLObject& xmlObject, DOMElement* domElement) const {
 135         const SimpleXMLObject& simpleXMLObject = dynamic_cast<const SimpleXMLObject&>(xmlObject);
 136
 137         if(simpleXMLObject.getValue()) {
 138             domElement->setTextContent(simpleXMLObject.getValue());
 139         }
 140     }
 141 };
 142
 143 class SimpleXMLObjectUnmarshaller : public AbstractXMLObjectUnmarshaller
 144 {
 145 public:
 146     SimpleXMLObjectUnmarshaller() {}
 147
 148 private:
 149     void processChildElement(XMLObject& parentXMLObject, XMLObject* childXMLObject, const DOMElement* root) const {
 150         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(parentXMLObject);
 151
 152         if (XMLHelper::isNodeNamed(root, SimpleXMLObject::NAMESPACE, SimpleXMLObject::LOCAL_NAME))
 153             simpleXMLObject.getSimpleXMLObjects().push_back(dynamic_cast<SimpleXMLObject*>(childXMLObject));
 154 #ifndef XMLTOOLING_NO_XMLSEC
 155         else if (XMLHelper::isNodeNamed(root, XMLConstants::XMLSIG_NS, Signature::LOCAL_NAME))
 156             simpleXMLObject.setSignature(dynamic_cast<Signature*>(childXMLObject));
 157 #endif
 158         else
 159             throw UnmarshallingException("Unknown child element cannot be added to parent object.");
 160     }
 161
 162     void processAttribute(XMLObject& xmlObject, const DOMAttr* attribute) const {
 163         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(xmlObject);
 164
 165         if (XMLHelper::isNodeNamed(attribute, NULL, SimpleXMLObject::ID_ATTRIB_NAME))
 166             simpleXMLObject.setId(attribute->getValue());
 167         else
 168             throw UnmarshallingException("Unknown attribute cannot be processed by parent object.");
 169     }
 170
 171     void processElementContent(XMLObject& xmlObject, const XMLCh* elementContent) const {
 172         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(xmlObject);
 173
 174         simpleXMLObject.setValue(elementContent);
 175     }
 176
 177 };
 178
 179 class WildcardXMLObjectMarshaller;
 180
 181 class WildcardXMLObject : public AbstractElementProxy, public AbstractAttributeExtensibleXMLObject
 182 {
 183     friend class WildcardXMLObjectMarshaller;
 184 public:
 185     WildcardXMLObject(const XMLCh* nsURI, const XMLCh* localName, const XMLCh* prefix)
 186         : AbstractDOMCachingXMLObject(nsURI, localName, prefix),
 187         AbstractElementProxy(nsURI, localName, prefix),
 188         AbstractAttributeExtensibleXMLObject(nsURI, localName, prefix) {}
 189     virtual ~WildcardXMLObject() {}
 190
 191     WildcardXMLObject* clone() const {
 192         auto_ptr<XMLObject> domClone(AbstractDOMCachingXMLObject::clone());
 193         WildcardXMLObject* ret=dynamic_cast<WildcardXMLObject*>(domClone.get());
 194         if (ret) {
 195             domClone.release();
 196             return ret;
 197         }
 198
 199         ret=new WildcardXMLObject(
 200             getElementQName().getNamespaceURI(),getElementQName().getLocalPart(),getElementQName().getPrefix()
 201             );
 202         ret->m_namespaces=m_namespaces;
 203         for (map<QName,XMLCh*>::const_iterator i=m_attributeMap.begin(); i!=m_attributeMap.end(); i++) {
 204             ret->m_attributeMap[i->first]=XMLString::replicate(i->second);
 205         }
 206         ret->setTextContent(getTextContent());
 207         xmltooling::clone(m_children, ret->m_children);
 208         return ret;
 209     }
 210 };
 211
 212 class WildcardXMLObjectBuilder : public XMLObjectBuilder
 213 {
 214 public:
 215     XMLObject* buildObject() const {
 216         throw XMLObjectException("No default builder available.");
 217     }
 218
 219     WildcardXMLObject* buildObject(const XMLCh* nsURI, const XMLCh* localName, const XMLCh* prefix) const {
 220         return new WildcardXMLObject(nsURI,localName,prefix);
 221     }
 222 };
 223
 224 class WildcardXMLObjectMarshaller : public AbstractXMLObjectMarshaller
 225 {
 226 public:
 227     WildcardXMLObjectMarshaller() : AbstractXMLObjectMarshaller() {}
 228
 229 private:
 230     void marshallAttributes(const XMLObject& xmlObject, DOMElement* domElement) const {
 231         const WildcardXMLObject& wcXMLObject = dynamic_cast<const WildcardXMLObject&>(xmlObject);
 232
 233         for (map<QName,XMLCh*>::const_iterator i=wcXMLObject.m_attributeMap.begin(); i!=wcXMLObject.m_attributeMap.end(); i++) {
 234             DOMAttr* attr=domElement->getOwnerDocument()->createAttributeNS(i->first.getNamespaceURI(),i->first.getLocalPart());
 235             if (i->first.hasPrefix())
 236                 attr->setPrefix(i->first.getPrefix());
 237             attr->setNodeValue(i->second);
 238             domElement->setAttributeNode(attr);
 239         }
 240     }
 241
 242     void marshallElementContent(const XMLObject& xmlObject, DOMElement* domElement) const {
 243         const WildcardXMLObject& wcXMLObject = dynamic_cast<const WildcardXMLObject&>(xmlObject);
 244
 245         if(wcXMLObject.getTextContent()) {
 246             domElement->appendChild(domElement->getOwnerDocument()->createTextNode(wcXMLObject.getTextContent()));
 247         }
 248     }
 249 };
 250
 251 class WildcardXMLObjectUnmarshaller : public AbstractXMLObjectUnmarshaller
 252 {
 253 public:
 254     WildcardXMLObjectUnmarshaller() {}
 255
 256 private:
 257     XMLObject* buildXMLObject(const DOMElement* domElement) const {
 258         const WildcardXMLObjectBuilder* builder =
 259             dynamic_cast<const WildcardXMLObjectBuilder*>(XMLObjectBuilder::getBuilder(domElement));
 260         if (builder)
 261             return builder->buildObject(domElement->getNamespaceURI(),domElement->getLocalName(),domElement->getPrefix());
 262         throw UnmarshallingException("Failed to locate WildcardObjectBuilder for element.");
 263     }
 264
 265     void processChildElement(XMLObject& parentXMLObject, XMLObject* childXMLObject, const DOMElement* root) const {
 266         WildcardXMLObject& wcXMLObject = dynamic_cast<WildcardXMLObject&>(parentXMLObject);
 267
 268         wcXMLObject.getXMLObjects().push_back(childXMLObject);
 269     }
 270
 271     void processAttribute(XMLObject& xmlObject, const DOMAttr* attribute) const {
 272         WildcardXMLObject& wcXMLObject = dynamic_cast<WildcardXMLObject&>(xmlObject);
 273
 274         QName q(attribute->getNamespaceURI(),attribute->getLocalName(),attribute->getPrefix());
 275         wcXMLObject.setAttribute(q,attribute->getNodeValue());
 276     }
 277
 278     void processElementContent(XMLObject& xmlObject, const XMLCh* elementContent) const {
 279         WildcardXMLObject& wcXMLObject = dynamic_cast<WildcardXMLObject&>(xmlObject);
 280
 281         wcXMLObject.setTextContent(elementContent);
 282     }
 283
 284 };
 285
 286 #if defined (_MSC_VER)
 287     #pragma warning( pop )
 288 #endif