kde-playground/kdepim/messagecomposer/tests/signencrypttest.cpp
2015-04-14 21:49:29 +00:00

155 lines
4.7 KiB
C++

/*
Copyright (C) 2009 Klaralvdalens Datakonsult AB, a KDAB Group company, info@kdab.net
Copyright (c) 2009 Leo Franchi <lfranchi@kde.org>
This library is free software; you can redistribute it and/or modify it
under the terms of the GNU Library General Public License as published by
the Free Software Foundation; either version 2 of the License, or (at your
option) any later version.
This library is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library General Public
License for more details.
You should have received a copy of the GNU Library General Public License
along with this library; see the file COPYING.LIB. If not, write to the
Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
02110-1301, USA.
*/
#include "signencrypttest.h"
#include <KDebug>
#include <qtest_kde.h>
#include "qtest_messagecomposer.h"
#include "cryptofunctions.h"
#include <akonadi/item.h>
#include <kmime/kmime_content.h>
#include <kleo/enum.h>
#include <kleo/keylistjob.h>
#include <kleo/cryptobackendfactory.h>
#include <kjob.h>
#include <messagecomposer/composer/composer.h>
#include <messagecomposer/job/encryptjob.h>
#include <messagecomposer/job/maintextjob.h>
#include <messagecomposer/job/signjob.h>
#include <messagecomposer/part/globalpart.h>
#include <messagecomposer/part/textpart.h>
#include <messageviewer/viewer/nodehelper.h>
#include <messagecore/tests/util.h>
#include <messagecore/helpers/nodehelper.h>
QTEST_KDEMAIN( SignEncryptTest, GUI )
void SignEncryptTest::initTestCase()
{
MessageCore::Test::setupEnv();
}
void SignEncryptTest::testContent()
{
std::vector< GpgME::Key > keys = MessageCore::Test::getKeys();
MessageComposer::Composer *composer = new MessageComposer::Composer;
MessageComposer::SignJob* sJob = new MessageComposer::SignJob( composer );
MessageComposer::EncryptJob* eJob = new MessageComposer::EncryptJob( composer );
QVERIFY( composer );
QVERIFY( sJob );
QVERIFY( eJob );
QList<QByteArray> charsets;
charsets << "us-ascii";
composer->globalPart()->setCharsets( charsets );
MessageComposer::TextPart* part = new MessageComposer::TextPart( this );
part->setWordWrappingEnabled(false);
part->setCleanPlainText( QString::fromLatin1("one flew over the cuckoo's nest"));
MessageComposer::MainTextJob *mainTextJob = new MessageComposer::MainTextJob( part, composer );
QVERIFY( composer );
QVERIFY( mainTextJob );
VERIFYEXEC( mainTextJob );
QStringList recipients;
recipients << QString::fromLocal8Bit( "test@kolab.org" );
sJob->setContent( mainTextJob->content() );
sJob->setSigningKeys( keys );
sJob->setCryptoMessageFormat( Kleo::OpenPGPMIMEFormat );
eJob->setCryptoMessageFormat( Kleo::OpenPGPMIMEFormat );
eJob->setRecipients( recipients );
eJob->setEncryptionKeys( keys );
eJob->appendSubjob( sJob );
VERIFYEXEC( eJob );
KMime::Content* result = eJob->content();
QVERIFY( result );
result->assemble();
kDebug() << "result:" << result->encodedContent();
ComposerTestUtil::verifySignatureAndEncryption(
result,
QString::fromLocal8Bit( "one flew over the cuckoo's nest" ).toUtf8(),
Kleo::OpenPGPMIMEFormat );
}
void SignEncryptTest::testHeaders()
{
std::vector< GpgME::Key > keys = MessageCore::Test::getKeys();
MessageComposer::Composer *composer = new MessageComposer::Composer;
MessageComposer::SignJob* sJob = new MessageComposer::SignJob( composer );
MessageComposer::EncryptJob* eJob = new MessageComposer::EncryptJob( composer );
QVERIFY( composer );
QVERIFY( sJob );
QVERIFY( eJob );
QByteArray data( QString::fromLocal8Bit( "one flew over the cuckoo's nest" ).toUtf8() );
KMime::Content* content = new KMime::Content;
content->setBody( data );
QStringList recipients;
recipients << QString::fromLocal8Bit( "test@kolab.org" );
sJob->setContent( content );
sJob->setSigningKeys( keys );
sJob->setCryptoMessageFormat( Kleo::OpenPGPMIMEFormat );
eJob->setCryptoMessageFormat( Kleo::OpenPGPMIMEFormat );
eJob->setRecipients( recipients );
eJob->setEncryptionKeys( keys );
eJob->appendSubjob( sJob );
VERIFYEXEC( eJob );
KMime::Content* result = eJob->content();
QVERIFY( result );
result->assemble();
QByteArray mimeType( "multipart/encrypted" );
QByteArray charset( "ISO-8859-1" );
QVERIFY( result->contentType( false ) );
QCOMPARE( result->contentType()->mimeType(), mimeType );
QCOMPARE( result->contentType()->charset(), charset );
QCOMPARE( result->contentType()->parameter( QString::fromLocal8Bit( "protocol" ) ), QString::fromLocal8Bit( "application/pgp-encrypted" ) );
QCOMPARE( result->contentTransferEncoding()->encoding(), KMime::Headers::CE7Bit );
}