2 files changed, 39 insertions, 1 deletions

diff --git a/tests/mailsynctest.cpp b/tests/mailsynctest.cpp
index 9c57f0a..8260978 100644
--- a/tests/mailsynctest.cpp
+++ b/tests/mailsynctest.cpp
@@ -344,7 +344,43 @@ void MailSyncTest::testFetchNewRemovedMessages()
     }
 }
 
-//TODO test flag sync
+void MailSyncTest::testFlagChange()
+{
+    Sink::Query query;
+    query.resources << mResourceInstanceIdentifier;
+    query.request<Mail::Subject>().request<Mail::Important>();
+
+    auto msg = KMime::Message::Ptr::create();
+    msg->subject(true)->fromUnicodeString("Foobar", "utf8");
+    msg->assemble();
+    auto messageIdentifier = createMessage(QStringList() << "test", msg->encodedContent(true));
+
+    Store::synchronize(query).exec().waitForFinished();
+    ResourceControl::flushMessageQueue(query.resources).exec().waitForFinished();
+
+    {
+        auto job = Store::fetchAll<Mail>(query).syncThen<void, QList<Mail::Ptr>>([](const QList<Mail::Ptr> &mails) {
+            QCOMPARE(mails.size(), 2);
+            QVERIFY(!mails.at(1)->getImportant());
+        });
+        VERIFYEXEC(job);
+    }
+
+    markAsImportant(QStringList() << "test", messageIdentifier);
+
+    // Ensure all local data is processed
+    VERIFYEXEC(Store::synchronize(query));
+    ResourceControl::flushMessageQueue(query.resources).exec().waitForFinished();
+
+    {
+        auto job = Store::fetchAll<Mail>(query).syncThen<void, QList<Mail::Ptr>>([](const QList<Mail::Ptr> &mails) {
+            QCOMPARE(mails.size(), 2);
+            QVERIFY(mails.at(1)->getImportant());
+        });
+        VERIFYEXEC(job);
+    }
+
+}
 
 void MailSyncTest::testFailingSync()
 {
diff --git a/tests/mailsynctest.h b/tests/mailsynctest.h
index 31d3f03..94643f6 100644
--- a/tests/mailsynctest.h
+++ b/tests/mailsynctest.h
@@ -50,6 +50,7 @@ protected:
     virtual void removeFolder(const QStringList &folderPath) = 0;
     virtual QByteArray createMessage(const QStringList &folderPath, const QByteArray &message) = 0;
     virtual void removeMessage(const QStringList &folderPath, const QByteArray &messageIdentifier) = 0;
+    virtual void markAsImportant(const QStringList &folderPath, const QByteArray &messageIdentifier) = 0;
 
 private slots:
     void initTestCase();
@@ -66,6 +67,7 @@ private slots:
     void testListMails();
     void testResyncMails();
     void testFetchNewRemovedMessages();
+    void testFlagChange();
 
     void testFailingSync();
 };