// Copyright (c) 2008-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 "ctestpopsaslauthhelper.h"

#if defined (__WINS__)

	#define PDD_NAME		_L("ECDRV")

	#define LDD_NAME		_L("ECOMM")

#else

	#define PDD_NAME		_L("EUART1")

	#define LDD_NAME		_L("ECOMM")

#endif

_LIT(KPopsComponent,"T_POPAuthCRAM-MD5Support");

_LIT(KPOPServer,"msexchange01.closedtest.intra");

_LIT8(KPOPUsernamePasswd, "pop001");

RTest testPop3Auth(KPopsComponent);

CTestPopSaslAuthHelper::CTestPopSaslAuthHelper()

	{

	}

CTestPopSaslAuthHelper ::~CTestPopSaslAuthHelper()

	{

	}

void CTestPopSaslAuthHelper::SetupL()

	{

	_LIT(KFunction, "SetupL");

	INFO_PRINTF1(KFunction);

	iResponseBuffer = HBufC8::NewL(KImMailMaxBufferSize);

	}

void CTestPopSaslAuthHelper::TearDownL()

	{

	_LIT(KFunction, "TearDownL");

	INFO_PRINTF1(KFunction);

	delete iResponseBuffer;

	delete iPopSettings;

	delete iActiveWaiter;

	delete iTestUtils;

	}

void CTestPopSaslAuthHelper::CreatePopSaslAuthMechanismL()

	{

	CreatePopTestAccountL();

	ConnectToRemoteServerL();

	TestForAuthCramMd5L();

	TestForAuthPlainL();

	TestForAuthLoginL();

	DisconnectL();

	}

void CTestPopSaslAuthHelper::CreatePopTestAccountL()

	{

	_LIT(KFunction, "CreatePopTestAccountL");

	INFO_PRINTF1(KFunction);

	TInt err;

	err=User::LoadPhysicalDevice(PDD_NAME);

	if (err!=KErrNone && err!=KErrAlreadyExists)

		User::Leave(err);

	err=User::LoadLogicalDevice(LDD_NAME);

	if (err!=KErrNone && err!=KErrAlreadyExists)

		User::Leave(err);

	iActiveWaiter = new(ELeave)CActiveWaiter();

	iTestUtils = CEmailTestUtils::NewL(testPop3Auth);

	iTestUtils->FileSession().SetSessionPath( _L( "C:\\" ) );

	iTestUtils->ClearEmailAccountsL();

	iTestUtils->CreateAllTestDirectories();

	iTestUtils->CleanMessageFolderL();

	iTestUtils->GoClientSideL();	

	if(!iPop3Service)

		{

		iPop3Service = iTestUtils->CreatePopServiceL();	

		}

	iEmailAccount = CEmailAccounts::NewLC();

	iPopSettings = new(ELeave) CImPop3Settings();

	CImIAPPreferences* imapIAP = CImIAPPreferences::NewLC();

	iEntrySelection = new (ELeave) CMsvEntrySelection;

	iEmailAccount->GetPopAccountL(iPop3Service, iPop3AccountId);

  	iEmailAccount->LoadPopSettingsL(iPop3AccountId, *iPopSettings);

	iPopSettings->SetServerAddressL(KPOPServer);

	iPopSettings->SetLoginNameL(KPOPUsernamePasswd /*iTestUtils->MachineName()*/);

	iPopSettings->SetPasswordL(KPOPUsernamePasswd /*iTestUtils->MachineName()*/);

	iPopSettings->SetPort(110);

	iPopSettings->SetPOP3Auth(ETrue);

	iPopSettings->SetFallBack(ETrue);

	iEmailAccount->SavePopSettingsL(iPop3AccountId, *iPopSettings);

	CleanupStack::PopAndDestroy(2, iEmailAccount);

	iTestUtils->GoServerSideL();

	iTestUtils->InstantiatePopServerMtmL();

	iPopServerMtm = iTestUtils->iPopServerMtm;

	}

//Connect to Remote Server 

void CTestPopSaslAuthHelper::ConnectToRemoteServerL()

	{

	_LIT(KFunction, "ConnectToRemoteServerL");

	INFO_PRINTF1(KFunction);

	TBuf8<128> parameter;

	iEntrySelection->AppendL(iPop3Service);

	TPop3Progress temp;	

	TPckgC<TPop3Progress> paramPack(temp);

	iPopServerMtm->StartCommandL(*iEntrySelection, KPOP3MTMConnect, parameter, iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	const TDesC8& progBuf = iTestUtils->iPopServerMtm->Progress();	

	paramPack.Set(progBuf);

	TPop3Progress progress=paramPack();	

	ASSERT_EQUALS( progress.iErrorCode, KErrNone );

	}

//Disconnecting from Remote Server

void CTestPopSaslAuthHelper::DisconnectL()

	{

	_LIT(KFunction, "DisconnectL");

	INFO_PRINTF1(KFunction);

	TBuf8<128> parameter;

	iEntrySelection->AppendL(iPop3Service);

	iPopServerMtm->StartCommandL(*iEntrySelection,KPOP3MTMDisconnect , parameter, iActiveWaiter->iStatus);

	iActiveWaiter->WaitActive();

	ASSERT_EQUALS( iActiveWaiter->iStatus.Int( ), KErrNone );

	}

void CTestPopSaslAuthHelper::TestForAuthCramMd5L()

	{

	// For CRAM-MD5

	_LIT(KFunction, "TestForAuthCramMd5L");

	INFO_PRINTF1(KFunction);

	iTestPopAuthMechanism = CPopAuthCramMd5MechanismHelper::NewL(*iPopSettings);

	CleanupStack::PushL(iTestPopAuthMechanism);

	TPtr8 popBuffer = iResponseBuffer->Des();

	//popBuffer will contain "AUTH CRAM-MD5"

	iTestPopAuthMechanism->GetNextClientMessageL(popBuffer);

	//INFO_PRINTF1(popBuffer);

	//set dummy server response

	_LIT(KServerResponse,"+ PDE4LjEyMjYzOTQ0MDZARDA3MDAzMS5wcm9kLmFkLnN5bWJpYW4uaW50cmE+");

	popBuffer.Append(KServerResponse);

	iTestPopAuthMechanism->SetLastServerMessageL(popBuffer, ETrue);

	//popBuffer will contain CRAM-MD5 authentication string

	iTestPopAuthMechanism->GetNextClientMessageL(popBuffer);

	//check the length of response buffer 

	TInt error = (popBuffer.Length() > 0 ) ? KErrNone : KErrArgument;

	ASSERT_EQUALS(error, KErrNone);	

	CleanupStack::PopAndDestroy(iTestPopAuthMechanism);

	}

void CTestPopSaslAuthHelper::TestForAuthPlainL()

	{

	// For PLAIN

	_LIT(KFunction, "TestForAuthPlainL");

	INFO_PRINTF1(KFunction);

	iTestPopAuthMechanism = CPopAuthPlainMechanismHelper::NewL(*iPopSettings);

	CleanupStack::PushL(iTestPopAuthMechanism);

	TPtr8 popBuffer = iResponseBuffer->Des();

	iTestPopAuthMechanism->GetNextClientMessageL(popBuffer);

	//check the length of response buffer 

	TInt error = (popBuffer.Length() > 0 ) ? KErrNone : KErrArgument;

	ASSERT_EQUALS(error, KErrNone);	

	CleanupStack::PopAndDestroy(iTestPopAuthMechanism);

	}

void CTestPopSaslAuthHelper::TestForAuthLoginL()

	{

	// For LOGIN

	_LIT(KFunction, "TestForAuthLoginL");

	INFO_PRINTF1(KFunction);

	iTestPopAuthMechanism = CPopAuthLoginMechanismHelper::NewL(*iPopSettings);

	CleanupStack::PushL(iTestPopAuthMechanism);

	TPtr8 popBuffer = iResponseBuffer->Des();

	iTestPopAuthMechanism->GetNextClientMessageL(popBuffer);

	//set dummy server response for Username

	popBuffer.Zero();

	_LIT(KServerResponseForUsername,"+ VXNlcm5hbWU6");

	popBuffer.Append(KServerResponseForUsername);

	iTestPopAuthMechanism->SetLastServerMessageL(popBuffer, ETrue);

	//popBuffer will contain Username in base64 encoded format

	iTestPopAuthMechanism->GetNextClientMessageL(popBuffer);

	//check the length of response buffer 

	TInt error = (popBuffer.Length() > 0 ) ? KErrNone : KErrArgument;

	ASSERT_EQUALS(error, KErrNone);

	//set dummy server response for Password

	popBuffer.Zero();

	_LIT(KServerResponseForPasswd,"+ UGFzc3dvcmQ6");

	popBuffer.Append(KServerResponseForPasswd);

	iTestPopAuthMechanism->SetLastServerMessageL(popBuffer, ETrue);

	//popBuffer will contain Password in base64 encoded format

	iTestPopAuthMechanism->GetNextClientMessageL(popBuffer);

	//check the length of response buffer 

	error = (popBuffer.Length() > 0 ) ? KErrNone : KErrArgument;

	ASSERT_EQUALS(error, KErrNone);

	CleanupStack::PopAndDestroy(iTestPopAuthMechanism);	

	}

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

	{

	SUB_SUITE;

	ADD_ASYNC_TEST_STEP(CreatePopSaslAuthMechanismL);

	END_SUITE;

	}