From 36d8ef19716eac2766b8dd195386fdb7c4e90b53 Mon Sep 17 00:00:00 2001
From: Pierre Kancir <pierre.kancir@azurdrones.com>
Date: Thu, 23 Aug 2018 09:59:20 +0200
Subject: [PATCH] Tools: auto fetch parameter if in fetch_all list

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

diff --git a/Tools/autotest/common.py b/Tools/autotest/common.py
index 12cf61c096..2b1ccc5435 100644
--- a/Tools/autotest/common.py
+++ b/Tools/autotest/common.py
@@ -7,6 +7,7 @@ import shutil
 import sys
 import time
 import pexpect
+import fnmatch
 
 from pymavlink import mavwp, mavutil
 from pysim import util, vehicleinfo
@@ -482,6 +483,21 @@ class AutoTest(ABC):
         self.progress("FAILED TO AUTODISARM")
         return False
 
+    @staticmethod
+    def should_fetch_all_for_parameter_change(param_name):
+        if fnmatch.fnmatch(param_name, "*_ENABLE") or fnmatch.fnmatch(param_name, "*_ENABLED"):
+            return True
+        if param_name in ["ARSPD_TYPE",
+                          "ARSPD2_TYPE",
+                          "BATT2_MONITOR",
+                          "CAN_DRIVER",
+                          "COMPASS_PMOT_EN",
+                          "OSD_TYPE",
+                          "RSSI_TYPE",
+                          "WENC_TYPE"]:
+            return True
+        return False
+
     def set_parameter(self, name, value, add_to_context=True):
         """Set parameters from vehicle."""
         old_value = self.get_parameter(name, retry=2)
@@ -492,6 +508,8 @@ class AutoTest(ABC):
                 # yes, exactly equal.
                 if add_to_context:
                     self.context_get().parameters.append((name, old_value))
+                if self.should_fetch_all_for_parameter_change(name.upper()) and value == 1:
+                    self.fetch_parameters()
                 return
             self.progress("Param fetch returned incorrect value (%s) vs (%s)"
                           % (returned_value, value))