xmltoolingtest/XMLObjectBaseTestCase.h

   1 /*\r
   2  *  Copyright 2001-2005 Internet2\r
   3  * \r
   4  * Licensed under the Apache License, Version 2.0 (the "License");\r
   5  * you may not use this file except in compliance with the License.\r
   6  * You may obtain a copy of the License at\r
   7  *\r
   8  *     http://www.apache.org/licenses/LICENSE-2.0\r
   9  *\r
  10  * Unless required by applicable law or agreed to in writing, software\r
  11  * distributed under the License is distributed on an "AS IS" BASIS,\r
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  13  * See the License for the specific language governing permissions and\r
  14  * limitations under the License.\r
  15  */\r
  16 \r
  17 #include <cxxtest/TestSuite.h>\r
  18 #include <xmltooling/AbstractDOMCachingXMLObject.h>\r
  19 #include <xmltooling/exceptions.h>\r
  20 #include <xmltooling/XMLObjectBuilder.h>\r
  21 #include <xmltooling/XMLToolingConfig.h>\r
  22 #include <xmltooling/io/AbstractXMLObjectMarshaller.h>\r
  23 #include <xmltooling/io/AbstractXMLObjectUnmarshaller.h>\r
  24 #include <xmltooling/util/ParserPool.h>\r
  25 #include <xmltooling/util/XMLHelper.h>\r
  26 \r
  27 using namespace xmltooling;\r
  28 using namespace std;\r
  29 \r
  30 extern ParserPool* validatingPool;\r
  31 extern ParserPool* nonvalidatingPool;\r
  32 extern string data_path;\r
  33 \r
  34 #if defined (_MSC_VER)\r
  35     #pragma warning( push )\r
  36     #pragma warning( disable : 4250 4251 )\r
  37 #endif\r
  38 \r
  39 class SimpleXMLObjectUnmarshaller;\r
  40 \r
  41 class SimpleXMLObject : public AbstractDOMCachingXMLObject\r
  42 {\r
  43 public:\r
  44     static const XMLCh NAMESPACE[];\r
  45     static const XMLCh NAMESPACE_PREFIX[];\r
  46     static const XMLCh LOCAL_NAME[];\r
  47     static const XMLCh ID_ATTRIB_NAME[];\r
  48     \r
  49     SimpleXMLObject() : AbstractDOMCachingXMLObject(NAMESPACE, LOCAL_NAME, NAMESPACE_PREFIX), m_id(NULL), m_value(NULL) {}\r
  50     virtual ~SimpleXMLObject() {\r
  51         XMLString::release(&m_id);\r
  52         XMLString::release(&m_value);\r
  53         for_each(m_children.begin(), m_children.end(), cleanup<SimpleXMLObject>());\r
  54     }\r
  55     \r
  56     const XMLCh* getId() const { return m_id; }\r
  57     void setId(const XMLCh* id) { m_id=prepareForAssignment(m_id,id); }\r
  58 \r
  59     const XMLCh* getValue() const { return m_value; }\r
  60     void setValue(const XMLCh* value) { m_value=prepareForAssignment(m_value,value); }\r
  61     \r
  62     // TODO: Leave non-const, but wrap STL container to intercept adds. \r
  63     list<SimpleXMLObject*>& getSimpleXMLObjects() { return m_children; }\r
  64     \r
  65     bool hasChildren() const { return !m_children.empty(); }\r
  66     size_t getOrderedChildren(vector<XMLObject*>& children) const {\r
  67         children.assign(m_children.begin(),m_children.end());\r
  68         return children.size();\r
  69     }\r
  70     SimpleXMLObject* clone() const {\r
  71         SimpleXMLObject* ret=new SimpleXMLObject();\r
  72         ret->setId(m_id);\r
  73         ret->setValue(m_value);\r
  74         xmltooling::clone(m_children, ret->m_children);\r
  75         return ret;\r
  76     }\r
  77 \r
  78 private:\r
  79     XMLCh* m_id;\r
  80     XMLCh* m_value;\r
  81     list<SimpleXMLObject*> m_children;\r
  82     \r
  83     friend class SimpleXMLObjectUnmarshaller;\r
  84 };\r
  85 \r
  86 class SimpleXMLObjectBuilder : public XMLObjectBuilder\r
  87 {\r
  88 public:\r
  89     XMLObject* buildObject() const {\r
  90         return new SimpleXMLObject();\r
  91     }\r
  92 };\r
  93 \r
  94 class SimpleXMLObjectMarshaller : public AbstractXMLObjectMarshaller\r
  95 {\r
  96 public:\r
  97     SimpleXMLObjectMarshaller() : AbstractXMLObjectMarshaller(SimpleXMLObject::NAMESPACE, SimpleXMLObject::LOCAL_NAME) {}\r
  98 \r
  99 private:\r
 100     void marshallAttributes(const XMLObject& xmlObject, DOMElement* domElement) const {\r
 101         const SimpleXMLObject& simpleXMLObject = dynamic_cast<const SimpleXMLObject&>(xmlObject);\r
 102         \r
 103         if(simpleXMLObject.getId()) {\r
 104             domElement->setAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME, simpleXMLObject.getId());\r
 105             domElement->setIdAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME);\r
 106         }\r
 107     }\r
 108 \r
 109     void marshallElementContent(const XMLObject& xmlObject, DOMElement* domElement) const {\r
 110         const SimpleXMLObject& simpleXMLObject = dynamic_cast<const SimpleXMLObject&>(xmlObject);\r
 111 \r
 112         if(simpleXMLObject.getValue()) {\r
 113             domElement->setTextContent(simpleXMLObject.getValue());\r
 114         }\r
 115     }\r
 116 };\r
 117 \r
 118 class SimpleXMLObjectUnmarshaller : public AbstractXMLObjectUnmarshaller\r
 119 {\r
 120 public:\r
 121     SimpleXMLObjectUnmarshaller() : AbstractXMLObjectUnmarshaller(SimpleXMLObject::NAMESPACE, SimpleXMLObject::LOCAL_NAME) {}\r
 122 \r
 123 private:\r
 124     void processChildElement(XMLObject& parentXMLObject, XMLObject* childXMLObject) const {\r
 125         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(parentXMLObject);\r
 126 \r
 127         SimpleXMLObject* child = dynamic_cast<SimpleXMLObject*>(childXMLObject);\r
 128         if (child) {\r
 129             simpleXMLObject.m_children.push_back(child);\r
 130         }\r
 131         else {\r
 132             throw UnmarshallingException("Unknown child element cannot be added to parent object.");\r
 133         }\r
 134     }\r
 135 \r
 136     void processAttribute(XMLObject& xmlObject, const DOMAttr* attribute) const {\r
 137         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(xmlObject);\r
 138 \r
 139         if (XMLString::equals(attribute->getLocalName(),SimpleXMLObject::ID_ATTRIB_NAME)) {\r
 140             simpleXMLObject.setId(attribute->getValue());\r
 141         }\r
 142         else {\r
 143             throw UnmarshallingException("Unknown attribute cannot be processed by parent object.");\r
 144         }\r
 145     }\r
 146 \r
 147     void processElementContent(XMLObject& xmlObject, const XMLCh* elementContent) const {\r
 148         SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(xmlObject);\r
 149         \r
 150         simpleXMLObject.setValue(elementContent);\r
 151     }\r
 152 \r
 153 };\r
 154 \r
 155 \r
 156 #if defined (_MSC_VER)\r
 157     #pragma warning( pop )\r
 158 #endif\r