--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/email/imap4mtm/imapsession/test/src/ctestimapfetchflags.cpp	Thu Dec 17 08:44:11 2009 +0200
@@ -0,0 +1,328 @@
+// 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 "ctestimapfetchflags.h"
+
+#include "cfakeinputstream.h"
+#include "cfakeoutputstream.h"
+#include "cactivewaiter.h"
+
+#include "moutputstream.h"
+#include "cimapsession.h"
+#include "cimapsessionconsts.h"
+#include "cimapfolderinfo.h"
+#include "cimaputils.h"
+
+CTestImapFetchFlags::CTestImapFetchFlags()
+	: iInputStream(NULL)
+	, iOutputStream(NULL)
+	, iActiveWaiter(NULL)
+	, iImapSession(NULL)
+	{}
+
+CTestImapFetchFlags::~CTestImapFetchFlags()
+	{
+	delete iImapSession;
+	delete iActiveWaiter;
+	delete iOutputStream;
+	delete iInputStream;
+	CImapUtils::Delete();
+	}
+
+void CTestImapFetchFlags::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 CTestImapFetchFlags::TearDownL()
+	{
+	delete iImapSession;
+	iImapSession = NULL;
+	
+	delete iActiveWaiter;
+	iActiveWaiter = NULL;
+	
+	delete iOutputStream;
+	iOutputStream = NULL;
+	
+	delete iInputStream;
+	iInputStream = NULL;
+
+	CImapUtils::Delete();
+	}
+
+// Tests
+void CTestImapFetchFlags::TestFetchFlagsWithFlagsL()
+	{
+	INFO_PRINTF1(_L("TestFetchFlagsWithFlagsL"));
+	iInputStream->ResetInputStrings();
+
+	iInputStream->AppendInputStringL(_L8("* 12 EXISTS\r\n"));
+	iInputStream->AppendInputStringL(_L8("* 1 FETCH (UID 18 FLAGS (\\Seen \\Recent))\r\n"));
+	iInputStream->AppendInputStringL(_L8("* 2 FETCH (UID 19 FLAGS (\\Seen \\Draft \\Recent))\r\n"));
+	iInputStream->AppendInputStringL(_L8("* 3 FETCH (FLAGS (\\Seen \\Deleted) UID 20)\r\n"));
+	iInputStream->AppendInputStringL(_L8("3 OK completed\r\n"));
+		
+	RArrayMessageFlagInfo messageFlagInfo;
+	iImapSession->FetchFlagsL(iActiveWaiter->iStatus, _L8("18:20"), messageFlagInfo);
+	iActiveWaiter->WaitActive();
+
+	ASSERT_EQUALS(messageFlagInfo.Count(), 3);
+	ASSERT_EQUALS(messageFlagInfo[0].MessageUid(), (TUint)18);
+	ASSERT_TRUE(messageFlagInfo[0].QueryFlag(TMessageFlagInfo::ESeen));
+	ASSERT_FALSE(messageFlagInfo[0].QueryFlag(TMessageFlagInfo::EAnswered));
+	ASSERT_FALSE(messageFlagInfo[0].QueryFlag(TMessageFlagInfo::EFlagged));
+	ASSERT_FALSE(messageFlagInfo[0].QueryFlag(TMessageFlagInfo::EDeleted));
+	ASSERT_FALSE(messageFlagInfo[0].QueryFlag(TMessageFlagInfo::EDraft));
+	ASSERT_TRUE(messageFlagInfo[0].QueryFlag(TMessageFlagInfo::ERecent));
+	
+	ASSERT_EQUALS(messageFlagInfo[1].MessageUid(), (TUint)19);
+	ASSERT_TRUE(messageFlagInfo[1].QueryFlag(TMessageFlagInfo::ESeen));
+	ASSERT_FALSE(messageFlagInfo[1].QueryFlag(TMessageFlagInfo::EAnswered));
+	ASSERT_FALSE(messageFlagInfo[1].QueryFlag(TMessageFlagInfo::EFlagged));
+	ASSERT_FALSE(messageFlagInfo[1].QueryFlag(TMessageFlagInfo::EDeleted));
+	ASSERT_TRUE(messageFlagInfo[1].QueryFlag(TMessageFlagInfo::EDraft));
+	ASSERT_TRUE(messageFlagInfo[1].QueryFlag(TMessageFlagInfo::ERecent));
+	
+	ASSERT_EQUALS(messageFlagInfo[2].MessageUid(), (TUint)20);
+	ASSERT_TRUE(messageFlagInfo[2].QueryFlag(TMessageFlagInfo::ESeen));
+	ASSERT_FALSE(messageFlagInfo[2].QueryFlag(TMessageFlagInfo::EAnswered));
+	ASSERT_FALSE(messageFlagInfo[2].QueryFlag(TMessageFlagInfo::EFlagged));
+	ASSERT_TRUE(messageFlagInfo[2].QueryFlag(TMessageFlagInfo::EDeleted));
+	ASSERT_FALSE(messageFlagInfo[2].QueryFlag(TMessageFlagInfo::EDraft));
+	ASSERT_FALSE(messageFlagInfo[2].QueryFlag(TMessageFlagInfo::ERecent));
+
+	CImapFolderInfo* folderInfo = iImapSession->SelectedFolderInfo();
+	ASSERT_NOT_NULL(folderInfo);
+	ASSERT_FALSE(folderInfo->MessageFlagsChanged());	
+
+	messageFlagInfo.Reset();
+
+	INFO_PRINTF1(_L("Complete"));
+	}
+	
+/**
+Tests that unknown extension flags are succsfully ignored without crashing
+*/
+void CTestImapFetchFlags::TestFetchFlagsWithUnknownExtensionFlagsL()
+	{
+	INFO_PRINTF1(_L("TestFetchFlagsWithUnknownExtensionFlagsL"));
+	iInputStream->ResetInputStrings();
+
+	iInputStream->AppendInputStringL(_L8("* 12 EXISTS\r\n"));
+	// Try a couple of flag-extension's.
+	iInputStream->AppendInputStringL(_L8("* 1 FETCH (UID 18 FLAGS (\\FlagExtension \\Recent))\r\n"));
+	iInputStream->AppendInputStringL(_L8("* 2 FETCH (UID 19 FLAGS (\\Seen \\Unread \\Recent))\r\n"));
+	// Try a flag-keyword's (defined as an atom without the forward slash)
+	iInputStream->AppendInputStringL(_L8("* 3 FETCH (FLAGS (\\Seen Keyword) UID 20)\r\n"));
+	iInputStream->AppendInputStringL(_L8("3 OK completed\r\n"));
+		
+	RArrayMessageFlagInfo messageFlagInfo;
+	iImapSession->FetchFlagsL(iActiveWaiter->iStatus, _L8("18:20"), messageFlagInfo);
+	iActiveWaiter->WaitActive();
+
+	ASSERT_EQUALS(messageFlagInfo.Count(), 3);
+	ASSERT_EQUALS(messageFlagInfo[0].MessageUid(), (TUint)18);
+	ASSERT_FALSE(messageFlagInfo[0].QueryFlag(TMessageFlagInfo::ESeen));
+	ASSERT_FALSE(messageFlagInfo[0].QueryFlag(TMessageFlagInfo::EAnswered));
+	ASSERT_FALSE(messageFlagInfo[0].QueryFlag(TMessageFlagInfo::EFlagged));
+	ASSERT_FALSE(messageFlagInfo[0].QueryFlag(TMessageFlagInfo::EDeleted));
+	ASSERT_FALSE(messageFlagInfo[0].QueryFlag(TMessageFlagInfo::EDraft));
+	ASSERT_TRUE(messageFlagInfo[0].QueryFlag(TMessageFlagInfo::ERecent));
+	
+	ASSERT_EQUALS(messageFlagInfo[1].MessageUid(), (TUint)19);
+	ASSERT_TRUE(messageFlagInfo[1].QueryFlag(TMessageFlagInfo::ESeen));
+	ASSERT_FALSE(messageFlagInfo[1].QueryFlag(TMessageFlagInfo::EAnswered));
+	ASSERT_FALSE(messageFlagInfo[1].QueryFlag(TMessageFlagInfo::EFlagged));
+	ASSERT_FALSE(messageFlagInfo[1].QueryFlag(TMessageFlagInfo::EDeleted));
+	ASSERT_FALSE(messageFlagInfo[1].QueryFlag(TMessageFlagInfo::EDraft));
+	ASSERT_TRUE(messageFlagInfo[1].QueryFlag(TMessageFlagInfo::ERecent));
+	
+	ASSERT_EQUALS(messageFlagInfo[2].MessageUid(), (TUint)20);
+	ASSERT_TRUE(messageFlagInfo[2].QueryFlag(TMessageFlagInfo::ESeen));
+	ASSERT_FALSE(messageFlagInfo[2].QueryFlag(TMessageFlagInfo::EAnswered));
+	ASSERT_FALSE(messageFlagInfo[2].QueryFlag(TMessageFlagInfo::EFlagged));
+	ASSERT_FALSE(messageFlagInfo[2].QueryFlag(TMessageFlagInfo::EDeleted));
+	ASSERT_FALSE(messageFlagInfo[2].QueryFlag(TMessageFlagInfo::EDraft));
+	ASSERT_FALSE(messageFlagInfo[2].QueryFlag(TMessageFlagInfo::ERecent));
+
+	messageFlagInfo.Reset();
+
+	INFO_PRINTF1(_L("Complete"));
+	}	
+	
+/**
+Check that a flags response without a UID is classed as an unsolicitied
+response and is not stored in the message flag info.
+*/
+void CTestImapFetchFlags::TestFetchFlagsWithMissingUidL()
+	{
+	INFO_PRINTF1(_L("TestFetchFlagsWithMissingUidL"));
+	
+	iInputStream->ResetInputStrings();
+	iInputStream->AppendInputStringL(_L8("* 1 FETCH (FLAGS (\\Seen \\Recent))\r\n")); // Missing UID
+	iInputStream->AppendInputStringL(_L8("3 OK completed\r\n"));
+
+	RArrayMessageFlagInfo messageFlagInfo;
+	iImapSession->FetchFlagsL(iActiveWaiter->iStatus, _L8("18:20"), messageFlagInfo);
+	iActiveWaiter->WaitActive();
+
+	ASSERT_EQUALS(messageFlagInfo.Count(), 0);
+	messageFlagInfo.Reset();
+
+	INFO_PRINTF1(_L("Complete"));
+	}
+	
+/**
+Check that corrupt data is detected without causing a panic and without hanging.
+*/
+void CTestImapFetchFlags::TestFetchFlagsCorruptMissingFlagsL()
+	{
+	INFO_PRINTF1(_L("TestFetchFlagsCorruptMissingFlagsL"));
+	
+	iInputStream->ResetInputStrings();
+	iInputStream->AppendInputStringL(_L8("* 2 FETCH (UID 19)\r\n")); // Missing FLAGS
+	iInputStream->AppendInputStringL(_L8("3 OK completed\r\n"));
+		
+	RArrayMessageFlagInfo messageFlagInfo;
+	iImapSession->FetchFlagsL(iActiveWaiter->iStatus, _L8("18:20"), messageFlagInfo);
+	iActiveWaiter->WaitActive(KErrImapCorrupt);
+
+	messageFlagInfo.Reset();
+
+	INFO_PRINTF1(_L("Complete"));
+	}
+	
+/**
+Check that corrupt data is detected without causing a panic and without hanging.
+*/
+void CTestImapFetchFlags::TestFetchFlagsCorruptNoDataItemsL()
+	{
+	INFO_PRINTF1(_L("TestFetchFlagsCorruptNoDataItemsL"));
+	
+	iInputStream->ResetInputStrings();
+	iInputStream->AppendInputStringL(_L8("* 3 FETCH ()\r\n")); // No data items
+	iInputStream->AppendInputStringL(_L8("3 OK completed\r\n"));
+		
+	RArrayMessageFlagInfo messageFlagInfo;
+	iImapSession->FetchFlagsL(iActiveWaiter->iStatus, _L8("18:20"), messageFlagInfo);
+	iActiveWaiter->WaitActive(KErrImapCorrupt);
+	
+	messageFlagInfo.Reset();
+
+	INFO_PRINTF1(_L("Complete"));
+	}
+	
+/**
+Check that corrupt data is detected without causing a panic and without hanging.
+*/
+void CTestImapFetchFlags::TestFetchFlagsCorruptNoFetchDataL()
+	{
+	INFO_PRINTF1(_L("TestFetchFlagsCorruptNoFetchDataL"));
+	iInputStream->ResetInputStrings();
+
+	iInputStream->AppendInputStringL(_L8("* 3 FETCH\r\n")); // No FETCH data at all
+	iInputStream->AppendInputStringL(_L8("3 OK completed\r\n"));
+		
+	RArrayMessageFlagInfo messageFlagInfo;
+	iImapSession->FetchFlagsL(iActiveWaiter->iStatus, _L8("18:20"), messageFlagInfo);
+	iActiveWaiter->WaitActive(KErrImapCorrupt);
+	
+	messageFlagInfo.Reset();
+
+	INFO_PRINTF1(_L("Complete"));
+	}
+	
+/**
+Check that corrupt data is detected without causing a panic and without hanging.
+*/
+void CTestImapFetchFlags::TestFetchFlagsCorruptMismatchedTagL()
+	{
+	INFO_PRINTF1(_L("TestFetchFlagsCorruptMismatchedTagL"));
+	iInputStream->ResetInputStrings();
+
+	iInputStream->AppendInputStringL(_L8("* 1 FETCH (FLAGS (\\Seen \\Recent) UID 18)\r\n")); // OK
+	iInputStream->AppendInputStringL(_L8("4 OK completed\r\n")); // Mismatched
+		
+	RArrayMessageFlagInfo messageFlagInfo;
+	iImapSession->FetchFlagsL(iActiveWaiter->iStatus, _L8("18:20"), messageFlagInfo);
+	iActiveWaiter->WaitActive(KErrImapCorrupt);
+	
+	messageFlagInfo.Reset();
+
+	INFO_PRINTF1(_L("Complete"));
+	}
+	
+CTestSuite* CTestImapFetchFlags::CreateSuiteL(const TDesC& aName)
+// static
+	{
+	SUB_SUITE;
+	ADD_ASYNC_TEST_STEP(TestFetchFlagsWithFlagsL);
+	ADD_ASYNC_TEST_STEP(TestFetchFlagsWithUnknownExtensionFlagsL);
+	ADD_ASYNC_TEST_STEP(TestFetchFlagsWithMissingUidL);
+	ADD_ASYNC_TEST_STEP(TestFetchFlagsCorruptMissingFlagsL);
+	ADD_ASYNC_TEST_STEP(TestFetchFlagsCorruptNoDataItemsL);
+	ADD_ASYNC_TEST_STEP(TestFetchFlagsCorruptNoFetchDataL);
+	ADD_ASYNC_TEST_STEP(TestFetchFlagsCorruptMismatchedTagL);
+	END_SUITE;
+	}