Add support for calendar in the teststore

2 files changed, 71 insertions, 0 deletions

diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index 35b28559..d7dfe81d 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -1,6 +1,7 @@
 find_package(Qt5 REQUIRED NO_MODULE COMPONENTS QuickTest Network Quick)
 find_package(Sink CONFIG REQUIRED)
 find_package(KAsync CONFIG REQUIRED)
+find_package(KF5CalendarCore CONFIG REQUIRED)
 
 add_executable(kubetestrunner kubetestrunner.cpp)
 target_link_libraries(kubetestrunner
@@ -15,5 +16,6 @@ install(FILES qmldir DESTINATION ${QML_INSTALL_DIR}/org/kube/test)
 add_library(testplugin SHARED testplugin.cpp teststore.cpp)
 target_link_libraries(testplugin
     kubeframework
+    KF5::CalendarCore
 )
 install(TARGETS testplugin DESTINATION ${QML_INSTALL_DIR}/org/kube/test)
diff --git a/tests/teststore.cpp b/tests/teststore.cpp
index 6a5304ad..902014cd 100644
--- a/tests/teststore.cpp
+++ b/tests/teststore.cpp
@@ -21,9 +21,14 @@
 #include <sink/store.h>
 #include <sink/resourcecontrol.h>
 #include <sink/secretstore.h>
+
 #include <kmime/kmime_message.h>
 
+#include <KCalCore/Event>
+#include <KCalCore/ICalFormat>
+
 #include <QDebug>
+#include <QUuid>
 #include <QVariant>
 
 #include "framework/src/domain/mime/mailtemplates.h"
@@ -124,6 +129,65 @@ static void createFolder(const QVariantMap &object)
     });
 }
 
+static void createEvent(const QVariantMap &object, const QByteArray &calendarId = {})
+{
+    using Sink::ApplicationDomain::ApplicationDomainType;
+    using Sink::ApplicationDomain::Event;
+
+    auto sinkEvent = ApplicationDomainType::createEntity<Event>(object["resource"].toByteArray());
+
+    auto calcoreEvent = QSharedPointer<KCalCore::Event>::create();
+
+    if (object.contains("uid")) {
+        auto uid = object["uid"].toString();
+        sinkEvent.setUid(uid);
+        calcoreEvent->setUid(uid);
+    } else {
+        auto uid = QUuid::createUuid().toString();
+        sinkEvent.setUid(uid);
+        calcoreEvent->setUid(uid);
+    }
+
+    auto summary = object["summary"].toString();
+    sinkEvent.setSummary(summary);
+    calcoreEvent->setSummary(summary);
+
+    if (object.contains("description")) {
+        auto description = object["description"].toString();
+        sinkEvent.setDescription(description);
+        calcoreEvent->setDescription(description);
+    }
+
+    auto startTime = object["starts"].toDateTime();
+    auto endTime = object["ends"].toDateTime();
+    sinkEvent.setStartTime(startTime);
+    sinkEvent.setEndTime(endTime);
+
+    calcoreEvent->setDtStart(startTime);
+    calcoreEvent->setDtEnd(endTime);
+
+    auto ical = KCalCore::ICalFormat().toRawString(static_cast<QSharedPointer<KCalCore::Incidence>>(calcoreEvent));
+    sinkEvent.setIcal(ical);
+
+    sinkEvent.setCalendar(calendarId);
+
+    Sink::Store::create(sinkEvent).exec().waitForFinished();
+}
+
+static void createCalendar(const QVariantMap &object)
+{
+    using Sink::ApplicationDomain::Calendar;
+    using Sink::ApplicationDomain::ApplicationDomainType;
+
+    auto calendar = ApplicationDomainType::createEntity<Calendar>(object["resource"].toByteArray());
+    calendar.setName(object["name"].toString());
+    Sink::Store::create(calendar).exec().waitForFinished();
+
+    auto calendarId = calendar.identifier();
+    iterateOverObjects(object.value("events").toList(),
+        [calendarId](const QVariantMap &object) { createEvent(object, calendarId); });
+}
+
 void TestStore::setup(const QVariantMap &map)
 {
     using namespace Sink::ApplicationDomain;
@@ -143,6 +207,9 @@ void TestStore::setup(const QVariantMap &map)
             } else if (object["type"] == "mailtransport") {
                 resource.setResourceType("sink.mailtransport");
                 resource.setProperty("testmode", true);
+            } else if (object["type"] == "caldav") {
+                resource.setResourceType("sink.caldav");
+                resource.setProperty("testmode", true);
             } else {
                 Q_ASSERT(false);
             }
@@ -166,6 +233,8 @@ void TestStore::setup(const QVariantMap &map)
         createMail(map);
     });
 
+    iterateOverObjects(map.value("calendars").toList(), createCalendar);
+
     Sink::ResourceControl::flushMessageQueue(resources).exec().waitForFinished();
 }