diff -r 2981cb3aa489 -r 8b14b30db193 email/imap4mtm/imapsession/test/src/ctestimapstore.cpp
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/email/imap4mtm/imapsession/test/src/ctestimapstore.cpp	Wed Nov 03 22:55:44 2010 +0530
@@ -0,0 +1,213 @@
+// 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 "ctestimapstore.h"
+
+#include "cfakeinputstream.h"
+#include "cfakeoutputstream.h"
+#include "cactivewaiter.h"
+
+#include "moutputstream.h"
+#include "cimapsession.h"
+#include "cimaputils.h"
+#include "cimapfolderinfo.h"
+
+_LIT(KMessageIdAndFlags, "%d %S");
+
+CTestImapStore::CTestImapStore()
+	: iInputStream(NULL)
+	, iOutputStream(NULL)
+	, iActiveWaiter(NULL)
+	, iImapSession(NULL)
+	{}
+
+CTestImapStore::~CTestImapStore()
+	{
+	delete iImapSession;
+	delete iActiveWaiter;
+	delete iOutputStream;
+	delete iInputStream;
+	CImapUtils::Delete();
+	}
+
+void CTestImapStore::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);
+	
+	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();
+	
+	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;
+	}
+	
+void CTestImapStore::TearDownL()
+	{
+	delete iImapSession;
+	iImapSession = NULL;
+	
+	delete iActiveWaiter;
+	iActiveWaiter = NULL;
+	
+	delete iOutputStream;
+	iOutputStream = NULL;
+	
+	delete iInputStream;
+	iInputStream = NULL;
+
+	CImapUtils::Delete();
+	}
+
+// Tests
+void CTestImapStore::TestStoreWithFlagsL()
+	{
+	INFO_PRINTF1(_L("TestStoreWithFlagsL"));
+	iInputStream->ResetInputStrings();
+	iInputStream->AppendInputStringL(_L8("* 2 FETCH (FLAGS (\\Deleted \\Seen) UID 2)\r\n"));
+	iInputStream->AppendInputStringL(_L8("* 3 FETCH (FLAGS (\\Deleted) UID 3)\r\n"));
+	iInputStream->AppendInputStringL(_L8("* 4 FETCH (FLAGS (\\Deleted \\Flagged \\Seen) UID 4)\r\n"));
+	iInputStream->AppendInputStringL(_L8("* 5 FETCH (FLAGS (\\Answered \\Flagged \\Deleted \\Seen \\Draft) UID 5)\r\n"));
+	iInputStream->AppendInputStringL(_L8("* 6 FETCH (FLAGS (\\Answered \\Flagged \\Deleted \\Seen \\Draft))\r\n"));
+	iInputStream->AppendInputStringL(_L8("3 OK STORE completed\r\n"));
+	
+	RArrayMessageFlagInfo messageFlagInfo;
+	iImapSession->StoreL(iActiveWaiter->iStatus, _L8("2:5"), _L8("+FLAGS"), _L8("(Deleted)"), 0, messageFlagInfo);
+	iActiveWaiter->WaitActive();
+
+	INFO_PRINTF1(_L("FLAGS LIST"));
+	
+	TInt countMessageFlagInfo = messageFlagInfo.Count();
+	ASSERT_EQUALS(countMessageFlagInfo, 4);
+	
+	for (TInt i = 0; i < countMessageFlagInfo; ++i)
+		{
+		// create account Id
+		TMessageFlagInfo entry = messageFlagInfo[i];
+		TUint messageUid = entry.MessageUid();
+		TBuf<100> mesageUidAndFlags;
+		TPtrC flag;
+		if (entry.QueryFlag(TMessageFlagInfo::EDeleted) && entry.QueryFlag(TMessageFlagInfo::EFlagged) 
+			&& entry.QueryFlag(TMessageFlagInfo::ESeen) && entry.QueryFlag(TMessageFlagInfo::EDraft) && entry.QueryFlag(TMessageFlagInfo::EAnswered))
+			{
+			// Check the other flags are false
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::ERecent));
+
+			// Check that this is the expected entry
+			ASSERT_EQUALS(i, 3);
+			
+			flag.Set(_L("Answered Flagged Deleted Seen Draft"));
+			mesageUidAndFlags.AppendFormat(KMessageIdAndFlags, messageUid, &flag);
+			INFO_PRINTF1(mesageUidAndFlags);	
+			}
+		else if (entry.QueryFlag(TMessageFlagInfo::EDeleted) && entry.QueryFlag(TMessageFlagInfo::EFlagged) && entry.QueryFlag(TMessageFlagInfo::ESeen))
+			{
+			// Check the other flags are false
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::EAnswered));			
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::EDraft));
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::ERecent));
+
+			// Check that this is the expected entry
+			ASSERT_EQUALS(i, 2);
+			
+			flag.Set(_L("Deleted Flagged Seen"));
+			mesageUidAndFlags.AppendFormat(KMessageIdAndFlags, messageUid, &flag);
+			INFO_PRINTF1(mesageUidAndFlags);	
+			}
+		else if (entry.QueryFlag(TMessageFlagInfo::EDeleted) && entry.QueryFlag(TMessageFlagInfo::ESeen))
+			{
+			// Check the other flags are false
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::EAnswered));
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::EFlagged));			
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::EDraft));
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::ERecent));
+
+			// Check that this is the expected entry
+			ASSERT_EQUALS(i, 0);
+			
+			flag.Set(_L("Deleted Seen"));
+			mesageUidAndFlags.AppendFormat(KMessageIdAndFlags, messageUid, &flag);
+			INFO_PRINTF1(mesageUidAndFlags);	
+			}
+		else if (entry.QueryFlag(TMessageFlagInfo::EDeleted))
+			{
+			// Check the other flags are false
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::ESeen));
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::EAnswered));
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::EFlagged));
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::EDraft));
+			ASSERT_FALSE(entry.QueryFlag(TMessageFlagInfo::ERecent));
+
+			// Check that this is the expected entry
+			ASSERT_EQUALS(i, 1);
+			
+			flag.Set(_L("Deleted"));
+			mesageUidAndFlags.AppendFormat(KMessageIdAndFlags, messageUid, &flag);
+			INFO_PRINTF1(mesageUidAndFlags);	
+			}
+		}
+	messageFlagInfo.Reset();
+	messageFlagInfo.Close();
+	INFO_PRINTF1(_L("Complete"));
+	}
+	
+CTestSuite* CTestImapStore::CreateSuiteL(const TDesC& aName)
+// static
+	{
+	SUB_SUITE;
+	ADD_ASYNC_TEST_STEP(TestStoreWithFlagsL);
+	END_SUITE;
+	}