// Copyright (c) 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:

//

/**

 @file

 @test

 @internalComponent - Internal Symbian test code

*/

#include "testprocgoodsession.h"

#include "tsysmon_steposrestarttest.h"

#include <ssm/ssmstate.h>

#include <ssm/ssmdomaindefs.h>

#include <ssm/ssmstateawaresession.h>

#include <sysmonclisess.h>

#include <startupproperties.h>

#include <e32cmn.h>

#include <f32file.h>

_LIT(KSysMonDontRendezvousRunCountFile, "C:\\testdata\\runcounts\\tsysmon_app_dontrendezvous.exe");

CStepOSRestart::CStepOSRestart(const TDesC& aTestName): iTestName(aTestName)

	{

    if (aTestName == KCTestCaseCriticalNoRetries)

        SetTestStepName(KCTestCaseCriticalNoRetriesStep);

    else if (aTestName == KCTestCaseRestartOS)

        SetTestStepName(KCTestCaseRestartOSStep);

    else

        {

        // If the test script is none of the above then we would fail the test

        TEST(KErrArgument);

        }

	}

CStepOSRestart::~CStepOSRestart()

    {

    iSsmStateAwareSess.Close();

    if (iAsyncCallbackForStartTest)

        delete iAsyncCallbackForStartTest;

    if(iAsyncCallbackForStopScheduler)

        delete iAsyncCallbackForStopScheduler;

    }

void CStepOSRestart::StateChanged(TSsmState aSsmState)

    {

    }

void CStepOSRestart::CallStopSchedulerL()

	{

	TSsmState currenState;

	currenState = iSsmStateAwareSess.State();

	TUint16 currentMainState = currenState.MainState();

	// Active scheduler is stopped once the system goes into shutdown state, else it issues a request for callback again

	// Once the system gets into normal state start the state transition test

	if (currentMainState == ESsmFail || currentMainState == ESsmShutdown)

		{

 		RDebug::Print(_L("\nStateTran Test: Stop Active Scheduler\n"));

 		TInt runCount = CSysMonTestHelper::ReadRunCountL(KSysMonDontRendezvousRunCountFile);

 		if((iTestName == KCTestCaseCriticalNoRetries) && (1 != runCount))

 		    SetTestStepResult(EFail);

 		else if((iTestName == KCTestCaseRestartOS) && (2 != runCount))

 		    SetTestStepResult(EFail);

		CActiveScheduler::Stop();

		}

	else

		{

		iAsyncCallbackForStopScheduler->CallBack();

		}

	}

static TInt StopScheduler(TAny* aStateTranFromNormal)

 	{

 	CStepOSRestart* stateTran = reinterpret_cast<CStepOSRestart*>(aStateTranFromNormal);

 	TRAPD(err,stateTran->CallStopSchedulerL());

 	return err;

 	}

TVerdict CStepOSRestart::doTestStepL()

	{

    TInt err =0;

    if (iTestName == KCTestCaseCriticalNoRetries)

        {

        TRAP(err, TestCriticalNoRetriesL());

        }

    else if (iTestName == KCTestCaseRestartOS)

        {

        TRAP(err, TestRestartOSL());

        }

    TEST(KErrNone == err);

    return TestStepResult();    

	}

void CStepOSRestart::TestCriticalNoRetriesL()

    {

	CActiveScheduler* sched=new(ELeave) CActiveScheduler;

	CleanupStack::PushL(sched);

	CActiveScheduler::Install(sched);

	RSysMonSession sysmon;

	sysmon.OpenL();

	CleanupClosePushL(sysmon);

	CStartupProperties* props = CStartupProperties::NewLC(_L("tsysmon_app_dontrendezvous.exe"), KNullDesC);

	props->SetMonitored(ETrue);

	props->SetStartupType(EStartProcess);

	props->SetStartMethod(EWaitForStart);

	props->SetNoOfRetries(1);

	props->SetTimeout(0);

	props->SetRecoveryParams(ECriticalNoRetries, 0);

	RProcess slave1;

	slave1.Create(_L("tsysmon_app_dontrendezvous.exe"), _L("5000"));

	CleanupClosePushL(slave1);

	TSsmState currenState;

	// Connects to RSsmStateAwareSession in order to enquire about the current system state

	TInt err = iSsmStateAwareSess.Connect(KSM2UiServicesDomain3);

	TEST(err == KErrNone);

	if (err == KErrNone)

		RDebug::Print(_L("\nStateTran Test: RSsmStateAwareSession sess Connect %d\n"), err);

	else

		{

		RDebug::Print(_L("\nStateTran Test: RSsmStateAwareSession sess Connect Failed\n"));	

		User::Leave(err);

		}

	currenState = iSsmStateAwareSess.State();

	RDebug::Print(_L("\nStateTran Test: Current System State is %d\n"), currenState.MainState());

	TEST(currenState.MainState() == ESsmNormal);

	iAsyncCallbackForStopScheduler =  new(ELeave) CAsyncCallBack(CActive::EPriorityIdle);

	TCallBack stopSchedulerCallback(StopScheduler, this);

	iAsyncCallbackForStopScheduler->Set(stopSchedulerCallback);

	// call the callback function which keeps looping until the system state gets into fail state.

	iAsyncCallbackForStopScheduler->CallBack();

	// Notifier used to get the notification when system gets to fail state

	CSsmStateAwareSession* notif_for_failstate = CSsmStateAwareSession::NewLC(KSM2UiServicesDomain3);

	notif_for_failstate->AddSubscriberL(*this);

    TRequestStatus status;

    slave1.Rendezvous(status);

    slave1.Resume();

    User::WaitForRequest(status);

	// Register with SysMon

	sysmon.MonitorL(*props, slave1);

	slave1.Terminate(KErrNone);

	sched->Start();

	CleanupStack::PopAndDestroy(5);

	}

void CStepOSRestart::TestRestartOSL()

    {

    CActiveScheduler* sched=new(ELeave) CActiveScheduler;

    CleanupStack::PushL(sched);

    CActiveScheduler::Install(sched);

    RSysMonSession sysmon;

    sysmon.OpenL();

    CleanupClosePushL(sysmon);

    CStartupProperties* props = CStartupProperties::NewLC(_L("tsysmon_app_dontrendezvous.exe"), KNullDesC);

    props->SetMonitored(ETrue);

    props->SetStartupType(EStartProcess);

    props->SetStartMethod(EWaitForStart);

    props->SetNoOfRetries(1);

    props->SetTimeout(0);

    props->SetRecoveryParams(ERestartOS, 0);

    RProcess slave1;

    slave1.Create(_L("tsysmon_app_dontrendezvous.exe"), _L("5000"));

    CleanupClosePushL(slave1);

    TSsmState currenState;

    // Connects to RSsmStateAwareSession in order to enquire about the current system state

    TInt err = iSsmStateAwareSess.Connect(KSM2UiServicesDomain3);

    TEST(err == KErrNone);

    if (err == KErrNone)

        RDebug::Print(_L("\nStateTran Test: RSsmStateAwareSession sess Connect %d\n"), err);

    else

        {

        RDebug::Print(_L("\nStateTran Test: RSsmStateAwareSession sess Connect Failed\n")); 

        User::Leave(err);

        }

    currenState = iSsmStateAwareSess.State();

    RDebug::Print(_L("\nStateTran Test: Current System State is %d\n"), currenState.MainState());

    TEST(currenState.MainState() == ESsmNormal);

    iAsyncCallbackForStopScheduler =  new(ELeave) CAsyncCallBack(CActive::EPriorityIdle);

    TCallBack stopSchedulerCallback(StopScheduler, this);

    iAsyncCallbackForStopScheduler->Set(stopSchedulerCallback);

    // call the callback function which keeps looping until the system state gets into fail state.

    iAsyncCallbackForStopScheduler->CallBack();

    // Notifier used to get the notification when system gets to fail state

    CSsmStateAwareSession* notif_for_failstate = CSsmStateAwareSession::NewLC(KSM2UiServicesDomain3);

    notif_for_failstate->AddSubscriberL(*this);

    TRequestStatus status;

    slave1.Rendezvous(status);

    slave1.Resume();

    User::WaitForRequest(status);

    // Register with SysMon

    sysmon.MonitorL(*props, slave1);

    slave1.Terminate(KErrNone);

    sched->Start();

    CleanupStack::PopAndDestroy(5);

    }