[Fuchsia] Allow device reboot to fail before paving/flashing

If a device is paved/flashed with a bad image, this will ensure that
the device gets repaved during a test run.

Bug: 1405525
Change-Id: I3120d9b54eff018b1b2ffca0e3260a35cabc3646
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/4162197
Reviewed-by: Zijie He <[email protected]>
Auto-Submit: Chong Gu <[email protected]>
Commit-Queue: Chong Gu <[email protected]>
Cr-Commit-Position: refs/heads/main@{#1092088}
NOKEYCHECK=True
GitOrigin-RevId: bc5cf4a55aa0ab4812a5fa39162acd982e63f197

Author: Chong Gu

fuchsia/test/flash_device.py

@@ -33,7 +33,11 @@ def _get_system_info(target: Optional[str]) -> Tuple[str, str]:
     if running_unattended():
         with ScopedFfxConfig('discovery.zedboot.enabled', 'true'):
             run_ffx_command(('target', 'reboot'), target_id=target)
-        run_ffx_command(('target', 'wait', '-t', '180'), target)
+        wait_cmd = run_ffx_command(('target', 'wait', '-t', '180'),
+                                   target,
+                                   check=False)
+        if wait_cmd.returncode != 0:
+            return ('', '')
 
     info_cmd = run_ffx_command(('target', 'show', '--json'),
                                target_id=target,

fuchsia/test/flash_device_unittests.py

@@ -208,6 +208,17 @@ def test_update_with_serial_num(self) -> None:
                                 should_pave=False)
         self.assertEqual(self._ffx_mock.call_count, 4)
 
+    def test_reboot_failure(self) -> None:
+        """Test update when |serial_num| is specified."""
+        self._ffx_mock.return_value.returncode = 1
+        with mock.patch('time.sleep'), \
+                mock.patch('os.path.exists', return_value=True), \
+                mock.patch('flash_device.running_unattended',
+                           return_value=True):
+            required, _ = flash_device.update_required('check',
+                                                       _TEST_IMAGE_DIR, None)
+            self.assertEqual(required, True)
+
     # pylint: disable=no-self-use
     def test_update_calls_paving_if_specified(self) -> None:
         """Test update calls pave if specified."""