Merge "Add test for DE (user) snapshot and restore." (e5981fae) · Commits · e / os / android_frameworks_base

tests/RollbackTest/StagedRollbackTest/src/com/android/tests/rollback/host/StagedRollbackTest.java

+44 −0

Original line number	Diff line number	Diff line
		@@ -328,6 +328,46 @@ public class StagedRollbackTest extends BaseHostJUnit4Test {
		assertNull(getDevice().pullFile(newFilePath4));
		}

		/**
		* Tests that data in DE (user) apex data directory is restored when apex is rolled back.
		*/
		@Test
		public void testRollbackApexDataDirectories_DeUser() throws Exception {
		pushTestApex();

		// Push files to apex data directory
		String oldFilePath1 = apexDataDirDeUser(
		APK_IN_APEX_TESTAPEX_NAME, 0) + "/" + TEST_FILENAME_1;
		String oldFilePath2 =
		apexDataDirDeUser(APK_IN_APEX_TESTAPEX_NAME, 0) + TEST_SUBDIR + TEST_FILENAME_2;
		assertTrue(getDevice().pushString(TEST_STRING_1, oldFilePath1));
		assertTrue(getDevice().pushString(TEST_STRING_2, oldFilePath2));

		// Install new version of the APEX with rollback enabled
		runPhase("testRollbackApexDataDirectories_Phase1");
		getDevice().reboot();

		// Replace files in data directory
		getDevice().deleteFile(oldFilePath1);
		getDevice().deleteFile(oldFilePath2);
		String newFilePath3 =
		apexDataDirDeUser(APK_IN_APEX_TESTAPEX_NAME, 0) + "/" + TEST_FILENAME_3;
		String newFilePath4 =
		apexDataDirDeUser(APK_IN_APEX_TESTAPEX_NAME, 0) + TEST_SUBDIR + TEST_FILENAME_4;
		assertTrue(getDevice().pushString(TEST_STRING_3, newFilePath3));
		assertTrue(getDevice().pushString(TEST_STRING_4, newFilePath4));

		// Roll back the APEX
		runPhase("testRollbackApexDataDirectories_Phase2");
		getDevice().reboot();

		// Verify that old files have been restored and new files are gone
		assertEquals(TEST_STRING_1, getDevice().pullFileContents(oldFilePath1));
		assertEquals(TEST_STRING_2, getDevice().pullFileContents(oldFilePath2));
		assertNull(getDevice().pullFile(newFilePath3));
		assertNull(getDevice().pullFile(newFilePath4));
		}

		/**
		* Tests that data in CE apex data directory is restored when apex is rolled back.
		*/
		@@ -382,6 +422,10 @@ public class StagedRollbackTest extends BaseHostJUnit4Test {
		return String.format("/data/misc/apexdata/%s", apexName);
		}

		private static String apexDataDirDeUser(String apexName, int userId) {
		return String.format("/data/misc_de/%d/apexdata/%s", userId, apexName);
		}

		private static String apexDataDirCe(String apexName, int userId) {
		return String.format("/data/misc_ce/%d/apexdata/%s", userId, apexName);
		}