Revert "Migrate remaining chrome/android native_java_unittest to javatests"

This reverts commit c3907c3e.

Reason for revert: Observed multiple test failures in chrome_pulic_test_apk on android builders, since the following builds:
* https://ci.chromium.org/p/chromium/builders/ci/android-marshmallow-x86-rel/973
* https://ci.chromium.org/p/chromium/builders/ci/android-pie-x86-rel/1894
* https://ci.chromium.org/p/chromium/builders/ci/android-arm64-proguard-rel/2437

Original change's description:
> Migrate remaining chrome/android native_java_unittest to javatests
> 
> native_java_unittests are going away, so these tests need to be migrated
> to javatests.
> 
> Bug: 1103344
> Change-Id: I49a99d1cc1e3170d4af01af8e1a75e9b09249fc5
> Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2353391
> Commit-Queue: Michael Thiessen <mthiesse@chromium.org>
> Reviewed-by: Yaron Friedman <yfriedman@chromium.org>
> Cr-Commit-Position: refs/heads/master@{#799264}

TBR=yfriedman@chromium.org,mthiesse@chromium.org

Change-Id: Ia64e53afde3d7fdeecb1a4500c74bf6e0ad69ffa
No-Presubmit: true
No-Tree-Checks: true
No-Try: true
Bug: 1103344
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2363925Reviewed-by: Haiyang Pan <hypan@google.com>
Commit-Queue: Haiyang Pan <hypan@google.com>
Cr-Commit-Position: refs/heads/master@{#799411}

chrome/android/BUILD.gn

@@ -948,6 +948,7 @@ android_library("chrome_test_java") {
     ":browser_java_test_support",
     ":chrome_app_java_resources",
     ":chrome_test_util_java",
+    ":native_java_unittests_java",
     ":partner_location_descriptor_proto_java",
     "$google_play_services_package:google_play_services_base_java",
     "$google_play_services_package:google_play_services_basement_java",
@@ -973,7 +974,6 @@ android_library("chrome_test_java") {
     "//chrome/browser/android/crypto:java",
     "//chrome/browser/android/lifecycle:java",
     "//chrome/browser/browser_controls/android:java",
-    "//chrome/browser/contextmenu:java",
     "//chrome/browser/download/android:java",
     "//chrome/browser/engagement/android:java",
     "//chrome/browser/enterprise/util:java",
@@ -1049,7 +1049,6 @@ android_library("chrome_test_java") {
     "//components/download/public/common:public_java",
     "//components/embedder_support/android:browser_context_java",
     "//components/embedder_support/android:content_view_java",
-    "//components/embedder_support/android:context_menu_java",
     "//components/embedder_support/android:util_java",
     "//components/embedder_support/android:web_contents_delegate_java",
     "//components/embedder_support/android:web_contents_delegate_javatests",
@@ -1087,7 +1086,6 @@ android_library("chrome_test_java") {
     "//components/schema_org/common:mojom_java",
     "//components/search_engines/android:java",
     "//components/security_interstitials/content/android:java",
-    "//components/security_state/content/android:java",
     "//components/security_state/core:security_state_enums_java",
     "//components/signin/core/browser:signin_enums_java",
     "//components/signin/core/browser/android:java",
@@ -3205,6 +3203,78 @@ generate_jni("chrome_jni_headers") {
   }
 }
 
+native_java_unittests_tests = [
+  "native_java_unittests/src/org/chromium/chrome/browser/contextmenu/ChromeContextMenuPopulatorTest.java",
+  "native_java_unittests/src/org/chromium/chrome/browser/installedapp/InstalledAppProviderTest.java",
+  "native_java_unittests/src/org/chromium/chrome/browser/notifications/NotificationPlatformBridgeUnitTest.java",
+  "native_java_unittests/src/org/chromium/chrome/browser/partnercustomizations/PartnerBrowserCustomizationsUnitTest.java",
+  "native_java_unittests/src/org/chromium/chrome/browser/payments/AndroidPaymentAppFinderUnitTest.java",
+  "native_java_unittests/src/org/chromium/chrome/browser/payments/AndroidPaymentAppUnitTest.java",
+  "native_java_unittests/src/org/chromium/chrome/browser/payments/PaymentManifestVerifierTest.java",
+  "native_java_unittests/src/org/chromium/chrome/browser/tabmodel/TabPersistentStoreUnitTest.java",
+  "native_java_unittests/src/org/chromium/chrome/browser/toolbar/ToolbarSecurityIconTest.java",
+]
+
+android_library("native_java_unittests_java") {
+  testonly = true
+  deps = [
+    ":browser_java_test_support",
+    ":chrome_app_java_resources",
+    ":chrome_test_util_java",
+    "//base:base_java",
+    "//base:base_java_test_support",
+    "//base:jni_java",
+    "//base/test:test_support_java",
+    "//chrome/android:chrome_java",
+    "//chrome/browser/contextmenu:java",
+    "//chrome/browser/flags:java",
+    "//chrome/browser/image_fetcher:java",
+    "//chrome/browser/preferences:java",
+    "//chrome/browser/profiles/android:java",
+    "//chrome/browser/tab:java",
+    "//chrome/browser/tabmodel:java",
+    "//chrome/browser/ui/android/favicon:java",
+    "//chrome/browser/util:java",
+    "//chrome/test/android:chrome_java_test_support",
+    "//components/embedder_support/android:context_menu_java",
+    "//components/embedder_support/android:util_java",
+    "//components/omnibox/browser:browser_java",
+    "//components/payments/content/android:java",
+    "//components/payments/mojom:mojom_java",
+    "//components/query_tiles:java",
+    "//components/search_engines/android:java",
+    "//components/security_state/content/android:java",
+    "//components/security_state/core:security_state_enums_java",
+    "//content/public/android:content_java",
+    "//content/public/test/android:android_test_message_pump_support_java",
+    "//content/public/test/android:content_java_test_support",
+    "//third_party/android_deps:androidx_collection_collection_java",
+    "//third_party/android_deps:com_google_code_findbugs_jsr305_java",
+    "//third_party/blink/public:blink_headers_java",
+    "//third_party/blink/public/mojom:android_mojo_bindings_java",
+    "//third_party/hamcrest:hamcrest_java",
+    "//third_party/junit",
+    "//third_party/mockito:mockito_java",
+    "//ui/android:ui_java",
+    "//ui/android:ui_java_test_support",
+    "//url:gurl_java",
+    "//url:origin_java",
+    "//url/mojom:url_mojom_gurl_java",
+  ]
+
+  sources = native_java_unittests_tests
+
+  annotation_processor_deps = [ "//base/android/jni_generator:jni_processor" ]
+}
+
+# See https://bugs.chromium.org/p/chromium/issues/detail?id=908819 for why we
+# can't put 'java' in the name here.
+generate_jni("native_j_unittests_jni_headers") {
+  testonly = true
+  sources = native_java_unittests_tests
+  feature_list_file = "//chrome/browser/flags/android/java/src/org/chromium/chrome/browser/flags/ChromeFeatureList.java"
+}
+
 # Used by android-binary-size trybot to validate expectations.
 if (_enable_libs_and_assets_verification || _enable_manifest_verification) {
   group("validate_expectations") {

chrome/android/chrome_test_java_sources.gni

@@ -104,7 +104,6 @@ chrome_test_java_sources = [
   "javatests/src/org/chromium/chrome/browser/compositor/overlays/strip/TabStripTest.java",
   "javatests/src/org/chromium/chrome/browser/contacts_picker/ContactDetailsTest.java",
   "javatests/src/org/chromium/chrome/browser/contacts_picker/ContactsPickerDialogTest.java",
-  "javatests/src/org/chromium/chrome/browser/contextmenu/ChromeContextMenuPopulatorTest.java",
   "javatests/src/org/chromium/chrome/browser/contextmenu/ContextMenuTest.java",
   "javatests/src/org/chromium/chrome/browser/contextmenu/RevampedContextMenuHeaderViewTest.java",
   "javatests/src/org/chromium/chrome/browser/contextmenu/RevampedContextMenuItemViewTest.java",
@@ -246,7 +245,6 @@ chrome_test_java_sources = [
   "javatests/src/org/chromium/chrome/browser/init/StartupTabPreloaderTest.java",
   "javatests/src/org/chromium/chrome/browser/init/StartupTabPreloaderUnitTest.java",
   "javatests/src/org/chromium/chrome/browser/input/SelectPopupOtherContentViewTest.java",
-  "javatests/src/org/chromium/chrome/browser/installedapp/InstalledAppProviderTest.java",
   "javatests/src/org/chromium/chrome/browser/instantapps/InstantAppsHandlerTest.java",
   "javatests/src/org/chromium/chrome/browser/interstitials/LookalikeInterstitialTest.java",
   "javatests/src/org/chromium/chrome/browser/jsdialog/JavascriptAppModalDialogTest.java",
@@ -277,7 +275,6 @@ chrome_test_java_sources = [
   "javatests/src/org/chromium/chrome/browser/notifications/NotificationBuilderBaseTest.java",
   "javatests/src/org/chromium/chrome/browser/notifications/NotificationPlatformBridgeIntentTest.java",
   "javatests/src/org/chromium/chrome/browser/notifications/NotificationPlatformBridgeTest.java",
-  "javatests/src/org/chromium/chrome/browser/notifications/NotificationPlatformBridgeUnitTest.java",
   "javatests/src/org/chromium/chrome/browser/notifications/NotificationTestRule.java",
   "javatests/src/org/chromium/chrome/browser/notifications/NotificationTestUtil.java",
   "javatests/src/org/chromium/chrome/browser/notifications/NotificationWrapperBuilderFactoryTest.java",
@@ -347,7 +344,6 @@ chrome_test_java_sources = [
   "javatests/src/org/chromium/chrome/browser/paint_preview/services/PaintPreviewTabServiceTest.java",
   "javatests/src/org/chromium/chrome/browser/partnercustomizations/BasePartnerBrowserCustomizationIntegrationTestRule.java",
   "javatests/src/org/chromium/chrome/browser/partnercustomizations/BasePartnerBrowserCustomizationUnitTestRule.java",
-  "javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerBrowserCustomizationsUnitTest.java",
   "javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableBookmarksEditingUnitTest.java",
   "javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableIncognitoModeIntegrationTest.java",
   "javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerDisableIncognitoModeUnitTest.java",
@@ -359,8 +355,6 @@ chrome_test_java_sources = [
   "javatests/src/org/chromium/chrome/browser/password_manager/settings/PasswordSettingsTest.java",
   "javatests/src/org/chromium/chrome/browser/password_manager/settings/PasswordViewingTypeTest.java",
   "javatests/src/org/chromium/chrome/browser/payments/AndroidPaymentAppFinderTest.java",
-  "javatests/src/org/chromium/chrome/browser/payments/AndroidPaymentAppFinderUnitTest.java",
-  "javatests/src/org/chromium/chrome/browser/payments/AndroidPaymentAppUnitTest.java",
   "javatests/src/org/chromium/chrome/browser/payments/CurrencyFormatterTest.java",
   "javatests/src/org/chromium/chrome/browser/payments/ExpandablePaymentHandlerChangePaymentMethodTest.java",
   "javatests/src/org/chromium/chrome/browser/payments/ExpandablePaymentHandlerTest.java",
@@ -374,7 +368,6 @@ chrome_test_java_sources = [
   "javatests/src/org/chromium/chrome/browser/payments/PaymentHandlerEnableDelegationsTest.java",
   "javatests/src/org/chromium/chrome/browser/payments/PaymentManifestDownloaderTest.java",
   "javatests/src/org/chromium/chrome/browser/payments/PaymentManifestParserTest.java",
-  "javatests/src/org/chromium/chrome/browser/payments/PaymentManifestVerifierTest.java",
   "javatests/src/org/chromium/chrome/browser/payments/PaymentRequestAbortTest.java",
   "javatests/src/org/chromium/chrome/browser/payments/PaymentRequestBillingAddressTest.java",
   "javatests/src/org/chromium/chrome/browser/payments/PaymentRequestBillingAddressWithoutPhoneTest.java",
@@ -563,7 +556,6 @@ chrome_test_java_sources = [
   "javatests/src/org/chromium/chrome/browser/tabmodel/TabModelSelectorTabModelObserverTest.java",
   "javatests/src/org/chromium/chrome/browser/tabmodel/TabModelSelectorTabObserverTest.java",
   "javatests/src/org/chromium/chrome/browser/tabmodel/TabPersistentStoreTest.java",
-  "javatests/src/org/chromium/chrome/browser/tabmodel/TabPersistentStoreUnitTest.java",
   "javatests/src/org/chromium/chrome/browser/tabmodel/TabWindowManagerTest.java",
   "javatests/src/org/chromium/chrome/browser/tabmodel/TestTabModelDirectory.java",
   "javatests/src/org/chromium/chrome/browser/tabmodel/UndoTabModelTest.java",
@@ -574,7 +566,6 @@ chrome_test_java_sources = [
   "javatests/src/org/chromium/chrome/browser/toolbar/LocationBarModelTest.java",
   "javatests/src/org/chromium/chrome/browser/toolbar/ToolbarProgressBarIntegrationTest.java",
   "javatests/src/org/chromium/chrome/browser/toolbar/ToolbarProgressBarTest.java",
-  "javatests/src/org/chromium/chrome/browser/toolbar/ToolbarSecurityIconTest.java",
   "javatests/src/org/chromium/chrome/browser/toolbar/ToolbarTest.java",
   "javatests/src/org/chromium/chrome/browser/toolbar/bottom/BottomToolbarTest.java",
   "javatests/src/org/chromium/chrome/browser/toolbar/bottom/BottomToolbarWithStartSurfaceTest.java",

chrome/android/javatests/DEPS

@@ -37,7 +37,6 @@ include_rules = [
   "+components/permissions/android/java/src/org/chromium/components/permissions",
   "+components/safe_browsing/android/java/src/org/chromium/components/safe_browsing",
   "+components/search_engines",
-  "+components/security_state/content/android/java",
   "+components/sync/android/java/src/org/chromium/components/sync",
   "+components/sync/test/fake_server/android",
   # We should only depend on the util package of something that lives in

chrome/android/javatests/src/org/chromium/chrome/browser/contextmenu/ChromeContextMenuPopulatorTest.java → chrome/android/native_java_unittests/src/org/chromium/chrome/browser/contextmenu/ChromeContextMenuPopulatorTest.java

@@ -11,19 +11,13 @@ import static org.mockito.Mockito.when;
 
 import android.util.Pair;
 
-import androidx.test.filters.SmallTest;
-
-import org.junit.Before;
-import org.junit.Test;
-import org.junit.runner.RunWith;
 import org.mockito.Mock;
 import org.mockito.Mockito;
 import org.mockito.MockitoAnnotations;
 
 import org.chromium.base.ContextUtils;
-import org.chromium.base.test.BaseJUnit4ClassRunner;
-import org.chromium.base.test.UiThreadTest;
-import org.chromium.base.test.util.Batch;
+import org.chromium.base.annotations.CalledByNative;
+import org.chromium.base.annotations.CalledByNativeJavaTest;
 import org.chromium.blink_public.common.ContextMenuDataMediaType;
 import org.chromium.chrome.R;
 import org.chromium.chrome.browser.IntentHandler;
@@ -34,7 +28,6 @@ import org.chromium.chrome.browser.flags.ChromeFeatureList;
 import org.chromium.chrome.browser.share.ShareDelegate;
 import org.chromium.components.embedder_support.contextmenu.ContextMenuParams;
 import org.chromium.components.search_engines.TemplateUrlService;
-import org.chromium.content_public.browser.test.NativeLibraryTestUtils;
 import org.chromium.ui.base.MenuSourceType;
 
 import java.util.Arrays;
@@ -44,8 +37,6 @@ import java.util.List;
 /**
  * Unit tests for the context menu logic of Chrome.
  */
-@RunWith(BaseJUnit4ClassRunner.class)
-@Batch(Batch.UNIT_TESTS)
 public class ChromeContextMenuPopulatorTest {
     private static final String PAGE_URL = "http://www.blah.com";
     private static final String LINK_URL = "http://www.blah.com/other_blah";
@@ -64,12 +55,13 @@ public class ChromeContextMenuPopulatorTest {
 
     private ChromeContextMenuPopulator mPopulator;
 
-    @Before
+    @CalledByNative
+    private ChromeContextMenuPopulatorTest() {}
+
+    @CalledByNative
     public void setUp() {
         MockitoAnnotations.initMocks(this);
 
-        NativeLibraryTestUtils.loadNativeLibraryNoBrowserProcess();
-
         when(mItemDelegate.getPageUrl()).thenReturn(PAGE_URL);
         when(mItemDelegate.isIncognitoSupported()).thenReturn(true);
         when(mItemDelegate.isOpenInOtherWindowSupported()).thenReturn(true);
@@ -122,9 +114,7 @@ public class ChromeContextMenuPopulatorTest {
         }
     }
 
-    @Test
-    @SmallTest
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testHttpLink() {
         FirstRunStatus.setFirstRunFlowComplete(false);
         ContextMenuParams contextMenuParams = new ContextMenuParams(0, 0, PAGE_URL, LINK_URL,
@@ -161,9 +151,7 @@ public class ChromeContextMenuPopulatorTest {
         checkMenuOptions(contextMenuParams, expected4);
     }
 
-    @Test
-    @SmallTest
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testHttpLinkWithPreviewTabEnabled() {
         ContextMenuParams contextMenuParams = new ContextMenuParams(0, 0, PAGE_URL, LINK_URL,
                 LINK_TEXT, "", "", "", null, false, 0, 0, MenuSourceType.MENU_SOURCE_TOUCH);
@@ -196,9 +184,7 @@ public class ChromeContextMenuPopulatorTest {
         checkMenuOptions(contextMenuParams, expected3);
     }
 
-    @Test
-    @SmallTest
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testMailLink() {
         FirstRunStatus.setFirstRunFlowComplete(false);
         ContextMenuParams contextMenuParams =
@@ -234,9 +220,7 @@ public class ChromeContextMenuPopulatorTest {
         checkMenuOptions(contextMenuParams, expected4);
     }
 
-    @Test
-    @SmallTest
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testTelLink() {
         FirstRunStatus.setFirstRunFlowComplete(false);
         ContextMenuParams contextMenuParams =
@@ -273,9 +257,7 @@ public class ChromeContextMenuPopulatorTest {
         checkMenuOptions(contextMenuParams, expected4);
     }
 
-    @Test
-    @SmallTest
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testVideoLink() {
         FirstRunStatus.setFirstRunFlowComplete(false);
         String sourceUrl = "http://www.blah.com/";
@@ -316,9 +298,7 @@ public class ChromeContextMenuPopulatorTest {
         checkMenuOptions(contextMenuParams, expected4Tab1, expected4Tab2);
     }
 
-    @Test
-    @SmallTest
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testImageHiFi() {
         FirstRunStatus.setFirstRunFlowComplete(false);
         ContextMenuParams contextMenuParams = new ContextMenuParams(0,
@@ -353,9 +333,7 @@ public class ChromeContextMenuPopulatorTest {
         checkMenuOptions(contextMenuParams, expected4);
     }
 
-    @Test
-    @SmallTest
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testHttpLinkWithImageHiFi() {
         FirstRunStatus.setFirstRunFlowComplete(false);
         ContextMenuParams contextMenuParams = new ContextMenuParams(0,

chrome/android/javatests/src/org/chromium/chrome/browser/notifications/NotificationPlatformBridgeUnitTest.java → chrome/android/native_java_unittests/src/org/chromium/chrome/browser/notifications/NotificationPlatformBridgeUnitTest.java

@@ -10,31 +10,20 @@ import static org.junit.Assert.assertTrue;
 
 import android.app.Notification;
 
-import androidx.test.filters.SmallTest;
-
-import org.junit.Before;
-import org.junit.Test;
-import org.junit.runner.RunWith;
-
-import org.chromium.base.test.BaseJUnit4ClassRunner;
-import org.chromium.base.test.util.Batch;
+import org.chromium.base.annotations.CalledByNative;
+import org.chromium.base.annotations.CalledByNativeJavaTest;
 import org.chromium.base.test.util.Feature;
 import org.chromium.chrome.browser.notifications.channels.ChromeChannelDefinitions;
 import org.chromium.chrome.browser.notifications.channels.SiteChannelsManager;
-import org.chromium.content_public.browser.test.NativeLibraryTestUtils;
 
 import java.util.Arrays;
 
 /**
  * Unit tests for NotificationPlatformBridge.
  */
-@RunWith(BaseJUnit4ClassRunner.class)
-@Batch(Batch.UNIT_TESTS)
 public class NotificationPlatformBridgeUnitTest {
-    @Before
-    public void setUp() {
-        NativeLibraryTestUtils.loadNativeLibraryNoBrowserProcess();
-    }
+    @CalledByNative
+    private NotificationPlatformBridgeUnitTest() {}
 
     /**
      * Verifies that the getOriginFromTag method returns the origin for valid input, and null for
@@ -43,8 +32,7 @@ public class NotificationPlatformBridgeUnitTest {
      * This is defined in functions in notification_id_generator.cc.
      */
     @Feature({"Browser", "Notifications"})
-    @SmallTest
-    @Test
+    @CalledByNativeJavaTest
     public void testGetOriginFromNotificationTag() {
         // The common case.
         assertEquals("https://example.com",
@@ -73,8 +61,7 @@ public class NotificationPlatformBridgeUnitTest {
      * and null for any other channel or a null channel id.
      */
     @Feature({"Browser", "Notifications"})
-    @SmallTest
-    @Test
+    @CalledByNativeJavaTest
     public void testGetOriginFromChannelId() {
         // Returns the expected origin for a channel id associated with a particular origin.
         assertEquals("https://example.com",
@@ -95,8 +82,7 @@ public class NotificationPlatformBridgeUnitTest {
      * Verifies that the makeDefaults method returns the generated notification defaults.
      */
     @Feature({"Browser", "Notifications"})
-    @SmallTest
-    @Test
+    @CalledByNativeJavaTest
     public void testMakeDefaults() {
         // 0 should be returned if pattern length is 0, silent is true, and vibration is enabled.
         assertEquals(0, NotificationPlatformBridge.makeDefaults(0, true, true));
@@ -122,8 +108,7 @@ public class NotificationPlatformBridgeUnitTest {
      * in Android notification.
      */
     @Feature({"Browser", "Notifications"})
-    @SmallTest
-    @Test
+    @CalledByNativeJavaTest
     public void testMakeVibrationPattern() {
         assertTrue(Arrays.equals(new long[] {0, 100, 200, 300},
                 NotificationPlatformBridge.makeVibrationPattern(new int[] {100, 200, 300})));

chrome/android/javatests/src/org/chromium/chrome/browser/partnercustomizations/PartnerBrowserCustomizationsUnitTest.java → chrome/android/native_java_unittests/src/org/chromium/chrome/browser/partnercustomizations/PartnerBrowserCustomizationsUnitTest.java

@@ -4,32 +4,21 @@
 
 package org.chromium.chrome.browser.partnercustomizations;
 
-import androidx.test.filters.SmallTest;
-
 import org.junit.Assert;
-import org.junit.Before;
-import org.junit.Test;
-import org.junit.runner.RunWith;
 
-import org.chromium.base.test.BaseJUnit4ClassRunner;
-import org.chromium.base.test.util.Batch;
+import org.chromium.base.annotations.CalledByNative;
+import org.chromium.base.annotations.CalledByNativeJavaTest;
 import org.chromium.base.test.util.Feature;
-import org.chromium.content_public.browser.test.NativeLibraryTestUtils;
 
 /**
  * Unit tests for {@link PartnerBrowserCustomizations}.
  */
-@RunWith(BaseJUnit4ClassRunner.class)
-@Batch(Batch.UNIT_TESTS)
 public class PartnerBrowserCustomizationsUnitTest {
-    @Before
-    public void setUp() {
-        NativeLibraryTestUtils.loadNativeLibraryNoBrowserProcess();
-    }
+    @CalledByNative
+    private PartnerBrowserCustomizationsUnitTest() {}
 
     @Feature({"Homepage"})
-    @SmallTest
-    @Test
+    @CalledByNativeJavaTest
     public void testIsValidHomepage() {
         Assert.assertTrue(PartnerBrowserCustomizations.isValidHomepage(
                 "chrome-native://newtab/path#fragment"));

chrome/android/javatests/src/org/chromium/chrome/browser/payments/AndroidPaymentAppFinderUnitTest.java → chrome/android/native_java_unittests/src/org/chromium/chrome/browser/payments/AndroidPaymentAppFinderUnitTest.java

@@ -13,20 +13,12 @@ import android.content.pm.ServiceInfo;
 import android.content.pm.Signature;
 import android.os.Bundle;
 
-import androidx.test.filters.SmallTest;
-
-import org.junit.Before;
-import org.junit.Rule;
-import org.junit.Test;
-import org.junit.runner.RunWith;
 import org.mockito.ArgumentMatcher;
 import org.mockito.ArgumentMatchers;
 import org.mockito.Mockito;
 
-import org.chromium.base.test.BaseJUnit4ClassRunner;
-import org.chromium.base.test.UiThreadTest;
-import org.chromium.base.test.util.Batch;
-import org.chromium.chrome.test.ChromeBrowserTestRule;
+import org.chromium.base.annotations.CalledByNative;
+import org.chromium.base.annotations.CalledByNativeJavaTest;
 import org.chromium.components.payments.PackageManagerDelegate;
 import org.chromium.components.payments.PaymentApp;
 import org.chromium.components.payments.PaymentAppFactoryParams;
@@ -35,7 +27,6 @@ import org.chromium.components.payments.PaymentManifestParser;
 import org.chromium.components.payments.PaymentManifestWebDataService;
 import org.chromium.components.payments.WebAppManifestSection;
 import org.chromium.content_public.browser.WebContents;
-import org.chromium.content_public.browser.test.NativeLibraryTestUtils;
 import org.chromium.payments.mojom.PaymentDetailsModifier;
 import org.chromium.payments.mojom.PaymentMethodData;
 import org.chromium.url.GURL;
@@ -47,21 +38,12 @@ import java.util.List;
 import java.util.Map;
 
 /** Tests for the native Android payment app finder. */
-@RunWith(BaseJUnit4ClassRunner.class)
-@Batch(AndroidPaymentAppFinderUnitTest.PAYMENTS_BROWSER_UNIT_TESTS)
 public class AndroidPaymentAppFinderUnitTest {
-    // Collection of payments unit tests that require the browser process to be initialized.
-    static final String PAYMENTS_BROWSER_UNIT_TESTS = "PaymentsBrowserUnitTests";
     private static final IntentArgumentMatcher sPayIntentArgumentMatcher =
             new IntentArgumentMatcher(new Intent("org.chromium.intent.action.PAY"));
 
-    @Rule
-    public ChromeBrowserTestRule mTestRule = new ChromeBrowserTestRule();
-
-    @Before
-    public void setUp() {
-        NativeLibraryTestUtils.loadNativeLibraryAndInitBrowserProcess();
-    }
+    @CalledByNative
+    private AndroidPaymentAppFinderUnitTest() {}
 
     /**
      * Argument matcher that matches Intents using |filterEquals| method.
@@ -126,9 +108,7 @@ public class AndroidPaymentAppFinderUnitTest {
         Mockito.verify(delegate).onDoneCreatingPaymentApps(/*factory=*/null);
     }
 
-    @SmallTest
-    @Test
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testNoValidPaymentMethodNames() {
         verifyNoAppsFound(findApps(new String[] {"unknown-payment-method-name",
                                            "http://not.secure.payment.method.name.com", "https://"},
@@ -137,9 +117,7 @@ public class AndroidPaymentAppFinderUnitTest {
                 Mockito.mock(PackageManagerDelegate.class)));
     }
 
-    @SmallTest
-    @Test
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testQueryWithoutApps() {
         PackageManagerDelegate packageManagerDelegate = Mockito.mock(PackageManagerDelegate.class);
         Mockito.when(packageManagerDelegate.getActivitiesThatCanRespondToIntentWithMetaData(
@@ -155,9 +133,7 @@ public class AndroidPaymentAppFinderUnitTest {
                         ArgumentMatchers.any(ApplicationInfo.class), ArgumentMatchers.anyInt());
     }
 
-    @SmallTest
-    @Test
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testQueryWithoutMetaData() {
         List<ResolveInfo> activities = new ArrayList<>();
         ResolveInfo alicePay = new ResolveInfo();
@@ -182,9 +158,7 @@ public class AndroidPaymentAppFinderUnitTest {
                         ArgumentMatchers.any(ApplicationInfo.class), ArgumentMatchers.anyInt());
     }
 
-    @SmallTest
-    @Test
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testQueryWithoutLabel() {
         List<ResolveInfo> activities = new ArrayList<>();
         ResolveInfo alicePay = new ResolveInfo();
@@ -212,9 +186,7 @@ public class AndroidPaymentAppFinderUnitTest {
                         ArgumentMatchers.any(ApplicationInfo.class), ArgumentMatchers.anyInt());
     }
 
-    @SmallTest
-    @Test
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testQueryUnsupportedPaymentMethod() {
         PackageManagerDelegate packageManagerDelegate = installPaymentApps(
                 new String[] {"com.alicepay.app"}, new String[] {"unsupported-payment-method"});
@@ -254,9 +226,7 @@ public class AndroidPaymentAppFinderUnitTest {
         return packageManagerDelegate;
     }
 
-    @SmallTest
-    @Test
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testQueryDifferentPaymentMethod() {
         PackageManagerDelegate packageManagerDelegate =
                 installPaymentApps(new String[] {"com.alicepay.app"}, new String[] {"basic-card"});
@@ -270,9 +240,7 @@ public class AndroidPaymentAppFinderUnitTest {
                         ArgumentMatchers.any(ApplicationInfo.class), ArgumentMatchers.anyInt());
     }
 
-    @SmallTest
-    @Test
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testQueryNoPaymentMethod() {
         PackageManagerDelegate packageManagerDelegate =
                 installPaymentApps(new String[] {"com.alicepay.app"}, new String[] {"basic-card"});
@@ -285,9 +253,7 @@ public class AndroidPaymentAppFinderUnitTest {
                         ArgumentMatchers.any(ApplicationInfo.class), ArgumentMatchers.anyInt());
     }
 
-    @SmallTest
-    @Test
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testQueryBasicCardsWithTwoApps() {
         List<ResolveInfo> activities = new ArrayList<>();
         ResolveInfo alicePay = new ResolveInfo();
@@ -348,9 +314,7 @@ public class AndroidPaymentAppFinderUnitTest {
         Mockito.verify(delegate).onDoneCreatingPaymentApps(/*factory=*/null);
     }
 
-    @SmallTest
-    @Test
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testQueryBobPayWithOneAppThatHasIsReadyToPayService() {
         List<ResolveInfo> activities = new ArrayList<>();
         ResolveInfo bobPay = new ResolveInfo();

chrome/android/javatests/src/org/chromium/chrome/browser/payments/AndroidPaymentAppUnitTest.java → chrome/android/native_java_unittests/src/org/chromium/chrome/browser/payments/AndroidPaymentAppUnitTest.java

@@ -8,22 +8,15 @@ import android.app.Activity;
 import android.content.Intent;
 import android.os.Bundle;
 
-import androidx.test.filters.SmallTest;
-
 import org.junit.Assert;
-import org.junit.Before;
-import org.junit.Test;
-import org.junit.runner.RunWith;
 import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;
 
-import org.chromium.base.test.BaseJUnit4ClassRunner;
-import org.chromium.base.test.UiThreadTest;
-import org.chromium.base.test.util.Batch;
+import org.chromium.base.annotations.CalledByNative;
+import org.chromium.base.annotations.CalledByNativeJavaTest;
 import org.chromium.components.payments.PayerData;
 import org.chromium.components.payments.PaymentApp;
 import org.chromium.components.payments.SupportedDelegations;
-import org.chromium.content_public.browser.test.NativeLibraryTestUtils;
 import org.chromium.content_public.browser.test.util.CriteriaHelper;
 import org.chromium.payments.mojom.PaymentCurrencyAmount;
 import org.chromium.payments.mojom.PaymentDetailsModifier;
@@ -37,8 +30,6 @@ import java.util.HashMap;
 import java.util.Map;
 
 /** Tests for the native Android payment app finder. */
-@RunWith(BaseJUnit4ClassRunner.class)
-@Batch(Batch.UNIT_TESTS)
 public class AndroidPaymentAppUnitTest {
     @Mock
     private AndroidPaymentApp.Launcher mLauncherMock;
@@ -50,14 +41,10 @@ public class AndroidPaymentAppUnitTest {
     private boolean mReadyToPayQueryFinished;
     private boolean mInvokePaymentAppFinished;
 
-    @Before
-    public void setUp() {
-        NativeLibraryTestUtils.loadNativeLibraryNoBrowserProcess();
-    }
+    @CalledByNative
+    private AndroidPaymentAppUnitTest() {}
 
-    @SmallTest
-    @Test
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testSuccessfulPaymentWithoutIsReadyToPayService() throws Exception {
         runTest(Activity.RESULT_OK);
         Assert.assertNull(mErrorMessage);
@@ -65,9 +52,7 @@ public class AndroidPaymentAppUnitTest {
         Assert.assertEquals("{}", mPaymentDetails);
     }
 
-    @SmallTest
-    @Test
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testCancelledPaymentWithoutIsReadyToPayService() throws Exception {
         runTest(Activity.RESULT_CANCELED);
         Assert.assertEquals("Payment app returned RESULT_CANCELED code. This is how payment apps "

chrome/android/native_java_unittests/src/org/chromium/chrome/browser/payments/OWNERS

+# TEAM: payments-dev@chromium.org
+# COMPONENT: UI>Browser>Payments
+
+file://components/payments/OWNERS

chrome/android/javatests/src/org/chromium/chrome/browser/payments/PaymentManifestVerifierTest.java → chrome/android/native_java_unittests/src/org/chromium/chrome/browser/payments/PaymentManifestVerifierTest.java

@@ -9,19 +9,12 @@ import android.content.pm.PackageInfo;
 import android.content.pm.ResolveInfo;
 import android.content.pm.Signature;
 
-import androidx.test.filters.SmallTest;
-
 import org.junit.Assert;
-import org.junit.Before;
-import org.junit.Rule;
-import org.junit.Test;
-import org.junit.runner.RunWith;
 import org.mockito.Mockito;
 
-import org.chromium.base.test.BaseJUnit4ClassRunner;
-import org.chromium.base.test.util.Batch;
+import org.chromium.base.annotations.CalledByNative;
+import org.chromium.base.annotations.CalledByNativeJavaTest;
 import org.chromium.chrome.browser.payments.PaymentManifestVerifier.ManifestVerifyCallback;
-import org.chromium.chrome.test.ChromeBrowserTestRule;
 import org.chromium.components.payments.PackageManagerDelegate;
 import org.chromium.components.payments.PaymentManifestDownloader;
 import org.chromium.components.payments.PaymentManifestParser;
@@ -29,7 +22,6 @@ import org.chromium.components.payments.PaymentManifestWebDataService;
 import org.chromium.components.payments.PaymentManifestWebDataService.PaymentManifestWebDataServiceCallback;
 import org.chromium.components.payments.WebAppManifestSection;
 import org.chromium.content_public.browser.WebContents;
-import org.chromium.content_public.browser.test.NativeLibraryTestUtils;
 import org.chromium.url.GURL;
 import org.chromium.url.Origin;
 
@@ -37,21 +29,19 @@ import java.util.HashSet;
 import java.util.Set;
 
 /** A test for the verifier of a payment app manifest. */
-@RunWith(BaseJUnit4ClassRunner.class)
-@Batch(AndroidPaymentAppFinderUnitTest.PAYMENTS_BROWSER_UNIT_TESTS)
 public class PaymentManifestVerifierTest {
     private static final String ERROR_MESSAGE = "This is an error message.";
 
-    private Origin mTestOrigin;
-    private GURL mMethodName;
-    private ResolveInfo mAlicePay;
-    private ResolveInfo mBobPay;
-    private Set<ResolveInfo> mMatchingApps;
-    private PaymentManifestDownloader mDownloader;
-    private PaymentManifestWebDataService mWebDataService;
-    private PaymentManifestParser mParser;
-    private PackageManagerDelegate mPackageManagerDelegate;
-    private ManifestVerifyCallback mCallback;
+    private final Origin mTestOrigin;
+    private final GURL mMethodName;
+    private final ResolveInfo mAlicePay;
+    private final ResolveInfo mBobPay;
+    private final Set<ResolveInfo> mMatchingApps;
+    private final PaymentManifestDownloader mDownloader;
+    private final PaymentManifestWebDataService mWebDataService;
+    private final PaymentManifestParser mParser;
+    private final PackageManagerDelegate mPackageManagerDelegate;
+    private final ManifestVerifyCallback mCallback;
 
     // SHA256("01020304050607080900"):
     public static final byte[][] BOB_PAY_SIGNATURE_FINGERPRINTS = {{(byte) 0x9A, (byte) 0x89,
@@ -62,13 +52,8 @@ public class PaymentManifestVerifierTest {
             (byte) 0x09, (byte) 0xC4, (byte) 0x74, (byte) 0xF5, (byte) 0x93, (byte) 0xFB}};
     public static final Signature BOB_PAY_SIGNATURE = new Signature("01020304050607080900");
 
-    @Rule
-    public ChromeBrowserTestRule mTestRule = new ChromeBrowserTestRule();
-
-    @Before
-    public void setUp() {
-        NativeLibraryTestUtils.loadNativeLibraryAndInitBrowserProcess();
-
+    @CalledByNative
+    private PaymentManifestVerifierTest() {
         mTestOrigin = PaymentManifestDownloader.createOpaqueOriginForTest();
         mMethodName = new GURL("https://example.com");
 
@@ -147,8 +132,7 @@ public class PaymentManifestVerifierTest {
         mCallback = Mockito.mock(ManifestVerifyCallback.class);
     }
 
-    @SmallTest
-    @Test
+    @CalledByNativeJavaTest
     public void testUnableToDownloadPaymentMethodManifest() {
         PaymentManifestVerifier verifier = new PaymentManifestVerifier(mTestOrigin, mMethodName,
                 mMatchingApps, null /* supportedOrigins */,
@@ -172,8 +156,7 @@ public class PaymentManifestVerifierTest {
                 .onValidDefaultPaymentApp(Mockito.any(GURL.class), Mockito.any(ResolveInfo.class));
     }
 
-    @SmallTest
-    @Test
+    @CalledByNativeJavaTest
     public void testUnableToDownloadWebAppManifest() {
         PaymentManifestVerifier verifier = new PaymentManifestVerifier(mTestOrigin, mMethodName,
                 mMatchingApps, null /* supportedOrigins */,
@@ -206,8 +189,7 @@ public class PaymentManifestVerifierTest {
         Mockito.verify(mCallback).onFinishedUsingResources();
     }
 
-    @SmallTest
-    @Test
+    @CalledByNativeJavaTest
     public void testUnableToParsePaymentMethodManifest() {
         PaymentManifestVerifier verifier = new PaymentManifestVerifier(mTestOrigin, mMethodName,
                 mMatchingApps, null /* supportedOrigins */, mWebDataService,
@@ -227,8 +209,7 @@ public class PaymentManifestVerifierTest {
         Mockito.verify(mCallback).onFinishedUsingResources();
     }
 
-    @SmallTest
-    @Test
+    @CalledByNativeJavaTest
     public void testUnableToParseWebAppManifest() {
         PaymentManifestVerifier verifier = new PaymentManifestVerifier(mTestOrigin, mMethodName,
                 mMatchingApps, null /* supportedOrigins */, mWebDataService,
@@ -256,8 +237,7 @@ public class PaymentManifestVerifierTest {
         Mockito.verify(mCallback).onFinishedUsingResources();
     }
 
-    @SmallTest
-    @Test
+    @CalledByNativeJavaTest
     public void testBobPayAllowed() {
         PaymentManifestVerifier verifier = new PaymentManifestVerifier(mTestOrigin, mMethodName,
                 mMatchingApps, null /* supportedOrigins */, mWebDataService, mDownloader, mParser,
@@ -280,8 +260,7 @@ public class PaymentManifestVerifierTest {
     }
 
     /** If a single web app manifest fails to download, all downloads should be aborted. */
-    @SmallTest
-    @Test
+    @CalledByNativeJavaTest
     public void testFirstOfTwoManifestsFailsToDownload() {
         CountingParser parser = new CountingParser() {
             @Override
@@ -333,8 +312,7 @@ public class PaymentManifestVerifierTest {
     }
 
     /** If a single web app manifest fails to parse, all downloads should be aborted. */
-    @SmallTest
-    @Test
+    @CalledByNativeJavaTest
     public void testFirstOfTwoManifestsFailsToParse() {
         CountingParser parser = new CountingParser() {
             @Override

chrome/android/javatests/src/org/chromium/chrome/browser/tabmodel/TabPersistentStoreUnitTest.java → chrome/android/native_java_unittests/src/org/chromium/chrome/browser/tabmodel/TabPersistentStoreUnitTest.java

@@ -18,21 +18,14 @@ import static org.mockito.Mockito.when;
 
 import android.text.TextUtils;
 
-import androidx.test.filters.SmallTest;
-
-import org.junit.After;
 import org.junit.Assert;
-import org.junit.Before;
-import org.junit.Test;
-import org.junit.runner.RunWith;
 import org.mockito.ArgumentMatcher;
 import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;
 
+import org.chromium.base.annotations.CalledByNative;
+import org.chromium.base.annotations.CalledByNativeJavaTest;
 import org.chromium.base.task.TaskRunner;
-import org.chromium.base.test.BaseJUnit4ClassRunner;
-import org.chromium.base.test.UiThreadTest;
-import org.chromium.base.test.util.Batch;
 import org.chromium.base.test.util.Feature;
 import org.chromium.chrome.browser.tab.Tab;
 import org.chromium.chrome.browser.tab.TabImpl;
@@ -43,7 +36,6 @@ import org.chromium.chrome.browser.tabmodel.TabPersistentStore.TabPersistentStor
 import org.chromium.chrome.browser.tabmodel.TabPersistentStore.TabRestoreDetails;
 import org.chromium.components.embedder_support.util.UrlConstants;
 import org.chromium.content_public.browser.LoadUrlParams;
-import org.chromium.content_public.browser.test.NativeLibraryTestUtils;
 import org.chromium.content_public.browser.test.util.CriteriaHelper;
 
 import java.io.IOException;
@@ -53,8 +45,6 @@ import java.util.concurrent.atomic.AtomicBoolean;
 /**
  * Unit tests for the tab persistent store logic.
  */
-@RunWith(BaseJUnit4ClassRunner.class)
-@Batch(Batch.UNIT_TESTS)
 public class TabPersistentStoreUnitTest {
     private static final Integer REGULAR_TAB_ID_1 = 1;
     private static final Integer INCOGNITO_TAB_ID_1 = 21;
@@ -91,12 +81,13 @@ public class TabPersistentStoreUnitTest {
 
     private TabPersistentStore mPersistentStore;
 
-    @Before
+    @CalledByNative
+    private TabPersistentStoreUnitTest() {}
+
+    @CalledByNative
     public void setUp() {
         MockitoAnnotations.initMocks(this);
 
-        NativeLibraryTestUtils.loadNativeLibraryNoBrowserProcess();
-
         when(mIncognitoTabModel.isIncognito()).thenReturn(true);
         when(mTabModelSelector.getModel(false)).thenReturn(mNormalTabModel);
         when(mTabModelSelector.getModel(true)).thenReturn(mIncognitoTabModel);
@@ -110,18 +101,17 @@ public class TabPersistentStoreUnitTest {
         when(mPersistencePolicy.performInitialization(any(TaskRunner.class))).thenReturn(false);
     }
 
-    @After
+    @CalledByNative
     public void tearDown() throws Exception {
         // Flush pending PersistentStore tasks.
         final AtomicBoolean flushed = new AtomicBoolean(false);
         if (mPersistentStore != null) {
             mPersistentStore.getTaskRunnerForTests().postTask(() -> { flushed.set(true); });
-            CriteriaHelper.pollUiThread(() -> flushed.get());
+            CriteriaHelper.pollUiThreadNested(() -> flushed.get());
         }
     }
 
-    @Test
-    @SmallTest
+    @CalledByNativeJavaTest
     @Feature("TabPersistentStore")
     public void testNtpSaveBehavior() {
         when(mNormalTabModel.index()).thenReturn(TabList.INVALID_TAB_INDEX);
@@ -172,8 +162,7 @@ public class TabPersistentStoreUnitTest {
         assertTrue(mPersistentStore.isTabPendingSave(ntpWithAllTheNavsTab));
     }
 
-    @Test
-    @SmallTest
+    @CalledByNativeJavaTest
     @Feature("TabPersistentStore")
     public void testNotActiveEmptyNtpIgnoredDuringRestore() {
         mPersistentStore = new TabPersistentStore(
@@ -187,8 +176,7 @@ public class TabPersistentStoreUnitTest {
         verifyZeroInteractions(mNormalTabCreator);
     }
 
-    @Test
-    @SmallTest
+    @CalledByNativeJavaTest
     @Feature("TabPersistentStore")
     public void testActiveEmptyNtpNotIgnoredDuringRestore() {
         when(mTabModelSelector.isIncognitoSelected()).thenReturn(false);
@@ -213,8 +201,7 @@ public class TabPersistentStoreUnitTest {
                         eq(TabLaunchType.FROM_RESTORE), (Tab) isNull());
     }
 
-    @Test
-    @SmallTest
+    @CalledByNativeJavaTest
     @Feature("TabPersistentStore")
     public void testNtpFromMergeWithNoStateNotIgnoredDuringMerge() {
         when(mTabModelSelector.isIncognitoSelected()).thenReturn(false);
@@ -245,8 +232,7 @@ public class TabPersistentStoreUnitTest {
                         eq(TabLaunchType.FROM_RESTORE), (Tab) isNull());
     }
 
-    @Test
-    @SmallTest
+    @CalledByNativeJavaTest
     @Feature("TabPersistentStore")
     public void testNtpWithStateNotIgnoredDuringRestore() {
         mPersistentStore = new TabPersistentStore(
@@ -261,8 +247,7 @@ public class TabPersistentStoreUnitTest {
         verify(mNormalTabCreator).createFrozenTab(eq(ntpState), eq(1), anyInt());
     }
 
-    @Test
-    @SmallTest
+    @CalledByNativeJavaTest
     @Feature("TabPersistentStore")
     public void testActiveEmptyIncognitoNtpNotIgnoredDuringRestore() {
         when(mTabModelSelector.isIncognitoSelected()).thenReturn(true);
@@ -287,8 +272,7 @@ public class TabPersistentStoreUnitTest {
                         eq(TabLaunchType.FROM_RESTORE), (Tab) isNull());
     }
 
-    @Test
-    @SmallTest
+    @CalledByNativeJavaTest
     @Feature("TabPersistentStore")
     public void testNotActiveIncognitoNtpIgnoredDuringRestore() {
         mPersistentStore = new TabPersistentStore(
@@ -302,8 +286,7 @@ public class TabPersistentStoreUnitTest {
         verifyZeroInteractions(mIncognitoTabCreator);
     }
 
-    @Test
-    @SmallTest
+    @CalledByNativeJavaTest
     @Feature("TabPersistentStore")
     public void testActiveEmptyIncognitoNtpIgnoredDuringRestoreIfIncognitoLoadingIsDisabled() {
         mPersistentStore = new TabPersistentStore(
@@ -317,9 +300,7 @@ public class TabPersistentStoreUnitTest {
         verifyZeroInteractions(mIncognitoTabCreator);
     }
 
-    @Test
-    @SmallTest
-    @UiThreadTest
+    @CalledByNativeJavaTest
     @Feature("TabPersistentStore")
     public void testSerializeTabModelSelector() throws IOException {
         setupSerializationTestMocks();
@@ -342,9 +323,7 @@ public class TabPersistentStoreUnitTest {
                 metadata.incognitoModelMetadata.urls.get(1));
     }
 
-    @Test
-    @SmallTest
-    @UiThreadTest
+    @CalledByNativeJavaTest
     @Feature("TabPersistentStore")
     public void testSerializeTabModelSelector_tabsBeingRestored() throws IOException {
         setupSerializationTestMocks();

chrome/android/javatests/src/org/chromium/chrome/browser/toolbar/ToolbarSecurityIconTest.java → chrome/android/native_java_unittests/src/org/chromium/chrome/browser/toolbar/ToolbarSecurityIconTest.java

@@ -9,33 +9,22 @@ import static org.mockito.Mockito.doReturn;
 import static org.mockito.Mockito.spy;
 import static org.mockito.Mockito.when;
 
-import androidx.test.filters.SmallTest;
-
-import org.junit.Before;
-import org.junit.Rule;
-import org.junit.Test;
-import org.junit.runner.RunWith;
 import org.mockito.Mock;
 import org.mockito.MockitoAnnotations;
 
 import org.chromium.base.ContextUtils;
-import org.chromium.base.test.BaseJUnit4ClassRunner;
-import org.chromium.base.test.UiThreadTest;
-import org.chromium.base.test.util.Batch;
-import org.chromium.base.test.util.JniMocker;
+import org.chromium.base.annotations.CalledByNative;
+import org.chromium.base.annotations.CalledByNativeJavaTest;
 import org.chromium.chrome.R;
 import org.chromium.chrome.browser.omnibox.LocationBarLayout;
 import org.chromium.chrome.browser.tab.TabImpl;
 import org.chromium.components.security_state.ConnectionSecurityLevel;
 import org.chromium.components.security_state.SecurityStateModel;
 import org.chromium.components.security_state.SecurityStateModelJni;
-import org.chromium.content_public.browser.test.NativeLibraryTestUtils;
 
 /**
  * Unit tests for {@link LocationBarLayout} class.
  */
-@RunWith(BaseJUnit4ClassRunner.class)
-@Batch(Batch.UNIT_TESTS)
 public final class ToolbarSecurityIconTest {
     private static final boolean IS_SMALL_DEVICE = true;
     private static final boolean IS_OFFLINE_PAGE = true;
@@ -45,9 +34,6 @@ public final class ToolbarSecurityIconTest {
             new int[] {ConnectionSecurityLevel.NONE, ConnectionSecurityLevel.WARNING,
                     ConnectionSecurityLevel.DANGEROUS, ConnectionSecurityLevel.SECURE};
 
-    @Rule
-    public JniMocker mocker = new JniMocker();
-
     @Mock
     private TabImpl mTab;
     @Mock
@@ -56,20 +42,23 @@ public final class ToolbarSecurityIconTest {
     @Mock
     private LocationBarModel mLocationBarModel;
 
-    @Before
+    @CalledByNative
+    private ToolbarSecurityIconTest() {}
+
+    @CalledByNative
     public void setUp() {
         MockitoAnnotations.initMocks(this);
-
-        NativeLibraryTestUtils.loadNativeLibraryNoBrowserProcess();
-
-        mocker.mock(SecurityStateModelJni.TEST_HOOKS, mSecurityStateMocks);
+        SecurityStateModelJni.TEST_HOOKS.setInstanceForTesting(mSecurityStateMocks);
         mLocationBarModel = spy(new LocationBarModel(ContextUtils.getApplicationContext()));
         mLocationBarModel.initializeWithNative();
     }
 
-    @Test
-    @SmallTest
-    @UiThreadTest
+    @CalledByNative
+    public void tearDown() {
+        SecurityStateModelJni.TEST_HOOKS.setInstanceForTesting(null);
+    }
+
+    @CalledByNativeJavaTest
     public void testGetSecurityLevel() {
         assertEquals(ConnectionSecurityLevel.NONE,
                 mLocationBarModel.getSecurityLevel(null, !IS_OFFLINE_PAGE, null));
@@ -102,9 +91,7 @@ public final class ToolbarSecurityIconTest {
                 mLocationBarModel.getSecurityLevel(mTab, !IS_OFFLINE_PAGE, null));
     }
 
-    @Test
-    @SmallTest
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testGetSecurityIconResource() {
         when(mSecurityStateMocks.shouldShowDangerTriangleForWarningLevel()).thenReturn(false);
         for (int securityLevel : SECURITY_LEVELS) {
@@ -181,9 +168,7 @@ public final class ToolbarSecurityIconTest {
                         !IS_SMALL_DEVICE, !IS_OFFLINE_PAGE, !IS_PREVIEW, !IS_PAINT_PREVIEW));
     }
 
-    @Test
-    @SmallTest
-    @UiThreadTest
+    @CalledByNativeJavaTest
     public void testGetSecurityIconResourceForMarkHttpAsDangerWarning() {
         when(mSecurityStateMocks.shouldShowDangerTriangleForWarningLevel()).thenReturn(false);
         assertEquals(0,

chrome/browser/contextmenu/chrome_context_menu_populator_unittest.cc

+// Copyright 2019 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.
+
+#include "base/android/jni_android.h"
+#include "chrome/android/native_j_unittests_jni_headers/ChromeContextMenuPopulatorTest_jni.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using base::android::AttachCurrentThread;
+
+class ChromeContextMenuPopulatorTest : public ::testing::Test {
+ public:
+  ChromeContextMenuPopulatorTest()
+      : j_test_(Java_ChromeContextMenuPopulatorTest_Constructor(
+            AttachCurrentThread())) {}
+
+  void SetUp() override {
+    Java_ChromeContextMenuPopulatorTest_setUp(AttachCurrentThread(), j_test_);
+  }
+
+  const base::android::ScopedJavaGlobalRef<jobject>& j_test() {
+    return j_test_;
+  }
+
+ private:
+  base::android::ScopedJavaGlobalRef<jobject> j_test_;
+};
+
+JAVA_TESTS(ChromeContextMenuPopulatorTest, j_test())

chrome/browser/installedapp/installed_app_provider_unittest.cc

+// Copyright 2020 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.
+
+#include "base/android/jni_android.h"
+#include "chrome/android/native_j_unittests_jni_headers/InstalledAppProviderTest_jni.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using base::android::AttachCurrentThread;
+
+class InstalledAppProviderTest : public ::testing::Test {
+ public:
+  InstalledAppProviderTest()
+      : j_test_(
+            Java_InstalledAppProviderTest_Constructor(AttachCurrentThread())) {}
+
+  void SetUp() override {
+    Java_InstalledAppProviderTest_setUp(AttachCurrentThread(), j_test_);
+  }
+
+  void TearDown() override {
+    Java_InstalledAppProviderTest_tearDown(AttachCurrentThread(), j_test_);
+  }
+
+  const base::android::ScopedJavaGlobalRef<jobject>& j_test() {
+    return j_test_;
+  }
+
+ private:
+  base::android::ScopedJavaGlobalRef<jobject> j_test_;
+};
+
+JAVA_TESTS(InstalledAppProviderTest, j_test())

chrome/browser/notifications/notification_platform_bridge_android_unittest.cc

+// Copyright 2020 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.
+
+#include "base/android/jni_android.h"
+#include "chrome/android/native_j_unittests_jni_headers/NotificationPlatformBridgeUnitTest_jni.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using base::android::AttachCurrentThread;
+
+class NotificationPlatformBridgeUnitTest : public ::testing::Test {
+ public:
+  NotificationPlatformBridgeUnitTest()
+      : j_test_(Java_NotificationPlatformBridgeUnitTest_Constructor(
+            AttachCurrentThread())) {}
+
+  const base::android::ScopedJavaGlobalRef<jobject>& j_test() {
+    return j_test_;
+  }
+
+ private:
+  base::android::ScopedJavaGlobalRef<jobject> j_test_;
+};
+
+JAVA_TESTS(NotificationPlatformBridgeUnitTest, j_test())

chrome/browser/partnercustomizations/partner_browser_customizations_unittest.cc

+// Copyright 2020 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.
+
+#include "base/android/jni_android.h"
+#include "chrome/android/native_j_unittests_jni_headers/PartnerBrowserCustomizationsUnitTest_jni.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using base::android::AttachCurrentThread;
+
+class PartnerBrowserCustomizationsUnitTest : public ::testing::Test {
+ public:
+  PartnerBrowserCustomizationsUnitTest()
+      : j_test_(Java_PartnerBrowserCustomizationsUnitTest_Constructor(
+            AttachCurrentThread())) {}
+
+  const base::android::ScopedJavaGlobalRef<jobject>& j_test() {
+    return j_test_;
+  }
+
+ private:
+  base::android::ScopedJavaGlobalRef<jobject> j_test_;
+};
+
+JAVA_TESTS(PartnerBrowserCustomizationsUnitTest, j_test())

chrome/browser/payments/BUILD.gn

@@ -80,9 +80,15 @@ source_set("unittests") {
   ]
 
   if (is_android) {
-    sources += [ "android/payment_app_service_bridge_unittest.cc" ]
+    sources += [
+      "android/android_payment_app_finder_unittest.cc",
+      "android/android_payment_app_unittest.cc",
+      "android/payment_app_service_bridge_unittest.cc",
+      "android/payment_manifest_verifier_unittest.cc",
+    ]
 
     deps += [
+      "//chrome/android:native_j_unittests_jni_headers",
       "//components/payments/content",
       "//components/payments/content:utils",
     ]

chrome/browser/payments/android/android_payment_app_finder_unittest.cc

+// Copyright 2020 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.
+
+#include "base/android/jni_android.h"
+#include "chrome/android/native_j_unittests_jni_headers/AndroidPaymentAppFinderUnitTest_jni.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using base::android::AttachCurrentThread;
+
+class AndroidPaymentAppFinderUnitTest : public ::testing::Test {
+ public:
+  AndroidPaymentAppFinderUnitTest()
+      : j_test_(Java_AndroidPaymentAppFinderUnitTest_Constructor(
+            AttachCurrentThread())) {}
+
+  const base::android::ScopedJavaGlobalRef<jobject>& j_test() {
+    return j_test_;
+  }
+
+ private:
+  base::android::ScopedJavaGlobalRef<jobject> j_test_;
+};
+
+JAVA_TESTS(AndroidPaymentAppFinderUnitTest, j_test())

chrome/browser/payments/android/android_payment_app_unittest.cc

+// Copyright 2020 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.
+
+#include "base/android/jni_android.h"
+#include "chrome/android/native_j_unittests_jni_headers/AndroidPaymentAppUnitTest_jni.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+class AndroidPaymentAppUnitTest : public ::testing::Test {
+ public:
+  AndroidPaymentAppUnitTest()
+      : j_test_(Java_AndroidPaymentAppUnitTest_Constructor(
+            base::android::AttachCurrentThread())) {}
+
+  const base::android::ScopedJavaGlobalRef<jobject>& j_test() {
+    return j_test_;
+  }
+
+ private:
+  base::android::ScopedJavaGlobalRef<jobject> j_test_;
+};
+
+JAVA_TESTS(AndroidPaymentAppUnitTest, j_test())

chrome/browser/payments/android/payment_manifest_verifier_unittest.cc

+// Copyright 2020 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.
+
+#include "base/android/jni_android.h"
+#include "chrome/android/native_j_unittests_jni_headers/PaymentManifestVerifierTest_jni.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using base::android::AttachCurrentThread;
+
+class PaymentManifestVerifierTest : public ::testing::Test {
+ public:
+  PaymentManifestVerifierTest()
+      : j_test_(Java_PaymentManifestVerifierTest_Constructor(
+            AttachCurrentThread())) {}
+
+  const base::android::ScopedJavaGlobalRef<jobject>& j_test() {
+    return j_test_;
+  }
+
+ private:
+  base::android::ScopedJavaGlobalRef<jobject> j_test_;
+};
+
+JAVA_TESTS(PaymentManifestVerifierTest, j_test())

chrome/browser/tabmodel/tab_persistent_store_unittest.cc

+// Copyright 2020 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.
+
+#include "base/android/jni_android.h"
+#include "chrome/android/native_j_unittests_jni_headers/TabPersistentStoreUnitTest_jni.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using base::android::AttachCurrentThread;
+
+class TabPersistentStoreUnitTest : public ::testing::Test {
+ public:
+  TabPersistentStoreUnitTest()
+      : j_test_(Java_TabPersistentStoreUnitTest_Constructor(
+            AttachCurrentThread())) {}
+
+  void SetUp() override {
+    Java_TabPersistentStoreUnitTest_setUp(AttachCurrentThread(), j_test_);
+  }
+
+  void TearDown() override {
+    Java_TabPersistentStoreUnitTest_tearDown(AttachCurrentThread(), j_test_);
+  }
+
+  const base::android::ScopedJavaGlobalRef<jobject>& j_test() {
+    return j_test_;
+  }
+
+ private:
+  base::android::ScopedJavaGlobalRef<jobject> j_test_;
+};
+
+JAVA_TESTS(TabPersistentStoreUnitTest, j_test())

chrome/browser/toolbar/toolbar_security_icon_unittest.cc

+// Copyright 2020 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.
+
+#include "base/android/jni_android.h"
+#include "chrome/android/native_j_unittests_jni_headers/ToolbarSecurityIconTest_jni.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using base::android::AttachCurrentThread;
+
+class ToolbarSecurityIconTest : public ::testing::Test {
+ public:
+  ToolbarSecurityIconTest()
+      : j_test_(
+            Java_ToolbarSecurityIconTest_Constructor(AttachCurrentThread())) {}
+
+  void SetUp() override {
+    Java_ToolbarSecurityIconTest_setUp(AttachCurrentThread(), j_test_);
+  }
+
+  void TearDown() override {
+    Java_ToolbarSecurityIconTest_tearDown(AttachCurrentThread(), j_test_);
+  }
+
+  const base::android::ScopedJavaGlobalRef<jobject>& j_test() {
+    return j_test_;
+  }
+
+ private:
+  base::android::ScopedJavaGlobalRef<jobject> j_test_;
+};
+
+JAVA_TESTS(ToolbarSecurityIconTest, j_test())

chrome/test/BUILD.gn

@@ -4032,11 +4032,15 @@ test("unit_tests") {
       "../browser/autofill/autofill_save_card_infobar_delegate_mobile_unittest.cc",
       "../browser/autofill/credit_card_accessory_controller_impl_unittest.cc",
       "../browser/autofill/manual_filling_controller_impl_unittest.cc",
+      "../browser/contextmenu/chrome_context_menu_populator_unittest.cc",
+      "../browser/installedapp/installed_app_provider_unittest.cc",
       "../browser/media/android/cdm/media_drm_origin_id_manager_unittest.cc",
       "../browser/metrics/chrome_android_metrics_provider_unittest.cc",
+      "../browser/notifications/notification_platform_bridge_android_unittest.cc",
       "../browser/offline_pages/prefetch/notifications/prefetch_notification_service_impl_unittest.cc",
       "../browser/optimization_guide/android/optimization_guide_bridge_unittest.cc",
       "../browser/page_load_metrics/observers/android_page_load_metrics_observer_unittest.cc",
+      "../browser/partnercustomizations/partner_browser_customizations_unittest.cc",
       "../browser/password_manager/android/account_chooser_dialog_android_unittest.cc",
       "../browser/password_manager/android/all_passwords_bottom_sheet_controller_unittest.cc",
       "../browser/password_manager/android/auto_signin_first_run_dialog_android_unittest.cc",
@@ -4046,6 +4050,8 @@ test("unit_tests") {
       "../browser/password_manager/android/save_password_infobar_delegate_android_unittest.cc",
       "../browser/password_manager/android/update_password_infobar_delegate_android_unittest.cc",
       "../browser/permissions/permission_prompt_android_unittest.cc",
+      "../browser/tabmodel/tab_persistent_store_unittest.cc",
+      "../browser/toolbar/toolbar_security_icon_unittest.cc",
       "../browser/touch_to_fill/touch_to_fill_controller_unittest.cc",
       "../browser/translate/translate_manager_render_view_host_android_unittest.cc",
     ]
@@ -4055,6 +4061,8 @@ test("unit_tests") {
       "//chrome:chrome_android_core",
       "//chrome/android:app_hooks_java",
       "//chrome/android:chrome_java",
+      "//chrome/android:native_j_unittests_jni_headers",
+      "//chrome/android:native_java_unittests_java",
       "//chrome/android/features/media_router:java",
       "//chrome/browser/optimization_guide/android:native_j_unittests_jni_headers",
       "//chrome/browser/optimization_guide/android:native_java_unittests",