1 /* 2 * Copyright (C) 2016 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 package android.support.v4.app; 17 18 import android.os.SystemClock; 19 20 import org.mockito.exceptions.base.MockitoAssertionError; 21 import org.mockito.internal.verification.VerificationModeFactory; 22 import org.mockito.internal.verification.api.VerificationData; 23 import org.mockito.invocation.Invocation; 24 import org.mockito.verification.VerificationMode; 25 26 import java.util.List; 27 28 public class CtsMockitoUtils { CtsMockitoUtils()29 private CtsMockitoUtils() {} 30 within(long timeout)31 public static VerificationMode within(long timeout) { 32 return new Within(timeout); 33 } 34 35 public static class Within implements VerificationMode { 36 private static final long TIME_SLICE = 50; 37 private final long mTimeout; 38 Within(long timeout)39 public Within(long timeout) { 40 mTimeout = timeout; 41 } 42 43 @Override verify(VerificationData data)44 public void verify(VerificationData data) { 45 long timeout = mTimeout; 46 MockitoAssertionError errorToRethrow = null; 47 // Loop in the same way we do in PollingCheck, sleeping and then testing for the target 48 // invocation 49 while (timeout > 0) { 50 SystemClock.sleep(TIME_SLICE); 51 52 try { 53 final List<Invocation> actualInvocations = data.getAllInvocations(); 54 // Iterate over all invocations so far to see if we have a match 55 for (Invocation invocation : actualInvocations) { 56 if (data.getWanted().matches(invocation)) { 57 // Found our match within our timeout. Mark all invocations as verified 58 markAllInvocationsAsVerified(data); 59 // and return 60 return; 61 } 62 } 63 } catch (MockitoAssertionError assertionError) { 64 errorToRethrow = assertionError; 65 } 66 67 timeout -= TIME_SLICE; 68 } 69 70 if (errorToRethrow != null) { 71 throw errorToRethrow; 72 } 73 74 throw new MockitoAssertionError("Timed out while waiting " + mTimeout + "ms for " 75 + data.getWanted().toString()); 76 } 77 78 @Override description(String description)79 public VerificationMode description(String description) { 80 return VerificationModeFactory.description(this, description); 81 } 82 markAllInvocationsAsVerified(VerificationData data)83 private void markAllInvocationsAsVerified(VerificationData data) { 84 for (Invocation invocation : data.getAllInvocations()) { 85 invocation.markVerified(); 86 data.getWanted().captureArgumentsFrom(invocation); 87 } 88 } 89 } 90 91 } 92