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

content/public/android/junit/src/org/chromium/content/browser/installedapp/InstalledAppProviderTest.java

 // Copyright 2017 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.content.browser.installedapp;
 
 import android.content.Context;
 import android.content.pm.ApplicationInfo;
 import android.content.pm.PackageManager;
 import android.content.pm.PackageManager.NameNotFoundException;
 import android.content.res.AssetManager;
 import android.content.res.Resources;
 import android.os.Bundle;
 
+import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
-
 import org.robolectric.RuntimeEnvironment;
 import org.robolectric.annotation.Config;
 import org.robolectric.res.builder.DefaultPackageManager;
 
+import org.chromium.base.ThreadUtils;
 import org.chromium.base.test.util.Feature;
 import org.chromium.installedapp.mojom.InstalledAppProvider;
 import org.chromium.installedapp.mojom.RelatedApplication;
 import org.chromium.testing.local.CustomShadowAsyncTask;
 import org.chromium.testing.local.LocalRobolectricTestRunner;
 
 import java.net.URI;
 import java.net.URISyntaxException;
 import java.util.HashMap;
 
@@ skipping 253 matching lines @@
                         for (int i = 0; i < installedRelatedApps.length; i++) {
                             Assert.assertEquals(
                                     expectedInstalledRelatedApps[i], installedRelatedApps[i]);
                         }
                     }
                 });
     }
 
     @Before
     public void setUp() {
+        // Avoid triggering asserts in InstalledAppProviderImpl that check they are being run off
+        // the UI thread (since this is a single-threaded test).
+        ThreadUtils.setThreadAssertsDisabledForTesting(true);
         mPackageManager = new FakePackageManager();
         RuntimeEnvironment.setRobolectricPackageManager(mPackageManager);
         mFrameUrlDelegate = new FakeFrameUrlDelegate(URL_ON_ORIGIN);
         mInstalledAppProvider =
                 new InstalledAppProviderTestImpl(mFrameUrlDelegate, RuntimeEnvironment.application);
     }
 
+    @After
+    public void tearDown() {
+        ThreadUtils.setThreadAssertsDisabledForTesting(false);
+    }
+
     /**
      * Origin of the page using the API is missing certain parts of the URI.
      */
     @Test
     @Feature({"InstalledApp"})
     public void testOriginMissingParts() {
         RelatedApplication manifestRelatedApps[] = new RelatedApplication[] {
                 createRelatedApplication(PLATFORM_ANDROID, PACKAGE_NAME_1, null)};
         setAssetStatement(PACKAGE_NAME_1, NAMESPACE_WEB, RELATION_HANDLE_ALL_URLS, ORIGIN);
         RelatedApplication[] expectedInstalledRelatedApps = new RelatedApplication[] {};
@@ skipping 519 matching lines @@
         // Non-installed app.
         manifestRelatedApps = new RelatedApplication[] {
                 createRelatedApplication(PLATFORM_ANDROID, PACKAGE_NAME_2, null)};
         expectedInstalledRelatedApps = new RelatedApplication[] {};
         verifyInstalledApps(manifestRelatedApps, expectedInstalledRelatedApps);
         // This expectation is based on HMAC_SHA256(salt, packageName encoded in UTF-8), taking the
         // low 10 bits of the first two bytes of the result / 100.
         Assert.assertEquals(5, mInstalledAppProvider.getLastDelayMillis());
     }
 }