Bug fixes and API changes from second unit test.

[shibboleth/cpp-xmltooling.git] / xmltoolingtest / XMLObjectBaseTestCase.h

diff --git a/xmltoolingtest/XMLObjectBaseTestCase.h b/xmltoolingtest/XMLObjectBaseTestCase.h
index 99e878a..d0fb2d0 100644 (file)
--- a/xmltoolingtest/XMLObjectBaseTestCase.h
+++ b/xmltoolingtest/XMLObjectBaseTestCase.h
@@ -15,7 +15,12 @@
  */\r
 \r
 #include <cxxtest/TestSuite.h>\r
+#include <xmltooling/AbstractDOMCachingXMLObject.h>\r
+#include <xmltooling/exceptions.h>\r
+#include <xmltooling/XMLObjectBuilder.h>\r
 #include <xmltooling/XMLToolingConfig.h>\r
+#include <xmltooling/io/AbstractXMLObjectMarshaller.h>\r
+#include <xmltooling/io/AbstractXMLObjectUnmarshaller.h>\r
 #include <xmltooling/util/ParserPool.h>\r
 #include <xmltooling/util/XMLHelper.h>\r
 \r
@@ -25,3 +30,124 @@ using namespace std;
 extern ParserPool* validatingPool;\r
 extern ParserPool* nonvalidatingPool;\r
 \r
+#if defined (_MSC_VER)\r
+    #pragma warning( push )\r
+    #pragma warning( disable : 4250 4251 )\r
+#endif\r
+\r
+class SimpleXMLObjectUnmarshaller;\r
+\r
+class SimpleXMLObject : public AbstractDOMCachingXMLObject\r
+{\r
+public:\r
+    static const XMLCh NAMESPACE[];\r
+    static const XMLCh NAMESPACE_PREFIX[];\r
+    static const XMLCh LOCAL_NAME[];\r
+    static const XMLCh ID_ATTRIB_NAME[];\r
+    \r
+    SimpleXMLObject() : AbstractDOMCachingXMLObject(NAMESPACE, LOCAL_NAME, NAMESPACE_PREFIX), m_id(NULL), m_value(NULL) {}\r
+    virtual ~SimpleXMLObject() {\r
+        XMLString::release(&m_id);\r
+        XMLString::release(&m_value);\r
+        for_each(m_children.begin(), m_children.end(), cleanup<SimpleXMLObject>());\r
+    }\r
+    \r
+    const XMLCh* getId() const { return m_id; }\r
+    void setId(const XMLCh* id) { m_id=prepareForAssignment(m_id,id); }\r
+\r
+    const XMLCh* getValue() const { return m_value; }\r
+    void setValue(const XMLCh* value) { m_value=prepareForAssignment(m_value,value); }\r
+    \r
+    const list<SimpleXMLObject*>& getSimpleXMLObjects() const { return m_children; }\r
+    \r
+    bool hasChildren() const { return !m_children.empty(); }\r
+    size_t getOrderedChildren(vector<XMLObject*>& children) const {\r
+        children.assign(m_children.begin(),m_children.end());\r
+        return children.size();\r
+    }\r
+    SimpleXMLObject* clone() const {\r
+        SimpleXMLObject* ret=new SimpleXMLObject();\r
+        ret->setId(m_id);\r
+        ret->setValue(m_value);\r
+        xmltooling::clone(m_children, ret->m_children);\r
+        return ret;\r
+    }\r
+\r
+private:\r
+    XMLCh* m_id;\r
+    XMLCh* m_value;\r
+    list<SimpleXMLObject*> m_children;\r
+    \r
+    friend class SimpleXMLObjectUnmarshaller;\r
+};\r
+\r
+class SimpleXMLObjectBuilder : public XMLObjectBuilder\r
+{\r
+public:\r
+    XMLObject* buildObject() const {\r
+        return new SimpleXMLObject();\r
+    }\r
+};\r
+\r
+class SimpleXMLObjectMarshaller : public AbstractXMLObjectMarshaller\r
+{\r
+public:\r
+    SimpleXMLObjectMarshaller() : AbstractXMLObjectMarshaller(SimpleXMLObject::NAMESPACE, SimpleXMLObject::LOCAL_NAME) {}\r
+\r
+private:\r
+    void marshallAttributes(const XMLObject& xmlObject, DOMElement* domElement) const {\r
+        const SimpleXMLObject& simpleXMLObject = dynamic_cast<const SimpleXMLObject&>(xmlObject);\r
+        \r
+        if(simpleXMLObject.getId()) {\r
+            domElement->setAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME, simpleXMLObject.getId());\r
+            domElement->setIdAttributeNS(NULL, SimpleXMLObject::ID_ATTRIB_NAME);\r
+        }\r
+    }\r
+\r
+    void marshallElementContent(const XMLObject& xmlObject, DOMElement* domElement) const {\r
+        const SimpleXMLObject& simpleXMLObject = dynamic_cast<const SimpleXMLObject&>(xmlObject);\r
+\r
+        if(simpleXMLObject.getValue()) {\r
+            domElement->setTextContent(simpleXMLObject.getValue());\r
+        }\r
+    }\r
+};\r
+\r
+class SimpleXMLObjectUnmarshaller : public AbstractXMLObjectUnmarshaller\r
+{\r
+public:\r
+    SimpleXMLObjectUnmarshaller() : AbstractXMLObjectUnmarshaller(SimpleXMLObject::NAMESPACE, SimpleXMLObject::LOCAL_NAME) {}\r
+\r
+private:\r
+    void processChildElement(XMLObject& parentXMLObject, XMLObject* childXMLObject) const {\r
+        SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(parentXMLObject);\r
+\r
+        SimpleXMLObject* child = dynamic_cast<SimpleXMLObject*>(childXMLObject);\r
+        if (child) {\r
+            simpleXMLObject.m_children.push_back(child);\r
+        }\r
+        else {\r
+            throw UnmarshallingException("Unknown child element cannot be added to parent object.");\r
+        }\r
+    }\r
+\r
+    void processAttribute(XMLObject& xmlObject, const DOMAttr* attribute) const {\r
+        SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(xmlObject);\r
+\r
+        if (XMLString::equals(attribute->getLocalName(),SimpleXMLObject::ID_ATTRIB_NAME)) {\r
+            simpleXMLObject.setId(attribute->getValue());\r
+        }\r
+    }\r
+\r
+    void processElementContent(XMLObject& xmlObject, const XMLCh* elementContent) const {\r
+        SimpleXMLObject& simpleXMLObject = dynamic_cast<SimpleXMLObject&>(xmlObject);\r
+        \r
+        simpleXMLObject.setValue(elementContent);\r
+    }\r
+\r
+};\r
+\r
+\r
+#if defined (_MSC_VER)\r
+    #pragma warning( pop )\r
+#endif\r