/*
  Copyright (c) 2009 Constantin Berzan <exit3219@gmail.com>
  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 "composertest.h"

#include "qtest_messagecomposer.h"

#include <KDebug>
#include <qtest_kde.h>

#include <kmime/kmime_headers.h>
using namespace KMime;

#include <messagecomposer/composer/composer.h>
#include <messagecomposer/part/globalpart.h>
#include <messagecomposer/part/infopart.h>
#include <messagecomposer/part/textpart.h>
using namespace MessageComposer;

#include <messagecore/attachment/attachmentpart.h>
#include <boost/shared_ptr.hpp>
using MessageCore::AttachmentPart;

QTEST_KDEMAIN( ComposerTest, GUI )

void ComposerTest::testAttachments()
{
  Composer *composer = new Composer;
  fillComposerData( composer );
  AttachmentPart::Ptr attachment = AttachmentPart::Ptr( new AttachmentPart );
  attachment->setData( "abc" );
  attachment->setMimeType( "x-some/x-type" );
  composer->addAttachmentPart( attachment );

  QVERIFY( composer->exec() );
  QCOMPARE( composer->resultMessages().size(), 1 );
  KMime::Message::Ptr message = composer->resultMessages().first();
  kDebug() << message->encodedContent();
  delete composer;
  composer = 0;

  // multipart/mixed
  {
    QVERIFY( message->contentType( false ) );
    QCOMPARE( message->contentType()->mimeType(), QByteArray( "multipart/mixed" ) );
    QCOMPARE( message->contents().count(), 2 );
    // text/plain
    {
      Content *plain = message->contents().at( 0 );
      QVERIFY( plain->contentType( false ) );
      QCOMPARE( plain->contentType()->mimeType(), QByteArray( "text/plain" ) );
    }
    // x-some/x-type (attachment)
    {
      Content *plain = message->contents().at( 1 );
      QVERIFY( plain->contentType( false ) );
      QCOMPARE( plain->contentType()->mimeType(), QByteArray( "x-some/x-type" ) );
    }
  }
}

void ComposerTest::testAutoSave()
{
  Composer *composer = new Composer;
  fillComposerData( composer );
  AttachmentPart::Ptr attachment = AttachmentPart::Ptr( new AttachmentPart );
  attachment->setData( "abc" );
  attachment->setMimeType( "x-some/x-type" );
  composer->addAttachmentPart( attachment );

  // This tests if autosave in crash mode works without invoking an event loop, since using an
  // event loop in the crash handler would be a pretty bad idea
  composer->setAutoSave( true );
  composer->start();
  QVERIFY( composer->finished() );
  QCOMPARE( composer->resultMessages().size(), 1 );
}

void ComposerTest::testNonAsciiHeaders()
{
  Composer *composer = new Composer;
  fillComposerData( composer );

  const QString mailbox = QLatin1String(" <bla@example.com>");
  const QString fromDisplayName = QString::fromUtf8("Hellö");
  const QString toDisplayName = QString::fromUtf8("æſłĸð");
  const QString ccDisplayName = QString::fromUtf8("Вася");
  const QString bccDisplayName = QString::fromUtf8("ĸłſðđøþĸµ»«„¢þµ¢”«ł„·ĸ”");
  const QString replyToDisplayName = QString::fromUtf8("æĸſłð˝đВасяðæĸđ");
  const QString from = fromDisplayName + mailbox;
  const QString to = toDisplayName + mailbox;
  const QString cc = ccDisplayName + mailbox;
  const QString bcc = bccDisplayName + mailbox;
  const QString replyto = replyToDisplayName + mailbox;

  composer->infoPart()->setFrom( from );
  composer->infoPart()->setTo( QStringList() << to );
  composer->infoPart()->setCc( QStringList() << cc );
  composer->infoPart()->setBcc( QStringList() << bcc );
  composer->infoPart()->setReplyTo( replyto );

  QVERIFY( composer->exec() );
  QCOMPARE( composer->resultMessages().size(), 1 );
  const KMime::Message::Ptr message = composer->resultMessages().first();
  kDebug() << message->encodedContent();
  message->assemble();
  message->parse();
  QCOMPARE( message->bcc( false )->displayNames().size(), 1 );
  QCOMPARE( message->to( false )->displayNames().size(), 1 );
  QCOMPARE( message->cc( false )->displayNames().size(), 1 );
  QCOMPARE( message->from( false )->displayNames().size(), 1 );
  QCOMPARE( message->replyTo( false )->displayNames().size(), 1 );
  QCOMPARE( message->from()->displayNames().first().toUtf8(), fromDisplayName.toUtf8() );
  QCOMPARE( message->to()->displayNames().first().toUtf8(), toDisplayName.toUtf8() );
  QCOMPARE( message->cc()->displayNames().first().toUtf8(), ccDisplayName.toUtf8() );
  QCOMPARE( message->bcc()->displayNames().first().toUtf8(), bccDisplayName.toUtf8() );
  QCOMPARE( message->replyTo()->displayNames().first().toUtf8(), replyToDisplayName.toUtf8() );
  delete composer;
  composer = 0;
}

void ComposerTest::testBug271192()
{
  const QString displayName = QString::fromUtf8( "Интернет-компания example" );
  const QString mailbox = QLatin1String( "example@example.com" );
  Composer *composer = new Composer;
  fillComposerData( composer );
  composer->infoPart()->setTo( QStringList() << (displayName + QLatin1String(" <") + mailbox + QLatin1String(">") ) );
  QVERIFY( composer->exec() );
  QCOMPARE( composer->resultMessages().size(), 1 );
  const KMime::Message::Ptr message = composer->resultMessages().first();
  QCOMPARE( message->to()->displayNames().size(), 1 );
  QCOMPARE( message->to()->displayNames().first().toUtf8(), displayName.toUtf8() );
}

void ComposerTest::fillComposerData( Composer* composer )
{
  composer->globalPart()->setFallbackCharsetEnabled( true );
  composer->infoPart()->setFrom( QString::fromLatin1( "me@me.me" ) );
  composer->infoPart()->setTo( QStringList( QString::fromLatin1( "you@you.you" ) ) );
  composer->textPart()->setWrappedPlainText( QString::fromLatin1( "All happy families are alike; each unhappy family is unhappy in its own way." ) );
}