Began to implement collection handling.

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

diff --git a/xmltoolingtest/XMLObjectBaseTestCase.h b/xmltoolingtest/XMLObjectBaseTestCase.h
index 99e878a..d69e1c0 100644 (file)
--- a/xmltoolingtest/XMLObjectBaseTestCase.h
+++ b/xmltoolingtest/XMLObjectBaseTestCase.h
@@ -15,8 +15,14 @@
  */\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/XMLObjectChildrenList.h>\r
 #include <xmltooling/util/XMLHelper.h>\r
 \r
 using namespace xmltooling;\r
@@ -24,4 +30,134 @@ using namespace std;
 \r
 extern ParserPool* validatingPool;\r
 extern ParserPool* nonvalidatingPool;\r
+extern string data_path;\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
+    }\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
+    // TODO: Leave non-const, but wrap STL container to intercept adds. \r
+    ListOf(SimpleXMLObject) getSimpleXMLObjects() {\r
+        return ListOf(SimpleXMLObject)(this, m_simples, m_children, m_children.end());\r
+    }\r
+    \r
+    SimpleXMLObject* clone() const {\r
+        auto_ptr<XMLObject> domClone(AbstractDOMCachingXMLObject::clone());\r
+        SimpleXMLObject* ret=dynamic_cast<SimpleXMLObject*>(domClone.get());\r
+        if (ret) {\r
+            domClone.release();\r
+            return ret;\r
+        }\r
+\r
+        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
+    vector<SimpleXMLObject*> m_simples;\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
+            simpleXMLObject.m_simples.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
+        else {\r
+            throw UnmarshallingException("Unknown attribute cannot be processed by parent object.");\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