Android: Remove GetApplicationContext part 2

chrome/android/javatests/src/org/chromium/chrome/browser/invalidation/InvalidationServiceTest.java

 // Copyright 2013 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.invalidation;
 
 import android.content.Intent;
 import android.support.test.InstrumentationRegistry;
 import android.support.test.filters.SmallTest;
 import android.support.test.rule.UiThreadTestRule;
 
 import com.google.ipc.invalidation.external.client.types.ObjectId;
 
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Rule;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 
+import org.chromium.base.ContextUtils;
 import org.chromium.base.test.BaseJUnit4ClassRunner;
 import org.chromium.base.test.util.Feature;
 import org.chromium.chrome.test.invalidation.IntentSavingContext;
 import org.chromium.components.invalidation.InvalidationService;
 import org.chromium.components.sync.AndroidSyncSettings;
 import org.chromium.components.sync.ModelType;
 import org.chromium.components.sync.ModelTypeHelper;
 import org.chromium.components.sync.notifier.InvalidationIntentProtocol;
 import org.chromium.components.sync.test.util.MockSyncContentResolverDelegate;
 import org.chromium.content.browser.test.NativeLibraryTestRule;
 
 import java.util.Set;
 
 /**
  * Tests for {@link InvalidationService}.
  */
 @RunWith(BaseJUnit4ClassRunner.class)
 public class InvalidationServiceTest {
     @Rule
     public NativeLibraryTestRule mActivityTestRule = new NativeLibraryTestRule();
 
     @Rule
     public UiThreadTestRule mUiThreadTestRule = new UiThreadTestRule();
 
-    private IntentSavingContext mContext;
+    private IntentSavingContext mAppContext;
 
     @Before
     public void setUp() throws Exception {
         mActivityTestRule.loadNativeLibraryAndInitBrowserProcess();
-        mContext = new IntentSavingContext(
-                InstrumentationRegistry.getInstrumentation().getTargetContext());
+        mAppContext = new IntentSavingContext(InstrumentationRegistry.getInstrumentation()
+                                                      .getTargetContext()
+                                                      .getApplicationContext());
+        // TODO(wnwen): Remove mAppContext and just use application context.
         // We don't want to use the system content resolver, so we override it.
         MockSyncContentResolverDelegate delegate = new MockSyncContentResolverDelegate();
         // Android master sync can safely always be on.
         delegate.setMasterSyncAutomatically(true);
-        AndroidSyncSettings.overrideForTests(mContext, delegate);
+        AndroidSyncSettings.overrideForTests(mAppContext, delegate);
+        ContextUtils.initApplicationContextForTests(mAppContext);
     }
 
     @Test
     @SmallTest
     @Feature({"Sync"})
     public void testSetRegisteredObjectIds() throws Throwable {
         mUiThreadTestRule.runOnUiThread(new Runnable() {
             @Override
             public void run() {
-                InvalidationService service = InvalidationServiceFactory.getForTest(mContext);
+                InvalidationService service = InvalidationServiceFactory.getForTest();
                 ObjectId bookmark = ModelTypeHelper.toObjectId(ModelType.BOOKMARKS);
                 service.setRegisteredObjectIds(new int[] {1, 2, bookmark.getSource()},
                         new String[] {"a", "b", new String(bookmark.getName())});
-                Assert.assertEquals(1, mContext.getNumStartedIntents());
+                Assert.assertEquals(1, mAppContext.getNumStartedIntents());
 
                 // Validate destination.
-                Intent intent = mContext.getStartedIntent(0);
+                Intent intent = mAppContext.getStartedIntent(0);
                 validateIntentComponent(intent);
                 Assert.assertEquals(InvalidationIntentProtocol.ACTION_REGISTER, intent.getAction());
 
                 // Validate registered object ids. The bookmark object should not be registered
                 // since it is a Sync type.
                 Assert.assertNull(intent.getStringArrayListExtra(
                         InvalidationIntentProtocol.EXTRA_REGISTERED_TYPES));
                 Set<ObjectId> objectIds = InvalidationIntentProtocol.getRegisteredObjectIds(intent);
                 Assert.assertEquals(2, objectIds.size());
                 Assert.assertTrue(objectIds.contains(ObjectId.newInstance(1, "a".getBytes())));
                 Assert.assertTrue(objectIds.contains(ObjectId.newInstance(2, "b".getBytes())));
             }
         });
     }
 
     /**
      * Asserts that {@code intent} is destined for the correct component.
      */
     private static void validateIntentComponent(Intent intent) {
         Assert.assertNotNull(intent.getComponent());
         Assert.assertEquals(ChromeInvalidationClientService.class.getName(),
                 intent.getComponent().getClassName());
     }
 
 }