Rely on isAttachment to collect attachments

1 files changed, 4 insertions, 3 deletions

diff --git a/framework/src/domain/mime/mimetreeparser/autotests/attachmenttest.cpp b/framework/src/domain/mime/mimetreeparser/autotests/attachmenttest.cpp
index 47b5cbce..789c9049 100644
--- a/framework/src/domain/mime/mimetreeparser/autotests/attachmenttest.cpp
+++ b/framework/src/domain/mime/mimetreeparser/autotests/attachmenttest.cpp
@@ -22,6 +22,7 @@
 #include "setupenv.h"
 
 #include <qtest.h>
+#include <QDebug>
 
 using namespace MimeTreeParser;
 
@@ -58,10 +59,10 @@ void AttachmentTest::testEncryptedAttachment()
     ObjectTreeParser otp(&nodeHelper);
     otp.parseObjectTree(msg.data());
     otp.decryptParts();
+    otp.print();
 
-    auto attachments = msg->attachments();
-    auto encAtts = nodeHelper.attachmentsOfExtraContents();
-    QCOMPARE(attachments.size() + encAtts.size(), 2);
+    auto attachmentParts = otp.collectAttachmentParts();
+    QCOMPARE(attachmentParts.size(), 2);
 }
 
 #include "attachmenttest.moc"