--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/testexecfw/symbianunittestfw/sutfw/sutfwui/sutfwconsoleui/src/symbianunittestcommandlineparser.cpp	Mon Mar 08 15:03:44 2010 +0800
@@ -0,0 +1,290 @@
+/*
+* 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 "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 "symbianunittestcommandlineparser.h"
+#include "symbianunittestdllnameparser.h"
+#include "symbianunittestrunner.h"
+
+_LIT( KHelpKey, "-help" );
+_LIT( KHelpKeyShort, "-h" );
+_LIT( KTestsKey, "-tests=" );
+_LIT( KTestsKeyShort, "-t=" );
+_LIT( KTestCasesKey, "-cases=" );
+_LIT( KTestCasesKeyShort, "-c=" );
+_LIT( KAllocKey, "-alloc" );
+_LIT( KAllocKeyShort, "-a" );
+_LIT( KOutputKey, "-output=" );
+_LIT( KOutputKeyShort, "-o=" );
+_LIT( KNoPromptKey, "-noprompt" );
+_LIT( KTimeoutKey, "-timeout=" );
+_LIT( KTimeoutKeyShort, "-to=" );
+
+
+_LIT( KOutputFileName, "SymbianUnitTestResults" );
+_LIT( KDefaultOutputFormat, "html" );
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+CSymbianUnitTestCommandLineParser* CSymbianUnitTestCommandLineParser::NewLC()
+    {
+    CSymbianUnitTestCommandLineParser* self = 
+        new( ELeave )CSymbianUnitTestCommandLineParser();
+    CleanupStack::PushL( self );
+    self->ConstructL();
+    return self;
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+CSymbianUnitTestCommandLineParser* CSymbianUnitTestCommandLineParser::NewL()
+    {
+    CSymbianUnitTestCommandLineParser* self = 
+        CSymbianUnitTestCommandLineParser::NewLC();
+    CleanupStack::Pop( self );
+    return self;
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+CSymbianUnitTestCommandLineParser::CSymbianUnitTestCommandLineParser()
+    {
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+void CSymbianUnitTestCommandLineParser::ConstructL()
+    {
+    iArguments = CCommandLineArguments::NewL();   
+    TPtrC dummy;
+    iShowHelp =  FindArgument( KHelpKey, KHelpKeyShort, dummy );
+    iAllocFailureSimulation = FindArgument( KAllocKey, KAllocKeyShort, dummy );
+    iNoPrompt = FindArgument( KNoPromptKey, KNoPromptKey, dummy );
+    SetOutputFormatL();
+    SetTimeoutL();
+    SetTestDllNamesL();
+    SetTestCaseNamesL();
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+CSymbianUnitTestCommandLineParser::~CSymbianUnitTestCommandLineParser()
+    {
+    delete iTestDllNames;
+    delete iTestCaseNames;
+    delete iOutputFormat;
+    delete iArguments;
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+TBool CSymbianUnitTestCommandLineParser::ShowHelp() const
+    {
+    return iShowHelp;
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+const MDesCArray& CSymbianUnitTestCommandLineParser::TestDllNames() const
+    {
+    return *iTestDllNames;
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+const CDesCArray& CSymbianUnitTestCommandLineParser::TestCaseNames() const
+    {
+    return *iTestCaseNames;
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+TBool CSymbianUnitTestCommandLineParser::MemoryAllocationFailureSimulation() const
+    {
+    return iAllocFailureSimulation;
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+const TDesC& CSymbianUnitTestCommandLineParser::OutputFileName() const
+    {
+    return KOutputFileName;
+    }
+    
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+const TDesC& CSymbianUnitTestCommandLineParser::OutputFormat() const
+    {
+    return *iOutputFormat;
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+TInt CSymbianUnitTestCommandLineParser::Timeout() const
+    {
+    return iTimeout;
+    }
+
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+TBool CSymbianUnitTestCommandLineParser::PromptUser() const
+    {
+    return !iNoPrompt;
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+void CSymbianUnitTestCommandLineParser::SetOutputFormatL()
+    {
+    HBufC* outputFormat = NULL;
+    TPtrC outputFormatPtr;
+    if ( FindArgument( KOutputKey, KOutputKeyShort, outputFormatPtr ) )
+        {
+        outputFormat = outputFormatPtr.AllocL();
+        }
+    else
+        {
+        outputFormat = KDefaultOutputFormat().AllocL();
+        }
+    delete iOutputFormat;
+    iOutputFormat = outputFormat;
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+void CSymbianUnitTestCommandLineParser::SetTimeoutL()
+    {
+    iTimeout = KSymbianUnitTestDefaultTimeout; //use default
+    TPtrC timeoutPtr;
+    TBool hasTimeout = FindArgument( KTimeoutKey, KTimeoutKeyShort, timeoutPtr );
+    if ( hasTimeout)
+        {
+        TLex timeoutLex(timeoutPtr);
+	TInt timeout;
+        TInt ret = timeoutLex.Val(timeout);
+	if (ret == KErrNone && timeout >= 0) 
+	    {
+	    iTimeout = timeout;
+	    }
+        }
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+void CSymbianUnitTestCommandLineParser::SetTestDllNamesL()
+    {
+    CDesCArray* testDllNames = new( ELeave )CDesCArrayFlat( 1 );
+    CleanupStack::PushL( testDllNames );
+    TPtrC testDllNamesPtr;
+    if ( FindArgument( KTestsKey, KTestsKeyShort, testDllNamesPtr ) )
+        {
+        TSymbianUnitTestDllNameParser parser;
+        parser.Parse( testDllNamesPtr );
+        TPtrC testDllNamePtr;
+        while ( parser.GetNext( testDllNamePtr ) == KErrNone )
+            {
+            testDllNames->AppendL( testDllNamePtr );
+            }
+        }
+    iShowHelp = ( testDllNames->Count() == 0 );
+    CleanupStack::Pop( testDllNames );
+    delete iTestDllNames;
+    iTestDllNames = testDllNames;
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+void CSymbianUnitTestCommandLineParser::SetTestCaseNamesL()
+    {
+    CDesCArray* testCaseNames = new( ELeave )CDesCArrayFlat( 1 );
+    CleanupStack::PushL( testCaseNames );
+    TPtrC testCaseNamesPtr;
+    if ( FindArgument( KTestCasesKey, KTestCasesKeyShort, testCaseNamesPtr ) )
+        {
+	//reuse the testdll parser here for case names
+        TSymbianUnitTestDllNameParser parser;
+        parser.Parse( testCaseNamesPtr );
+        TPtrC testCaseNamePtr;
+        while ( parser.GetNext( testCaseNamePtr ) == KErrNone )
+            {
+            testCaseNames->AppendL( testCaseNamePtr );
+            }
+        }
+    CleanupStack::Pop( testCaseNames );
+    delete iTestCaseNames;
+    iTestCaseNames = testCaseNames;
+    }
+
+// -----------------------------------------------------------------------------
+//
+// -----------------------------------------------------------------------------
+//
+TBool CSymbianUnitTestCommandLineParser::FindArgument( 
+    const TDesC& aKey,
+    const TDesC& aShortKey,
+    TPtrC& aValue )
+    {
+    TBool found( EFalse );
+    for ( TInt i = 0; i < iArguments->Count(); i++ )
+        {
+        TPtrC arg( iArguments->Arg( i ) );
+        if ( arg.FindF( aKey ) == 0 || arg.FindF( aShortKey ) == 0 )
+            {
+            found = ETrue;
+            TInt equalsPos = arg.Locate( '=' );
+            if ( equalsPos > 0 & equalsPos < arg.Length() - 1 ) 
+                {
+                aValue.Set( arg.Mid( equalsPos + 1 ) );
+                }
+            }
+        }
+    return found;
+    }