blob: cd27c392dd73d7d77c6d8d6a1fb11bbb14eb8c51 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
/*
* 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.invocation.Invocation;
import org.mockito.quality.Strictness;
import org.mockito.stubbing.Answer;
import org.mockito.stubbing.OngoingStubbing;
import java.util.List;
import static org.mockito.internal.exceptions.Reporter.incorrectUseOfApi;
public class OngoingStubbingImpl<T> extends BaseStubbing<T> {
private final InvocationContainerImpl invocationContainer;
private Strictness strictness;
public OngoingStubbingImpl(InvocationContainerImpl invocationContainer) {
this.invocationContainer = invocationContainer;
}
@Override
public OngoingStubbing<T> thenAnswer(Answer<?> answer) {
if(!invocationContainer.hasInvocationForPotentialStubbing()) {
throw incorrectUseOfApi();
}
invocationContainer.addAnswer(answer, strictness);
return new ConsecutiveStubbing<T>(invocationContainer);
}
@Override
public OngoingStubbing<T> then(Answer<?> answer) {
return thenAnswer(answer);
}
public List<Invocation> getRegisteredInvocations() {
//TODO interface for tests
return invocationContainer.getInvocations();
}
@Override
@SuppressWarnings("unchecked")
public <M> M getMock() {
return (M) invocationContainer.invokedMock();
}
public void setStrictness(Strictness strictness) {
this.strictness = strictness;
}
}
|