/*
 * 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 <Swiften/Parser/PayloadParsers/RosterParser.h>
#include <Swiften/Parser/PayloadParsers/UnitTest/PayloadsParserTester.h>

using namespace Swift;

class RosterParserTest : public CppUnit::TestFixture
{
        CPPUNIT_TEST_SUITE(RosterParserTest);
        CPPUNIT_TEST(testParse);
        CPPUNIT_TEST(testParse_ItemWithUnknownContent);
        CPPUNIT_TEST(testParse_WithVersion);
        CPPUNIT_TEST(testParse_WithEmptyVersion);
        CPPUNIT_TEST_SUITE_END();

    public:
        void testParse() {
            PayloadsParserTester parser;
            CPPUNIT_ASSERT(parser.parse(
                "<query xmlns='jabber:iq:roster'>"
                "    <item jid='foo@bar.com' name='Foo @ Bar' subscription='from' ask='subscribe'>"
                "        <group>Group 1</group>"
                "        <group>Group 2</group>"
                "    </item>"
                " <item jid='baz@blo.com' name='Baz'/>"
                "</query>"));

            RosterPayload* payload = dynamic_cast<RosterPayload*>(parser.getPayload().get());

            CPPUNIT_ASSERT(!payload->getVersion());
            const RosterPayload::RosterItemPayloads& items = payload->getItems();

            CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(2), items.size());

            CPPUNIT_ASSERT_EQUAL(JID("foo@bar.com"), items[0].getJID());
            CPPUNIT_ASSERT_EQUAL(std::string("Foo @ Bar"), items[0].getName());
            CPPUNIT_ASSERT_EQUAL(RosterItemPayload::From, items[0].getSubscription());
            CPPUNIT_ASSERT(items[0].getSubscriptionRequested());
            CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(2), items[0].getGroups().size());
            CPPUNIT_ASSERT_EQUAL(std::string("Group 1"), items[0].getGroups()[0]);
            CPPUNIT_ASSERT_EQUAL(std::string("Group 2"), items[0].getGroups()[1]);

            CPPUNIT_ASSERT_EQUAL(JID("baz@blo.com"), items[1].getJID());
            CPPUNIT_ASSERT_EQUAL(std::string("Baz"), items[1].getName());
            CPPUNIT_ASSERT_EQUAL(RosterItemPayload::None, items[1].getSubscription());
            CPPUNIT_ASSERT(!items[1].getSubscriptionRequested());
            CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(0), items[1].getGroups().size());
        }

        void testParse_ItemWithUnknownContent() {
            PayloadsParserTester parser;
            CPPUNIT_ASSERT(parser.parse(
                "<query xmlns='jabber:iq:roster'>"
                "    <item jid='foo@bar.com' name='Foo @ Bar' subscription='from' ask='subscribe'>"
                "        <group>Group 1</group>"
                "        <foo xmlns=\"http://example.com\"><bar>Baz</bar></foo>"
                "        <group>Group 2</group>"
                "        <baz><fum>foo</fum></baz>"
                "    </item>"
                "</query>"));

            RosterPayload* payload = dynamic_cast<RosterPayload*>(parser.getPayload().get());
            const RosterPayload::RosterItemPayloads& items = payload->getItems();

            CPPUNIT_ASSERT_EQUAL(static_cast<size_t>(1), items.size());
            CPPUNIT_ASSERT_EQUAL(std::string("Group 1"), items[0].getGroups()[0]);
            CPPUNIT_ASSERT_EQUAL(std::string("Group 2"), items[0].getGroups()[1]);
            CPPUNIT_ASSERT_EQUAL(std::string(
                "<foo xmlns=\"http://example.com\"><bar xmlns=\"http://example.com\">Baz</bar></foo>"
                "<baz xmlns=\"jabber:iq:roster\"><fum xmlns=\"jabber:iq:roster\">foo</fum></baz>"
                ), items[0].getUnknownContent());
        }

        void testParse_WithVersion() {
            PayloadsParserTester parser;
            CPPUNIT_ASSERT(parser.parse("<query xmlns='jabber:iq:roster' ver='ver10'/>"));

            RosterPayload* payload = dynamic_cast<RosterPayload*>(parser.getPayload().get());
            CPPUNIT_ASSERT(payload->getVersion());
            CPPUNIT_ASSERT_EQUAL(std::string("ver10"), *payload->getVersion());
        }

        void testParse_WithEmptyVersion() {
            PayloadsParserTester parser;
            CPPUNIT_ASSERT(parser.parse("<query xmlns='jabber:iq:roster' ver=''/>"));

            RosterPayload* payload = dynamic_cast<RosterPayload*>(parser.getPayload().get());
            CPPUNIT_ASSERT(payload->getVersion());
            CPPUNIT_ASSERT_EQUAL(std::string(""), *payload->getVersion());
        }
};

CPPUNIT_TEST_SUITE_REGISTRATION(RosterParserTest);
