InstalledAppProviderImpl: Assert that code is run off the UI thread.

base/android/java/src/org/chromium/base/ThreadUtils.java

 // Copyright 2012 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;
 
 import android.os.Handler;
 import android.os.Looper;
 import android.os.Process;
 
 import org.chromium.base.annotations.CalledByNative;
 
 import java.util.concurrent.Callable;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.FutureTask;
 
 /**
  * Helper methods to deal with threading related tasks.
  */
 public class ThreadUtils {
 
     private static final Object sLock = new Object();
 
     private static boolean sWillOverride;
 
     private static Handler sUiThreadHandler;
 
+    private static boolean sThreadAssertsDisabled;
+
     public static void setWillOverrideUiThread() {
         synchronized (sLock) {
             sWillOverride = true;
         }
     }
 
     @VisibleForTesting
     public static void setUiThread(Looper looper) {
         synchronized (sLock) {
             if (looper == null) {
@@ skipping 147 matching lines @@
      * @param task The Runnable to run
      * @param delayMillis The delay in milliseconds until the Runnable will be run
      */
     @VisibleForTesting
     public static void postOnUiThreadDelayed(Runnable task, long delayMillis) {
         getUiThreadHandler().postDelayed(task, delayMillis);
     }
 
     /**
      * Throw an exception (when DCHECKs are enabled) if currently not running on the UI thread.
+     *
+     * Can be disabled by setThreadAssertsDisabledForTesting(true).
      */
     public static void assertOnUiThread() {
-        if (BuildConfig.DCHECK_IS_ON && !runningOnUiThread()) {
-            throw new IllegalStateException("Must be called on the Ui thread.");
-        }
+        if (sThreadAssertsDisabled) return;
+
+        assert runningOnUiThread() : "Must be called on the UI thread.";
     }
 
     /**
      * Throw an exception (regardless of build) if currently not running on the UI thread.
      *
+     * Can be disabled by setThreadAssertsEnabledForTesting(false).
+     *
      * @see #assertOnUiThread()
      */
     public static void checkUiThread() {
-        if (!runningOnUiThread()) {
-            throw new IllegalStateException("Must be called on the Ui thread.");
+        if (!sThreadAssertsDisabled && !runningOnUiThread()) {
+            throw new IllegalStateException("Must be called on the UI thread.");
         }
     }
 
     /**
+     * Throw an exception (when DCHECKs are enabled) if currently running on the UI thread.
+     *
+     * Can be disabled by setThreadAssertsDisabledForTesting(true).
+     */
+    public static void assertOnBackgroundThread() {
+        if (sThreadAssertsDisabled) return;
+
+        assert !runningOnUiThread() : "Must be called on a thread other than UI.";
+    }
+
+    /**
+     * Disables thread asserts.
+     *
+     * Can be used by tests where code that normally runs multi-threaded is going to run
+     * single-threaded for the test (otherwise asserts that are valid in production would fail in
+     * those tests).
+     */
+    public static void setThreadAssertsDisabledForTesting(boolean disabled) {
+        sThreadAssertsDisabled = disabled;
+    }
+
+    /**
      * @return true iff the current thread is the main (UI) thread.
      */
     public static boolean runningOnUiThread() {
         return getUiThreadHandler().getLooper() == Looper.myLooper();
     }
 
     public static Looper getUiThreadLooper() {
         return getUiThreadHandler().getLooper();
     }
 
     /**
      * Set thread priority to audio.
      */
     @CalledByNative
     public static void setThreadPriorityAudio(int tid) {
         Process.setThreadPriority(tid, Process.THREAD_PRIORITY_AUDIO);
     }
 
     /**
      * Checks whether Thread priority is THREAD_PRIORITY_AUDIO or not.
      * @param tid Thread id.
      * @return true for THREAD_PRIORITY_AUDIO and false otherwise.
      */
     @CalledByNative
     private static boolean isThreadPriorityAudio(int tid) {
         return Process.getThreadPriority(tid) == Process.THREAD_PRIORITY_AUDIO;
     }
 }