summaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
authorSunny Goyal <sunnygoyal@google.com>2016-05-05 14:37:05 -0700
committerSunny Goyal <sunnygoyal@google.com>2016-05-10 15:27:24 -0700
commitc190dbfb784c0bca629b6966ce314d735cf98db3 (patch)
treeb7df71ce2f6dea4698a84c8e7fabe0aac2362595 /tests
parent78508ec5ab69367e52eef1e8c3273ad6e2c40fc6 (diff)
downloadandroid_packages_apps_Trebuchet-c190dbfb784c0bca629b6966ce314d735cf98db3.tar.gz
android_packages_apps_Trebuchet-c190dbfb784c0bca629b6966ce314d735cf98db3.tar.bz2
android_packages_apps_Trebuchet-c190dbfb784c0bca629b6966ce314d735cf98db3.zip
Adding full backup handler for launcher3
Bug: 28302573 Change-Id: If0889a3b00b88b134b19d7f790159dfafdcdb004
Diffstat (limited to 'tests')
-rw-r--r--tests/src/com/android/launcher3/LauncherBackupAgentTest.java74
1 files changed, 74 insertions, 0 deletions
diff --git a/tests/src/com/android/launcher3/LauncherBackupAgentTest.java b/tests/src/com/android/launcher3/LauncherBackupAgentTest.java
new file mode 100644
index 000000000..798f0c25d
--- /dev/null
+++ b/tests/src/com/android/launcher3/LauncherBackupAgentTest.java
@@ -0,0 +1,74 @@
+package com.android.launcher3;
+
+import android.content.ContentValues;
+import android.database.Cursor;
+import android.database.sqlite.SQLiteDatabase;
+import android.test.AndroidTestCase;
+import android.test.suitebuilder.annotation.MediumTest;
+
+import com.android.launcher3.LauncherProvider.DatabaseHelper;
+import com.android.launcher3.LauncherSettings.Favorites;
+
+/**
+ * Tests for {@link LauncherBackupAgent}
+ */
+@MediumTest
+public class LauncherBackupAgentTest extends AndroidTestCase {
+
+ public void testGetProfileId() throws Exception {
+ SQLiteDatabase db = new MyDatabaseHelper(23).getWritableDatabase();
+ assertEquals(23, new LauncherBackupAgent().getDefaultProfileId(db));
+ }
+
+ public void testMigrateProfileId() throws Exception {
+ SQLiteDatabase db = new MyDatabaseHelper(42).getWritableDatabase();
+ // Add some dummy data
+ for (int i = 0; i < 5; i++) {
+ ContentValues values = new ContentValues();
+ values.put(Favorites._ID, i);
+ values.put(Favorites.TITLE, "item " + i);
+ db.insert(Favorites.TABLE_NAME, null, values);
+ }
+ // Verify item add
+ assertEquals(5, getCount(db, "select * from favorites where profileId = 42"));
+
+ new LauncherBackupAgent().migrateProfileId(db, 33);
+
+ // verify data migrated
+ assertEquals(0, getCount(db, "select * from favorites where profileId = 42"));
+ assertEquals(5, getCount(db, "select * from favorites where profileId = 33"));
+
+ // Verify default value changed
+ ContentValues values = new ContentValues();
+ values.put(Favorites._ID, 100);
+ values.put(Favorites.TITLE, "item 100");
+ db.insert(Favorites.TABLE_NAME, null, values);
+ assertEquals(6, getCount(db, "select * from favorites where profileId = 33"));
+ }
+
+ private int getCount(SQLiteDatabase db, String sql) {
+ Cursor c = db.rawQuery(sql, null);
+ try {
+ return c.getCount();
+ } finally {
+ c.getCount();
+ }
+ }
+
+ private class MyDatabaseHelper extends DatabaseHelper {
+
+ private final long mProfileId;
+
+ public MyDatabaseHelper(long profileId) {
+ super(getContext(), null, null, null);
+ mProfileId = profileId;
+ }
+
+ @Override
+ public long getDefaultUserSerial() {
+ return mProfileId;
+ }
+
+ protected void onEmptyDbCreated() { }
+ }
+}