1 files changed, 39 insertions, 18 deletions

diff --git a/examples/carddavresource/tests/carddavtest.cpp b/examples/carddavresource/tests/carddavtest.cpp
index 3da03a6..f75a30a 100644
--- a/examples/carddavresource/tests/carddavtest.cpp
+++ b/examples/carddavresource/tests/carddavtest.cpp
@@ -11,6 +11,7 @@
 #include <KDAV2/DavUrl>
 #include <KDAV2/DavItemCreateJob>
 #include <KDAV2/DavCollectionsFetchJob>
+#include <KDAV2/DavCollectionCreateJob>
 #include <KDAV2/DavCollection>
 #include <KContacts/Addressee>
 #include <KContacts/VCardConverter>
@@ -36,21 +37,22 @@ class CardDavTest : public QObject
 
     QByteArray mResourceInstanceIdentifier;
 
-    void createContact(const QString &firstname, const QString &lastname)
+    void createContact(const QString &firstname, const QString &lastname, const QString &collectionName)
     {
         QUrl mainUrl(QStringLiteral("http://localhost/dav/addressbooks/user/doe"));
         mainUrl.setUserName(QStringLiteral("doe"));
         mainUrl.setPassword(QStringLiteral("doe"));
 
-
         KDAV2::DavUrl davUrl(mainUrl, KDAV2::CardDav);
 
         auto *job = new KDAV2::DavCollectionsFetchJob(davUrl);
         job->exec();
 
         const auto collectionUrl = [&] {
-            if (!job->collections().isEmpty()) {
-                return job->collections().first().url().url();
+            for (const auto &col : job->collections()) {
+                if (col.displayName() == collectionName) {
+                    return col.url().url();
+                }
             }
             return QUrl{};
         }();
@@ -68,6 +70,22 @@ class CardDavTest : public QObject
         }
     }
 
+    void createCollection(const QString &name)
+    {
+        QUrl mainUrl(QStringLiteral("http://localhost/dav/addressbooks/user/doe/") + name);
+        mainUrl.setUserName(QStringLiteral("doe"));
+        mainUrl.setPassword(QStringLiteral("doe"));
+
+        KDAV2::DavUrl davUrl(mainUrl, KDAV2::CardDav);
+        KDAV2::DavCollection collection{davUrl, name, KDAV2::DavCollection::Contacts};
+
+        auto createJob = new KDAV2::DavCollectionCreateJob(collection);
+        createJob->exec();
+        if (createJob->error()) {
+            qWarning() << createJob->errorString();
+        }
+    }
+
     void resetTestEnvironment()
     {
         system("resetmailbox.sh");
@@ -95,10 +113,25 @@ private slots:
         VERIFYEXEC(Sink::ResourceControl::start(mResourceInstanceIdentifier));
     }
 
+    void testSyncAddressbooks()
+    {
+        createCollection("addressbook2");
+
+        Sink::SyncScope scope;
+        scope.setType<Addressbook>();
+        scope.resourceFilter(mResourceInstanceIdentifier);
+
+        VERIFYEXEC(Sink::Store::synchronize(scope));
+        VERIFYEXEC(Sink::ResourceControl::flushMessageQueue(mResourceInstanceIdentifier));
+        const auto addressbooks = Sink::Store::read<Addressbook>(Sink::Query().resourceFilter(mResourceInstanceIdentifier));
+        QCOMPARE(addressbooks.size(), 2);
+    }
+
     void testSyncContacts()
     {
-        createContact("john", "doe");
-        createContact("jane", "doe");
+        //We want the extra collection from the previous test in this test.
+        createContact("john", "doe", "personal");
+        createContact("jane", "doe", "personal");
         Sink::SyncScope scope;
         scope.setType<Sink::ApplicationDomain::Contact>();
         scope.resourceFilter(mResourceInstanceIdentifier);
@@ -117,18 +150,6 @@ private slots:
         }
     }
 
-    void testSyncAddressbooks()
-    {
-        Sink::SyncScope scope;
-        scope.setType<Addressbook>();
-        scope.resourceFilter(mResourceInstanceIdentifier);
-
-        VERIFYEXEC(Sink::Store::synchronize(scope));
-        VERIFYEXEC(Sink::ResourceControl::flushMessageQueue(mResourceInstanceIdentifier));
-        const auto addressbooks = Sink::Store::read<Addressbook>(Sink::Query().resourceFilter(mResourceInstanceIdentifier));
-        QCOMPARE(addressbooks.size(), 1);
-    }
-
     void testAddContact()
     {
         Sink::SyncScope scope;