summaryrefslogtreecommitdiff
path: root/src/main/java/org/mockito/internal/creation/bytebuddy/InlineByteBuddyMockMaker.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/org/mockito/internal/creation/bytebuddy/InlineByteBuddyMockMaker.java')
-rw-r--r--src/main/java/org/mockito/internal/creation/bytebuddy/InlineByteBuddyMockMaker.java13
1 files changed, 12 insertions, 1 deletions
diff --git a/src/main/java/org/mockito/internal/creation/bytebuddy/InlineByteBuddyMockMaker.java b/src/main/java/org/mockito/internal/creation/bytebuddy/InlineByteBuddyMockMaker.java
index 42f10ce..dfe2061 100644
--- a/src/main/java/org/mockito/internal/creation/bytebuddy/InlineByteBuddyMockMaker.java
+++ b/src/main/java/org/mockito/internal/creation/bytebuddy/InlineByteBuddyMockMaker.java
@@ -14,6 +14,7 @@ import org.mockito.internal.util.Platform;
import org.mockito.internal.util.concurrent.WeakConcurrentMap;
import org.mockito.invocation.MockHandler;
import org.mockito.mock.MockCreationSettings;
+import org.mockito.plugins.InlineMockMaker;
import java.io.File;
import java.io.FileOutputStream;
@@ -87,7 +88,7 @@ import static org.mockito.internal.util.StringUtil.join;
* support this feature.
*/
@Incubating
-public class InlineByteBuddyMockMaker implements ClassCreatingMockMaker {
+public class InlineByteBuddyMockMaker implements ClassCreatingMockMaker, InlineMockMaker {
private static final Instrumentation INSTRUMENTATION;
@@ -276,6 +277,16 @@ public class InlineByteBuddyMockMaker implements ClassCreatingMockMaker {
}
@Override
+ public void clearMock(Object mock) {
+ mocks.remove(mock);
+ }
+
+ @Override
+ public void clearAllMocks() {
+ mocks.clear();
+ }
+
+ @Override
public TypeMockability isTypeMockable(final Class<?> type) {
return new TypeMockability() {
@Override