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