7 files changed, 132 insertions, 266 deletions

diff --git a/common/clientapi.cpp b/common/clientapi.cpp
index f99ebb8..839e77b 100644
--- a/common/clientapi.cpp
+++ b/common/clientapi.cpp
@@ -95,7 +95,7 @@ KAsync::Job<void> Store::synchronize(const Akonadi2::Query &query)
     .template each<void, QByteArray>([query](const QByteArray &resource, KAsync::Future<void> &future) {
         auto resourceAccess = QSharedPointer<Akonadi2::ResourceAccess>::create(resource);
         resourceAccess->open();
-        resourceAccess->synchronizeResource(true, false).then<void>([&future]() {
+        resourceAccess->synchronizeResource(query.syncOnDemand, query.processAll).then<void>([&future, resourceAccess]() {
             future.setFinished();
         }).exec();
     })
diff --git a/common/facade.h b/common/facade.h
index eb55c98..5be1c73 100644
--- a/common/facade.h
+++ b/common/facade.h
@@ -44,19 +44,15 @@ class QueryRunner : public QObject
 {
     Q_OBJECT
 public:
-    typedef std::function<KAsync::Job<qint64>(qint64 oldRevision)> QueryFunction;
+    typedef std::function<KAsync::Job<void>()> QueryFunction;
 
-    QueryRunner(const Akonadi2::Query &query) : mLatestRevision(0) {};
+    QueryRunner(const Akonadi2::Query &query) {};
     /**
      * Starts query
      */
     KAsync::Job<void> run(qint64 newRevision = 0)
     {
-        //TODO: JOBAPI: that last empty .then should not be necessary
-        //TODO: remove newRevision
-        return queryFunction(mLatestRevision + 1).then<void, qint64>([this](qint64 revision) {
-            mLatestRevision = revision;
-        }).then<void>([](){});
+        return queryFunction();
     }
 
     /**
@@ -74,12 +70,11 @@ public slots:
     void revisionChanged(qint64 newRevision)
     {
         Trace() << "New revision: " << newRevision;
-        run(newRevision).exec();
+        run().exec();
     }
 
 private:
     QueryFunction queryFunction;
-    qint64 mLatestRevision;
 };
 
 static inline ResultSet fullScan(const Akonadi2::Storage::Transaction &transaction, const QByteArray &bufferType)
@@ -125,10 +120,9 @@ public:
      * @param resourceIdentifier is the identifier of the resource instance
      * @param adaptorFactory is the adaptor factory used to generate the mappings from domain to resource types and vice versa
      */
-    GenericFacade(const QByteArray &resourceIdentifier, const DomainTypeAdaptorFactoryInterface::Ptr &adaptorFactory = DomainTypeAdaptorFactoryInterface::Ptr(), const QSharedPointer<EntityStorage<DomainType> > storage = QSharedPointer<EntityStorage<DomainType> >(), const QSharedPointer<Akonadi2::ResourceAccessInterface> resourceAccess = QSharedPointer<Akonadi2::ResourceAccessInterface>())
+    GenericFacade(const QByteArray &resourceIdentifier, const DomainTypeAdaptorFactoryInterface::Ptr &adaptorFactory = DomainTypeAdaptorFactoryInterface::Ptr(), const QSharedPointer<Akonadi2::ResourceAccessInterface> resourceAccess = QSharedPointer<Akonadi2::ResourceAccessInterface>())
         : Akonadi2::StoreFacade<DomainType>(),
         mResourceAccess(resourceAccess),
-        mStorage(storage),
         mDomainTypeAdaptorFactory(adaptorFactory),
         mResourceInstanceIdentifier(resourceIdentifier)
     {
@@ -177,48 +171,28 @@ public:
     //TODO JOBAPI return job from sync continuation to execute it as subjob?
     KAsync::Job<void> load(const Akonadi2::Query &query, const QSharedPointer<Akonadi2::ResultProviderInterface<typename DomainType::Ptr> > &resultProvider) Q_DECL_OVERRIDE
     {
-        {
-            QSet<QByteArray> remainingFilters;
-            auto filter = [remainingFilters, query](const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &domainObject) -> bool {
-                for (const auto &filterProperty : remainingFilters) {
-                    //TODO implement other comparison operators than equality
-                    if (domainObject->getProperty(filterProperty) != query.propertyFilter.value(filterProperty)) {
-                        return false;
-                    }
-                }
-                return true;
-            };
-
-            auto fetchEntities = [this, query, resultProvider, filter](const QByteArray &parent) {
-                Trace() << "Running fetchEntities" << parent;
-                Akonadi2::Storage storage(Akonadi2::storageLocation(), mResourceInstanceIdentifier);
-                storage.setDefaultErrorHandler([](const Akonadi2::Storage::Error &error) {
-                    Warning() << "Error during query: " << error.store << error.message;
-                });
+        auto fetchEntities = [this, query, resultProvider](const QByteArray &parent) {
+            Trace() << "Fetching initial set for parent:" << parent;
 
-                auto transaction = storage.createTransaction(Akonadi2::Storage::ReadOnly);
+            Akonadi2::Storage storage(Akonadi2::storageLocation(), mResourceInstanceIdentifier);
+            storage.setDefaultErrorHandler([](const Akonadi2::Storage::Error &error) {
+                Warning() << "Error during query: " << error.store << error.message;
+            });
 
-                auto modifiedQuery = query;
-                modifiedQuery.propertyFilter.insert("parent", parent);
-                //TODO
-                QSet<QByteArray> appliedFilters;
-                auto resultSet = Akonadi2::ApplicationDomain::TypeImplementation<DomainType>::queryIndexes(modifiedQuery, mResourceInstanceIdentifier, appliedFilters, transaction);
-                QSet<QByteArray> remainingFilters = query.propertyFilter.keys().toSet() - appliedFilters;
+            auto transaction = storage.createTransaction(Akonadi2::Storage::ReadOnly);
 
-                //We do a full scan if there were no indexes available to create the initial set.
-                if (appliedFilters.isEmpty()) {
-                    //TODO this should be replaced by an index lookup as well
-                    resultSet = fullScan(transaction, bufferTypeForDomainType());
-                }
-                auto filteredSet = filterSet(resultSet, filter, transaction, true);
-                replaySet(filteredSet, resultProvider);
-                resultProvider->setRevision(Akonadi2::Storage::maxRevision(transaction));
-                qint64 newRevision = Akonadi2::Storage::maxRevision(transaction);
-                //TODO send newRevision to resource
-                // mResourceAccess->sendRevisionReplayedCommand(newRevision);
-            };
-            resultProvider->setFetcher(fetchEntities);
-        }
+            auto modifiedQuery = query;
+            modifiedQuery.propertyFilter.insert("parent", parent);
+
+            QSet<QByteArray> remainingFilters;
+            auto resultSet = loadInitialResultSet(parent, modifiedQuery, transaction, remainingFilters);
+            auto filteredSet = filterSet(resultSet, getFilter(remainingFilters, query), transaction, true);
+            replaySet(filteredSet, resultProvider);
+            const qint64 newRevision = Akonadi2::Storage::maxRevision(transaction);
+            resultProvider->setRevision(newRevision);
+            mResourceAccess->sendRevisionReplayedCommand(newRevision);
+        };
+        resultProvider->setFetcher(fetchEntities);
 
         auto runner = QSharedPointer<QueryRunner>::create(query);
         QWeakPointer<Akonadi2::ResultProviderInterface<typename DomainType::Ptr> > weakResultProvider = resultProvider;
@@ -233,8 +207,6 @@ public:
                     return;
                 }
                 executeQuery(query, resultProvider).template then<void, qint64>([&future, this](qint64 queriedRevision) {
-                    //TODO set revision in result provider?
-                    //TODO update all existing results with new revision
                     mResourceAccess->sendRevisionReplayedCommand(queriedRevision);
                     future.setFinished();
                 }).exec();
@@ -249,27 +221,12 @@ public:
             mResourceAccess->open();
             QObject::connect(mResourceAccess.data(), &Akonadi2::ResourceAccess::revisionChanged, runner.data(), &QueryRunner::revisionChanged);
         }
+        return KAsync::null<void>();
 
         //We have to capture the runner to keep it alive
-        return synchronizeResource(query).template then<void>([runner](KAsync::Future<void> &future) {
-            future.setFinished();
-        },
-        [](int error, const QString &errorString) {
-            Warning() << "Error during sync " << error << errorString;
-        });
     }
 
 private:
-    KAsync::Job<void> synchronizeResource(const Akonadi2::Query &query)
-    {
-        //TODO check if a sync is necessary
-        //TODO Only sync what was requested
-        //TODO timeout
-        if (query.syncOnDemand || query.processAll) {
-            return mResourceAccess->synchronizeResource(query.syncOnDemand, query.processAll);
-        }
-        return KAsync::null<void>();
-    }
 
     //TODO move into result provider?
     void replaySet(ResultSet &resultSet, const QSharedPointer<Akonadi2::ResultProviderInterface<typename DomainType::Ptr> > &resultProvider)
@@ -280,17 +237,14 @@ private:
                 Trace() << "Got creation";
                 //TODO Only copy in result provider
                 resultProvider->add(Akonadi2::ApplicationDomain::ApplicationDomainType::getInMemoryRepresentation<DomainType>(*value).template staticCast<DomainType>());
-                // modelResult->add();
                 break;
             case Akonadi2::Operation_Modification:
                 Trace() << "Got modification";
                 resultProvider->modify(Akonadi2::ApplicationDomain::ApplicationDomainType::getInMemoryRepresentation<DomainType>(*value).template staticCast<DomainType>());
-                // modelResult->modify();
                 break;
             case Akonadi2::Operation_Removal:
                 Trace() << "Got removal";
                 resultProvider->remove(Akonadi2::ApplicationDomain::ApplicationDomainType::getInMemoryRepresentation<DomainType>(*value).template staticCast<DomainType>());
-                // modelResult->remove();
                 break;
             }
             return true;
@@ -320,13 +274,28 @@ private:
         });
     }
 
-    ResultSet loadIncrementalResultSet(qint64 baseRevision, const Akonadi2::Query &query, Akonadi2::Storage::Transaction &transaction, QSet<QByteArray> &remainingFilters)
+    ResultSet loadInitialResultSet(const QByteArray &parent, const Akonadi2::Query &query, Akonadi2::Storage::Transaction &transaction, QSet<QByteArray> &remainingFilters)
     {
+        Trace() << "Fetching initial set for parent:" << parent;
+        //TODO
+        QSet<QByteArray> appliedFilters;
+        auto resultSet = Akonadi2::ApplicationDomain::TypeImplementation<DomainType>::queryIndexes(query, mResourceInstanceIdentifier, appliedFilters, transaction);
+        remainingFilters = query.propertyFilter.keys().toSet() - appliedFilters;
+
+        //We do a full scan if there were no indexes available to create the initial set.
+        if (appliedFilters.isEmpty()) {
+            //TODO this should be replaced by an index lookup as well
+            resultSet = fullScan(transaction, bufferTypeForDomainType());
+        }
+        return resultSet;
+    }
 
+    ResultSet loadIncrementalResultSet(qint64 baseRevision, const Akonadi2::Query &query, Akonadi2::Storage::Transaction &transaction, QSet<QByteArray> &remainingFilters)
+    {
+        Trace() << "Loading incremental result set starting from revision: " << baseRevision;
         const auto bufferType = bufferTypeForDomainType();
         auto revisionCounter = QSharedPointer<qint64>::create(baseRevision);
-        //TODO apply filter from index
-        return ResultSet([bufferType, revisionCounter, &transaction, this]() -> QByteArray {
+        return ResultSet([bufferType, revisionCounter, &transaction]() -> QByteArray {
             const qint64 topRevision = Akonadi2::Storage::maxRevision(transaction);
             //Spit out the revision keys one by one.
             while (*revisionCounter <= topRevision) {
@@ -354,7 +323,7 @@ private:
         //Read through the source values and return whatever matches the filter
         std::function<bool(std::function<void(const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &, Akonadi2::Operation)>)> generator = [this, resultSetPtr, &transaction, filter, initialQuery](std::function<void(const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &, Akonadi2::Operation)> callback) -> bool {
             while (resultSetPtr->next()) {
-                //TODO only necessary if we actually want to filter or neew the operation type (but not a big deal if we do it always I guess)
+                //readEntity is only necessary if we actually want to filter or know the operation type (but not a big deal if we do it always I guess)
                 readEntity(transaction, resultSetPtr->id(), [this, filter, callback, initialQuery](const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &domainObject, Akonadi2::Operation operation) {
                     //Always remove removals, they probably don't match due to non-available properties
                     if (filter(domainObject) || operation == Akonadi2::Operation_Removal) {
@@ -374,6 +343,20 @@ private:
         return ResultSet(generator);
     }
 
+
+    std::function<bool(const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &domainObject)> getFilter(const QSet<QByteArray> remainingFilters, const Akonadi2::Query &query)
+    {
+        return [remainingFilters, query](const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &domainObject) -> bool {
+            for (const auto &filterProperty : remainingFilters) {
+                //TODO implement other comparison operators than equality
+                if (domainObject->getProperty(filterProperty) != query.propertyFilter.value(filterProperty)) {
+                    return false;
+                }
+            }
+            return true;
+        };
+    }
+
     virtual KAsync::Job<qint64> executeQuery(const Akonadi2::Query &query, const QSharedPointer<Akonadi2::ResultProviderInterface<typename DomainType::Ptr> > &resultProvider)
     {
         /*
@@ -384,16 +367,6 @@ private:
         const qint64 baseRevision = resultProvider->revision() + 1;
         Trace() << "Running query " << baseRevision;
         QSet<QByteArray> remainingFilters;
-        auto filter = [remainingFilters, query, baseRevision](const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &domainObject) -> bool {
-            for (const auto &filterProperty : remainingFilters) {
-                //TODO implement other comparison operators than equality
-                if (domainObject->getProperty(filterProperty) != query.propertyFilter.value(filterProperty)) {
-                    return false;
-                }
-            }
-            return true;
-        };
-        qint64 newRevision = 0;
 
         Trace() << "Fetching updates";
         Akonadi2::Storage storage(Akonadi2::storageLocation(), mResourceInstanceIdentifier);
@@ -404,10 +377,10 @@ private:
         auto transaction = storage.createTransaction(Akonadi2::Storage::ReadOnly);
 
         auto resultSet = loadIncrementalResultSet(baseRevision, query, transaction, remainingFilters);
-        auto filteredSet = filterSet(resultSet, filter, transaction, false);
+        auto filteredSet = filterSet(resultSet, getFilter(remainingFilters, query), transaction, false);
         replaySet(filteredSet, resultProvider);
         resultProvider->setRevision(Akonadi2::Storage::maxRevision(transaction));
-        newRevision = Akonadi2::Storage::maxRevision(transaction);
+        qint64 newRevision = Akonadi2::Storage::maxRevision(transaction);
 
         return KAsync::start<qint64>([=]() -> qint64 {
             return newRevision;
diff --git a/common/modelresult.h b/common/modelresult.h
index 756f4d6..eabb868 100644
--- a/common/modelresult.h
+++ b/common/modelresult.h
@@ -111,6 +111,10 @@ public:
     {
         auto childId = qHash(value->identifier());
         auto id = parentId(value);
+        //Ignore updates we get before the initial fetch is done
+        if (!mEntityChildrenFetched[id]) {
+            return;
+        }
         auto parent = createIndexFromId(id);
         qDebug() << "Added entity " << childId;
         const auto keys = mTree[id];
@@ -131,6 +135,10 @@ public:
     {
         auto childId = qHash(value->identifier());
         auto id = parentId(value);
+        //Ignore updates we get before the initial fetch is done
+        if (!mEntityChildrenFetched[id]) {
+            return;
+        }
         auto parent = createIndexFromId(id);
         qDebug() << "Modified entity" << childId;
         auto i = mTree[id].indexOf(childId);
diff --git a/common/resultprovider.h b/common/resultprovider.h
index 43d21a4..0d23127 100644
--- a/common/resultprovider.h
+++ b/common/resultprovider.h
@@ -23,6 +23,9 @@
 #include <functional>
 #include <memory>
 #include "threadboundary.h"
+#include "resultset.h"
+#include "log.h"
+#include "modelresult.h"
 
 using namespace async;
 
@@ -117,18 +120,6 @@ public:
         // mResultEmitter->clear();
     }
 
-    // QSharedPointer<ResultEmitter<T> > emitter()
-    // {
-    //     if (!mResultEmitter) {
-    //         //We have to go over a separate var and return that, otherwise we'd delete the emitter immediately again
-    //         auto sharedPtr = QSharedPointer<ResultEmitter<T> >(new ResultEmitter<T>, [this](ResultEmitter<T> *emitter){ done(); delete emitter; });
-    //         mResultEmitter = sharedPtr;
-    //         return sharedPtr;
-    //     }
-    //
-    //     return mResultEmitter.toStrongRef();
-    // }
-
     /**
      * For lifetimemanagement only.
      * We keep the runner alive as long as the result provider exists.
@@ -162,40 +153,6 @@ public:
         mQueryRunner = runner;
     }
 
-    // qint64 fetch(const ResultSet &resultSet)
-    // {
-    //     //Fetch a bunch
-    //     //
-    //     // Akonadi2::Storage storage(Akonadi2::storageLocation(), mResourceInstanceIdentifier);
-    //     // storage.setDefaultErrorHandler([](const Akonadi2::Storage::Error &error) {
-    //     //     Warning() << "Error during query: " << error.store << error.message;
-    //     // });
-    //     //
-    //     // auto transaction = storage.createTransaction(Akonadi2::Storage::ReadOnly);
-    //
-    //     // Log() << "Querying" << baseRevision << Akonadi2::Storage::maxRevision(transaction);
-    //     // auto resultSet = getResultSet(query, transaction, baseRevision);
-    //     while (resultSet.next([this](const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &value, Akonadi2::Operation operation) -> bool {
-    //         switch (operation) {
-    //         case Akonadi2::Operation_Creation:
-    //             Trace() << "Got creation";
-    //             //TODO Only copy in result provider
-    //             add(Akonadi2::ApplicationDomain::ApplicationDomainType::getInMemoryRepresentation<T>(*value).template staticCast<T>());
-    //             break;
-    //         case Akonadi2::Operation_Modification:
-    //             Trace() << "Got modification";
-    //             modify(Akonadi2::ApplicationDomain::ApplicationDomainType::getInMemoryRepresentation<T>(*value).template staticCast<T>());
-    //             break;
-    //         case Akonadi2::Operation_Removal:
-    //             Trace() << "Got removal";
-    //             remove(Akonadi2::ApplicationDomain::ApplicationDomainType::getInMemoryRepresentation<T>(*value).template staticCast<T>());
-    //             break;
-    //         }
-    //         return true;
-    //     })){};
-    //     // return Akonadi2::Storage::maxRevision(transaction);
-    // }
-
 private:
     void done()
     {
@@ -212,129 +169,6 @@ private:
     std::function<void()> mOnDoneCallback;
 };
 
-
-
-
-
-
-template<class T>
-class SyncResultProvider : public ResultProviderInterface<T> {
-public:
-    void add(const T &value)
-    {
-        mResultEmitter->addHandler(value);
-    }
-
-    void modify(const T &value)
-    {
-        mResultEmitter->modifyHandler(value);
-    }
-
-    void remove(const T &value)
-    {
-        mResultEmitter->removeHandler(value);
-    }
-
-    void initialResultSetComplete()
-    {
-        mResultEmitter->initialResultSetComplete();
-    }
-
-    void complete()
-    {
-        mResultEmitter->complete();
-    }
-
-    void clear()
-    {
-        mResultEmitter->clear();
-    }
-
-    QSharedPointer<ResultEmitter<T> > emitter()
-    {
-        if (!mResultEmitter) {
-            //We have to go over a separate var and return that, otherwise we'd delete the emitter immediately again
-            auto sharedPtr = QSharedPointer<ResultEmitter<T> >(new ResultEmitter<T>, [this](ResultEmitter<T> *emitter){ done(); delete emitter; });
-            mResultEmitter = sharedPtr;
-            return sharedPtr;
-        }
-
-        return mResultEmitter.toStrongRef();
-    }
-
-    /**
-     * For lifetimemanagement only.
-     * We keep the runner alive as long as the result provider exists.
-     */
-    void setFacade(const std::shared_ptr<void> &facade)
-    {
-        mFacade = facade;
-    }
-
-    void onDone(const std::function<void()> &callback)
-    {
-        mOnDoneCallback = callback;
-    }
-
-    bool isDone() const
-    {
-        //The existance of the emitter currently defines wether we're done or not.
-        return mResultEmitter.toStrongRef().isNull();
-    }
-
-    // qint64 fetch(const ResultSet &resultSet)
-    // {
-    //     //Fetch a bunch
-    //     //
-    //     // Akonadi2::Storage storage(Akonadi2::storageLocation(), mResourceInstanceIdentifier);
-    //     // storage.setDefaultErrorHandler([](const Akonadi2::Storage::Error &error) {
-    //     //     Warning() << "Error during query: " << error.store << error.message;
-    //     // });
-    //     //
-    //     // auto transaction = storage.createTransaction(Akonadi2::Storage::ReadOnly);
-    //
-    //     // Log() << "Querying" << baseRevision << Akonadi2::Storage::maxRevision(transaction);
-    //     // auto resultSet = getResultSet(query, transaction, baseRevision);
-    //     while (resultSet.next([this](const Akonadi2::ApplicationDomain::ApplicationDomainType::Ptr &value, Akonadi2::Operation operation) -> bool {
-    //         switch (operation) {
-    //         case Akonadi2::Operation_Creation:
-    //             Trace() << "Got creation";
-    //             //TODO Only copy in result provider
-    //             add(Akonadi2::ApplicationDomain::ApplicationDomainType::getInMemoryRepresentation<T>(*value).template staticCast<T>());
-    //             break;
-    //         case Akonadi2::Operation_Modification:
-    //             Trace() << "Got modification";
-    //             modify(Akonadi2::ApplicationDomain::ApplicationDomainType::getInMemoryRepresentation<T>(*value).template staticCast<T>());
-    //             break;
-    //         case Akonadi2::Operation_Removal:
-    //             Trace() << "Got removal";
-    //             remove(Akonadi2::ApplicationDomain::ApplicationDomainType::getInMemoryRepresentation<T>(*value).template staticCast<T>());
-    //             break;
-    //         }
-    //         return true;
-    //     })){};
-    //     // return Akonadi2::Storage::maxRevision(transaction);
-    // }
-
-private:
-    void done()
-    {
-        qWarning() << "done";
-        if (mOnDoneCallback) {
-            mOnDoneCallback();
-            mOnDoneCallback = std::function<void()>();
-        }
-    }
-
-    QWeakPointer<ResultEmitter<T> > mResultEmitter;
-    std::shared_ptr<void> mFacade;
-    std::function<void()> mOnDoneCallback;
-    QSharedPointer<ThreadBoundary> mThreadBoundary;
-};
-
-
-
-
 /*
 * The promise side for the result emitter
 */
@@ -434,18 +268,18 @@ public:
     }
 
     /**
-        * For lifetimemanagement only.
-        * We keep the runner alive as long as the result provider exists.
-        */
+     * For lifetimemanagement only.
+     * We keep the runner alive as long as the result provider exists.
+     */
     void setQueryRunner(const QSharedPointer<QObject> &runner)
     {
         mQueryRunner = runner;
     }
 
     /**
-        * For lifetimemanagement only.
-        * We keep the runner alive as long as the result provider exists.
-        */
+     * For lifetimemanagement only.
+     * We keep the runner alive as long as the result provider exists.
+     */
     void setFacade(const std::shared_ptr<void> &facade)
     {
         mFacade = facade;
@@ -463,6 +297,11 @@ public:
         return mResultEmitter.toStrongRef().isNull();
     }
 
+    void setFetcher(const std::function<void(const QByteArray &parent)> &fetcher)
+    {
+        fetcher(QByteArray());
+    }
+
 private:
     void done()
     {
diff --git a/tests/dummyresourcetest.cpp b/tests/dummyresourcetest.cpp
index d027266..3b90e6c 100644
--- a/tests/dummyresourcetest.cpp
+++ b/tests/dummyresourcetest.cpp
@@ -64,6 +64,9 @@ private Q_SLOTS:
         query.syncOnDemand = false;
         query.processAll = true;
 
+        //Ensure all local data is processed
+        Akonadi2::Store::synchronize(query).exec().waitForFinished();
+
         query.propertyFilter.insert("uid", "testuid");
         async::SyncListResult<Akonadi2::ApplicationDomain::Event::Ptr> result(Akonadi2::Store::load<Akonadi2::ApplicationDomain::Event>(query));
         result.exec();
@@ -88,6 +91,9 @@ private Q_SLOTS:
         query.syncOnDemand = false;
         query.processAll = true;
 
+        //Ensure all local data is processed
+        Akonadi2::Store::synchronize(query).exec().waitForFinished();
+
         query.propertyFilter.insert("uid", "testuid");
         async::SyncListResult<Akonadi2::ApplicationDomain::Event::Ptr> result(Akonadi2::Store::load<Akonadi2::ApplicationDomain::Event>(query));
         result.exec();
@@ -114,6 +120,9 @@ private Q_SLOTS:
         query.syncOnDemand = false;
         query.processAll = true;
 
+        //Ensure all local data is processed
+        Akonadi2::Store::synchronize(query).exec().waitForFinished();
+
         query.propertyFilter.insert("summary", "summaryValue2");
         async::SyncListResult<Akonadi2::ApplicationDomain::Event::Ptr> result(Akonadi2::Store::load<Akonadi2::ApplicationDomain::Event>(query));
         result.exec();
@@ -145,6 +154,9 @@ private Q_SLOTS:
         query.syncOnDemand = true;
         query.processAll = true;
 
+        //Ensure all local data is processed
+        Akonadi2::Store::synchronize(query).exec().waitForFinished();
+
         async::SyncListResult<Akonadi2::ApplicationDomain::Event::Ptr> result(Akonadi2::Store::load<Akonadi2::ApplicationDomain::Event>(query));
         result.exec();
         QVERIFY(!result.isEmpty());
@@ -160,6 +172,9 @@ private Q_SLOTS:
         query.syncOnDemand = true;
         query.processAll = true;
 
+        //Ensure all local data is processed
+        Akonadi2::Store::synchronize(query).exec().waitForFinished();
+
         async::SyncListResult<Akonadi2::ApplicationDomain::Mail::Ptr> result(Akonadi2::Store::load<Akonadi2::ApplicationDomain::Mail>(query));
         result.exec();
         QVERIFY(!result.isEmpty());
@@ -182,6 +197,9 @@ private Q_SLOTS:
         query.processAll = true;
         query.propertyFilter.insert("uid", "testuid");
 
+        //Ensure all local data is processed
+        Akonadi2::Store::synchronize(query).exec().waitForFinished();
+
         //Test create
         Akonadi2::ApplicationDomain::Event event2;
         {
@@ -198,6 +216,9 @@ private Q_SLOTS:
         event2.setProperty("summary", "summaryValue2");
         Akonadi2::Store::modify<Akonadi2::ApplicationDomain::Event>(event2).exec().waitForFinished();
 
+        //Ensure all local data is processed
+        Akonadi2::Store::synchronize(query).exec().waitForFinished();
+
         //Test modify
         {
             async::SyncListResult<Akonadi2::ApplicationDomain::Event::Ptr> result(Akonadi2::Store::load<Akonadi2::ApplicationDomain::Event>(query));
@@ -210,6 +231,9 @@ private Q_SLOTS:
 
         Akonadi2::Store::remove<Akonadi2::ApplicationDomain::Event>(event2).exec().waitForFinished();
 
+        //Ensure all local data is processed
+        Akonadi2::Store::synchronize(query).exec().waitForFinished();
+
         //Test remove
         {
             async::SyncListResult<Akonadi2::ApplicationDomain::Event::Ptr> result(Akonadi2::Store::load<Akonadi2::ApplicationDomain::Event>(query));
diff --git a/tests/querytest.cpp b/tests/querytest.cpp
index 9f4b3bb..47d977b 100644
--- a/tests/querytest.cpp
+++ b/tests/querytest.cpp
@@ -57,11 +57,33 @@ private Q_SLOTS:
         query.syncOnDemand = false;
         query.processAll = true;
 
-        auto model = new ModelResult<Akonadi2::ApplicationDomain::Mail>(query, QList<QByteArray>() << "summary" << "uid");
+        auto model = Akonadi2::Store::loadModel<Akonadi2::ApplicationDomain::Mail>(query);
         model->fetchMore(QModelIndex());
         QTRY_COMPARE(model->rowCount(), 1);
     }
 
+    void testSingleWithDelay()
+    {
+        //Setup
+        {
+            Akonadi2::ApplicationDomain::Mail mail("org.kde.dummy.instance1");
+            Akonadi2::Store::create<Akonadi2::ApplicationDomain::Mail>(mail).exec().waitForFinished();
+        }
+
+        //Test
+        Akonadi2::Query query;
+        query.resources << "org.kde.dummy.instance1";
+        query.syncOnDemand = false;
+        query.processAll = true;
+        query.liveQuery = true;
+
+        auto model = Akonadi2::Store::loadModel<Akonadi2::ApplicationDomain::Mail>(query);
+        QTest::qWait(200);
+        model->fetchMore(QModelIndex());
+        QVERIFY(model->rowCount() < 2);
+        QTRY_COMPARE(model->rowCount(), 1);
+    }
+
     // void testTree()
     // {
     //     //Setup
diff --git a/tests/testimplementations.h b/tests/testimplementations.h
index eee78b0..1436c68 100644
--- a/tests/testimplementations.h
+++ b/tests/testimplementations.h
@@ -85,8 +85,8 @@ public Q_SLOTS:
 class TestResourceFacade : public Akonadi2::GenericFacade<Akonadi2::ApplicationDomain::Event>
 {
 public:
-    TestResourceFacade(const QByteArray &instanceIdentifier, const QSharedPointer<EntityStorage<Akonadi2::ApplicationDomain::Event> > storage, const QSharedPointer<Akonadi2::ResourceAccessInterface> resourceAccess)
-        : Akonadi2::GenericFacade<Akonadi2::ApplicationDomain::Event>(instanceIdentifier, QSharedPointer<TestEventAdaptorFactory>::create(), storage, resourceAccess)
+    TestResourceFacade(const QByteArray &instanceIdentifier, const QSharedPointer<Akonadi2::ResourceAccessInterface> resourceAccess)
+        : Akonadi2::GenericFacade<Akonadi2::ApplicationDomain::Event>(instanceIdentifier, QSharedPointer<TestEventAdaptorFactory>::create(), resourceAccess)
     {
 
     }