Raman Tenneti
d0a660cf22
Fixed check-style errors. These files are AOSP specific files and they don't exist in the internal code. See https://source.android.com/setup/contribute/respectful-code for reference BUG=161896447 Test: m and Launcher3Tests $ find out -name "Launcher*.apk" -print -exec ls -l {} \; out/target/product/bonito/testcases/Launcher3Tests/arm64/Launcher3Tests.apk -rw-r----- 1 rtenneti primarygroup 947525 Aug 4 16:22 out/target/product/bonito/testcases/Launcher3Tests/arm64/Launcher3Tests.apk out/target/product/bonito/system_ext/priv-app/Launcher3QuickStep/Launcher3QuickStep.apk -rw-r----- 1 rtenneti primarygroup 9249956 Aug 4 16:20 out/target/product/bonito/system_ext/priv-app/Launcher3QuickStep/Launcher3QuickStep.apk Change-Id: I428eaa50ada96567a3b583eca1aa8ee2d8b85e6b
101 lines
3.2 KiB
Java
101 lines
3.2 KiB
Java
package com.android.launcher3.provider;
|
|
|
|
import static org.junit.Assert.assertEquals;
|
|
|
|
import android.content.ContentValues;
|
|
import android.database.Cursor;
|
|
import android.database.sqlite.SQLiteDatabase;
|
|
|
|
import androidx.test.InstrumentationRegistry;
|
|
import androidx.test.filters.MediumTest;
|
|
import androidx.test.runner.AndroidJUnit4;
|
|
|
|
import com.android.launcher3.LauncherProvider.DatabaseHelper;
|
|
import com.android.launcher3.LauncherSettings.Favorites;
|
|
|
|
import org.junit.Test;
|
|
import org.junit.runner.RunWith;
|
|
|
|
/**
|
|
* Tests for {@link RestoreDbTask}
|
|
*/
|
|
@MediumTest
|
|
@RunWith(AndroidJUnit4.class)
|
|
public class RestoreDbTaskTest {
|
|
|
|
@Test
|
|
public void testGetProfileId() throws Exception {
|
|
SQLiteDatabase db = new MyDatabaseHelper(23).getWritableDatabase();
|
|
assertEquals(23, new RestoreDbTask().getDefaultProfileId(db));
|
|
}
|
|
|
|
@Test
|
|
public void testMigrateProfileId() throws Exception {
|
|
SQLiteDatabase db = new MyDatabaseHelper(42).getWritableDatabase();
|
|
// Add some mock 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 RestoreDbTask().migrateProfileId(db, 42, 33);
|
|
|
|
// verify data migrated
|
|
assertEquals(0, getCount(db, "select * from favorites where profileId = 42"));
|
|
assertEquals(5, getCount(db, "select * from favorites where profileId = 33"));
|
|
}
|
|
|
|
@Test
|
|
public void testChangeDefaultColumn() throws Exception {
|
|
SQLiteDatabase db = new MyDatabaseHelper(42).getWritableDatabase();
|
|
// Add some mock 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 default column is 42
|
|
assertEquals(5, getCount(db, "select * from favorites where profileId = 42"));
|
|
|
|
new RestoreDbTask().changeDefaultColumn(db, 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(1, getCount(db, "select * from favorites where profileId = 33"));
|
|
}
|
|
|
|
private int getCount(SQLiteDatabase db, String sql) {
|
|
try (Cursor c = db.rawQuery(sql, null)) {
|
|
return c.getCount();
|
|
}
|
|
}
|
|
|
|
private class MyDatabaseHelper extends DatabaseHelper {
|
|
|
|
private final long mProfileId;
|
|
|
|
MyDatabaseHelper(long profileId) {
|
|
super(InstrumentationRegistry.getContext(), null, null);
|
|
mProfileId = profileId;
|
|
}
|
|
|
|
@Override
|
|
public long getDefaultUserSerial() {
|
|
return mProfileId;
|
|
}
|
|
|
|
@Override
|
|
protected void handleOneTimeDataUpgrade(SQLiteDatabase db) { }
|
|
|
|
protected void onEmptyDbCreated() { }
|
|
}
|
|
}
|