1e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson/* 2e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson * Copyright (c) 2007 Mockito contributors 3e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson * This program is made available under the terms of the MIT License. 4e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson */ 5e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson 6e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonpackage org.mockito.internal.verification; 7e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson 8e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonimport java.util.List; 9e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson 10e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonimport org.mockito.exceptions.base.MockitoException; 11e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonimport org.mockito.internal.invocation.InvocationMatcher; 12e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonimport org.mockito.internal.verification.api.VerificationData; 13e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonimport org.mockito.internal.verification.api.VerificationDataInOrder; 14e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonimport org.mockito.internal.verification.api.VerificationInOrderMode; 15e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonimport org.mockito.internal.verification.checkers.MissingInvocationChecker; 16e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonimport org.mockito.internal.verification.checkers.MissingInvocationInOrderChecker; 17e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonimport org.mockito.internal.verification.checkers.NumberOfInvocationsChecker; 18e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonimport org.mockito.internal.verification.checkers.NumberOfInvocationsInOrderChecker; 19e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonimport org.mockito.invocation.Invocation; 20e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonimport org.mockito.verification.VerificationMode; 21e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson 22e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinsonpublic class Times implements VerificationInOrderMode, VerificationMode { 23e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson 24e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson final int wantedCount; 25e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson 26e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson public Times(int wantedNumberOfInvocations) { 27e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson if (wantedNumberOfInvocations < 0) { 28e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson throw new MockitoException("Negative value is not allowed here"); 29e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson } 30e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson this.wantedCount = wantedNumberOfInvocations; 31e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson } 32e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson 33e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson public void verify(VerificationData data) { 34e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson if (wantedCount > 0) { 35e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson MissingInvocationChecker missingInvocation = new MissingInvocationChecker(); 36e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson missingInvocation.check(data.getAllInvocations(), data.getWanted()); 37e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson } 38e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson NumberOfInvocationsChecker numberOfInvocations = new NumberOfInvocationsChecker(); 39e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson numberOfInvocations.check(data.getAllInvocations(), data.getWanted(), wantedCount); 40e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson } 41e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson 42e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson public void verifyInOrder(VerificationDataInOrder data) { 43e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson List<Invocation> allInvocations = data.getAllInvocations(); 44e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson InvocationMatcher wanted = data.getWanted(); 45e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson 46e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson if (wantedCount > 0) { 47e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson MissingInvocationInOrderChecker missingInvocation = new MissingInvocationInOrderChecker(); 48e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson missingInvocation.check(allInvocations, wanted, this, data.getOrderingContext()); 49e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson } 50e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson NumberOfInvocationsInOrderChecker numberOfCalls = new NumberOfInvocationsInOrderChecker(); 51e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson numberOfCalls.check(allInvocations, wanted, wantedCount, data.getOrderingContext()); 52e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson } 53e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson 54e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson @Override 55e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson public String toString() { 56e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson return "Wanted invocations count: " + wantedCount; 57e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson } 58e0ae5d7e87b1dd6e789803c1b9615a84bd7488b7Ian Parkinson}