| Index: chrome/android/javatests/src/org/chromium/chrome/browser/webapps/WebappActivityTestBase.java
|
| diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/webapps/WebappActivityTestBase.java b/chrome/android/javatests/src/org/chromium/chrome/browser/webapps/WebappActivityTestBase.java
|
| index ec1f26a1bd1fb623e199c01278aef335753d9c5c..e63b80f5f4cd68e990db62e09173a9e072aad64e 100644
|
| --- a/chrome/android/javatests/src/org/chromium/chrome/browser/webapps/WebappActivityTestBase.java
|
| +++ b/chrome/android/javatests/src/org/chromium/chrome/browser/webapps/WebappActivityTestBase.java
|
| @@ -92,16 +92,12 @@ public abstract class WebappActivityTestBase extends ChromeActivityTestCaseBase<
|
| protected void setUp() throws Exception {
|
| super.setUp();
|
|
|
| - // Register the webapp so when the data storage is opened, the test doesn't crash. There is
|
| - // no race condition with the retrieval as AsyncTasks are run sequentially on the background
|
| - // thread.
|
| - WebappRegistry.registerWebapp(
|
| - WEBAPP_ID, new WebappRegistry.FetchWebappDataStorageCallback() {
|
| - @Override
|
| - public void onWebappDataStorageRetrieved(WebappDataStorage storage) {
|
| - storage.updateFromShortcutIntent(createIntent());
|
| - }
|
| - });
|
| + // Register the webapp so when the data storage is opened, the test doesn't crash.
|
| + WebappRegistry.refreshSharedPrefsForTesting();
|
| + TestFetchStorageCallback callback = new TestFetchStorageCallback();
|
| + WebappRegistry.getInstance().register(WEBAPP_ID, callback);
|
| + callback.waitForCallback(0);
|
| + callback.getStorage().updateFromShortcutIntent(createIntent());
|
| }
|
|
|
| /**
|
|
|