1 files changed, 28 insertions, 7 deletions
diff --git a/Swiften/Parser/UnitTest/XMLParserTest.cpp b/Swiften/Parser/UnitTest/XMLParserTest.cpp
index 9e9012b..c026b4b 100644
--- a/Swiften/Parser/UnitTest/XMLParserTest.cpp
+++ b/Swiften/Parser/UnitTest/XMLParserTest.cpp
@@ -1,17 +1,18 @@
 /*
  * Copyright (c) 2010 Isode Limited.
  * All rights reserved.
  * See the COPYING file for more information.
  */
 
 #include <cppunit/extensions/HelperMacros.h>
 #include <cppunit/extensions/TestFactoryRegistry.h>
+#include <unordered_map>
 #include <vector>
 
 #include <string>
 #include <Swiften/Parser/XMLParserClient.h>
 #ifdef HAVE_EXPAT
 #include <Swiften/Parser/ExpatParser.h>
 #endif
 #ifdef HAVE_LIBXML
 #include <Swiften/Parser/LibXMLParser.h>
@@ -55,18 +56,21 @@ class XMLParserTest : public CppUnit::TestFixture {
             CPPUNIT_ASSERT_EQUAL(std::string("iq"), client_.events[0].data);
             CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), client_.events[0].attributes.getEntries().size());
             CPPUNIT_ASSERT_EQUAL(std::string("get"), client_.events[0].attributes.getAttribute("type"));
             CPPUNIT_ASSERT_EQUAL(std::string(), client_.events[0].ns);
 
             CPPUNIT_ASSERT_EQUAL(Client::StartElement, client_.events[1].type);
             CPPUNIT_ASSERT_EQUAL(std::string("query"), client_.events[1].data);
             CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(0), client_.events[1].attributes.getEntries().size());
             CPPUNIT_ASSERT_EQUAL(std::string("jabber:iq:version"), client_.events[1].ns);
+            CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), client_.events[1].namespaces.size());
+            CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), client_.events[1].namespaces.count(""));
+            CPPUNIT_ASSERT_EQUAL(std::string("jabber:iq:version"), client_.events[1].namespaces[""]);
 
             CPPUNIT_ASSERT_EQUAL(Client::EndElement, client_.events[2].type);
             CPPUNIT_ASSERT_EQUAL(std::string("query"), client_.events[2].data);
             CPPUNIT_ASSERT_EQUAL(std::string("jabber:iq:version"), client_.events[2].ns);
 
             CPPUNIT_ASSERT_EQUAL(Client::EndElement, client_.events[3].type);
             CPPUNIT_ASSERT_EQUAL(std::string("iq"), client_.events[3].data);
             CPPUNIT_ASSERT_EQUAL(std::string(), client_.events[3].ns);
         }
@@ -79,22 +83,25 @@ class XMLParserTest : public CppUnit::TestFixture {
                     "<name>Swift</name>"
                 "</query>"));
 
             CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(5), client_.events.size());
 
             CPPUNIT_ASSERT_EQUAL(Client::StartElement, client_.events[0].type);
             CPPUNIT_ASSERT_EQUAL(std::string("query"), client_.events[0].data);
             CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(0), client_.events[0].attributes.getEntries().size());
             CPPUNIT_ASSERT_EQUAL(std::string("jabber:iq:version"), client_.events[0].ns);
+            CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), client_.events[0].namespaces.size());
+            CPPUNIT_ASSERT_EQUAL(std::string("jabber:iq:version"), client_.events[0].namespaces[""]);
 
             CPPUNIT_ASSERT_EQUAL(Client::StartElement, client_.events[1].type);
             CPPUNIT_ASSERT_EQUAL(std::string("name"), client_.events[1].data);
             CPPUNIT_ASSERT_EQUAL(std::string("jabber:iq:version"), client_.events[1].ns);
+            CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(0), client_.events[1].namespaces.size());
 
             CPPUNIT_ASSERT_EQUAL(Client::CharacterData, client_.events[2].type);
             CPPUNIT_ASSERT_EQUAL(std::string("Swift"), client_.events[2].data);
 
             CPPUNIT_ASSERT_EQUAL(Client::EndElement, client_.events[3].type);
             CPPUNIT_ASSERT_EQUAL(std::string("name"), client_.events[3].data);
             CPPUNIT_ASSERT_EQUAL(std::string("jabber:iq:version"), client_.events[3].ns);
 
             CPPUNIT_ASSERT_EQUAL(Client::EndElement, client_.events[4].type);
@@ -155,18 +162,20 @@ class XMLParserTest : public CppUnit::TestFixture {
             ParserType testling(&client_);
 
             CPPUNIT_ASSERT(testling.parse("<p:x xmlns:p='bla'><p:y/></p:x>"));
 
             CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(4), client_.events.size());
 
             CPPUNIT_ASSERT_EQUAL(Client::StartElement, client_.events[0].type);
             CPPUNIT_ASSERT_EQUAL(std::string("x"), client_.events[0].data);
             CPPUNIT_ASSERT_EQUAL(std::string("bla"), client_.events[0].ns);
+            CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), client_.events[0].namespaces.size());
+            CPPUNIT_ASSERT_EQUAL(std::string("bla"), client_.events[0].namespaces["p"]);
 
             CPPUNIT_ASSERT_EQUAL(Client::StartElement, client_.events[1].type);
             CPPUNIT_ASSERT_EQUAL(std::string("y"), client_.events[1].data);
             CPPUNIT_ASSERT_EQUAL(std::string("bla"), client_.events[1].ns);
 
             CPPUNIT_ASSERT_EQUAL(Client::EndElement, client_.events[2].type);
             CPPUNIT_ASSERT_EQUAL(std::string("y"), client_.events[2].data);
             CPPUNIT_ASSERT_EQUAL(std::string("bla"), client_.events[2].ns);
 
@@ -256,18 +265,21 @@ class XMLParserTest : public CppUnit::TestFixture {
         void testParse_AttributeWithNamespace() {
             ParserType testling(&client_);
 
             CPPUNIT_ASSERT(testling.parse(
                 "<query xmlns='http://swift.im' xmlns:f='http://swift.im/f' f:attr='3'/>"));
 
             CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), client_.events[0].attributes.getEntries().size());
             CPPUNIT_ASSERT_EQUAL(std::string("attr"), client_.events[0].attributes.getEntries()[0].getAttribute().getName());
             CPPUNIT_ASSERT_EQUAL(std::string("http://swift.im/f"), client_.events[0].attributes.getEntries()[0].getAttribute().getNamespace());
+            CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(2), client_.events[0].namespaces.size());
+            CPPUNIT_ASSERT_EQUAL(std::string("http://swift.im"), client_.events[0].namespaces[""]);
+            CPPUNIT_ASSERT_EQUAL(std::string("http://swift.im/f"), client_.events[0].namespaces["f"]);
         }
 
         void testParse_BillionLaughs() {
             ParserType testling(&client_);
 
             CPPUNIT_ASSERT(!testling.parse(
                 "<?xml version=\"1.0\"?>"
                 "<!DOCTYPE lolz ["
                 "  <!ENTITY lol \"lol\">"
@@ -295,18 +307,19 @@ class XMLParserTest : public CppUnit::TestFixture {
 
             CPPUNIT_ASSERT(testling.parse(
                 "<foo:bar><bla/></foo:bar>"));
 
             CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(4), client_.events.size());
 
             CPPUNIT_ASSERT_EQUAL(Client::StartElement, client_.events[0].type);
             CPPUNIT_ASSERT_EQUAL(std::string("foo:bar"), client_.events[0].data);
             CPPUNIT_ASSERT_EQUAL(std::string(""), client_.events[0].ns);
+            CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(0), client_.events[0].namespaces.size());
 
             CPPUNIT_ASSERT_EQUAL(Client::StartElement, client_.events[1].type);
             CPPUNIT_ASSERT_EQUAL(std::string("bla"), client_.events[1].data);
             CPPUNIT_ASSERT_EQUAL(std::string(""), client_.events[1].ns);
 
             CPPUNIT_ASSERT_EQUAL(Client::EndElement, client_.events[2].type);
             CPPUNIT_ASSERT_EQUAL(std::string("bla"), client_.events[2].data);
             CPPUNIT_ASSERT_EQUAL(std::string(""), client_.events[2].ns);
 
@@ -322,50 +335,58 @@ class XMLParserTest : public CppUnit::TestFixture {
                 "<foo bar:baz='bla'/>"));
 
             CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), client_.events[0].attributes.getEntries().size());
             CPPUNIT_ASSERT_EQUAL(std::string("bar:baz"), client_.events[0].attributes.getEntries()[0].getAttribute().getName());
         }
 
     private:
         class Client : public XMLParserClient {
             public:
-                enum Type { StartElement, EndElement, CharacterData };
+                using NamespaceMap = std::unordered_map<std::string /* prefix */, std::string /* uri */>;
+                enum Type { StartElement, EndElement, CharacterData, NamespaceDefined };
                 struct Event {
                     Event(
                             Type type,
                             const std::string& data,
                             const std::string& ns,
-                            const AttributeMap& attributes)
-                                : type(type), data(data), ns(ns), attributes(attributes) {}
+                            const AttributeMap& attributes,
+                            NamespaceMap namespaces = {})
+                                : type(type), data(data), ns(ns), attributes(attributes), namespaces(std::move(namespaces)) {}
                     Event(Type type, const std::string& data, const std::string& ns = std::string())
                                 : type(type), data(data), ns(ns) {}
 
                     Type type;
                     std::string data;
                     std::string ns;
                     AttributeMap attributes;
+                    NamespaceMap namespaces;
                 };
 
                 Client() {}
 
-                virtual void handleStartElement(const std::string& element, const std::string& ns, const AttributeMap& attributes) {
-                    events.push_back(Event(StartElement, element, ns, attributes));
+                void handleStartElement(const std::string& element, const std::string& ns, const AttributeMap& attributes) override {
+                    events.push_back(Event(StartElement, element, ns, attributes, std::move(namespaces_)));
                 }
 
-                virtual void handleEndElement(const std::string& element, const std::string& ns) {
+                void handleEndElement(const std::string& element, const std::string& ns) override {
                     events.push_back(Event(EndElement, element, ns));
                 }
 
-                virtual void handleCharacterData(const std::string& data) {
+                void handleCharacterData(const std::string& data) override {
                     events.push_back(Event(CharacterData, data));
                 }
 
+                void handleNamespaceDeclaration(const std::string& prefix, const std::string& uri) override {
+                    namespaces_[prefix] = uri;
+                }
                 std::vector<Event> events;
+            private:
+                NamespaceMap namespaces_;
         } client_;
 };
 
 #ifdef HAVE_EXPAT
 CPPUNIT_TEST_SUITE_REGISTRATION(XMLParserTest<ExpatParser>);
 #endif
 #ifdef HAVE_LIBXML
 CPPUNIT_TEST_SUITE_REGISTRATION(XMLParserTest<LibXMLParser>);
 #endif