1 /* 2 * Copyright (C) 2015 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 17 package android.support.v4.testutils; 18 19 import android.view.View; 20 import android.view.ViewGroup; 21 22 import android.support.test.espresso.NoMatchingViewException; 23 import android.support.test.espresso.ViewAssertion; 24 25 import junit.framework.AssertionFailedError; 26 27 import static android.support.test.espresso.matcher.ViewMatchers.isDisplayed; 28 29 public class TestUtilsAssertions { 30 31 /** 32 * Returns an assertion that asserts that there is specified number of fully displayed 33 * children. 34 */ 35 public static ViewAssertion hasDisplayedChildren(final int expectedCount) { 36 return new ViewAssertion() { 37 @Override 38 public void check(final View foundView, NoMatchingViewException noViewException) { 39 if (noViewException != null) { 40 throw noViewException; 41 } else { 42 if (!(foundView instanceof ViewGroup)) { 43 throw new AssertionFailedError("View " + 44 foundView.getClass().getSimpleName() + " is not a ViewGroup"); 45 } 46 final ViewGroup foundGroup = (ViewGroup) foundView; 47 48 final int childrenCount = foundGroup.getChildCount(); 49 int childrenDisplayedCount = 0; 50 for (int i = 0; i < childrenCount; i++) { 51 if (isDisplayed().matches(foundGroup.getChildAt(i))) { 52 childrenDisplayedCount++; 53 } 54 } 55 56 if (childrenDisplayedCount != expectedCount) { 57 throw new AssertionFailedError("Expected " + expectedCount + 58 " displayed children, but found " + childrenDisplayedCount); 59 } 60 } 61 } 62 }; 63 } 64 }