/*
* soapUI, copyright (C) 2004-2011 eviware.com
*
* soapUI is free software; you can redistribute it and/or modify it under the
* terms of version 2.1 of the GNU Lesser General Public License as published by
* the Free Software Foundation.
*
* soapUI is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without
* even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
* See the GNU Lesser General Public License for more details at gnu.org.
*/
package com.eviware.soapui.model.propertyexpansion.resolvers;
import java.util.HashMap;
import java.util.Map;
import org.apache.log4j.Logger;
import com.eviware.soapui.SoapUI;
import com.eviware.soapui.impl.wsdl.teststeps.TestRequest;
import com.eviware.soapui.model.ModelItem;
import com.eviware.soapui.model.iface.Interface;
import com.eviware.soapui.model.iface.Operation;
import com.eviware.soapui.model.iface.Request;
import com.eviware.soapui.model.mock.MockOperation;
import com.eviware.soapui.model.mock.MockResponse;
import com.eviware.soapui.model.mock.MockRunContext;
import com.eviware.soapui.model.mock.MockService;
import com.eviware.soapui.model.project.Project;
import com.eviware.soapui.model.propertyexpansion.PropertyExpansionContext;
import com.eviware.soapui.model.testsuite.LoadTest;
import com.eviware.soapui.model.testsuite.LoadTestRunContext;
import com.eviware.soapui.model.testsuite.SamplerTestStep;
import com.eviware.soapui.model.testsuite.TestCase;
import com.eviware.soapui.model.testsuite.TestCaseRunContext;
import com.eviware.soapui.model.testsuite.TestStep;
import com.eviware.soapui.model.testsuite.TestSuite;
import com.eviware.soapui.security.SecurityTest;
import com.eviware.soapui.support.scripting.ScriptEnginePool;
import com.eviware.soapui.support.scripting.SoapUIScriptEngine;
import com.eviware.soapui.support.scripting.SoapUIScriptEngineRegistry;
import com.eviware.soapui.support.types.StringToObjectMap;
public class EvalPropertyResolver implements PropertyResolver
{
private Logger log = Logger.getLogger( EvalPropertyResolver.class );
private Map<String, ScriptEnginePool> scriptEnginePools = new HashMap<String, ScriptEnginePool>();
public String resolveProperty( PropertyExpansionContext context, String name, boolean globalOverride )
{
if( name.length() == 0 || name.charAt( 0 ) != '=' )
return null;
name = name.substring( 1 );
StringToObjectMap objects = new StringToObjectMap( context.getProperties() );
objects.put( "context", context );
objects.put( "log", SoapUI.ensureGroovyLog() );
if( context instanceof TestCaseRunContext )
{
TestCaseRunContext testCaseRunContext = ( TestCaseRunContext )context;
objects.put( "testRunner", testCaseRunContext.getTestRunner() );
objects.put( "testStep", testCaseRunContext.getCurrentStep() );
if( testCaseRunContext.getCurrentStep() instanceof SamplerTestStep )
{
objects.put( "request", ( ( SamplerTestStep )testCaseRunContext.getCurrentStep() ).getTestRequest() );
}
}
if( context instanceof LoadTestRunContext )
{
objects.put( "loadTestRunner", ( ( LoadTestRunContext )context ).getLoadTestRunner() );
}
if( context instanceof MockRunContext )
{
objects.put( "mockRunner", ( ( MockRunContext )context ).getMockRunner() );
}
ModelItem modelItem = context.getModelItem();
if( modelItem instanceof TestCase )
{
objects.put( "testCase", modelItem );
objects.put( "testSuite", ( ( TestCase )modelItem ).getTestSuite() );
objects.put( "project", ( ( TestCase )modelItem ).getTestSuite().getProject() );
}
else if( modelItem instanceof TestStep )
{
objects.put( "testStep", modelItem );
if( modelItem instanceof SamplerTestStep )
{
objects.put( "request", ( ( SamplerTestStep )modelItem ).getTestRequest() );
}
objects.put( "testCase", ( ( TestStep )modelItem ).getTestCase() );
objects.put( "testSuite", ( ( TestStep )modelItem ).getTestCase().getTestSuite() );
objects.put( "project", ( ( TestStep )modelItem ).getTestCase().getTestSuite().getProject() );
}
else if( modelItem instanceof TestSuite )
{
objects.put( "testSuite", modelItem );
objects.put( "project", ( ( TestSuite )modelItem ).getProject() );
}
if( modelItem instanceof LoadTest )
{
objects.put( "loadTest", modelItem );
objects.put( "testCase", ( ( LoadTest )modelItem ).getTestCase() );
objects.put( "testSuite", ( ( LoadTest )modelItem ).getTestCase().getTestSuite() );
objects.put( "project", ( ( LoadTest )modelItem ).getTestCase().getTestSuite().getProject() );
}
else if( modelItem instanceof Project )
{
objects.put( "project", modelItem );
}
else if( modelItem instanceof MockService )
{
objects.put( "mockService", modelItem );
objects.put( "project", ( ( MockService )modelItem ).getProject() );
}
else if( modelItem instanceof MockOperation )
{
objects.put( "mockOperation", modelItem );
objects.put( "mockService", ( ( MockOperation )modelItem ).getMockService() );
objects.put( "project", ( ( MockOperation )modelItem ).getMockService().getProject() );
}
else if( modelItem instanceof MockResponse )
{
objects.put( "mockResponse", modelItem );
objects.put( "mockOperation", ( ( MockResponse )modelItem ).getMockOperation() );
objects.put( "mockService", ( ( MockResponse )modelItem ).getMockOperation().getMockService() );
objects.put( "project", ( ( MockResponse )modelItem ).getMockOperation().getMockService().getProject() );
}
else if( modelItem instanceof Request )
{
objects.put( "request", modelItem );
if( modelItem instanceof TestRequest )
{
objects.put( "testStep", ( ( TestRequest )modelItem ).getTestStep() );
objects.put( "testCase", ( ( TestRequest )modelItem ).getTestStep().getTestCase() );
objects.put( "testSuite", ( ( TestRequest )modelItem ).getTestStep().getTestCase().getTestSuite() );
objects.put( "project", ( ( TestRequest )modelItem ).getTestStep().getTestCase().getTestSuite()
.getProject() );
}
}
else if( modelItem instanceof Operation )
{
objects.put( "operation", modelItem );
}
else if( modelItem instanceof Interface )
{
objects.put( "interface", modelItem );
}
else if( modelItem instanceof SecurityTest )
{
objects.put( "securityTest", modelItem );
}
if( modelItem != null )
{
objects.put( "modelItem", modelItem );
}
return doEval( name, modelItem, objects );
}
private String doEval( String name, ModelItem modelItem, StringToObjectMap objects )
{
String engineId = SoapUIScriptEngineRegistry.getScriptEngineId( modelItem );
synchronized( this )
{
if( !scriptEnginePools.containsKey( engineId ) )
scriptEnginePools.put( engineId, new ScriptEnginePool( engineId ) );
}
ScriptEnginePool scriptEnginePool = scriptEnginePools.get( engineId );
SoapUIScriptEngine scriptEngine = scriptEnginePool.getScriptEngine();
try
{
scriptEngine.setScript( name );
for( String key : objects.keySet() )
scriptEngine.setVariable( key, objects.get( key ) );
Object result = scriptEngine.run();
return result == null ? null : result.toString();
}
catch( Throwable e )
{
log.error( "Error evaluating script", e );
return e.getMessage();
}
finally
{
scriptEngine.clearVariables();
scriptEnginePool.returnScriptEngine( scriptEngine );
}
}
}