Add a 'scripts_data' provisioning variable

canonical · Jan 18, 2024 · 278d044 · 278d044
1 parent abafc0d
commit 278d044
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/upgrade_testing/configspec/_config.py b/upgrade_testing/configspec/_config.py
@@ -66,6 +66,8 @@ def _reader(self, details):
             )
         )
 
+        self.scripts_data = details.get("scripts_data", None)
+
         backend_args = details.get("backend_args", [])
         self.backend_args = [
             arg.format(scripts_location=self.scripts_location)

diff --git a/upgrade_testing/preparation/_hostprep.py b/upgrade_testing/preparation/_hostprep.py
@@ -16,6 +16,7 @@
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 #
 
+import json
 import logging
 import os
 import shutil
@@ -75,6 +76,11 @@ def prepare_test_environment(testsuite):
         scripts_path = os.path.join(temp_dir, "scripts")
         _copy_script_files(testsuite.scripts_location, scripts_path)
 
+        if hasattr(testsuite, "scripts_data"):
+            data_path = os.path.join(temp_dir, "scripts_data.json")
+            with open(data_path, "w") as f:
+                json.dump(testsuite.scripts_data, f)
+
         adt_base_path, adt_cmd = _get_adt_path(temp_dir)
 
         yield TestrunTempFiles(