// 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 "caltestlib.h"

#include <e32std.h>

#include <e32test.h>

#include <calattachment.h>

#include <calattachmentmanager.h>

#include <calentry.h>

#include <calentryview.h>

#include <calsession.h>

RTest test(_L("tcal_attach"));

_LIT(KCalendarFile, "tcal_attach");

_LIT(KExpectedFileName1, "tcal_attach.jpg");

_LIT(KExpectedFileName2, "tcal_attach0.jpg");

_LIT(KFileLocation1, "c:\\calendar_attachments\\tcal_attach.jpg");

_LIT(KFileLocation2, "c:\\calendar_attachments\\tcal_attach2.jpg");

_LIT(KFileLocation3, "c:\\calendar_attachments\\tcal_attach3.txt");

_LIT(KFileLocation4, "c:\\calendar_attachments\\tcal_attach4.doc");

_LIT(KFileLocation5, "c:\\calendar_attachments\\tcal_attach5.bmp");

_LIT(KFileLocation6, "c:\\calendar_attachments\\tcal_attach6.gif");

_LIT(KFileLocation7, "c:\\calendar_attachments\\tcal_attach7.xls");

_LIT(KFileLocation8, "c:\\calendar_attachments\\tcal_attach_huge.txt");

_LIT(KOriginalDrive, "z:");

_LIT(KCalendarAttachmentsDir, "c:\\private\\10003a5b\\tcal_attach_a");

class CTestApp : public CBase

	{

public:

	static CTestApp* NewLC();

	~CTestApp();

	void RunTestsL();

	void RunNonShareProtectedTestsL();

	void SetFileSessionShareProtectedL();

private:

	CTestApp();

	void ConstructL();

	// test functions

	void TestURIAttachmentL();

	void TestBinaryAttachmentL();

	void TestFileAttachmentL(const TDesC& aExpectedFileName);

	void TestCIDAttachmentL();

	void TestSortAttachmentsL();

	void TestMultipleBinaryAttachmentsL();

	void TestMultipleFileAttachmentsL();

	void TestInvalidAttachmentsL(); 

	void TestManyBinaryAttachmentsL();

	void TestComparingAttachmentsL();

	void TestDeletingAttachmentL();

	void TestCopyingEntryWithAttachmentsL();

	void TestCopyingEntryWithAttachments2L();

	void TestAttachmentIndexPersistenceL();

	void TestFetchingHugeFileL();

	void TestDeleteCalendarFileL();

	void TestSameAttachmentWithDifferentDriveL();

	// utility functions

	TCalLocalUid StoreEntryWithAttachmentL(CCalAttachment* aAttachment);

	TCalLocalUid StoreEntryL(CCalEntry* aEntry);

	void ClearAttachmentsL(TCalLocalUid aLuid);

	void InitialiseFileL(CFileMan* aFileMan, const TDesC& aFileName);

	void TestFileHandleL(RFile& aFile, TInt aExpectedSize);

	void CheckDriveAndValueL(const TDesC8& aUid, TDriveName aDriveName);

private:

	CCalTestLibrary* iTestLib;

	CCalAttachmentManager* iAttachmentManager;

	RPIMTestServer iPIMTestServer;

	TInt iEntryCount;

	};

CTestApp::CTestApp()

	{

	}

CTestApp::~CTestApp()

	{

	// try to delete files

	TRAP_IGNORE(iTestLib->DeleteFileL(KCalendarFile, ETrue));

	TRAP_IGNORE(iTestLib->DeleteFileL(KFileLocation1, EFalse));

	TRAP_IGNORE(iTestLib->DeleteFileL(KFileLocation2, EFalse));

	TRAP_IGNORE(iTestLib->DeleteFileL(KFileLocation3, EFalse));

	TRAP_IGNORE(iTestLib->DeleteFileL(KFileLocation4, EFalse));

	TRAP_IGNORE(iTestLib->DeleteFileL(KFileLocation5, EFalse));

	TRAP_IGNORE(iTestLib->DeleteFileL(KFileLocation6, EFalse));

	TRAP_IGNORE(iTestLib->DeleteFileL(KFileLocation7, EFalse));

	TRAP_IGNORE(iTestLib->DeleteFileL(KFileLocation8, EFalse));

	CFileMan* fileCopier=NULL;

	TRAP_IGNORE(fileCopier = CFileMan::NewL(iTestLib->FileSession()));

	TRAP_IGNORE(fileCopier->RmDir(KCalendarAttachmentsDir()));

	if(fileCopier)

		{

		delete fileCopier;

		}

	iPIMTestServer.Close();

	delete iAttachmentManager;

	delete iTestLib;

	}

void CTestApp::RunTestsL()

	{

	TestURIAttachmentL();

	TestBinaryAttachmentL();

	TestFileAttachmentL(KExpectedFileName1);

	//Save the same attachment file to an entry, the file name should be "filenamen" where 'n'is a trailing number 0,1,2,...to avoid redutancy.

    CFileMan* fileCopier = CFileMan::NewL(iTestLib->FileSession());

    CleanupStack::PushL(fileCopier);

	InitialiseFileL(fileCopier, KFileLocation1());

	CleanupStack::PopAndDestroy(fileCopier);

	TestFileAttachmentL(KExpectedFileName2);

	TestCIDAttachmentL();

	TestSortAttachmentsL();

	TestMultipleBinaryAttachmentsL();

	TestMultipleFileAttachmentsL();

	TestInvalidAttachmentsL();

	TestComparingAttachmentsL();

	TestDeletingAttachmentL();

	TestCopyingEntryWithAttachmentsL();

	TestCopyingEntryWithAttachments2L();

	TestAttachmentIndexPersistenceL();

	TestManyBinaryAttachmentsL();

	TestFetchingHugeFileL();

// 	This test need to add _EPOC_DRIVE_F in epoc.ini file

//  Example _EPOC_DRIVE_F = \epoc32\winscw\f_drive"

	TestSameAttachmentWithDifferentDriveL();

	TestDeleteCalendarFileL();

	}

CTestApp* CTestApp::NewLC()

	{

	CTestApp* self = new (ELeave) CTestApp;

	CleanupStack::PushL(self);

	self->ConstructL();

	return (self);

	}

void CTestApp::SetFileSessionShareProtectedL()

	{

	// required for passing attachment file handles to server

	User::LeaveIfError(iTestLib->FileSession().ShareProtected()); 

	}

void CTestApp::ConstructL()

	{	

	User::LeaveIfError(iPIMTestServer.Connect());

	iTestLib = CCalTestLibrary::NewL(EFalse);

	iTestLib->ReplaceFileL(KCalendarFile());

	iTestLib->OpenFileL(KCalendarFile());

	CCalSession& session = iTestLib->GetSession();

	iAttachmentManager = CCalAttachmentManager::NewL(session, *iTestLib);

	CActiveScheduler::Start();

	CFileMan* fileCopier = CFileMan::NewL(iTestLib->FileSession());

	CleanupStack::PushL(fileCopier);

	InitialiseFileL(fileCopier, KFileLocation1());

	InitialiseFileL(fileCopier, KFileLocation2());

	InitialiseFileL(fileCopier, KFileLocation3());

	InitialiseFileL(fileCopier, KFileLocation4());

	InitialiseFileL(fileCopier, KFileLocation5());

	InitialiseFileL(fileCopier, KFileLocation6());

	InitialiseFileL(fileCopier, KFileLocation7());

	fileCopier->RmDir(KCalendarAttachmentsDir());

	CleanupStack::PopAndDestroy(fileCopier);

	}

void CTestApp::InitialiseFileL(CFileMan* aFileMan, const TDesC& aFileName)

	{

	TInt err = iTestLib->FileSession().MkDirAll(aFileName);

	if (err != KErrAlreadyExists)

		{

		User::LeaveIfError(err);

		}

	// symmetric loacation, just change of drive 

	TFileName initialLocationOfFile = aFileName;

	initialLocationOfFile.Replace(0, 2, KOriginalDrive());

	// get current time

	TTime now;

	now.HomeTime();

	// clear any read only attribute if the file is present such we avoid errors at copy time

	aFileMan->Attribs(aFileName, 0, KEntryAttReadOnly, now);

	// copy the file from Z: drive to C: drive (ROM to RAM)	

	User::LeaveIfError(aFileMan->Copy(initialLocationOfFile, aFileName));

	// clear any read only attribute such we avoid errors at open time (true on real hardware)

	User::LeaveIfError(aFileMan->Attribs(aFileName, 0, KEntryAttReadOnly, now));	

	}

// create an appointment for tomorrow with this attachment

// dont push the attachment passed into this function - entry takes ownership

TCalLocalUid CTestApp::StoreEntryWithAttachmentL(CCalAttachment* aAttachment)

	{

	test.Printf(_L("Creating entry with attachment\n"));

	CleanupStack::PushL(aAttachment);

	_LIT8(KBaseGuid, "uid_");

	TBuf8<16> guid;

	guid.Copy(KBaseGuid);

	guid.AppendNum(iEntryCount++);

	HBufC8* guidHBuf = guid.AllocLC();

	CCalEntry* entry = iTestLib->CreateCalEntryL(CCalEntry::EAppt, guidHBuf);

	CleanupStack::Pop(guidHBuf);

	CleanupStack::PushL(entry);

	TTime now;

	now.HomeTime();

	now += TTimeIntervalDays(8);

	TCalTime calTime;

	calTime.SetTimeLocalL(now);

	entry->SetStartAndEndTimeL(calTime, calTime);

	entry->AddAttachmentL(*aAttachment);

	CleanupStack::Pop(entry);

	CleanupStack::Pop(aAttachment);

	CleanupStack::PushL(entry);

	TCalLocalUid localUid = StoreEntryL(entry);

	CleanupStack::PopAndDestroy(entry);

	return localUid;

	}

// store an entry

TCalLocalUid CTestApp::StoreEntryL(CCalEntry* aEntry)

	{

	test.Printf(_L("Storing entry\n"));

	RPointerArray<CCalEntry> entryArray;

	CleanupClosePushL(entryArray);

	entryArray.AppendL(aEntry);

	TInt success = 0;

	iTestLib->SynCGetEntryViewL().StoreL(entryArray, success);

	test(success == entryArray.Count());

	TCalLocalUid localUid = aEntry->LocalUidL();

	test.Printf(_L("entry local uid=%d\n"), localUid);

	CleanupStack::PopAndDestroy(&entryArray); // entryArray.Close()

	return localUid;

	}

// clear all attachments on this entry

void CTestApp::ClearAttachmentsL(TCalLocalUid aLuid)

	{

	test.Printf(_L("Clearing attachments on entry %d\n"), aLuid);

	CCalEntry* entry = iTestLib->SynCGetEntryViewL().FetchL(aLuid);

	CleanupStack::PushL(entry);

	while (entry->AttachmentCountL() > 0)

		{

		CCalAttachment* attach = entry->AttachmentL(0);

		entry->DeleteAttachmentL(*attach);

		}

	StoreEntryL(entry);

	CleanupStack::PopAndDestroy(entry);

	}

void CTestApp::TestFileHandleL(RFile& aFile, TInt aExpectedSize)

	{

	test.Printf(_L("Verifying file handle returned correctly\n"));

	// check file name correct

	TFileName name;

	User::LeaveIfError(aFile.FullName(name));

	name.LowerCase();

	test(name.Find(KCalendarAttachmentsDir()) != KErrNotFound);

	// check file size correct

	TInt actualSize;

	User::LeaveIfError(aFile.Size(actualSize));

	test(actualSize == aExpectedSize);

	// check file cannot be moved

	_LIT(KDummy, "c:\\tcal_attach.dummy");

	test(KErrAccessDenied == aFile.Rename(KDummy()));

	}

//

void CTestApp::TestURIAttachmentL()

	{

	test.Printf(_L("\n"));

	test.Printf(_L("URI attachment\n"));

	test.Printf(_L("==============\n"));

	_LIT8(KTestUri1, "ftp://ftp.bar.com/pub/docs/foo.doc");

	_LIT8(KMimeType1, "application/msword");

	_LIT(KName1, "document name");

	// set up attachment properties

	CCalAttachment* attachment = CCalAttachment::NewUriL(KTestUri1());

	CleanupStack::PushL(attachment);

	attachment->SetMimeTypeL(KMimeType1());

	attachment->SetLabelL(KName1());

	// store an entry with this attachment

	CleanupStack::Pop(attachment);

	TCalLocalUid luid = StoreEntryWithAttachmentL(attachment);

	// fetch the attachments on the entry

	CCalEntry* entry = iTestLib->SynCGetEntryViewL().FetchL(luid);

	CleanupStack::PushL(entry);

	test(entry->AttachmentCountL() == 1);

	CCalAttachment* fetchedAttachment = entry->AttachmentL(0);

	// check the attachment has been stored correctly 

	test.Printf(_L("Checking attachment details\n"));

	test(fetchedAttachment->Type() == CCalAttachment::EUri);

	test(fetchedAttachment->Value() == KTestUri1());

	test(fetchedAttachment->MimeType() == KMimeType1());

	test(fetchedAttachment->Label() == KName1());

	test(fetchedAttachment->FileAttachment() == NULL);

	CleanupStack::PopAndDestroy(entry);

	}

void CTestApp::TestBinaryAttachmentL()

	{

	test.Printf(_L("\n"));

	test.Printf(_L("Binary attachment\n"));

	test.Printf(_L("=================\n"));

	_LIT8(KBinaryData1, "fvgbuygbrvgbrfguyrt4fg67463t23875rt87ty8754yt89y96y0u6908yu896uyrtuyirfuygbcfyugyuvertgvrwsrv67er847rt87ei7rektyiufyhosiuydgdryiu");

	_LIT8(KMimeType1, "dummy/gibberish");

	_LIT(KName1, "binary data attachment name");

	// set up attachment properties

	HBufC8* data = KBinaryData1().AllocLC();

	CCalAttachment* attachment = CCalAttachment::NewFileL(data);

	CleanupStack::Pop(data);

	CleanupStack::PushL(attachment);

	attachment->SetMimeTypeL(KMimeType1());

	attachment->SetLabelL(KName1());

	// store an entry with this attachment

	CleanupStack::Pop(attachment);

	TCalLocalUid luid = StoreEntryWithAttachmentL(attachment);

	// fetch the attachments on the entry

	CCalEntry* entry = iTestLib->SynCGetEntryViewL().FetchL(luid);

	CleanupStack::PushL(entry);

	test(entry->AttachmentCountL() == 1);

	CCalAttachment* fetchedAttachment = entry->AttachmentL(0);

	// check the attachment has been stored correctly 

	test.Printf(_L("Checking attachment details\n"));

	test(fetchedAttachment->Type() == CCalAttachment::EFile);

	test(fetchedAttachment->MimeType() == KMimeType1());

	test(fetchedAttachment->Label() == KName1());

	test(fetchedAttachment->IsAttributeSet(CCalAttachment::EExportInline));

	test(fetchedAttachment->FileAttachment() != NULL);

	test(fetchedAttachment->FileAttachment()->Size() == KBinaryData1().Length());

	_LIT(KExpectedDrive, "C:");

	TDriveName driveName;

	driveName.Copy(fetchedAttachment->FileAttachment()->Drive());

	driveName.UpperCase();

	test(driveName == KExpectedDrive());

	test(fetchedAttachment->Value().Length() == 0);

	fetchedAttachment->FileAttachment()->LoadBinaryDataL();

	test(fetchedAttachment->Value() == KBinaryData1());

	RFile file;

	CleanupClosePushL(file);

	fetchedAttachment->FileAttachment()->FetchFileHandleL(file);

	TestFileHandleL(file, KBinaryData1().Length());

	CleanupStack::PopAndDestroy(&file);

	CleanupStack::PopAndDestroy(entry);

	}

void CTestApp::TestFileAttachmentL(const TDesC& aExpectedFileName)

	{

	test.Printf(_L("\n"));

	test.Printf(_L("File attachment\n"));

	test.Printf(_L("===============\n"));

	_LIT8(KMimeType1, "image/jpeg");

	_LIT(KName1, "snowman");

	RFile file;

	TInt err = file.Open(iTestLib->FileSession(), KFileLocation1(), EFileWrite);

	if (err != KErrNone)

		{

		test.Printf(_L("TestFileAttachmentL, failed to open file %S, err=%d\n"), &KFileLocation1(), err);

		User::Leave(err);

		}

	CleanupClosePushL(file);

	// set up attachment properties

	CCalAttachment* attachment = CCalAttachment::NewFileL(file);

	TTime lastModifiedUtc;

	file.Modified(lastModifiedUtc); // gets time in UTC

    TInt size;

	file.Size(size);

    CleanupStack::PopAndDestroy(&file); // file.Close()

	CleanupStack::PushL(attachment);

	attachment->SetMimeTypeL(KMimeType1());

	attachment->SetLabelL(KName1());

	test(attachment->FileAttachment() != NULL);

	attachment->FileAttachment()->SetLastModifiedTimeUtc(lastModifiedUtc);

	// store an entry with this attachment

	CleanupStack::Pop(attachment);

	TCalLocalUid luid = StoreEntryWithAttachmentL(attachment);

	// fetch the attachments on the entry

	CCalEntry* entry = iTestLib->SynCGetEntryViewL().FetchL(luid);

	CleanupStack::PushL(entry);

	test(entry->AttachmentCountL() == 1);

	CCalAttachment* fetchedAttachment = entry->AttachmentL(0);

	// check the attachment has been stored correctly 

	test.Printf(_L("Checking attachment details\n"));

	test(fetchedAttachment->Type() == CCalAttachment::EFile);

	test(fetchedAttachment->MimeType() == KMimeType1());

	test(fetchedAttachment->Label() == KName1());

	test(fetchedAttachment->IsAttributeSet(CCalAttachment::EExportInline) == EFalse);

	test(fetchedAttachment->FileAttachment() != NULL);

	test(fetchedAttachment->FileAttachment()->Size() == size);

	test(fetchedAttachment->FileAttachment()->LastModifiedTimeUtc() == lastModifiedUtc);

	_LIT(KExpectedDrive, "C:");

	TDriveName driveName;

	driveName.Copy(fetchedAttachment->FileAttachment()->Drive());

	driveName.UpperCase();

	test(driveName == KExpectedDrive());

	test(fetchedAttachment->Value().Length() == 0);

	fetchedAttachment->FileAttachment()->LoadBinaryDataL();

	test(fetchedAttachment->Value().Length() == size);

	RFile file2;

	CleanupClosePushL(file2);

	fetchedAttachment->FileAttachment()->FetchFileHandleL(file2);

	TBuf<100> name;

	file2.Name(name);

	test (aExpectedFileName == name);

	TestFileHandleL(file2, size);

	CleanupStack::PopAndDestroy(&file2);

	CleanupStack::PopAndDestroy(entry);

	}

void CTestApp::TestCIDAttachmentL()

	{

	test.Printf(_L("\n"));

	test.Printf(_L("CID attachment\n"));

	test.Printf(_L("==============\n"));

	_LIT8(KContentId1, "&£$rg$*£$r*(reuf(%ehtg*(e%$y*tgeght*e%y*(t£$*(rt*^85643856438765");

	_LIT8(KMimeType1, "image/jpeg");

	_LIT(KName1, "rodent");

	RFile file;

	TInt err = file.Open(iTestLib->FileSession(), KFileLocation2(), EFileWrite);

	if (err != KErrNone)

		{

		test.Printf(_L("TestCIDAttachmentL, failed to open file %S, err=%d\n"), &KFileLocation2(), err);

		User::Leave(err);

		}

	CleanupClosePushL(file);

	// set up attachment properties

	CCalAttachment* attachment = CCalAttachment::NewFileByContentIdL(KContentId1());

	CleanupStack::PushL(attachment);

	attachment->SetMimeTypeL(KMimeType1());

	attachment->SetLabelL(KName1());

	test(attachment->FileAttachment() != NULL);

	attachment->FileAttachment()->SetResourceL(file);

	TTime lastModifiedUtc;

	file.Modified(lastModifiedUtc); // gets time in UTC

	attachment->FileAttachment()->SetLastModifiedTimeUtc(lastModifiedUtc);

	TInt size;

	file.Size(size);

	// store an entry with this attachment

	CleanupStack::Pop(attachment);

	CleanupStack::PopAndDestroy(&file); // file.Close()

	TCalLocalUid luid = StoreEntryWithAttachmentL(attachment);