org.devocative.artemis.Context Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of artemis-core Show documentation
The newest version!
package org.devocative.artemis;

import com.fasterxml.jackson.annotation.JsonIgnore;
import org.devocative.artemis.log.ALog;

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

public class Context {
	private final Map globalVars = new HashMap<>();
	private final Map scenarioVars = new HashMap<>();
	private final Map vars = new HashMap<>();

	private Map cookies = Collections.emptyMap();

	@JsonIgnore
	private EVarScope scope;

	// ------------------------------

	public Context() {
		this(null);
	}

	public Context(Context parent) {
		if (parent != null) {
			this.globalVars.putAll(parent.globalVars);
			this.vars.putAll(parent.globalVars);
			setCookies(parent.cookies);
		}
	}

	// ------------------------------

	public void addVar(String name, Object value) {
		addVar(name, value, false);
	}

	public void addVar(String name, Object value, boolean store) {
		if (vars.containsKey(name)) {
			throw new RuntimeException("Duplicate Var for Context: " + name);
		}

		if (scope == null) {
			throw new RuntimeException("Null scope in 'addVar'");
		}

		if (scope == EVarScope.Assert) {
			throw new RuntimeException("Can't 'addVar' in  call");
		}

		if (store) {
			ALog.warn("[Groovy] Added Var Globally: {}", name);
			addVarByScope(name, value, EVarScope.Global);
		} else {
			addVarByScope(name, value, scope);
		}
	}

	public void putVar(String name, Object value) {
		putVar(name, value, false);
	}

	public void putVar(String name, Object value, boolean store) {
		if (scope == null) {
			throw new RuntimeException("Null scope in 'putVar'");
		}

		if (scope == EVarScope.Assert) {
			throw new RuntimeException("Can't 'putVar' in  call");
		}

		if (store) {
			ALog.warn("[Groovy] Put Var Globally: {}", name);
			addVarByScope(name, value, EVarScope.Global);
		} else {
			addVarByScope(name, value, scope);
		}
	}

	public Map getVars() {
		return Immutable.create(vars);
	}

	public Map getCookies() {
		return cookies;
	}

	public void setCookies(Map cookies) {
		this.cookies = Collections.unmodifiableMap(cookies);
	}

	// ---------------

	void runAtScope(EVarScope scope, Runnable code) {
		this.scope = scope;
		code.run();
		this.scope = null;
	}

	boolean containsVar(String name, EVarScope scope) {
		switch (scope) {
			case Global:
				return globalVars.containsKey(name);
			case Scenario:
				return scenarioVars.containsKey(name);
			case Request:
				break;
		}
		return vars.containsKey(name);
	}

	Object removeVar(String name, EVarScope scope) {
		Object result;
		result = vars.remove(name);

		switch (scope) {
			case Global:
				result = globalVars.remove(name);
				break;
			case Scenario:
				result = scenarioVars.remove(name);
				break;
		}

		return result;
	}

	void addVarByScope(String name, Object value, EVarScope scope) {
		if (name == null || name.trim().isEmpty()) {
			throw new TestFailedException("Var Attribute's Name Required");
		}

		vars.put(name, value);

		switch (scope) {
			case Global:
				globalVars.put(name, value);
				break;
			case Scenario:
				scenarioVars.put(name, value);
				break;
		}
	}

	void clearVars(EVarScope scope) {
		vars.clear();
		vars.putAll(globalVars);

		switch (scope) {
			case Scenario:
				scenarioVars.clear();
				break;
			case Request:
				vars.putAll(scenarioVars);
				break;
		}
	}
}