1 files changed, 29 insertions, 0 deletions

diff --git a/tests/querytest.cpp b/tests/querytest.cpp
index 714e549..3b62f56 100644
--- a/tests/querytest.cpp
+++ b/tests/querytest.cpp
@@ -250,6 +250,35 @@ private slots:
         QCOMPARE(model->rowCount(model->index(0, 0)), 1);
     }
 
+    void testIncrementalFolderTree()
+    {
+        // Setup
+        auto folder = ApplicationDomainType::createEntity<Folder>("sink.dummy.instance1");
+        VERIFYEXEC(Sink::Store::create<Folder>(folder));
+        // Ensure all local data is processed
+        VERIFYEXEC(Sink::ResourceControl::flushMessageQueue("sink.dummy.instance1"));
+
+        // Test
+        Sink::Query query{Sink::Query::LiveQuery};
+        query.resourceFilter("sink.dummy.instance1");
+        query.requestTree<Folder::Parent>();
+
+        // We fetch after the data is available and don't rely on the live query mechanism to deliver the actual data
+        auto model = Sink::Store::loadModel<Folder>(query);
+        QTRY_VERIFY(model->data(QModelIndex(), Sink::Store::ChildrenFetchedRole).toBool());
+        QCOMPARE(model->rowCount(), 1);
+        model->fetchMore(model->index(0, 0));
+
+        auto subfolder = ApplicationDomainType::createEntity<Folder>("sink.dummy.instance1");
+        subfolder.setParent(folder.identifier());
+        VERIFYEXEC(Sink::Store::create<Folder>(subfolder));
+        VERIFYEXEC(Sink::ResourceControl::flushMessageQueue("sink.dummy.instance1"));
+
+        model->fetchMore(model->index(0, 0));
+        QTRY_VERIFY(model->data(model->index(0, 0), Sink::Store::ChildrenFetchedRole).toBool());
+        QCOMPARE(model->rowCount(model->index(0, 0)), 1);
+    }
+
     void testMailByMessageId()
     {
         // Setup