Index: base/test/android/javatests/src/org/chromium/base/test/util/RestrictionSkipCheck.java |
diff --git a/base/test/android/javatests/src/org/chromium/base/test/util/RestrictionSkipCheck.java b/base/test/android/javatests/src/org/chromium/base/test/util/RestrictionSkipCheck.java |
new file mode 100644 |
index 0000000000000000000000000000000000000000..b4beef2bf4b3b8dc32ddcc66bdbabee70592c8d6 |
--- /dev/null |
+++ b/base/test/android/javatests/src/org/chromium/base/test/util/RestrictionSkipCheck.java |
@@ -0,0 +1,72 @@ |
+// Copyright 2016 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+package org.chromium.base.test.util; |
+ |
+import android.content.Context; |
+import android.net.ConnectivityManager; |
+import android.net.NetworkInfo; |
+import android.text.TextUtils; |
+ |
+import junit.framework.TestCase; |
+ |
+import org.chromium.base.Log; |
+import org.chromium.base.SysUtils; |
+ |
+import java.lang.reflect.Method; |
+ |
+/** |
+ * Checks if any restrictions exist and skip the test if it meets those restrictions. |
+ */ |
+public class RestrictionSkipCheck extends SkipCheck { |
+ |
+ private static final String TAG = "base_test"; |
+ |
+ private final Context mTargetContext; |
+ |
+ public RestrictionSkipCheck(Context targetContext) { |
+ mTargetContext = targetContext; |
+ } |
+ |
+ @Override |
+ public boolean shouldSkip(TestCase testCase) { |
+ Method method = getTestMethod(testCase); |
+ if (method == null) return true; |
+ |
+ for (Restriction restriction : getAnnotations(method, Restriction.class)) { |
+ for (String restrictionVal : restriction.value()) { |
+ if (restrictionApplies(restrictionVal)) { |
+ Log.i(TAG, "Test " + testCase.getClass().getName() + "#" |
+ + testCase.getName() + " skipped because of restriction " |
+ + restriction); |
+ return true; |
+ } |
+ } |
+ } |
+ return false; |
+ } |
+ |
+ protected boolean restrictionApplies(String restriction) { |
+ if (TextUtils.equals(restriction, Restriction.RESTRICTION_TYPE_LOW_END_DEVICE) |
+ && !SysUtils.isLowEndDevice()) { |
+ return true; |
+ } |
+ if (TextUtils.equals(restriction, Restriction.RESTRICTION_TYPE_NON_LOW_END_DEVICE) |
+ && SysUtils.isLowEndDevice()) { |
+ return true; |
+ } |
+ if (TextUtils.equals(restriction, Restriction.RESTRICTION_TYPE_INTERNET) |
+ && !isNetworkAvailable()) { |
+ return true; |
+ } |
+ return false; |
+ } |
+ |
+ private boolean isNetworkAvailable() { |
+ final ConnectivityManager connectivityManager = (ConnectivityManager) |
+ mTargetContext.getSystemService(Context.CONNECTIVITY_SERVICE); |
+ final NetworkInfo activeNetworkInfo = connectivityManager.getActiveNetworkInfo(); |
+ return activeNetworkInfo != null && activeNetworkInfo.isConnected(); |
+ } |
+} |