// Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).

// All rights reserved.

// This component and the accompanying materials are made available

// under the terms of the License "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 <e32def.h>

#include <e32cmn.h>

#include <e32base.h>

#include <f32file.h>

#include <e32cons.h>

#include <e32debug.h>

#define __E32TEST_EXTENSION__

#include <e32test.h>

#include "cmsdrive.h"

#include "tmsprintdrive.h"

#include "ttestutils.h"

#include "cblockdevicetester.h"

#include "tmslog.h"

extern CMsDrive* msDrive;

RTest test(_L("T_MSRMB"));

RFs fsSession;

class TTests

    {

public:

    TTests(TInt aDriveNumber);

    void FullFormat();

    void tRemovableSet();

    void tRemovableClr();

    void RemountDrive();

private:

    TInt iDriveNumber;

    };

TTests::TTests(TInt aDriveNumber)

:   iDriveNumber(aDriveNumber)

    {

    }

void TTests::RemountDrive()

    {

    TDriveInfo info;

    const TUint KMediaRemountForceMediaChange = 0x00000001;

    TRequestStatus changeStatus;

    fsSession.NotifyChange(ENotifyAll, changeStatus);

    test.Printf(_L("Remounting the drive\n"));

    TInt err = fsSession.RemountDrive(iDriveNumber, NULL, KMediaRemountForceMediaChange);

    test(err == KErrNotReady || err == KErrNone);

    do

        {

        test.Printf(_L("Waiting for media change...\n"));

        User::WaitForRequest(changeStatus);

        err = fsSession.Drive(info, iDriveNumber);

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

        fsSession.NotifyChange(ENotifyAll, changeStatus);

        }

    while (err == KErrNotReady);

    fsSession.NotifyChangeCancel(changeStatus);

    }

void TTests::tRemovableSet()

    {

    test.Start(_L("tRemovableSet\n"));

    TDriveInfo driveInfo;

    TInt err = fsSession.Drive(driveInfo);

    test(err == KErrNone);

    test.Printf(_L("DriveInfo DriveAtt= %x\n"), driveInfo.iDriveAtt);

    test(driveInfo.iDriveAtt&KDriveAttRemovable != 0);

    test.End();

    }

void TTests::tRemovableClr()

    {

    test.Start(_L("tRemovableClr\n"));

    TDriveInfo driveInfo;

    TInt err = fsSession.Drive(driveInfo);

    test(err == KErrNone);

    test.Printf(_L("DriveInfo DriveAtt= %x\n"), driveInfo.iDriveAtt);

    test(driveInfo.iDriveAtt&KDriveAttRemovable != 0);

    test.End();

    }

void TTests::FullFormat()

    {

    TInt counter;

    RFormat format;

    test.Next(_L("Test EFullFormat"));

    TInt err = format.Open(fsSession, msDrive->GetSessionPath(), EFullFormat, counter);

    test(err == KErrNone);

    while (counter)

        {

        err = format.Next(counter);

        test(err == KErrNone);

        }

    format.Close();

    }

void CallTestsL()

    {

    TInt driveNumber = msDrive->DriveNumber();

    // Print drive info

    TRAPD(err, TMsPrintDrive::VolInfoL(driveNumber));

    err = err;

    TTests t(driveNumber);

    test.Printf(_L("Preparing target drive..."));

    //t.FullFormat();

    CWrPrTester* wpTester = CWrPrTester::NewL(driveNumber);

    CleanupStack::PushL(wpTester);

    test.Printf(_L("Target drive ready.\n"));

    test.Start(_L("REMOVABLE"));

    test.Next(_L("Test RMB=Set\n"));

    err = wpTester->SetRemovableL();

    test(err == KErrNone);

    t.RemountDrive();

    t.tRemovableSet();

    test.Next(_L("Test RMB=Clr\n"));

    err = wpTester->ClrRemovableL();

    test(err == KErrNone);

    t.RemountDrive();

    t.tRemovableClr();

    test.End();

    CleanupStack::PopAndDestroy(wpTester);

    }