Add pathway to auto-migrate users when Chrome launches

chrome/android/javatests/src/org/chromium/chrome/browser/tabmodel/DocumentModeAssassinTest.java

 // Copyright 2016 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.tabmodel;
 
 import android.content.Context;
 import android.test.MoreAsserts;
 import android.test.suitebuilder.annotation.MediumTest;
 import android.test.suitebuilder.annotation.SmallTest;
@@ skipping 192 matching lines @@
                     writeDoneCallback.notifyCalled();
                 }
             }
         };
 
         File[] tabbedModeFilesBefore = mTabbedModeDirectory.getDataDirectory().listFiles();
         assertNotNull(tabbedModeFilesBefore);
         int numFilesBefore = tabbedModeFilesBefore.length;
         assertEquals(0, writeStartedCallback.getCallCount());
         assertEquals(0, writeDoneCallback.getCallCount());
+        final Context context = getInstrumentation().getTargetContext();
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
                 assassin.addObserver(observer);
-                assassin.writeTabModelMetadata(
-                        mTabbedModeDirectory.getDataDirectory(), testTabModel, migratedTabIds);
+                assassin.writeTabModelMetadata(testTabModel, migratedTabIds, context,
+                        mTabbedModeDirectory.getDataDirectory());
             }
         });
 
         // Wait and confirm that the tabbed mode metadata file was written out.
         writeStartedCallback.waitForCallback(0);
         writeDoneCallback.waitForCallback(0);
         File[] tabbedModeFilesAfter = mTabbedModeDirectory.getDataDirectory().listFiles();
         assertNotNull(tabbedModeFilesAfter);
         assertEquals(numFilesBefore + 1, tabbedModeFilesAfter.length);
 
         // Load up the metadata file via a TabPersistentStore to make sure that it contains all of
         // the migrated tab information.
         loadNativeLibraryAndInitBrowserProcess();
         TabPersistentStore.setBaseStateDirectory(mTabbedModeDirectory.getBaseDirectory());
 
-        Context context = getInstrumentation().getTargetContext();
         TestTabModelSelector selector = new TestTabModelSelector(context);
         TabPersistentStore store = selector.mTabPersistentStore;
         MockTabPersistentStoreObserver mockObserver = selector.mTabPersistentStoreObserver;
 
         // Load up the TabModel metadata.
         int numExpectedTabs = TEST_INFO.numRegularTabs + TEST_INFO.numIncognitoTabs;
         store.loadState();
         mockObserver.initializedCallback.waitForCallback(0, 1);
         assertEquals(numExpectedTabs, mockObserver.mTabCountAtStartup);
         mockObserver.detailsReadCallback.waitForCallback(0, TEST_INFO.contents.length);
@@ skipping 77 matching lines @@
                 TabState.SAVED_TAB_STATE_FILE_PREFIX + "_unparseable");
 
         // Kick off copying the tab states.
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
                 assassin.addObserver(observer);
                 assertEquals(0, copyStartedCallback.getCallCount());
                 assertEquals(0, copyDoneCallback.getCallCount());
                 assertEquals(0, copyCallback.getCallCount());
-                assassin.copyTabStateFiles(selectedTabId, mDocumentModeDirectory.getDataDirectory(),
+                assassin.copyTabStateFiles(selectedTabId, getInstrumentation().getTargetContext(),
+                        mDocumentModeDirectory.getDataDirectory(),
                         mTabbedModeDirectory.getDataDirectory());
             }
         });
         copyStartedCallback.waitForCallback(0);
 
         // Confirm that the first TabState file copied back is the selected one.
         copyCallback.waitForCallback(0);
         if (selectedTabId != Tab.INVALID_TAB_ID) assertEquals(selectedTabId, firstCopiedId.get());
 
         // Confirm that all the TabState files were copied over.
@@ skipping 11 matching lines @@
 
         for (int i = 0; i < TAB_STATE_INFO.length; i++) {
             boolean found = false;
             for (int j = 0; j < tabbedModeFilesAfter.length && !found; j++) {
                 found |= TAB_STATE_INFO[i].filename.equals(tabbedModeFilesAfter[j].getName());
             }
             assertTrue("Couldn't find file: " + TAB_STATE_INFO[i].filename, found);
         }
     }
 }