From ba3f95ad20d56c123809363aef7c318fb42efe78 Mon Sep 17 00:00:00 2001
From: Peter Barker <pbarker@barker.dropbear.id.au>
Date: Fri, 6 Sep 2019 11:13:47 +1000
Subject: [PATCH] Tools: autotest: add timeout parameter to
 detect_and_handle_reboot

---
 Tools/autotest/common.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Tools/autotest/common.py b/Tools/autotest/common.py
index 5a2f4b18c7..3457be02d4 100644
--- a/Tools/autotest/common.py
+++ b/Tools/autotest/common.py
@@ -349,12 +349,12 @@ class AutoTest(ABC):
         self.mavproxy.send("reboot\n")
         self.detect_and_handle_reboot(old_bootcount, required_bootcount=required_bootcount)
 
-    def detect_and_handle_reboot(self, old_bootcount, required_bootcount=None):
+    def detect_and_handle_reboot(self, old_bootcount, required_bootcount=None, timeout=10):
         tstart = time.time()
         if required_bootcount is None:
             required_bootcount = old_bootcount + 1
         while True:
-            if time.time() - tstart > 10:
+            if time.time() - tstart > timeout:
                 raise AutoTestTimeoutException("Did not detect reboot")
             try:
                 current_bootcount = self.get_parameter('STAT_BOOTCNT', timeout=1)