#include <gtest/gtest.h>

#include <QJsonArray>
#include <QJsonObject>

#include "Event.h"
#include "RoomEvent.h"
#include "StateEvent.h"

#include "AliasesEventContent.h"
#include "AvatarEventContent.h"
#include "CanonicalAliasEventContent.h"
#include "CreateEventContent.h"
#include "HistoryVisibilityEventContent.h"
#include "JoinRulesEventContent.h"
#include "MemberEventContent.h"
#include "NameEventContent.h"
#include "PowerLevelsEventContent.h"
#include "TopicEventContent.h"

using namespace matrix::events;

TEST(EventCollection, Deserialize)
{
        auto events = QJsonArray{
                QJsonObject{ { "content", QJsonObject{ { "name", "Name" } } },
                             { "event_id", "$asdfafdf8af:matrix.org" },
                             { "prev_content", QJsonObject{ { "name", "Previous Name" } } },
                             { "room_id", "!aasdfaeae23r9:matrix.org" },
                             { "sender", "@alice:matrix.org" },
                             { "origin_server_ts", 1323238293289323LL },
                             { "state_key", "" },
                             { "type", "m.room.name" } },
                QJsonObject{ { "content", QJsonObject{ { "topic", "Topic" } } },
                             { "event_id", "$asdfafdf8af:matrix.org" },
                             { "prev_content", QJsonObject{ { "topic", "Previous Topic" } } },
                             { "room_id", "!aasdfaeae23r9:matrix.org" },
                             { "state_key", "" },
                             { "sender", "@alice:matrix.org" },
                             { "origin_server_ts", 1323238293289323LL },
                             { "type", "m.room.topic" } },
        };

        for (const auto &event : events) {
                EventType ty = extractEventType(event.toObject());

                if (ty == EventType::RoomName) {
                        StateEvent<NameEventContent> name_event;
                        name_event.deserialize(event);

                        EXPECT_EQ(name_event.content().name(), "Name");
                        EXPECT_EQ(name_event.previousContent().name(), "Previous Name");
                } else if (ty == EventType::RoomTopic) {
                        StateEvent<TopicEventContent> topic_event;
                        topic_event.deserialize(event);

                        EXPECT_EQ(topic_event.content().topic(), "Topic");
                        EXPECT_EQ(topic_event.previousContent().topic(), "Previous Topic");
                } else {
                        ASSERT_EQ(false, true);
                }
        }
}

TEST(EventCollection, DeserializationException)
{
        // Using wrong event types.
        auto events = QJsonArray{
                QJsonObject{ { "content", QJsonObject{ { "name", "Name" } } },
                             { "event_id", "$asdfafdf8af:matrix.org" },
                             { "prev_content", QJsonObject{ { "name", "Previous Name" } } },
                             { "room_id", "!aasdfaeae23r9:matrix.org" },
                             { "sender", "@alice:matrix.org" },
                             { "origin_server_ts", 1323238293289323LL },
                             { "state_key", "" },
                             { "type", "m.room.topic" } },
                QJsonObject{ { "content", QJsonObject{ { "topic", "Topic" } } },
                             { "event_id", "$asdfafdf8af:matrix.org" },
                             { "prev_content", QJsonObject{ { "topic", "Previous Topic" } } },
                             { "room_id", "!aasdfaeae23r9:matrix.org" },
                             { "state_key", "" },
                             { "sender", "@alice:matrix.org" },
                             { "origin_server_ts", 1323238293289323LL },
                             { "type", "m.room.name" } },
        };

        for (const auto &event : events) {
                EventType ty = extractEventType(event.toObject());

                if (ty == EventType::RoomName) {
                        StateEvent<NameEventContent> name_event;

                        try {
                                name_event.deserialize(event);
                        } catch (const DeserializationException &e) {
                                ASSERT_STREQ("name key is missing", e.what());
                        }

                } else if (ty == EventType::RoomTopic) {
                        StateEvent<TopicEventContent> topic_event;

                        try {
                                topic_event.deserialize(event);
                        } catch (const DeserializationException &e) {
                                ASSERT_STREQ("topic key is missing", e.what());
                        }
                } else {
                        ASSERT_EQ(false, true);
                }
        }
}