/**
* Copyright 2001-2010 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.easymock.internal;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Stack;
import org.easymock.IArgumentMatcher;
import org.easymock.internal.matchers.And;
import org.easymock.internal.matchers.Not;
import org.easymock.internal.matchers.Or;
/**
* @author OFFIS, Tammo Freese
*/
public final class LastControl {
private static final String NO_MATCHERS_FOUND = "no matchers found.";
private static final ThreadLocal<MocksControl> threadToControl = new ThreadLocal<MocksControl>();
private static final ThreadLocal<Stack<Invocation>> threadToCurrentInvocation = new ThreadLocal<Stack<Invocation>>();
private static final ThreadLocal<Stack<IArgumentMatcher>> threadToArgumentMatcherStack = new ThreadLocal<Stack<IArgumentMatcher>>();
// ///CLOVER:OFF
private LastControl() {
}
// ///CLOVER:ON
public static void reportLastControl(final MocksControl control) {
if (control != null) {
threadToControl.set(control);
} else {
threadToControl.remove();
}
}
public static MocksControl lastControl() {
return threadToControl.get();
}
public static void reportMatcher(final IArgumentMatcher matcher) {
Stack<IArgumentMatcher> stack = threadToArgumentMatcherStack.get();
if (stack == null) {
stack = new Stack<IArgumentMatcher>();
threadToArgumentMatcherStack.set(stack);
}
stack.push(matcher);
}
public static List<IArgumentMatcher> pullMatchers() {
final Stack<IArgumentMatcher> stack = threadToArgumentMatcherStack.get();
if (stack == null) {
return null;
}
threadToArgumentMatcherStack.remove();
return new ArrayList<IArgumentMatcher>(stack);
}
public static void reportAnd(final int count) {
final Stack<IArgumentMatcher> stack = threadToArgumentMatcherStack.get();
assertState(stack != null, NO_MATCHERS_FOUND);
stack.push(new And(popLastArgumentMatchers(count)));
}
public static void reportNot() {
final Stack<IArgumentMatcher> stack = threadToArgumentMatcherStack.get();
assertState(stack != null, NO_MATCHERS_FOUND);
stack.push(new Not(popLastArgumentMatchers(1).get(0)));
}
private static List<IArgumentMatcher> popLastArgumentMatchers(final int count) {
final Stack<IArgumentMatcher> stack = threadToArgumentMatcherStack.get();
assertState(stack != null, NO_MATCHERS_FOUND);
assertState(stack.size() >= count, "" + count + " matchers expected, " + stack.size() + " recorded.");
final List<IArgumentMatcher> result = new LinkedList<IArgumentMatcher>();
result.addAll(stack.subList(stack.size() - count, stack.size()));
for (int i = 0; i < count; i++) {
stack.pop();
}
return result;
}
private static void assertState(final boolean toAssert, final String message) {
if (!toAssert) {
threadToArgumentMatcherStack.remove();
throw new IllegalStateException(message);
}
}
public static void reportOr(final int count) {
final Stack<IArgumentMatcher> stack = threadToArgumentMatcherStack.get();
assertState(stack != null, NO_MATCHERS_FOUND);
stack.push(new Or(popLastArgumentMatchers(count)));
}
public static Invocation getCurrentInvocation() {
final Stack<Invocation> stack = threadToCurrentInvocation.get();
if (stack == null || stack.empty()) {
return null;
}
return stack.lastElement();
}
public static void pushCurrentInvocation(final Invocation invocation) {
Stack<Invocation> stack = threadToCurrentInvocation.get();
if (stack == null) {
stack = new Stack<Invocation>();
threadToCurrentInvocation.set(stack);
}
stack.push(invocation);
}
public static void popCurrentInvocation() {
final Stack<Invocation> stack = threadToCurrentInvocation.get();
stack.pop();
}
}