-
-
Notifications
You must be signed in to change notification settings - Fork 2.5k
/
StubbedInvocationMatcher.java
63 lines (52 loc) · 1.91 KB
/
StubbedInvocationMatcher.java
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
57
58
59
60
61
62
63
/*
* 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.InvocationMatcher;
import org.mockito.invocation.DescribedInvocation;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.invocation.MatchableInvocation;
import org.mockito.quality.Strictness;
import org.mockito.stubbing.Answer;
import org.mockito.stubbing.Stubbing;
import java.io.Serializable;
import java.util.Queue;
import java.util.concurrent.ConcurrentLinkedQueue;
@SuppressWarnings("unchecked")
public class StubbedInvocationMatcher extends InvocationMatcher implements Serializable, Stubbing {
private static final long serialVersionUID = 4919105134123672727L;
private final Queue<Answer> answers = new ConcurrentLinkedQueue<Answer>();
private final Strictness strictness;
private DescribedInvocation usedAt;
public StubbedInvocationMatcher(Answer answer, MatchableInvocation invocation, Strictness strictness) {
super(invocation.getInvocation(), invocation.getMatchers());
this.strictness = strictness;
this.answers.add(answer);
}
public Object answer(InvocationOnMock invocation) throws Throwable {
//see ThreadsShareGenerouslyStubbedMockTest
Answer a;
synchronized(answers) {
a = answers.size() == 1 ? answers.peek() : answers.poll();
}
return a.answer(invocation);
}
public void addAnswer(Answer answer) {
answers.add(answer);
}
public void markStubUsed(DescribedInvocation usedAt) {
this.usedAt = usedAt;
}
public boolean wasUsed() {
return usedAt != null;
}
@Override
public String toString() {
return super.toString() + " stubbed with: " + answers;
}
@Override
public Strictness getStrictness() {
return strictness;
}
}