/*
 * Copyright (c) 2011-2016 Isode Limited.
 * All rights reserved.
 * See the COPYING file for more information.
 */

#include <cppunit/extensions/HelperMacros.h>
#include <cppunit/extensions/TestFactoryRegistry.h>

#include <Swiften/Base/foreach.h>
#include <Swiften/Elements/MUCDestroyPayload.h>
#include <Swiften/Parser/PayloadParsers/MUCUserPayloadParser.h>
#include <Swiften/Parser/PayloadParsers/UnitTest/PayloadsParserTester.h>

using namespace Swift;

class MUCUserPayloadParserTest : public CppUnit::TestFixture
{
        CPPUNIT_TEST_SUITE(MUCUserPayloadParserTest);
        CPPUNIT_TEST(testParseEmpty);
        CPPUNIT_TEST(testParse);
        CPPUNIT_TEST(testParseDestroy);
        CPPUNIT_TEST(testParseInvite);
        CPPUNIT_TEST_SUITE_END();

    public:
        MUCUserPayloadParserTest() {}

        void testParse() {
            PayloadsParserTester parser;

            CPPUNIT_ASSERT(parser.parse("<x xmlns=\"http://jabber.org/protocol/muc#user\"><status code='110'/><item affiliation=\"owner\" role=\"visitor\"><actor jid=\"kev@tester.lit\"/><reason>malice</reason></item><status code='210'/></x>"));

            bool found110 = false;
            bool found210 = false;

            MUCUserPayload::ref payload = std::dynamic_pointer_cast<MUCUserPayload>(parser.getPayload());

            foreach (MUCUserPayload::StatusCode status, payload->getStatusCodes()) {
                if (status.code == 110) found110 = true;
                if (status.code == 210) found210 = true;
            }

            MUCItem item = payload->getItems()[0];
            CPPUNIT_ASSERT_EQUAL(MUCOccupant::Owner, item.affiliation.get());
            CPPUNIT_ASSERT_EQUAL(MUCOccupant::Visitor, item.role.get());
            CPPUNIT_ASSERT_EQUAL(JID("kev@tester.lit"), item.actor.get());
            CPPUNIT_ASSERT_EQUAL(std::string("malice"), item.reason.get());
            CPPUNIT_ASSERT(found110);
            CPPUNIT_ASSERT(found210);
        }

        void testParseEmpty() {
            PayloadsParserTester parser;

            CPPUNIT_ASSERT(parser.parse("<x xmlns=\"http://jabber.org/protocol/muc#user\"/>"));

            MUCUserPayload::ref payload = std::dynamic_pointer_cast<MUCUserPayload>(parser.getPayload());
            CPPUNIT_ASSERT(payload);
            CPPUNIT_ASSERT(payload->getItems().empty());
        }

        void testParseDestroy() {
            PayloadsParserTester parser;

            CPPUNIT_ASSERT(parser.parse("<x xmlns=\"http://jabber.org/protocol/muc#user\"><destroy jid='alice@wonderland.lit'><reason>bert</reason></destroy></x>"));

            MUCUserPayload::ref payload = std::dynamic_pointer_cast<MUCUserPayload>(parser.getPayload());
            CPPUNIT_ASSERT(payload);
            MUCDestroyPayload::ref destroy = std::dynamic_pointer_cast<MUCDestroyPayload>(payload->getPayload());
            CPPUNIT_ASSERT(destroy);
            CPPUNIT_ASSERT_EQUAL(std::string("bert"), destroy->getReason());
            CPPUNIT_ASSERT_EQUAL(JID("alice@wonderland.lit"), destroy->getNewVenue());
        }

        void testParseInvite() {
            PayloadsParserTester parser;

            CPPUNIT_ASSERT(parser.parse("<x xmlns=\"http://jabber.org/protocol/muc#user\"><invite from='crone1@shakespeare.lit/desktop' to='alice@wonderland.lit/xxx'>      <reason>Hey Hecate, this is the place for all good witches!</reason>    </invite>    <password>cauldronburn</password></x>"));

            MUCUserPayload::ref payload = std::dynamic_pointer_cast<MUCUserPayload>(parser.getPayload());
            CPPUNIT_ASSERT(payload);
            CPPUNIT_ASSERT(payload->getInvite());
            CPPUNIT_ASSERT(payload->getPassword());
            CPPUNIT_ASSERT_EQUAL(std::string("cauldronburn"), *payload->getPassword());
            MUCUserPayload::Invite invite = *payload->getInvite();
            CPPUNIT_ASSERT_EQUAL(std::string("Hey Hecate, this is the place for all good witches!"), invite.reason);
            CPPUNIT_ASSERT_EQUAL(JID("crone1@shakespeare.lit/desktop"), invite.from);
            CPPUNIT_ASSERT_EQUAL(JID("alice@wonderland.lit/xxx"), invite.to);
        }

};

CPPUNIT_TEST_SUITE_REGISTRATION(MUCUserPayloadParserTest);