/*
 * Copyright (c) 2010-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/Serializer/PayloadSerializers/SecurityLabelSerializer.h>

using namespace Swift;

class SecurityLabelSerializerTest : public CppUnit::TestFixture {
        CPPUNIT_TEST_SUITE(SecurityLabelSerializerTest);
        CPPUNIT_TEST(testSerialize);
        CPPUNIT_TEST(testSerialize_EmptyLabel);
        CPPUNIT_TEST_SUITE_END();

    public:
        void testSerialize() {
            SecurityLabelSerializer testling;
            std::shared_ptr<SecurityLabel> securityLabel(new SecurityLabel());
            securityLabel->setDisplayMarking("SECRET");
            securityLabel->setForegroundColor("black");
            securityLabel->setBackgroundColor("red");
            securityLabel->setLabel("<esssecuritylabel xmlns=\"urn:xmpp:sec-label:ess:0\">MQYCAQQGASk=</esssecuritylabel>");
            securityLabel->addEquivalentLabel("<icismlabel xmlns=\"http://example.gov/IC-ISM/0\" classification=\"S\" ownerProducer=\"USA\" disseminationControls=\"FOUO\"/>");
            securityLabel->addEquivalentLabel("<esssecuritylabel xmlns=\"urn:xmpp:sec-label:ess:0\">MRUCAgD9DA9BcXVhIChvYnNvbGV0ZSk=</esssecuritylabel>");

            CPPUNIT_ASSERT_EQUAL(std::string(
                "<securitylabel xmlns=\"urn:xmpp:sec-label:0\">"
                    "<displaymarking bgcolor=\"red\" fgcolor=\"black\">SECRET</displaymarking>"
                    "<label>"
                        "<esssecuritylabel xmlns=\"urn:xmpp:sec-label:ess:0\">MQYCAQQGASk=</esssecuritylabel>"
                    "</label>"
                    "<equivalentlabel>"
                        "<icismlabel xmlns=\"http://example.gov/IC-ISM/0\" classification=\"S\" ownerProducer=\"USA\" disseminationControls=\"FOUO\"/>"
                    "</equivalentlabel>"
                    "<equivalentlabel>"
                        "<esssecuritylabel xmlns=\"urn:xmpp:sec-label:ess:0\">MRUCAgD9DA9BcXVhIChvYnNvbGV0ZSk=</esssecuritylabel>"
                    "</equivalentlabel>"
                "</securitylabel>"), testling.serialize(securityLabel));
        }

        void testSerialize_EmptyLabel() {
            SecurityLabelSerializer testling;
            std::shared_ptr<SecurityLabel> securityLabel(new SecurityLabel());
            securityLabel->setDisplayMarking("SECRET");
            securityLabel->setLabel("");

            CPPUNIT_ASSERT_EQUAL(std::string(
                "<securitylabel xmlns=\"urn:xmpp:sec-label:0\">"
                    "<displaymarking>SECRET</displaymarking>"
                    "<label></label>"
                "</securitylabel>"), testling.serialize(securityLabel));
        }
};

CPPUNIT_TEST_SUITE_REGISTRATION(SecurityLabelSerializerTest);