Change AccountManagerHelper initialization

chrome/android/junit/src/org/chromium/chrome/browser/superviseduser/SupervisedUserContentProviderUnitTest.java

 // Copyright 2015 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.chrome.browser.superviseduser;
 
 import static org.hamcrest.CoreMatchers.is;
 import static org.junit.Assert.assertThat;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.anyLong;
 import static org.mockito.ArgumentMatchers.anyString;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.doAnswer;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
 import android.accounts.Account;
 import android.content.Context;
 import android.content.pm.ProviderInfo;
 
 import org.junit.After;
 import org.junit.Before;
+import org.junit.BeforeClass;
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.Mockito;
 import org.mockito.invocation.InvocationOnMock;
 import org.mockito.stubbing.Answer;
 import org.robolectric.Robolectric;
 import org.robolectric.RuntimeEnvironment;
 import org.robolectric.annotation.Config;
 
@@ skipping 16 matching lines @@
 @RunWith(LocalRobolectricTestRunner.class)
 @Config(manifest = Config.NONE)
 public class SupervisedUserContentProviderUnitTest {
     @Rule
     public DisableHistogramsRule mDisableHistogramsRule = new DisableHistogramsRule();
 
     private SupervisedUserContentProvider mSupervisedUserContentProvider;
 
     private static final String DEFAULT_CALLING_PACKAGE = "com.example.some.app";
 
+    private static AccountManagerDelegate sMockDelegate;
+
+    @BeforeClass
+    public static void overrideAccountManagerHelper() {
+        sMockDelegate = mock(AccountManagerDelegate.class);
+        AccountManagerHelper.overrideAccountManagerHelperForTests(sMockDelegate);
+    }
+
     // Override methods that wrap things that can't be mocked (including native calls).
     private static class MySupervisedUserContentProvider extends SupervisedUserContentProvider {
         @Override
         void startForcedSigninProcessor(Context context, Runnable onComplete) {
             ChromeSigninController.get().setSignedInAccountName("Dummy");
             onComplete.run();
         }
 
         @Override
         void listenForChildAccountStatusChange(Callback<Boolean> callback) {
@@ skipping 141 matching lines @@
                         any(SupervisedUserContentProvider.SupervisedUserQueryReply.class),
                         eq("url"));
     }
 
     @SuppressWarnings("unchecked")
     @Test
     public void testShouldProceed_notSignedIn() throws ProcessInitException {
         // Mock things called during startup
         ChromeBrowserInitializer mockBrowserInitializer = mock(ChromeBrowserInitializer.class);
         ChromeBrowserInitializer.setForTesting(mockBrowserInitializer);
-        AccountManagerDelegate mockDelegate = mock(AccountManagerDelegate.class);
-        AccountManagerHelper.overrideAccountManagerHelperForTests(
-                RuntimeEnvironment.application, mockDelegate);
         Account account = new Account("Google", "Dummy");
-        when(mockDelegate.getAccountsByType("Google")).thenReturn(new Account[] {account});
+        when(sMockDelegate.getAccountsByType("Google")).thenReturn(new Account[] {account});
 
         WebRestrictionsResult result =
                 mSupervisedUserContentProvider.shouldProceed(DEFAULT_CALLING_PACKAGE, "url");
 
         assertThat(result.shouldProceed(), is(true));
         verify(mockBrowserInitializer).handleSynchronousStartup();
         verify(mSupervisedUserContentProvider)
                 .startForcedSigninProcessor(any(Context.class), any(Runnable.class));
         verify(mSupervisedUserContentProvider)
                 .listenForChildAccountStatusChange(any(Callback.class));
         verify(mSupervisedUserContentProvider)
                 .nativeShouldProceed(eq(5678L),
                         any(SupervisedUserContentProvider.SupervisedUserQueryReply.class),
                         eq("url"));
-
-        AccountManagerHelper.resetAccountManagerHelperForTests();
     }
 
     @Test
     public void testShouldProceed_cannotSignIn() {
         // Mock things called during startup
         ChromeBrowserInitializer mockBrowserInitializer = mock(ChromeBrowserInitializer.class);
         ChromeBrowserInitializer.setForTesting(mockBrowserInitializer);
-        AccountManagerDelegate mockDelegate = mock(AccountManagerDelegate.class);
-        AccountManagerHelper.overrideAccountManagerHelperForTests(
-                RuntimeEnvironment.application, mockDelegate);
         Account account = new Account("Google", "Dummy");
-        when(mockDelegate.getAccountsByType("Google")).thenReturn(new Account[] {account});
+        when(sMockDelegate.getAccountsByType("Google")).thenReturn(new Account[] {account});
 
         // Change the behavior of the forced sign-in processor to not sign in.
         doAnswer(new Answer<Void>() {
             @Override
             public Void answer(InvocationOnMock invocation) throws Throwable {
                 invocation.<Runnable>getArgument(1).run();
                 return null;
             }
         })
                 .when(mSupervisedUserContentProvider)
                 .startForcedSigninProcessor(any(Context.class), any(Runnable.class));
 
         WebRestrictionsResult result =
                 mSupervisedUserContentProvider.shouldProceed(DEFAULT_CALLING_PACKAGE, "url");
 
         assertThat(result.shouldProceed(), is(false));
         assertThat(result.getErrorInt(0), is(5));
-
-        AccountManagerHelper.resetAccountManagerHelperForTests();
     }
 
     @Test
     public void testShouldProceed_requestWhitelisted() {
         mSupervisedUserContentProvider.setNativeSupervisedUserContentProviderForTesting(1234L);
 
         // Modify the result of the native call to block any URL.
         doAnswer(new Answer<Void>() {
             @Override
             public Void answer(InvocationOnMock invocation) throws Throwable {
@@ skipping 18 matching lines @@
 
         WebRestrictionsResult wrongCallingPackage = mSupervisedUserContentProvider.shouldProceed(
                 DEFAULT_CALLING_PACKAGE, "https://accounts.google.com/reauth");
         assertThat(wrongCallingPackage.shouldProceed(), is(false));
 
         WebRestrictionsResult nullCallingPackage = mSupervisedUserContentProvider.shouldProceed(
                 null, "https://accounts.google.com/reauth");
         assertThat(nullCallingPackage.shouldProceed(), is(false));
     }
 }