// Copyright (c) 2006-2009 Nokia Corporation and/or its subsidiary(-ies).

// All rights reserved.

// This component and the accompanying materials are made available

// under the terms of "Eclipse Public License v1.0"

// which accompanies this distribution, and is available

// at the URL "http://www.eclipse.org/legal/epl-v10.html".

//

// Initial Contributors:

// Nokia Corporation - initial contribution.

//

// Contributors:

//

// Description:

//

#include "ctestimapsession.h"

#include "cfakeinputstream.h"

#include "cfakeoutputstream.h"

#include "cactivewaiter.h"

#include "moutputstream.h"

#include "cimapsession.h"

#include "cimapservergreetinginfo.h"

#include "cimapcapabilityinfo.h"

#include "cimapfolderinfo.h"

#include "cimaputils.h"

class CImapSettings;

class CImapMailStore;

CTestImapSession::CTestImapSession()

	: iInputStream(NULL)

	, iOutputStream(NULL)

	, iActiveWaiter(NULL)

	, iImapSession(NULL)

	{}

CTestImapSession::~CTestImapSession()

	{

	delete iImapSession;

	delete iActiveWaiter;

	delete iOutputStream;

	delete iInputStream;

	CImapUtils::Delete();

	}

void CTestImapSession::SetupL()

	{

	ASSERT(iInputStream == NULL);

	ASSERT(iOutputStream == NULL);

	ASSERT(iActiveWaiter == NULL);

	ASSERT(iImapSession == NULL);

	CImapUtils::CreateL();

	iInputStream = CFakeInputStream::NewL(Logger());

	iOutputStream = CFakeOutputStream::NewL(Logger());

	iActiveWaiter = new(ELeave)CActiveWaiter(Logger());

	CImapSettings* imapSettings=NULL; 

	CImapMailStore* imapMailStore=NULL;

	iImapSession = CImapSession::NewL(*imapSettings, *imapMailStore, *iInputStream, *iOutputStream);

	}

void CTestImapSession::TearDownL()

	{

	delete iImapSession;

	iImapSession = NULL;

	delete iActiveWaiter;

	iActiveWaiter = NULL;

	delete iOutputStream;

	iOutputStream = NULL;

	delete iInputStream;

	iInputStream = NULL;

	CImapUtils::Delete();

	}

void CTestImapSession::SetupServerGreetingL()

	{

	INFO_PRINTF1(_L("Setup: ServerGreeting"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* OK Microsoft Exchange 2000 IMAP4rev1 server version 6.0.6249.0 (lon-cn-exchng2k.msexchange2k.closedtest.intra) ready.\r\n"));

	ASSERT_EQUALS(CImapSession::EServerStateNone, iImapSession->ServerState());

	iImapSession->ReadServerGreetingL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	ASSERT_EQUALS(CImapSession::EServerStateNotAuthenticated, iImapSession->ServerState());

	INFO_PRINTF1(_L("Setup: Complete"));

	}

// Tests

void CTestImapSession::TestOneLineL()

	{

	SetupServerGreetingL();

	INFO_PRINTF1(_L("TestOneLineL"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("1 OK LOGOUT completed\r\n"));

	iImapSession->LogoutL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestTwoLinesSimpleL()

	{

	SetupServerGreetingL();

	INFO_PRINTF1(_L("TestTwoLinesSimpleL"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* BYE IMAP4rev1 Server logging out\r\n"));

	iInputStream->AppendInputStringL(_L8("1 OK LOGOUT completed\r\n"));

	iImapSession->LogoutL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestTwoLinesSplitBeforeFirstCrlfL()

	{

	SetupServerGreetingL();

	INFO_PRINTF1(_L("TestTwoLinesSplitBeforeFirstCrlfL"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* BYE IMAP4rev1 Serv"));

	iInputStream->AppendInputStringL(_L8("er logging out\r\n1 OK LOGOUT completed\r\n"));

	iImapSession->LogoutL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestTwoLinesSplitAfterFirstCrlfL()

	{

	SetupServerGreetingL();

	INFO_PRINTF1(_L("TestTwoLinesSplitAfterFirstCrlfL"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* BYE IMAP4rev1 Server logging out\r\n1 OK LOGOUT co"));

	iInputStream->AppendInputStringL(_L8("mpleted\r\n"));

	iImapSession->LogoutL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestTwoLinesSplitOverFirstCrlfL()

	{

	SetupServerGreetingL();

	INFO_PRINTF1(_L("TestTwoLinesSplitOverFirstCrlfL"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* BYE IMAP4rev1 Server logging out\r"));

	iInputStream->AppendInputStringL(_L8("\n1 OK LOGOUT completed\r\n"));

	iImapSession->LogoutL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestTwoLinesSplitHeavilyOverFirstCrlfL()

	{

	SetupServerGreetingL();

	INFO_PRINTF1(_L("TestTwoLinesSplitHeavilyOverFirstCrlfL"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* BYE IMAP4rev1 Server logging out\r"));

	iInputStream->AppendInputStringL(_L8("\n")); // <<== This line needs special handling in CImapSession

	iInputStream->AppendInputStringL(_L8("1 OK LOGOUT completed\r\n"));

	iImapSession->LogoutL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestTwoLinesInOneBlockL()

	{

	SetupServerGreetingL();

	INFO_PRINTF1(_L("TestTwoLinesInOneBlockL"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* BYE IMAP4rev1 Server logging out\r\n1 OK LOGOUT completed\r\n"));

	iImapSession->LogoutL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestSevenLinesSimpleL()

	{

	SetupServerGreetingL();

	INFO_PRINTF1(_L("TestSevenLinesSimpleL"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* BYE IMAP4rev1 Server logging out\r\n"));

	iInputStream->AppendInputStringL(_L8("* BYE second\r\n"));

	iInputStream->AppendInputStringL(_L8("* BYE third\r\n"));

	iInputStream->AppendInputStringL(_L8("* BYE fourth\r\n"));

	iInputStream->AppendInputStringL(_L8("* BYE fifth\r\n"));

	iInputStream->AppendInputStringL(_L8("* BYE sixth\r\n"));

	iInputStream->AppendInputStringL(_L8("1 OK LOGOUT completed\r\n"));

	iImapSession->LogoutL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestSevenLinesIn3ChunksL()

	{

	SetupServerGreetingL();

	INFO_PRINTF1(_L("TestSevenLinesIn3ChunksL"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* BYE IMAP4rev1 Server logging out\r\n* BYE second"));

	iInputStream->AppendInputStringL(_L8("\r\n* BYE third\r\n* BYE fourth\r\n* BYE fi"));

	iInputStream->AppendInputStringL(_L8("fth\r\n* BYE sixth\r\n1 OK LOGOUT completed\r\n"));

	iImapSession->LogoutL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestOneLineFragmentedL()

	{

	SetupServerGreetingL();

	INFO_PRINTF1(_L("TestOneLineFragmentedL"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("1 O"));

	iInputStream->AppendInputStringL(_L8("K LOG"));

	iInputStream->AppendInputStringL(_L8("O"));

	iInputStream->AppendInputStringL(_L8("UT co"));

	iInputStream->AppendInputStringL(_L8("mpleted"));

	iInputStream->AppendInputStringL(_L8("\r\n"));

	iImapSession->LogoutL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestOneLineHeavilyFragmentedL()

	{

	SetupServerGreetingL();

	INFO_PRINTF1(_L("TestOneLineHeavilyFragmentedL"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("1"));

	iInputStream->AppendInputStringL(_L8(" "));

	iInputStream->AppendInputStringL(_L8("O"));

	iInputStream->AppendInputStringL(_L8("K"));

	iInputStream->AppendInputStringL(_L8(" "));

	iInputStream->AppendInputStringL(_L8("L"));

	iInputStream->AppendInputStringL(_L8("O"));

	iInputStream->AppendInputStringL(_L8("G"));

	iInputStream->AppendInputStringL(_L8("O"));

	iInputStream->AppendInputStringL(_L8("U"));

	iInputStream->AppendInputStringL(_L8("T"));

	iInputStream->AppendInputStringL(_L8(" "));

	iInputStream->AppendInputStringL(_L8("c"));

	iInputStream->AppendInputStringL(_L8("o"));

	iInputStream->AppendInputStringL(_L8("m"));

	iInputStream->AppendInputStringL(_L8("p"));

	iInputStream->AppendInputStringL(_L8("l"));

	iInputStream->AppendInputStringL(_L8("e"));

	iInputStream->AppendInputStringL(_L8("t"));

	iInputStream->AppendInputStringL(_L8("e"));

	iInputStream->AppendInputStringL(_L8("d"));

	iInputStream->AppendInputStringL(_L8("\r"));

	iInputStream->AppendInputStringL(_L8("\n"));

	iImapSession->LogoutL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestLoginSequenceL()

	{

	INFO_PRINTF1(_L("TestLoginSequenceL"));

	INFO_PRINTF1(_L("...ServerGreeting"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* OK Microsoft Exchange 2000 IMAP4rev1 server version 6.0.6249.0 (lon-cn-exchng2k.msexchange2k.closedtest.intra) ready.\r\n"));

	ASSERT_EQUALS(CImapServerGreetingInfo::ETagNone, iImapSession->ServerGreetingInfo().ResponseTag());

	iImapSession->ReadServerGreetingL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	ASSERT_EQUALS(CImapServerGreetingInfo::ETagOk, iImapSession->ServerGreetingInfo().ResponseTag());

	INFO_PRINTF1(_L("...Capability"));

	iInputStream->ResetInputStrings();

#if (defined SYMBIAN_EMAIL_CAPABILITY_SUPPORT)

	iInputStream->AppendInputStringL(_L8("* CAPABILITY IMAP4rev1 STARTTLS AUTH=GSSAPI LOGINDISABLED BINARY\r\n"));

#else

	iInputStream->AppendInputStringL(_L8("* CAPABILITY IMAP4rev1 STARTTLS AUTH=GSSAPI LOGINDISABLED\r\n"));

#endif

	iInputStream->AppendInputStringL(_L8("1 OK CAPABILITY completed\r\n"));

	iImapSession->CapabilityL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	const CImapCapabilityInfo& capabilityInfo = iImapSession->CapabilityInfo();

	ASSERT_TRUE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EVersion4Rev1));

	ASSERT_TRUE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EStartTls));

	ASSERT_TRUE(capabilityInfo.QueryFlag(CImapCapabilityInfo::ELoginDisabled));

	ASSERT_FALSE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EIdle));

	ASSERT_FALSE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EAuthPlain));

#if (defined SYMBIAN_EMAIL_CAPABILITY_SUPPORT)

	ASSERT_TRUE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EBinaryCap));

#endif

	INFO_PRINTF1(_L("...StartTLS"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("2 OK Begin TLS negotiation now\r\n"));

	iImapSession->StartTlsL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();	

	INFO_PRINTF1(_L("...Capability"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* CAPABILITY IMAP4rev1 AUTH=GSSAPI AUTH=PLAIN IDLE\r\n"));

	iInputStream->AppendInputStringL(_L8("3 OK CAPABILITY completed\r\n"));

	iImapSession->CapabilityL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	ASSERT_TRUE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EVersion4Rev1));

	ASSERT_FALSE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EStartTls));

	ASSERT_FALSE(capabilityInfo.QueryFlag(CImapCapabilityInfo::ELoginDisabled));

	ASSERT_TRUE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EIdle));

	ASSERT_TRUE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EAuthPlain));

	INFO_PRINTF1(_L("...Login"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("4 OK [CAPABILITY IMAP4rev1 IDLE] LOGIN completed\r\n"));

	iImapSession->LoginL(iActiveWaiter->iStatus, _L8("username"), _L8("password"));

	iActiveWaiter->WaitActive();

	ASSERT_TRUE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EVersion4Rev1));

	ASSERT_FALSE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EStartTls));

	ASSERT_FALSE(capabilityInfo.QueryFlag(CImapCapabilityInfo::ELoginDisabled));

	ASSERT_TRUE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EIdle));

	ASSERT_FALSE(capabilityInfo.QueryFlag(CImapCapabilityInfo::EAuthPlain));

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestServerStateL()

	{

	INFO_PRINTF1(_L("TestServerStateL"));

	INFO_PRINTF1(_L("...ServerGreeting"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* OK Microsoft Exchange 2000 IMAP4rev1 server version 6.0.6249.0 (lon-cn-exchng2k.msexchange2k.closedtest.intra) ready.\r\n"));

	ASSERT_EQUALS(CImapSession::EServerStateNone, iImapSession->ServerState());

	iImapSession->ReadServerGreetingL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	ASSERT_EQUALS(CImapSession::EServerStateNotAuthenticated, iImapSession->ServerState());

	INFO_PRINTF1(_L("...Login"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("1 OK LOGIN completed\r\n"));

	iImapSession->LoginL(iActiveWaiter->iStatus, _L8("username"), _L8("password"));

	iActiveWaiter->WaitActive();

	ASSERT_EQUALS(CImapSession::EServerStateAuthenticated, iImapSession->ServerState());

	INFO_PRINTF1(_L("...Select inbox"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("2 OK [READ-WRITE] SELECT completed\r\n"));

	CImapFolderInfo* folderInfo = CImapFolderInfo::NewL();

	CleanupStack::PushL(folderInfo);

	folderInfo->SetNameL(_L16("inbox"));

	CleanupStack::Pop(folderInfo);

	iImapSession->SelectL(iActiveWaiter->iStatus, folderInfo);

	iActiveWaiter->WaitActive();

	ASSERT_EQUALS(CImapSession::EServerStateSelected, iImapSession->ServerState());

	ASSERT_EQUALS(folderInfo, iImapSession->SelectedFolderInfo());

	folderInfo = NULL;

	INFO_PRINTF1(_L("...Close"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("3 OK CLOSE completed\r\n"));

	iImapSession->CloseL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	ASSERT_EQUALS(CImapSession::EServerStateAuthenticated, iImapSession->ServerState());

	ASSERT_NULL(iImapSession->SelectedFolderInfo());

	INFO_PRINTF1(_L("...Logout"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("* BYE IMAP4rev1 Server logging out\r\n4 OK LOGOUT completed\r\n"));

	iImapSession->LogoutL(iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	ASSERT_EQUALS(CImapSession::EServerStateNone, iImapSession->ServerState());

	INFO_PRINTF1(_L("Complete"));

	}

void CTestImapSession::TestLiteralL()

	{

	INFO_PRINTF1(_L("TestLiteralL"));

	SetupServerGreetingL();

	INFO_PRINTF1(_L("...Login"));

	iInputStream->ResetInputStrings();

	iInputStream->AppendInputStringL(_L8("1 OK LOGIN completed\r\n"));

	iImapSession->LoginL(iActiveWaiter->iStatus, _L8("username"), _L8("password"));

	iActiveWaiter->WaitActive();

	INFO_PRINTF1(_L("...Status"));	

	iInputStream->ResetInputStrings();

/*

	iInputStream->AppendInputStringL(_L8("* STATUS {12}\r\nHello\r\nWorld (MESSAGES 231)\r\n"));

	iInputStream->AppendInputStringL(_L8("3 OK STATUS completed\r\n"));

*/

	iInputStream->AppendInputStringL(_L8("* STATUS {12}\r\nHello\r\nWor"));

	iInputStream->AppendInputStringL(_L8("ld (MESSAGES 231)\r\n"));

	iInputStream->AppendInputStringL(_L8("2 OK STATUS completed\r\n"));

	CImapFolderInfo* folderInfo = CImapFolderInfo::NewL();

	CleanupStack::PushL(folderInfo);

	iImapSession->StatusL(iActiveWaiter->iStatus, _L16("Hello\r\nWorld"), _L8("MESSAGES"), *folderInfo);

	iActiveWaiter->WaitActive();

	CleanupStack::PopAndDestroy(folderInfo);

	INFO_PRINTF1(_L("Complete"));

	}

CTestSuite* CTestImapSession::CreateSuiteL(const TDesC& aName)

// static

	{

	SUB_SUITE;

	ADD_ASYNC_TEST_STEP(TestOneLineL);

	ADD_ASYNC_TEST_STEP(TestTwoLinesSimpleL);

	ADD_ASYNC_TEST_STEP(TestTwoLinesSplitBeforeFirstCrlfL);

	ADD_ASYNC_TEST_STEP(TestTwoLinesSplitAfterFirstCrlfL);

	ADD_ASYNC_TEST_STEP(TestTwoLinesSplitOverFirstCrlfL);

	ADD_ASYNC_TEST_STEP(TestTwoLinesSplitHeavilyOverFirstCrlfL);

	ADD_ASYNC_TEST_STEP(TestTwoLinesInOneBlockL);

	ADD_ASYNC_TEST_STEP(TestSevenLinesSimpleL);

	ADD_ASYNC_TEST_STEP(TestSevenLinesIn3ChunksL);

	ADD_ASYNC_TEST_STEP(TestOneLineFragmentedL);	

	ADD_ASYNC_TEST_STEP(TestOneLineHeavilyFragmentedL);

	ADD_ASYNC_TEST_STEP(TestLoginSequenceL);

	ADD_ASYNC_TEST_STEP(TestServerStateL);

	ADD_ASYNC_TEST_STEP(TestLiteralL);

	END_SUITE;

	}