1/*
2 * Copyright (c) 2007 Mockito contributors
3 * This program is made available under the terms of the MIT License.
4 */
5
6package org.mockito.internal.verification;
7
8import java.util.List;
9
10import org.mockito.exceptions.Reporter;
11import org.mockito.exceptions.base.MockitoException;
12import org.mockito.internal.invocation.InvocationMatcher;
13import org.mockito.internal.invocation.InvocationMarker;
14import org.mockito.internal.invocation.InvocationsFinder;
15import org.mockito.internal.verification.api.VerificationData;
16import org.mockito.invocation.Invocation;
17import org.mockito.verification.VerificationMode;
18
19public class AtMost implements VerificationMode {
20
21    private final int maxNumberOfInvocations;
22    private final InvocationMarker invocationMarker = new InvocationMarker();
23
24    public AtMost(int maxNumberOfInvocations) {
25        if (maxNumberOfInvocations < 0) {
26            throw new MockitoException("Negative value is not allowed here");
27        }
28        this.maxNumberOfInvocations = maxNumberOfInvocations;
29    }
30
31    public void verify(VerificationData data) {
32        List<Invocation> invocations = data.getAllInvocations();
33        InvocationMatcher wanted = data.getWanted();
34
35        InvocationsFinder finder = new InvocationsFinder();
36        List<Invocation> found = finder.findInvocations(invocations, wanted);
37        int foundSize = found.size();
38        if (foundSize > maxNumberOfInvocations) {
39            new Reporter().wantedAtMostX(maxNumberOfInvocations, foundSize);
40        }
41
42        invocationMarker.markVerified(found, wanted);
43    }
44}