/*
* 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: 
*
*/
package com.nokia.sdt.referenceprojects.test;

import java.util.HashMap;
import java.util.Map;

import junit.extensions.TestSetup;
import junit.framework.*;

/**
 * Extend TestSetup with support for a Map with test state.
 * Since this class implements IStatefulTest is can receive,
 * and optionally supplement, state from an outer context,
 * or it can create a new map with state.
 * 
 * This class also passes state into any tests it wraps.
 */
public class StatefulTestSetup extends TestSetup implements IStatefulTest {

	private Map stateMap;
	
	public StatefulTestSetup(Test test) {
		super(test);
	}
	
	public void setState(Map state) {
		// Check we don't have some inconsistency where we've
		// the state is set more than once, or we've creating a map
		// and then later another map is set.
		Assert.assertNull(stateMap);
		stateMap = state;

	}
	
	public Map getStateMap() {
		if (stateMap == null) {
			stateMap = new HashMap();
		}
		return stateMap;
	}
	
	public void putState(String key, Object value) {
		getStateMap().put(key, value);
	}
	
	public Object getState(String key) {
		return getStateMap().get(key);
	}
	
	public void removeState(String key) {
		getStateMap().remove(key);
	}
	
	@Override
	public void basicRun(TestResult result) {
		if (fTest instanceof IStatefulTest) {
			((IStatefulTest)fTest).setState(stateMap);
		}
		super.basicRun(result);
	}
}