File indexing completed on 2024-09-22 04:49:57
0001 /* 0002 SPDX-FileCopyrightText: 2015-2024 Laurent Montel <montel@kde.org> 0003 0004 SPDX-License-Identifier: GPL-2.0-only 0005 */ 0006 0007 #include "filteractionrewriteheadertest.h" 0008 #include "../filteractions/filteractionrewriteheader.h" 0009 #include <KLineEdit> 0010 #include <QComboBox> 0011 #include <QLabel> 0012 #include <QTest> 0013 #include <QWidget> 0014 0015 FilterActionRewriteHeaderTest::FilterActionRewriteHeaderTest(QObject *parent) 0016 : QObject(parent) 0017 { 0018 } 0019 0020 FilterActionRewriteHeaderTest::~FilterActionRewriteHeaderTest() = default; 0021 0022 void FilterActionRewriteHeaderTest::shouldHaveDefaultValue() 0023 { 0024 MailCommon::FilterActionRewriteHeader filter; 0025 QWidget *w = filter.createParamWidget(nullptr); 0026 auto combo = w->findChild<QComboBox *>(QStringLiteral("combo")); 0027 QVERIFY(combo); 0028 QVERIFY(combo->count() > 0); 0029 0030 auto label = w->findChild<QLabel *>(QStringLiteral("label_replace")); 0031 QVERIFY(label); 0032 0033 auto regExpLineEdit = w->findChild<KLineEdit *>(QStringLiteral("search")); 0034 QVERIFY(regExpLineEdit); 0035 QVERIFY(regExpLineEdit->text().isEmpty()); 0036 0037 label = w->findChild<QLabel *>(QStringLiteral("label_with")); 0038 QVERIFY(label); 0039 0040 auto lineEdit = w->findChild<KLineEdit *>(QStringLiteral("replace")); 0041 QVERIFY(lineEdit); 0042 QVERIFY(lineEdit->text().isEmpty()); 0043 } 0044 0045 void FilterActionRewriteHeaderTest::shouldBeEmpty() 0046 { 0047 MailCommon::FilterActionRewriteHeader filter; 0048 QVERIFY(filter.isEmpty()); 0049 filter.argsFromString(QStringLiteral("foo\tbla")); 0050 QVERIFY(filter.isEmpty()); 0051 filter.argsFromString(QStringLiteral("foo\tbla\tkde")); 0052 QVERIFY(!filter.isEmpty()); 0053 } 0054 0055 void FilterActionRewriteHeaderTest::shouldNotExecuteActionWhenParameterIsEmpty() 0056 { 0057 MailCommon::FilterActionRewriteHeader filter(this); 0058 KMime::Message::Ptr msgPtr = KMime::Message::Ptr(new KMime::Message()); 0059 Akonadi::Item item; 0060 item.setPayload<KMime::Message::Ptr>(msgPtr); 0061 MailCommon::ItemContext context(item, true); 0062 0063 filter.argsFromString(QString()); 0064 QCOMPARE(filter.process(context, false), MailCommon::FilterAction::ErrorButGoOn); 0065 QCOMPARE(context.needsPayloadStore(), false); 0066 } 0067 0068 void FilterActionRewriteHeaderTest::shouldNotExecuteActionWhenValueIsEmpty() 0069 { 0070 MailCommon::FilterActionRewriteHeader filter(this); 0071 KMime::Message::Ptr msgPtr = KMime::Message::Ptr(new KMime::Message()); 0072 Akonadi::Item item; 0073 item.setPayload<KMime::Message::Ptr>(msgPtr); 0074 MailCommon::ItemContext context(item, true); 0075 0076 filter.argsFromString(QStringLiteral("foo")); 0077 QCOMPARE(filter.process(context, false), MailCommon::FilterAction::ErrorButGoOn); 0078 QCOMPARE(context.needsPayloadStore(), false); 0079 0080 filter.argsFromString(QStringLiteral("foo\tbla")); 0081 QCOMPARE(filter.process(context, false), MailCommon::FilterAction::ErrorButGoOn); 0082 QCOMPARE(context.needsPayloadStore(), false); 0083 } 0084 0085 void FilterActionRewriteHeaderTest::shouldRewriteHeader() 0086 { 0087 const QByteArray data = 0088 "From: foo@kde.org\n" 0089 "To: foo@kde.org\n" 0090 "Subject: test\n" 0091 "testheader: foo\n" 0092 "Date: Wed, 01 Apr 2015 09:33:01 +0200\n" 0093 "MIME-Version: 1.0\n" 0094 "Content-Type: text/plain; charset=\"us-ascii\"\n" 0095 "\n" 0096 "test"; 0097 const QByteArray output = 0098 "From: foo@kde.org\n" 0099 "To: foo@kde.org\n" 0100 "Subject: test\n" 0101 "Date: Wed, 01 Apr 2015 09:33:01 +0200\n" 0102 "MIME-Version: 1.0\n" 0103 "Content-Type: text/plain; charset=\"us-ascii\"\n" 0104 "testheader: bla\n" 0105 "\n" 0106 "test"; 0107 0108 MailCommon::FilterActionRewriteHeader filter(this); 0109 KMime::Message::Ptr msgPtr = KMime::Message::Ptr(new KMime::Message()); 0110 msgPtr->setContent(data); 0111 msgPtr->parse(); 0112 Akonadi::Item item; 0113 item.setPayload<KMime::Message::Ptr>(msgPtr); 0114 MailCommon::ItemContext context(item, true); 0115 0116 filter.argsFromString(QStringLiteral("testheader\tfoo\tbla")); 0117 QCOMPARE(filter.process(context, false), MailCommon::FilterAction::GoOn); 0118 QCOMPARE(context.needsPayloadStore(), true); 0119 QCOMPARE(msgPtr->encodedContent(), output); 0120 } 0121 0122 void FilterActionRewriteHeaderTest::shouldNotRewriteHeaderWhenHeaderNotFound() 0123 { 0124 const QByteArray data = 0125 "From: foo@kde.org\n" 0126 "To: foo@kde.org\n" 0127 "Subject: test\n" 0128 "Date: Wed, 01 Apr 2015 09:33:01 +0200\n" 0129 "MIME-Version: 1.0\n" 0130 "Content-Type: text/plain; charset=\"us-ascii\"\n" 0131 "\n" 0132 "test"; 0133 0134 MailCommon::FilterActionRewriteHeader filter(this); 0135 KMime::Message::Ptr msgPtr = KMime::Message::Ptr(new KMime::Message()); 0136 msgPtr->setContent(data); 0137 msgPtr->parse(); 0138 Akonadi::Item item; 0139 item.setPayload<KMime::Message::Ptr>(msgPtr); 0140 MailCommon::ItemContext context(item, true); 0141 0142 filter.argsFromString(QStringLiteral("testheader\tfoo\tbla")); 0143 QCOMPARE(filter.process(context, false), MailCommon::FilterAction::GoOn); 0144 QCOMPARE(context.needsPayloadStore(), false); 0145 QCOMPARE(msgPtr->encodedContent(), data); 0146 } 0147 0148 void FilterActionRewriteHeaderTest::shouldNotRewriteHeaderWhenRegexpNotFound() 0149 { 0150 const QByteArray data = 0151 "From: foo@kde.org\n" 0152 "To: foo@kde.org\n" 0153 "Subject: test\n" 0154 "Date: Wed, 01 Apr 2015 09:33:01 +0200\n" 0155 "MIME-Version: 1.0\n" 0156 "Content-Type: text/plain; charset=\"us-ascii\"\n" 0157 "testheader: bla\n" 0158 "\n" 0159 "test"; 0160 0161 MailCommon::FilterActionRewriteHeader filter(this); 0162 KMime::Message::Ptr msgPtr = KMime::Message::Ptr(new KMime::Message()); 0163 msgPtr->setContent(data); 0164 msgPtr->parse(); 0165 Akonadi::Item item; 0166 item.setPayload<KMime::Message::Ptr>(msgPtr); 0167 MailCommon::ItemContext context(item, true); 0168 0169 filter.argsFromString(QStringLiteral("testheader\tfoo\tbla")); 0170 QCOMPARE(filter.process(context, false), MailCommon::FilterAction::GoOn); 0171 QCOMPARE(context.needsPayloadStore(), false); 0172 QCOMPARE(msgPtr->encodedContent(), data); 0173 } 0174 0175 void FilterActionRewriteHeaderTest::shouldHaveRequiredPart() 0176 { 0177 MailCommon::FilterActionRewriteHeader filter; 0178 QCOMPARE(filter.requiredPart(), MailCommon::SearchRule::CompleteMessage); 0179 } 0180 0181 QTEST_MAIN(FilterActionRewriteHeaderTest) 0182 0183 #include "moc_filteractionrewriteheadertest.cpp"