org.mockito.internal.stubbing.OngoingStubbingImpl Maven / Gradle / Ivy
/*
* Copyright (c) 2007 Mockito contributors
* This program is made available under the terms of the MIT License.
*/
package org.mockito.internal.stubbing;
import org.mockito.internal.invocation.Invocation;
import org.mockito.stubbing.Answer;
import org.mockito.stubbing.DeprecatedOngoingStubbing;
import org.mockito.stubbing.OngoingStubbing;
import java.util.List;
public class OngoingStubbingImpl extends BaseStubbing {
private final InvocationContainerImpl invocationContainerImpl;
public OngoingStubbingImpl(InvocationContainerImpl invocationContainerImpl) {
this.invocationContainerImpl = invocationContainerImpl;
}
public OngoingStubbing thenAnswer(Answer answer) {
invocationContainerImpl.addAnswer(answer);
return new ConsecutiveStubbing(invocationContainerImpl);
}
public DeprecatedOngoingStubbing toAnswer(Answer answer) {
invocationContainerImpl.addAnswer(answer);
return new ConsecutiveStubbing(invocationContainerImpl);
}
public List getRegisteredInvocations() {
//TODO interface for tests
return invocationContainerImpl.getInvocations();
}
}