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/AbstractComplexElement.h>
  19 #include <xmltooling/ElementProxy.h>
  20 #include <xmltooling/exceptions.h>
  21 #include <xmltooling/XMLObjectBuilder.h>
  22 #include <xmltooling/XMLToolingConfig.h>
  23 #include <xmltooling/io/AbstractXMLObjectMarshaller.h>
  24 #include <xmltooling/io/AbstractXMLObjectUnmarshaller.h>
  25 #include <xmltooling/impl/AnyElement.h>
  26 #include <xmltooling/impl/UnknownElement.h>
  27 #ifndef XMLTOOLING_NO_XMLSEC
  28     #include <xmltooling/signature/Signature.h>
  29 #endif
  30 #include <xmltooling/util/ParserPool.h>
  31 #include <xmltooling/util/XMLConstants.h>
  32 #include <xmltooling/util/XMLHelper.h>
  33 #include <xmltooling/util/XMLObjectChildrenList.h>
  34
  35 using namespace xmlsignature;
  36 using namespace xmltooling;
  37 using namespace std;
  38
  39 extern ParserPool* validatingPool;
  40 extern ParserPool* nonvalidatingPool;
  41 extern string data_path;
  42
  43 #if defined (_MSC_VER)
  44     #pragma warning( push )
  45     #pragma warning( disable : 4250 4251 )
  46 #endif
  47
  48 class SimpleXMLObject
  49     : public AbstractComplexElement,
  50         public AbstractDOMCachingXMLObject,
  51         public AbstractXMLObjectMarshaller,
  52         public AbstractXMLObjectUnmarshaller
  53 {
  54 protected:
  55     SimpleXMLObject(const SimpleXMLObject& src) : AbstractXMLObject(src), AbstractDOMCachingXMLObject(src),
  56         m_id(XMLString::replicate(src.m_id)), m_value(XMLString::replicate(src.m_value)) {
  57 #ifndef XMLTOOLING_NO_XMLSEC
  58         m_children.push_back(NULL);
  59         m_signature=m_children.begin();
  60 #endif
  61         VectorOf(SimpleXMLObject) mine=getSimpleXMLObjects();
  62         for (vector<SimpleXMLObject*>::const_iterator i=src.m_simples.begin(); i!=src.m_simples.end(); i++) {
  63             mine.push_back((*i) ? (*i)->clone() : NULL);
  64         }
  65     }
  66
  67 public:
  68     static const XMLCh NAMESPACE[];
  69     static const XMLCh NAMESPACE_PREFIX[];
  70     static const XMLCh LOCAL_NAME[];
  71     static const XMLCh DERIVED_NAME[];
  72     static const XMLCh TYPE_NAME[];
  73     static const XMLCh ID_ATTRIB_NAME[];
  74
  75     SimpleXMLObject(
  76         const XMLCh* nsURI=NULL, const XMLCh* localName=NULL, const XMLCh* prefix=NULL, const QName* schemaType=NULL
  77         ) : AbstractXMLObject(nsURI, localName, prefix, schemaType), m_id(NULL), m_value(NULL) {
  78 #ifndef XMLTOOLING_NO_XMLSEC
  79         m_children.push_back(NULL);
  80         m_signature=m_children.begin();
  81 #endif
  82     }
  83
  84     virtual ~SimpleXMLObject() {
  85         XMLString::release(&m_id);
  86         XMLString::release(&m_value);
  87     }
  88
  89     SimpleXMLObject* clone() const {
  90         auto_ptr<XMLObject> domClone(AbstractDOMCachingXMLObject::clone());
  91         SimpleXMLObject* ret=dynamic_cast<SimpleXMLObject*>(domClone.get());
  92         if (ret) {
  93             domClone.release();
  94             return ret;
  95         }
  96
  97         return new SimpleXMLObject(*this);
  98     }
  99
 100     const XMLCh* getId() const { return m_id; }
 101     void setId(const XMLCh* id) { m_id=prepareForAssignment(m_id,id); }
 102
 103     const XMLCh* getValue() const { return m_value; }
 104     void setValue(const XMLCh* value) { m_value=prepareForAssignment(m_value,value); }
 105
 106 #ifndef XMLTOOLING_NO_XMLSEC
 107     Signature* getSignature() const {
 108         return dynamic_cast<Signature*>(*m_signature);
 109     }
 110
 111     void setSignature(Signature* sig) {
 112         *m_signature=prepareForAssignment(*m_signature,sig);
 113     }
 114 #endif
 115
 116     VectorOf(SimpleXMLObject) getSimpleXMLObjects() {
 117         return VectorOf(SimpleXMLObject)(this, m_simples, &m_children, m_children.end());
 118     }
 119
 120     const std::vector<SimpleXMLObject*>& getSimpleXMLObjects() const {
 121         return m_simples;
 122     }
 123
 124 protected:
 125     void marshallAttributes(DOMElement* domElement) const {
 126         if(getId()) {
 127             domElement->setAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME, getId());
 128             domElement->setIdAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME);
 129         }
 130     }
 131
 132     void marshallElementContent(DOMElement* domElement) const {
 133         if(getValue()) {
 134             domElement->setTextContent(getValue());
 135         }
 136     }
 137
 138     void processChildElement(XMLObject* childXMLObject, const DOMElement* root) {
 139         SimpleXMLObject* simple=dynamic_cast<SimpleXMLObject*>(childXMLObject);
 140         if (simple) {
 141             getSimpleXMLObjects().push_back(simple);
 142             return;
 143         }
 144
 145 #ifndef XMLTOOLING_NO_XMLSEC
 146         Signature* sig=dynamic_cast<Signature*>(childXMLObject);
 147         if (sig) {
 148             setSignature(sig);
 149             return;
 150         }
 151 #endif
 152
 153         throw UnmarshallingException("Unknown child element cannot be added to parent object.");
 154     }
 155
 156     void processAttribute(const DOMAttr* attribute) {
 157         if (XMLHelper::isNodeNamed(attribute, NULL, SimpleXMLObject::ID_ATTRIB_NAME))
 158             setId(attribute->getValue());
 159         else
 160             throw UnmarshallingException("Unknown attribute cannot be processed by parent object.");
 161     }
 162
 163     void processElementContent(const XMLCh* elementContent) {
 164         setValue(elementContent);
 165     }
 166
 167 private:
 168     XMLCh* m_id;
 169     XMLCh* m_value;
 170     vector<SimpleXMLObject*> m_simples;
 171 #ifndef XMLTOOLING_NO_XMLSEC
 172     list<XMLObject*>::iterator m_signature;
 173 #endif
 174 };
 175
 176 class SimpleXMLObjectBuilder : public XMLObjectBuilder
 177 {
 178 public:
 179     SimpleXMLObject* buildObject() const {
 180         return buildObject(SimpleXMLObject::NAMESPACE, SimpleXMLObject::LOCAL_NAME, SimpleXMLObject::NAMESPACE_PREFIX);
 181     }
 182
 183     SimpleXMLObject* buildObject(
 184         const XMLCh* nsURI, const XMLCh* localName, const XMLCh* prefix=NULL, const QName* schemaType=NULL
 185         ) const {
 186         return new SimpleXMLObject(nsURI, localName, prefix, schemaType);
 187     }
 188 };
 189
 190 #if defined (_MSC_VER)
 191     #pragma warning( pop )
 192 #endif