org.mockito.internal.verification.Times Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of mockito-core Show documentation
Show all versions of mockito-core Show documentation
Mockito mock objects library core API and implementation
/*
* Copyright (c) 2007 Mockito contributors
* This program is made available under the terms of the MIT License.
*/
package org.mockito.internal.verification;
import static org.mockito.internal.verification.checkers.MissingInvocationChecker.checkMissingInvocation;
import static org.mockito.internal.verification.checkers.NumberOfInvocationsChecker.checkNumberOfInvocations;
import java.util.List;
import org.mockito.exceptions.base.MockitoException;
import org.mockito.internal.verification.api.VerificationData;
import org.mockito.internal.verification.api.VerificationDataInOrder;
import org.mockito.internal.verification.api.VerificationInOrderMode;
import org.mockito.invocation.Invocation;
import org.mockito.invocation.MatchableInvocation;
import org.mockito.verification.VerificationMode;
public class Times implements VerificationInOrderMode, VerificationMode {
final int wantedCount;
public Times(int wantedNumberOfInvocations) {
if (wantedNumberOfInvocations < 0) {
throw new MockitoException("Negative value is not allowed here");
}
this.wantedCount = wantedNumberOfInvocations;
}
@Override
public void verify(VerificationData data) {
List invocations = data.getAllInvocations();
MatchableInvocation wanted = data.getTarget();
if (wantedCount > 0) {
checkMissingInvocation(data.getAllInvocations(), data.getTarget());
}
checkNumberOfInvocations(invocations, wanted, wantedCount);
}
@Override
public void verifyInOrder(VerificationDataInOrder data) {
List allInvocations = data.getAllInvocations();
MatchableInvocation wanted = data.getWanted();
if (wantedCount > 0) {
checkMissingInvocation(allInvocations, wanted, data.getOrderingContext());
}
checkNumberOfInvocations(allInvocations, wanted, wantedCount, data.getOrderingContext());
}
@Override
public String toString() {
return "Wanted invocations count: " + wantedCount;
}
@Override
public VerificationMode description(String description) {
return VerificationModeFactory.description(this, description);
}
}