commit c0e9df6dd6f50aae681b5d403ec836951e17446d
author Sunny Goyal <sunnygoyal@google.com> Thu Jan 16 14:54:36 2020 -0800
committer Sunny Goyal <sunnygoyal@google.com> Thu Jan 16 14:57:39 2020 -0800
tree 03edf6dfcdf12f4247fc85eaeb4fac9dad468673
parent 9be1cfde7965412293a57bff54b094a7871227be

Removing some unnecessary LauncherProvider overrides

Change-Id: Ia07919309bd4fe942af152881642fe24de9b00ee

robolectric_tests/src/com/android/launcher3/model/BackupRestoreTest.java

diff --git a/robolectric_tests/src/com/android/launcher3/model/BackupRestoreTest.java b/robolectric_tests/src/com/android/launcher3/model/BackupRestoreTest.java
index 83b8599..6223760 100644
--- a/robolectric_tests/src/com/android/launcher3/model/BackupRestoreTest.java
+++ b/robolectric_tests/src/com/android/launcher3/model/BackupRestoreTest.java
@@ -23,12 +23,14 @@
 
 import android.database.sqlite.SQLiteDatabase;
 
+import com.android.launcher3.provider.RestoreDbTask;
 import com.android.launcher3.util.LauncherModelHelper;
 import com.android.launcher3.util.LauncherRoboTestRunner;
 
 import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
+import org.robolectric.RuntimeEnvironment;
 import org.robolectric.annotation.LooperMode;
 
 /**
@@ -44,7 +46,8 @@
     @Before
     public void setUp() {
         mModelHelper = new LauncherModelHelper();
-        mDb = mModelHelper.provider.getDbWithRestoreDbTask();
+        RestoreDbTask.setPending(RuntimeEnvironment.application, true);
+        mDb = mModelHelper.provider.getDb();
     }
 
     @Test