gooftool: Add --test_umount option in for Tast testing

Add --test_umount in `gooftool wipe_in_place` for Tast testing.

BUG=b:147337631
TEST=manually run and passed new created Tast test

Change-Id: I5582fa25da5ccbb40ffe389342a478f35e36c918
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/factory/+/1989545
Tested-by: Meng-Huan Yu <menghuan@chromium.org>
Auto-Submit: Meng-Huan Yu <menghuan@chromium.org>
Reviewed-by: Wei-Han Chen <stimim@chromium.org>
Commit-Queue: Meng-Huan Yu <menghuan@chromium.org>
diff --git a/py/gooftool/commands.py b/py/gooftool/commands.py
index a534f05..2e9e161 100755
--- a/py/gooftool/commands.py
+++ b/py/gooftool/commands.py
@@ -252,6 +252,11 @@
          'Each item should be a sub-command of gooftool. '
          'e.g. "gooftool verify --skip_list verify_tpm clear_gbb_flags".')
 
+_test_umount_cmd_arg = CmdArg(
+    '--test_umount', action='store_true',
+    help='(For testing only) Only umount rootfs and stateful partition '
+         'instead of running full wiping and cutoff process.')
+
 _rlz_embargo_end_date_offset_cmd_arg = CmdArg(
     '--embargo_offset', type=int, default=7, choices=list(range(7, 15)),
     help='Change the offset of embargo end date, cannot less than 7 days or '
@@ -475,7 +480,8 @@
          _shopfloor_url_args_cmd_arg,
          _station_ip_cmd_arg,
          _station_port_cmd_arg,
-         _wipe_finish_token_cmd_arg)
+         _wipe_finish_token_cmd_arg,
+         _test_umount_cmd_arg)
 def WipeInPlace(options):
   """Start factory wipe directly without reboot."""
 
@@ -483,7 +489,8 @@
                                    options.shopfloor_url,
                                    options.station_ip,
                                    options.station_port,
-                                   options.wipe_finish_token)
+                                   options.wipe_finish_token,
+                                   options.test_umount)
 
 @Command('wipe_init',
          CmdArg('--wipe_args', help='arguments for clobber-state'),
@@ -495,7 +502,8 @@
          _station_ip_cmd_arg,
          _station_port_cmd_arg,
          _wipe_finish_token_cmd_arg,
-         _keep_developer_mode_flag_after_clobber_state_cmd_arg)
+         _keep_developer_mode_flag_after_clobber_state_cmd_arg,
+         _test_umount_cmd_arg)
 def WipeInit(options):
   GetGooftool(options).WipeInit(
       options.wipe_args,
@@ -507,7 +515,8 @@
       options.station_ip,
       options.station_port,
       options.wipe_finish_token,
-      options.keep_developer_mode_flag_after_clobber_state)
+      options.keep_developer_mode_flag_after_clobber_state,
+      options.test_umount)
 
 @Command('verify',
          CmdArg('--no_write_protect', action='store_true',