// Copyright (c) 2007-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 <e32test.h>

#include <e32panic.h>

#include <f32file.h>

#include <bautils.h>

#include <swi/swispubsubdefs.h>

#include <ecom/ecom.h>

#include "EComUidCodes.h"

#include "Interface.h" // interface to Plugins

//Test utils for copying the resolver to C

#include "../EcomTestUtils/EcomTestUtils.h"

#include "../EcomTestUtils/TPropertyManager.h"

using namespace Swi;

REComSession EComSess;

LOCAL_D RTest TEST(_L("ECom SWI Test"));

_LIT(KEComExDllOnZ,		"Z:\\ramonly\\EComSwiExample.dll");

_LIT(KEComExDllOnC,		"C:\\sys\\bin\\EComSwiExample.dll");

_LIT(KEComRscFileOnC,	"C:\\resource\\plugins\\EComSwiExample.rsc");

_LIT(KEComRscFileOnZ,	"Z:\\ramonly\\EComSwiExample.rsc");

_LIT(KEComRscPath,		"C:\\resource\\plugins\\");

const TInt KWaitDuration = 5000000; //  delay in usecs

#define UNUSED_VAR(a) a = a

inline LOCAL_C TInt DeleteTestPlugin()

	{

	TRAPD(err, EComTestUtils::FileManDeleteFileL(KEComExDllOnC));

	if((err == KErrNone)||(err == KErrNotFound))

		{

		TRAP(err, EComTestUtils::FileManDeleteFileL(KEComRscFileOnC));

		}

	if(err == KErrNotFound)

		{

		err = KErrNone;

		}

	return err;

	}

LOCAL_C TInt DeletePluginFolder()

	{

	TRAPD(err, EComTestUtils::FileManDeleteDirL(KEComRscPath));

	User::After(1000000);

	return err;

	}

/**

Copies the Resolver Plugins to C:\ drive

*/

LOCAL_C TInt CopyPluginsL()

    {

	// Copy the dlls and .rsc files on to RAM

	TRAPD(err, EComTestUtils::FileManCopyFileL(KEComExDllOnZ, KEComExDllOnC));

 	TEST(err==KErrNone, __LINE__);

 	TRAP(err, EComTestUtils::FileManCopyFileL(KEComRscFileOnZ, KEComRscFileOnC));

 	TEST(err==KErrNone, __LINE__);

 	User::After(1000000);

 	return err;

	}

LOCAL_C void RegisterForNotificationL(TRequestStatus& aNotifyStatus)

	{

    REComSession& ecomSession = REComSession::OpenL();

	CleanupClosePushL(ecomSession);

	ecomSession.NotifyOnChange(aNotifyStatus);

	TEST.Printf(_L("Request notification on ECom registration data change\n"));

	TInt result = aNotifyStatus.Int();

	TEST(result == KRequestPending);

	ecomSession.Close();

	CleanupStack::PopAndDestroy();

	}

LOCAL_C void WaitForNotificationL(TBool aExpected, TRequestStatus& aNotifyStatus)

	{

    //Wait for ECOM Server notification to arrive with timeout

	RTimer timer;

	CleanupClosePushL(timer);

	User::LeaveIfError(timer.CreateLocal());

	TRequestStatus timerStatus;

	//Wait for ECom notification

	timer.After(timerStatus, KWaitDuration);

	TEST(timerStatus.Int() == KRequestPending);

    TEST.Printf(_L("Waiting for notification from ECOM Server...\n"));

    User::WaitForRequest(timerStatus, aNotifyStatus);

    if(aExpected)

    	{

	    //Verify that we recieved notification from ECom

		TEST(timerStatus.Int() == KRequestPending);

		TEST(aNotifyStatus.Int() == KErrNone);

    	}

    else

    	{

	    //Verify that we have not recieved notification from ECom

		TEST(timerStatus.Int() == KErrNone);

		TEST(aNotifyStatus.Int() == KRequestPending);

    	}

	timer.Cancel();

	CleanupStack::PopAndDestroy();

	}

LOCAL_C void FindImplementationsL(TInt aExpected)

	{

    REComSession& ecomSession = REComSession::OpenL();

	CleanupClosePushL(ecomSession);

	//Get a list of available implementations

	TUid interfaceUid={0x10009DD9};

	RImplInfoPtrArray ifArray;

	ecomSession.ListImplementationsL(interfaceUid,ifArray);

	//Verify that the expected number of implementations were found

	TInt count = ifArray.Count();

	TEST(count == aExpected);

	TEST.Printf(_L("%d Implementations found...\n"),count);

	//cleanup

	ifArray.ResetAndDestroy();

	ecomSession.Close();

	CleanupStack::PopAndDestroy();

	}

/**

@SYMTestCaseID          SYSLIB-ECOM-CT-3602

@SYMTestCaseDesc		Tests notification processing when SWI does not exist

						DEF108840: Undesireable interaction between ECOM and SWI

@SYMTestActions  	    Request notifcation of ECom registry change

						Copy plugin to c:\ and check that notification is recieved

@SYMTestExpectedResults Notification should be recieved as normal if SWI is not present

@SYMDEF                 DEF108840

*/

LOCAL_C void DoSWITest1L()

	{

	TRequestStatus notifyStatus;

	RegisterForNotificationL(notifyStatus);

	FindImplementationsL(0);

	//Copy plugin file to c:

	TEST.Printf(_L("Copy Plugin Files...\n"));

	CopyPluginsL();

	WaitForNotificationL(ETrue,notifyStatus);

	FindImplementationsL(1);

	}

/**

@SYMTestCaseID          SYSLIB-ECOM-CT-3547

@SYMTestCaseDesc	    Tests notification processing during SWI for

						DEF108840: Undesireable interaction between ECOM and SWI

@SYMTestActions  	    Set P&S variable to indicate SWI in progress.

						Request notifcation of ECom registry change

						Copy plugin to c:\ and check that notification is not recieved

						Clear P&S variable to indicate SWI completion and verify that notification is recieved.

@SYMTestExpectedResults Notification should not be recieved while SWI is in progress, but should be

						recieved when SWI completes

@SYMDEF                 DEF108840

*/

LOCAL_C void DoSWITest2L()

	{

	//Set SWI as running

	TInt r = PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisInstall);

	TEST.Printf(_L("SWI Started\n"));

	TRequestStatus notifyStatus;

	RegisterForNotificationL(notifyStatus);

	FindImplementationsL(0);

	//Copy plugin file to c:

	TEST.Printf(_L("Copy Plugin Files...\n"));

	CopyPluginsL();

    //Verify that the wait timed out - we didn't recieve notification

	WaitForNotificationL(EFalse,notifyStatus);

	FindImplementationsL(0);

	//Set SWI as complete

	r = PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisNone);

	TEST.Printf(_L("SWI Complete\n"));

	WaitForNotificationL(ETrue,notifyStatus);

	FindImplementationsL(1);

	}

/**

@SYMTestCaseID          SYSLIB-ECOM-CT-3668

@SYMTestCaseDesc	    Tests notification processing during SWI for

						INC110470 ECOM does not notify when plug-ins get uninstalled

@SYMTestActions  	    Uses P&S variables to simulate SWI install and uninstall

						Request notification of ECom registry change

						Verify that notifications are recieved after both install and uninstall

@SYMTestExpectedResults Notification should be recieved after both SWI install and uninstall

@SYMDEF                 INC110470

*/

LOCAL_C void DoSWITest3L()

	{

	TRequestStatus notifyStatus;

	RegisterForNotificationL(notifyStatus);

	FindImplementationsL(0);

	//Set SWI as running

	TInt r = PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisInstall);

	TEST.Printf(_L("SWI Started\n"));

	//Copy plugin file to c:

	TEST.Printf(_L("Copy Plugin Files...\n"));

	CopyPluginsL();

	//Set SWI as complete

	r = PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisNone);

	TEST.Printf(_L("SWI Complete\n"));

	WaitForNotificationL(ETrue,notifyStatus);

	FindImplementationsL(1);

	RegisterForNotificationL(notifyStatus);

	//Set SWI as running

	r = PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisUninstall);

	TEST.Printf(_L("SWI Started\n"));

	//delete files from c:

	TEST.Printf(_L("Delete Plugin Files...\n"));

	DeleteTestPlugin();

	//Delete plugin folder so that drive is unmounted by ECom

	DeletePluginFolder();

	//Set SWI as complete

	r = PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisNone);

	TEST.Printf(_L("SWI Complete\n"));

	WaitForNotificationL(ETrue,notifyStatus);

	FindImplementationsL(0);

	RegisterForNotificationL(notifyStatus);

	//Set SWI as running

	r = PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisInstall);

	TEST.Printf(_L("SWI Started\n"));

	//Copy plugin file to c:

	TEST.Printf(_L("Copy Plugin Files...\n"));

	CopyPluginsL();

	//Set SWI as complete

	r = PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisNone);

	TEST.Printf(_L("SWI Complete\n"));

	WaitForNotificationL(ETrue,notifyStatus);

	FindImplementationsL(1);

	}

/**

@SYMTestCaseID          SYSLIB-ECOM-CT-3669

@SYMTestCaseDesc	    Tests robustness of notification processing during SWI

@SYMTestActions  	    Uses P&S variables to simulate SWI install and uninstall

						Request notification of ECom registry change

						Verify that notifications are recieved after both install and uninstall even

						if a SWI scan is pending when SWI begins

@SYMTestExpectedResults Notification should be recieved after both SWI install and uninstall

@SYMDEF                 INC110470

*/

LOCAL_C void DoSWITest4L()

	{

	TRequestStatus notifyStatus;

	RegisterForNotificationL(notifyStatus);

	//Set SWI as complete

	PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisNone);

	TEST.Printf(_L("SWI Complete\n"));

	FindImplementationsL(0);

	//Set SWI as running

	TInt r = PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisInstall);

	TEST.Printf(_L("SWI Started\n"));

	//Copy plugin file to c:

	TEST.Printf(_L("Copy Plugin Files...\n"));

	CopyPluginsL();

	//Set SWI as complete

	r = PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisNone);

	TEST.Printf(_L("SWI Complete\n"));

	WaitForNotificationL(ETrue,notifyStatus);

	FindImplementationsL(1);

	RegisterForNotificationL(notifyStatus);

	//Set SWI as running

	r = PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisUninstall);

	TEST.Printf(_L("SWI Started\n"));

	//Delete plugin files

	TEST.Printf(_L("Delete Plugin Files...\n"));

	DeleteTestPlugin();

	//Set SWI as complete

	r = PropertyManager::SetProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,ESASwisNone);

	TEST.Printf(_L("SWI Complete\n"));

	//Wait again for ECom notification

	WaitForNotificationL(ETrue,notifyStatus);

	FindImplementationsL(0);

	}

LOCAL_C TInt SetupTest()

	{

	//Ensure plugin files are not on C:

	TInt res = DeleteTestPlugin();

	TEST.Printf(_L("Deleting test plugin...\n"));

	//Wait to ensure files are deleted

	User::After(2000000);

	//Create an ECom session to ensure ECom is up and running

	EComSess = REComSession::OpenL();

	//Wait to ensure ECom startup has occurred

	User::After(2000000);

	return res;

	}

LOCAL_C void RunTestL()

	{

	__UHEAP_MARK;

	TInt res = SetupTest();

	TEST(res == KErrNone);

	res = PropertyManager::DeleteProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue);

	TEST(res == KErrNone);

	//Run the test

	TEST.Next(_L(" @SYMTestCaseID:SYSLIB-ECOM-CT-3602 ECom SWI Test - SWI Not Present "));

	DoSWITest1L();

	//Cleanup

	EComSess.Close();

	res = SetupTest();

	TEST(res == KErrNone);

	res = PropertyManager::DefineProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue,RProperty::EInt);

	TEST(res == KErrNone);

	TEST.Next(_L(" @SYMTestCaseID:SYSLIB-ECOM-CT-3547 ECom SWI Test - SWI Present "));

	DoSWITest2L();

	//Cleanup

	EComSess.Close();

	res = SetupTest();

	TEST(res == KErrNone);

	TEST.Next(_L(" @SYMTestCaseID:SYSLIB-ECOM-CT-3668 ECom SWI Uninstall Test - SWI Present "));

	DoSWITest3L();

	//Cleanup

	EComSess.Close();

	res = SetupTest();

	TEST(res == KErrNone);

	TEST.Next(_L(" @SYMTestCaseID:SYSLIB-ECOM-CT-3669 ECom SWI Robustness Test "));

	DoSWITest4L();

	res = PropertyManager::DeleteProperty(KUidSystemCategory, KSAUidSoftwareInstallKeyValue);

	TEST(res == KErrNone);

	//Cleanup

	EComSess.Close();

	REComSession::FinalClose();

	//Ensure plugin files are not on C:

	res = DeleteTestPlugin();

	TEST(res == KErrNone);

	__UHEAP_MARKEND;

	}

GLDEF_C TInt E32Main()

	{

	__UHEAP_MARK;

	TEST.Title();

	TEST.Start(_L("ECom SWI tests."));

	CTrapCleanup* cleanup = CTrapCleanup::New();

	CActiveScheduler* scheduler = new(ELeave)CActiveScheduler;

	CActiveScheduler::Install(scheduler);

	TRAPD(err,RunTestL());

	TEST(err==KErrNone, __LINE__);

	delete scheduler;

	delete cleanup;

	TEST.End();

	TEST.Close();

	__UHEAP_MARKEND;

	return(0);

	}