Expose an CLI option of post script in advanced reboot test to user

tests/common/fixtures/advanced_reboot.py

@@ -70,6 +70,7 @@ def __extractTestParam(self):
         self.cleanupOldSonicImages = self.request.config.getoption("--cleanup_old_sonic_images")
         self.readyTimeout = self.request.config.getoption("--ready_timeout")
         self.replaceFastRebootScript = self.request.config.getoption("--replace_fast_reboot_script")
+        self.postRebootCheckScript = self.request.config.getoption("--post_reboot_check_script")
 
     def getHostMaxLen(self):
         '''
@@ -501,6 +502,10 @@ def tearDown(self):
 
         self.__runScript(['remove_ip.sh'], self.ptfhost)
 
+        if self.postRebootCheckScript:
+            logger.info('Run the post reboot check script')
+            self.__runScript([self.postRebootCheckScript], self.duthost)
+
         if not self.stayInTargetImage:
             self.__restorePrevImage()
 

tests/platform_tests/args/advanced_reboot_args.py

@@ -84,3 +84,11 @@ def add_advanced_reboot_args(parser):
         default=False,
         help="Replace fast-reboot script on DUT",
     )
+
+    parser.addoption(
+        "--post_reboot_check_script",
+        action="store",
+        type=str,
+        default=None,
+        help="Script for checking additional states on DUT"
+    )