// 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:
// Test for CImPruneMessage class. 
// This class removes body text and attachment data from messages, leaving the message structure intact
// 
//

#include "emailtestutils.h"
#include <miutlog.h>
#include <cacheman.h>
#include <cimprunemessage.h>

_LIT(K_PLAIN_MSG, "c:\\mailtest\\rfc822\\plain_text.txt");
_LIT(K_HTML_ATTACHMENT_MSG, "c:\\mailtest\\rfc822\\html_with_attachment.txt" );
_LIT(K_TEST_INFO, "T_CImPruneMessage Test");
_LIT(K_DESCRIPTION, "Testing pruning of messages containing text and attachment.");
_LIT(K_MSG_NOT_PRUNED,"\nWarning email not pruned\n* BODY text or attachment data still exists!*---Message entry id: %d--");
_LIT(K_MSG_PRUNED,"\nSuccess: email  pruned! Message entry id: %d ");
#define KErrMsgNotPruned 5601

RTest test(K_TEST_INFO);
LOCAL_D TMsvId pop3Service;
LOCAL_D CTrapCleanup* theCleanup;
LOCAL_D CEmailTestUtils* testUtils;
LOCAL_D	TInt globalError = KErrNone;


LOCAL_C void InitL()
	{
	CActiveScheduler* scheduler = new (ELeave) CActiveScheduler;
	CActiveScheduler::Install(scheduler);
	CleanupStack::PushL(scheduler);

	testUtils = CEmailTestUtils::NewLC(test);
	testUtils->CreateAllTestDirectories();

	testUtils->WriteComment(K_TEST_INFO);
	
	testUtils->FileSession().SetSessionPath(_L("C:\\"));
	testUtils->CleanMessageFolderL();
	testUtils->ClearEmailAccountsL();
	testUtils->GoServerSideL();
	}
	
LOCAL_C void Closedown()
	{
	CleanupStack::PopAndDestroy(2);  //testUtils, scheduler
	}

LOCAL_C void TestMessagePartsL(CMsvEntry& aEntry)
	{
	CMsvEntrySelection* sel = aEntry.ChildrenL();
	CleanupStack::PushL(sel);
	
	TInt count = sel->Count();
	for(TInt ii=0; ii<sel->Count(); ++ii)
		{
		aEntry.SetEntryL(sel->At(ii));

		CMsvStore* store = aEntry.ReadStoreL();
		CleanupStack::PushL(store);		
		if (store->IsPresentL(KMsvEntryRichTextBody))
			{
			// body text exist	
	    	globalError = KErrMsgNotPruned;
			}

		MMsvAttachmentManager& attachmentMgr = store->AttachmentManagerL();		
		if(attachmentMgr.AttachmentCount() > 0)
			{
			CMsvAttachment* attachmentInfo = attachmentMgr.GetAttachmentInfoL(0);
			CleanupStack::PushL(attachmentInfo);

			if ((attachmentInfo->Type() == CMsvAttachment::EMsvFile) && (attachmentInfo->Size() != 0))
				{
				// attachment data exist	
		   	 	globalError = KErrMsgNotPruned;			
				}
			CleanupStack::PopAndDestroy(attachmentInfo);	
			}
		CleanupStack::PopAndDestroy(store);	

		
		// recursively test other message parts
		TestMessagePartsL(aEntry);
		}
	CleanupStack::PopAndDestroy(sel);
	}

LOCAL_C void doMainL()
	{
	// CImPruneMessage class removes body text and attachment data from messages, 
	// leaving the message structure intact	
	
	InitL();

	testUtils->FileSession().SetSessionPath(_L("c:\\"));

	testUtils->GoClientSideL();
	pop3Service = testUtils->CreatePopServiceL();

	testUtils->GoServerSideL();
  
	testUtils->CreateMessageL(K_PLAIN_MSG, pop3Service, pop3Service);
	testUtils->CreateMessageL(K_HTML_ATTACHMENT_MSG, pop3Service, pop3Service);
	testUtils->CreateMessageL(K_PLAIN_MSG, pop3Service, pop3Service);	
	testUtils->CreateMessageL(K_HTML_ATTACHMENT_MSG, pop3Service, pop3Service);	
 
	test.Printf(_L("\nPerforming Prune Message Tests\n"));
	
	testUtils->GoClientSideL();

	testUtils->iMsvEntry->SetEntryL(pop3Service);
	TMsvEntry entry = testUtils->iMsvEntry->Entry();
	entry.SetVisible(ETrue);
	testUtils->iMsvEntry->ChangeL(entry);
	
	CImPruneMessage* pruneMessage = CImPruneMessage::NewL(*testUtils->iMsvEntry, testUtils->FileSession());
	CleanupStack::PushL(pruneMessage);
	CTestActive* testActive = new (ELeave) CTestActive();
	CleanupStack::PushL(testActive);

	testActive->StartL();

	testUtils->TestStart(0, K_DESCRIPTION);

	pruneMessage->StartL(pop3Service, testActive->iStatus);

	CActiveScheduler::Start();

	
	// Get a list of pop messages (created above)
	testUtils->iMsvEntry->SetEntryL(pop3Service);
	CMsvEntrySelection* popMessages = testUtils->iMsvEntry->ChildrenL();	
    CleanupStack::PushL(popMessages);
    
    TInt messageCount = popMessages->Count();
   	TMsvId msgId;
   	TBuf<128> msgFail;
   	TBuf<128> msgPass;    
    
   	for(TInt ii=0; ii<messageCount; ++ii)
		{
		// Check that there are no body text and no data in the attachment file for this message
    	msgId = popMessages->At(ii);
    	testUtils->iMsvEntry->SetEntryL(msgId);
  
  		// Go through the message parts and check that body text and attachment data are removed		    		
	    TestMessagePartsL(*testUtils->iMsvEntry);
	    
		if (globalError == KErrNone)
			{		
		    //email successfully prunned
		    msgPass.Format(K_MSG_PRUNED,msgId);		    	
		    testUtils->WriteComment(msgPass);
			}			
		else
			{
		    msgFail.Format(K_MSG_NOT_PRUNED,msgId);
		    testUtils->WriteComment(msgFail);	
		    break;	
			}	   	
	   	}
			
	testUtils->TestFinish(0, globalError);
	
	if (globalError == KErrNone)
		{
		testUtils->TestHarnessCompleted();
		}
	else
		{
		testUtils->TestHarnessFailed(globalError);
		}

	CleanupStack::PopAndDestroy(3,pruneMessage); // pruneMessage, testActive, popMessages

	Closedown();
	}

GLDEF_C TInt E32Main()
	{	
	__UHEAP_MARK;
	test.Start(_L("T_CImPruneMessage Test\n"));
	theCleanup=CTrapCleanup::New();
	TRAPD(ret,doMainL());		
	test(ret==KErrNone);
	delete theCleanup;	
	test.End();
	test.Close();
	__UHEAP_MARKEND;
	User::Heap().Check();
	return(KErrNone);
	}