mockit.internal.expectations.InstanceBasedMatching Maven / Gradle / Ivy
/*
* Copyright (c) 2006 JMockit developers
* This file is subject to the terms of the MIT license (see LICENSE.txt).
*/
package mockit.internal.expectations;
import java.util.*;
import javax.annotation.*;
import mockit.internal.util.*;
import static mockit.internal.util.Utilities.containsReference;
final class InstanceBasedMatching
{
@Nonnull private final List> mockedTypesToMatchOnInstances;
InstanceBasedMatching() { mockedTypesToMatchOnInstances = new LinkedList<>(); }
void discoverMockedTypesToMatchOnInstances(@Nonnull List> targetClasses) {
int numClasses = targetClasses.size();
if (numClasses > 1) {
for (int i = 0; i < numClasses; i++) {
Class targetClass = targetClasses.get(i);
if (targetClasses.lastIndexOf(targetClass) > i) {
addMockedTypeToMatchOnInstance(targetClass);
}
}
}
}
private void addMockedTypeToMatchOnInstance(@Nonnull Class mockedType) {
if (!containsReference(mockedTypesToMatchOnInstances, mockedType)) {
mockedTypesToMatchOnInstances.add(mockedType);
}
}
boolean isToBeMatchedOnInstance(@Nonnull Object mock) {
Class mockedClass = GeneratedClasses.getMockedClass(mock);
return containsReference(mockedTypesToMatchOnInstances, mockedClass);
}
}