// Copyright (c) 1998-2010 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 <s32file.h>

#include <s32mem.h>

#include <e32test.h>

const TInt KTestCleanupStack=0x20;

// This is a path specification and should not be used as is

_LIT(KFileLocationSpec, "Z:\\STOR-TST\\T_FSTRM.DAT");

const TUint8* KTestData=_S8("0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ");

const TInt KTestLength=36;

const TInt KTestTotal=KTestLength*(KTestLength+1);

const TPtrC8 KTestDes(KTestData,KTestLength);

LOCAL_D CTrapCleanup* TheTrapCleanup;

LOCAL_D RTest test(_L("t_storfstrm"));

LOCAL_D RFs TheFs;

LOCAL_D TFileName TheTempFile;

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1162

@SYMTestCaseDesc	    Tests for writing to a stream.

@SYMTestPriority 	    High

@SYMTestActions  	    Attempt for writing to a stream

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testWriteL(RWriteStream& aStream)

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1162 Writing... "));

	for (TInt i=0;i<=KTestLength;++i)

		{

		aStream.WriteL(KTestDes,i);

		aStream.WriteL(&KTestData[i],KTestLength-i);

		}

	}

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1163

@SYMTestCaseDesc	    Tests for reading from a stream.

@SYMTestPriority 	    High

@SYMTestActions  	    Attempt for reading from a stream

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testReadL(RReadStream& aStream)

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1163 Reading... "));

	for (TInt i=KTestLength;i>=0;--i)

		{

		TBuf8<KTestLength+1> buf;

		aStream.ReadL(buf,i);

		test(buf.Length()==i);

		buf.SetMax();

		aStream.ReadL(&buf[i],KTestLength-i);

		buf.SetLength(KTestLength);

		test(buf==KTestDes);

		}

	}

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1164

@SYMTestCaseDesc	    Tests for skipping data on a stream.

@SYMTestPriority 	    High

@SYMTestActions  	    Tests for skipping data while reading from a stream

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testSkipL(RReadStream& aStream)

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1164 Skipping... "));

	for (TInt i=0;i<=KTestLength;++i)

		{

		aStream.ReadL(i);

		aStream.ReadL(KTestLength-i);

		}

	}

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1165

@SYMTestCaseDesc	    Tests a stream is at end-of-file.

@SYMTestPriority 	    High

@SYMTestActions  	    Tests for end of file while reading from a stream

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testEofL(RReadStream& aStream)

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1165 At end "));

	TUint8 b;

	test(aStream.Source()->ReadL(&b,1)==0);

	}

//

// Test copying from one stream to another.

//

LOCAL_C void testCopyL(RWriteStream& aWriteStream,RReadStream& aReadStream)

	{

	test.Next(_L("Copying"));

	for (TInt i=KTestLength;i>=0;--i)

		{

		aWriteStream.WriteL(aReadStream,i);

		aReadStream.ReadL(aWriteStream,KTestLength-i);

		}

	}

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1166

@SYMTestCaseDesc	    Tests for writing to a file stream.

						Tests RFileWriteStream::WriteL() function

@SYMTestPriority 	    High

@SYMTestActions  	    Tests for writing to replaced,temporary,opened,created file.

						Tests for creating an already existing file.

						Tests for panic while deleting a file.

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testWriteL()

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1166 Writing to attached file "));

	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	

	TParse parse;

	parse.Set(drive.Name(), &KFileLocationSpec, NULL);

	RFile file;

	if (file.Replace(TheFs,parse.NameAndExt(),EFileWrite)!=KErrNone)

		test.Panic(_L("Replacing file"));

	RFile f=file;

	RFileWriteStream out(f);

	testWriteL(out);

	out.CommitL();

	out.Attach(file);

	testWriteL(out);

	out.Close();

//

	test.Next(_L("Writing to replaced file"));

	test(out.Replace(TheFs,parse.NameAndExt(),EFileWrite)==KErrNone);

	testWriteL(out);

	out.Close();

//

	test.Next(_L("Writing to temp file"));

	test(out.Temp(TheFs,parse.DriveAndPath(),TheTempFile,EFileWrite)==KErrNone);

	testWriteL(out);

	out.CommitL();

	out.Close();

//

	test.Next(_L("Writing to opened file"));

	test(out.Open(TheFs,parse.NameAndExt(),EFileWrite)==KErrNone);

	testWriteL(out);

//

	test.Next(_L("Failing to create existing file"));

	test(out.Create(TheFs,TheTempFile,EFileWrite)==KErrAlreadyExists);

	testWriteL(out);

	out.Close();

	if (TheFs.Delete(parse.NameAndExt())!=KErrNone)

		test.Panic(_L("Deleting file"));

//

	test.Next(_L("Writing to created file"));

	test(out.Create(TheFs,parse.NameAndExt(),EFileWrite)==KErrNone);

	testWriteL(out);

	out.CommitL();

	out.Close();

	}

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1167

@SYMTestCaseDesc	    Tests reading from a file stream.

@SYMTestPriority 	    High

@SYMTestActions  	    Attempt for reading from an attached file

						Attempt for reading from opened file

						Attempt for reading from temp file.Test for end of file error

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testReadL()

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1167 Reading from attached file "));

	TParsePtrC parse(KFileLocationSpec);

	RFile file;

	if (file.Open(TheFs,parse.NameAndExt(),EFileRead)!=KErrNone)

		test.Panic(_L("Opening file"));

	RFile f=file;

	RFileReadStream in(f);

	testReadL(in);

	testEofL(in);

	in.Attach(file);

	testReadL(in);

	testEofL(in);

	in.Close();

//

	test.Next(_L("Reading from opened file"));

	test(in.Open(TheFs,parse.NameAndExt(),EFileRead)==KErrNone);

	testReadL(in);

	testEofL(in);

	in.Close();

//

	test.Next(_L("Reading from temp file"));

	test(in.Open(TheFs,TheTempFile,EFileRead)==KErrNone);

	testReadL(in);

	testEofL(in);

	in.Close();

	}

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1168

@SYMTestCaseDesc	    Skipping on a file stream test

@SYMTestPriority 	    High

@SYMTestActions  	    Attempt for skipping data while reading from a stream

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testSkipL()

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1168 Skipping using small transfers "));

	TParsePtrC parse(KFileLocationSpec);

	RFileReadStream in;

	test(in.Open(TheFs,parse.NameAndExt(),EFileRead)==KErrNone);

	testSkipL(in);

	testEofL(in);

	in.Close();

//

	test.Next(_L("Skipping using a single big transfer"));

	test(in.Open(TheFs,parse.NameAndExt(),EFileRead)==KErrNone);

	in.ReadL(KTestTotal);

	testEofL(in);

	in.Close();

	}

/**

@SYMTestCaseID          SYSLIB-STORE-CT-1169

@SYMTestCaseDesc	    Copying from one file stream to another test

@SYMTestPriority 	    High

@SYMTestActions  	    Attempt for copying using small transfers.

                        Attempt for copying until end of file.

						Attempt for end of file error

@SYMTestExpectedResults Test must not fail

@SYMREQ                 REQ0000

*/

LOCAL_C void testCopyL()

	{

	test.Next(_L(" @SYMTestCaseID:SYSLIB-STORE-CT-1169 Copying using small transfers "));

	TParsePtrC parse(KFileLocationSpec);

	RFileReadStream in;

	RFileWriteStream out;

	test(in.Open(TheFs,parse.NameAndExt(),EFileRead)==KErrNone);

	test(out.Replace(TheFs,TheTempFile,EFileWrite)==KErrNone);

	testCopyL(out,in);

	testEofL(in);

	out.CommitL();

	out.Close();

	in.Close();

	test(in.Open(TheFs,TheTempFile,EFileRead)==KErrNone);

	testReadL(in);

	testEofL(in);

	in.Close();

//

	test.Next(_L("Copying using a single big transfer"));

	test(out.Replace(TheFs,TheTempFile,EFileWrite)==KErrNone);

	test(in.Open(TheFs,parse.NameAndExt(),EFileRead)==KErrNone);

	in.ReadL(out,KTestTotal);

	testEofL(in);

	in.Close();

	test(in.Open(TheFs,parse.NameAndExt(),EFileRead)==KErrNone);

	out.WriteL(in,KTestTotal);

	testEofL(in);

	in.Close();

	out.CommitL();

	out.Close();

	test(in.Open(TheFs,TheTempFile,EFileRead)==KErrNone);

	testReadL(in);

	testReadL(in);

	testEofL(in);

	in.Close();

//

	test.Next(_L("Copying until end of file"));

	test(out.Replace(TheFs,TheTempFile,EFileWrite)==KErrNone);

	test(in.Open(TheFs,parse.NameAndExt(),EFileRead)==KErrNone);

	in.ReadL(out);

	testEofL(in);

	in.Close();

	test(in.Open(TheFs,parse.NameAndExt(),EFileRead)==KErrNone);

	out.WriteL(in);

	testEofL(in);

	in.Close();

	out.CommitL();

	out.Close();

	test(in.Open(TheFs,TheTempFile,EFileRead)==KErrNone);

	testReadL(in);

	testReadL(in);

	testEofL(in);

	in.Close();

	}

//

// Prepare the test directory.

//

LOCAL_C void setupTestDirectory()

    {

	TInt r=TheFs.Connect();

	test(r==KErrNone);

//

	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	

	TParse parse;

	parse.Set(drive.Name(), &KFileLocationSpec, NULL);

	r=TheFs.MkDir(parse.DriveAndPath());

	test(r==KErrNone||r==KErrAlreadyExists);

	r=TheFs.SetSessionPath(parse.DriveAndPath());

	test(r==KErrNone);

	}

//

// Initialise the cleanup stack.

//

LOCAL_C void setupCleanup()

    {

	TheTrapCleanup=CTrapCleanup::New();

	test(TheTrapCleanup!=NULL);

	TRAPD(r,\

		{\

		for (TInt i=KTestCleanupStack;i>0;i--)\

			CleanupStack::PushL((TAny*)1);\

		test(r==KErrNone);\

		CleanupStack::Pop(KTestCleanupStack);\

		});

	test(r==KErrNone);

	}

LOCAL_C void DeleteDataFile(const TDesC& aFullName)

	{

	RFs fsSession;

	TInt err = fsSession.Connect();

	if(err == KErrNone)

		{

		TEntry entry;

		if(fsSession.Entry(aFullName, entry) == KErrNone)

			{

			RDebug::Print(_L("Deleting \"%S\" file.\n"), &aFullName);

			err = fsSession.SetAtt(aFullName, 0, KEntryAttReadOnly);

			if(err != KErrNone)

				{

				RDebug::Print(_L("Error %d changing \"%S\" file attributes.\n"), err, &aFullName);

				}

			err = fsSession.Delete(aFullName);

			if(err != KErrNone)

				{

				RDebug::Print(_L("Error %d deleting \"%S\" file.\n"), err, &aFullName);

				}

			}

		fsSession.Close();

		}

	else

		{

		RDebug::Print(_L("Error %d connecting file session. File: %S.\n"), err, &aFullName);

		}

	}

class RTestReadStream : public RReadStream

	{

public:

	RTestReadStream(MStreamBuf* aSource) :

		RReadStream(aSource)

		{

		}

	void Attach(MStreamBuf* aSource)

		{

		RReadStream::Attach(aSource);

		}

	void Detach()

		{

		RReadStream::Detach();

		}

	};

class RTestWriteStream : public RWriteStream

	{

public:

	RTestWriteStream(MStreamBuf* aSink) :

		RWriteStream(aSink)

		{

		}

	void Attach(MStreamBuf* aSink)

		{

		RWriteStream::Attach(aSink);

		}

	void Detach()

		{

		RWriteStream::Detach();

		}

	};

/**

@SYMTestCaseID          PDS-STORE-CT-4064

@SYMTestCaseDesc        RReadStream, RWriteStream, Pop() and Detach() test.

@SYMTestActions			The test calls Pop() and Detach() methods of RReadStream and RWriteStream classes.         

@SYMTestPriority        High

@SYMTestExpectedResults Test must not fail

*/

void StreamDetachTestL()

	{

	test.Next(_L("@SYMTestCaseID:PDS-STORE-CT-4064: RReadStream, RWriteStream, Pop() and Detach() test"));

	TBuf8<100> buf;

	TDesBuf desBuf;

	desBuf.Set(buf);

	MStreamBuf* mbuf = &desBuf; 

	_LIT8(KStr, "1234567890");

	RTestWriteStream wstrm(mbuf);

	wstrm.PushL();

	wstrm.Detach();

	wstrm.Attach(mbuf);

	TRAPD(err, wstrm.WriteL(KStr));

	test(err == KErrNone);

	TRAP(err, wstrm.CommitL());

	test(err == KErrNone);

	wstrm.Pop();

	wstrm.Close();

	RTestReadStream rstrm(mbuf);

	rstrm.PushL();

	rstrm.Detach();

	rstrm.Attach(mbuf);

	TBuf8<100> buf2;

	TRAP(err, rstrm.ReadL(buf2, KStr().Length()));

	test(err == KErrNone);

	rstrm.Pop();

	rstrm.Close();

	test(KStr() == buf2);

	}

//

// Test file-based streams.

//

GLDEF_C TInt E32Main()

    {

	test.Title();

	setupTestDirectory();

	setupCleanup();

	__UHEAP_MARK;

//

	test.Start(_L("Test file-based streams"));

	TRAPD(r,testWriteL());

	test(r==KErrNone);

	TRAP(r,testReadL());

	test(r==KErrNone);

	TRAP(r,testSkipL());

	test(r==KErrNone);

	TRAP(r,testCopyL());

	test(r==KErrNone);

	TRAP(r, StreamDetachTestL());

	test(r==KErrNone);

	//deletion of data files must be before call to .End() - DEF047652

	TDriveUnit drive(static_cast<TUint>(RFs::GetSystemDrive()));	

	TParse parse;

	parse.Set(drive.Name(), &KFileLocationSpec, NULL);

	::DeleteDataFile(parse.FullName());

	test.End();

//

	__UHEAP_MARKEND;

	delete TheTrapCleanup;

	if (TheFs.Delete(TheTempFile)!=KErrNone)

		test.Panic(_L("Deleting temp file"));

	TheFs.Close();

	test.Close();

	return 0;

    }