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