31     def setUp(self):  argument
33         self.assertTrue(serial_number, "$ANDROID_SERIAL is empty.")
34         self.dut = utils.AndroidDevice(serial_number)
35         self.adb = ADB(serial_number)
36         self.isCheckpoint_ = self.isCheckpoint()
38     def getFstab(self):  argument
40         self.adb.Execute(["wait-for-device"], timeout=900)
41         self.adb.Execute(["root"])
44             out, err, return_code = self.dut.Execute("getprop ro.boot." + prop)
50                 out, err, return_code = self.dut.Execute("cat " + filename + extension)
58     def isCheckpoint(self):  argument
59         fstab = self.getFstab().splitlines()
73     def reboot(self):  argument
74         self.adb.Execute(["reboot"])
76           self.adb.Execute(["wait-for-device"], timeout=900)
77         except self.adb.AdbError as e:
78           self.fail("Exception thrown waiting for device:" + e.msg())
84             self.adb.Execute(["root"])
91             self.dut.Execute("ls");
96     def checkBooted(self):  argument
98           out, err, return_code = self.dut.Execute("getprop sys.boot_completed")
101             self.assertEqual(boot_completed, 1)
106         self.fail("sys.boot_completed not set")
108     def testCheckpointEnabled(self):  argument
109         out, err, return_code = self.dut.Execute("getprop ro.product.first_api_level")
112           self.assertTrue(first_api_level < 29 or self.isCheckpoint_,
117     def testRollback(self):  argument
118         if not self.isCheckpoint_:
121         self.adb.Execute(["root"])
125         self.checkBooted()
128         self.dut.Execute("setprop persist.vold.dont_commit_checkpoint 1")
129         self.dut.Execute("echo " + self._ORIGINALVALUE + " > " + self._CHECKPOINTTESTFILE)
130         out, err, return_code = self.dut.Execute("vdc checkpoint startCheckpoint 1")
131         self.assertEqual(return_code, 0)
132         self.reboot()
135         self.dut.Execute("echo " + self._MODIFIEDVALUE + " > " + self._CHECKPOINTTESTFILE)
136         self.reboot()
139         out, err, return_code = self.dut.Execute("cat " + self._CHECKPOINTTESTFILE)
140         self.assertEqual(out.strip(), self._ORIGINALVALUE)
143         self.dut.Execute("setprop persist.vold.dont_commit_checkpoint 0")
144         out, err, return_code = self.dut.Execute("vdc checkpoint commitChanges")
145         self.assertEqual(return_code, 0)
146         self.reboot()
147         self.dut.Execute("rm " + self._CHECKPOINTTESTFILE)
149     def testCommit(self):  argument
150         if not self.isCheckpoint_:
153         self.adb.Execute(["root"])
157         self.checkBooted()
160         self.dut.Execute("setprop persist.vold.dont_commit_checkpoint 1")
161         self.dut.Execute("echo " + self._ORIGINALVALUE + " > " + self._CHECKPOINTTESTFILE)
162         out, err, return_code = self.dut.Execute("vdc checkpoint startCheckpoint 1")
163         self.assertEqual(return_code, 0)
164         self.reboot()
167         self.dut.Execute("echo " + self._MODIFIEDVALUE + " > " + self._CHECKPOINTTESTFILE)
168         self.dut.Execute("setprop persist.vold.dont_commit_checkpoint 0")
169         out, err, return_code = self.dut.Execute("vdc checkpoint commitChanges")
170         self.assertEqual(return_code, 0)
171         self.reboot()
174         out, err, return_code = self.dut.Execute("cat " + self._CHECKPOINTTESTFILE)
175         self.assertEqual(out.strip(), self._MODIFIEDVALUE)
178         self.dut.Execute("rm " + self._CHECKPOINTTESTFILE)