/*
* JMockit Expectations
* Copyright (c) 2006-2010 Rogério Liesenfeld
* All rights reserved.
*
* Permission is hereby granted, free of charge, to any person obtaining
* a copy of this software and associated documentation files (the
* "Software"), to deal in the Software without restriction, including
* without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to
* permit persons to whom the Software is furnished to do so, subject to
* the following conditions:
*
* The above copyright notice and this permission notice shall be
* included in all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*/
package mockit.internal.expectations.invocation;
import java.util.*;
import mockit.*;
import mockit.internal.expectations.invocation.InvocationResult.*;
public final class InvocationResults
{
private final ExpectedInvocation invocation;
private final InvocationConstraints constraints;
private InvocationResult currentResult;
private InvocationResult lastResult;
private int resultCount;
public InvocationResults(ExpectedInvocation invocation, InvocationConstraints constraints)
{
this.invocation = invocation;
this.constraints = constraints;
}
public void addReturnValue(Object value)
{
InvocationResult result =
value instanceof Delegate ? new DelegatedResult((Delegate) value) : new ReturnValueResult(value);
addResult(result);
}
public void addReturnValues(Object... values)
{
for (Object value : values) {
addReturnValue(value);
}
}
public void addResults(Collection<?> values)
{
for (Object value : values) {
if (value instanceof Throwable) {
addThrowable((Throwable) value);
}
else {
addReturnValue(value);
}
}
}
public void addDeferredReturnValues(Iterator<?> values)
{
InvocationResult result = new DeferredReturnValues(values);
addResult(result);
constraints.setUnlimitedMaxInvocations();
}
public void addDeferredResults(Iterator<?> values)
{
InvocationResult result = new DeferredResults(values);
addResult(result);
constraints.setUnlimitedMaxInvocations();
}
public void addThrowable(Throwable t)
{
addResult(new ThrowableResult(t));
}
public void addResult(InvocationResult result)
{
resultCount++;
constraints.adjustMaxInvocations(resultCount);
if (currentResult == null) {
currentResult = result;
lastResult = result;
}
else {
lastResult.next = result;
lastResult = result;
}
}
public Object produceResult(Object invokedObject, Object[] invocationArgs) throws Throwable
{
InvocationResult resultToBeProduced = currentResult;
InvocationResult nextResult = resultToBeProduced.next;
if (nextResult != null) {
currentResult = nextResult;
}
Object result = resultToBeProduced.produceResult(invokedObject, invocation, constraints, invocationArgs);
return result;
}
}