CART-831 TEST: add support for arbitrary envariables in yaml files

src/tests/ftest/cart/no_pmix/cart_nopmix_multictx_one_node.py

@@ -46,6 +46,7 @@ def setUp(self):
         print("Running setup\n")
         self.utils = CartUtils()
         self.env = self.utils.get_env(self)
+        self.utils.set_other_env_vars(self)
         crt_phy_addr = self.params.get("CRT_PHY_ADDR_STR", '/run/defaultENV/')
         ofi_interface = self.params.get("OFI_INTERFACE", '/run/defaultENV/')
         ofi_ctx_num = self.params.get("CRT_CTX_NUM", '/run/defaultENV/')
@@ -61,6 +62,7 @@ def tearDown(self):
 
         super(CartNoPmixOneNodeTest, self).tearDown()
         self.utils.cleanup_processes()
+        self.utils.unset_other_env_vars(self)
 
     def test_cart_no_pmix(self):
         """

src/tests/ftest/cart/no_pmix/cart_nopmix_multictx_one_node.yaml

@@ -8,6 +8,10 @@ defaultENV:
   OFI_INTERFACE: "eth0"
   CRT_CTX_NUM: "8"
   CRT_CTX_SHARE_ADDR: "0"
+  other_env_vars:
+    - ARBITRARY_ENV_VAR1: apple
+    - ARBITRARY_ENV_VAR2: orange
+    - ARBITRARY_ENV_VAR3: banana pear
 tests: !mux
   no_pmix_multi_ctx:
     name: no_pmix_multi_ctx

src/tests/ftest/cart/util/cart_utils.py

@@ -105,6 +105,34 @@ def stop_process(proc):
 
         return procrtn
 
+    @staticmethod
+    def set_other_env_vars(cartobj):
+        """ import env vars from other_env_var param """
+        other_env_vars = cartobj.params.get("other_env_vars",
+                                            "/run/defaultENV/")
+        if other_env_vars is None:
+            print("other_env_vars was not set in yaml file.\n")
+            return
+
+        for kv_pair in other_env_vars:
+            key, value = kv_pair[0]
+            print("Adding {}={} to environment.\n".format(key, value))
+            os.environ[key] = value
+
+    @staticmethod
+    def unset_other_env_vars(cartobj):
+        """ import env vars from other_env_var param """
+        other_env_vars = cartobj.params.get("other_env_vars",
+                                            "/run/defaultENV/")
+        if other_env_vars is None:
+            print("other_env_vars was not set in yaml file.\n")
+            return
+
+        for kv_pair in other_env_vars:
+            key = kv_pair[0][0]
+            print("Removing key {} from environment.\n".format(key))
+            del os.environ[key]
+
     # What is special about pylint's 15 variable limit?
     # pylint: disable=too-many-locals
     def get_env(self, cartobj):