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

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):

src/tests/ftest/cart/utils_test/set_other_env_vars.py

@@ -0,0 +1,73 @@
+#!/usr/bin/python
+'''
+  (C) Copyright 2018-2021 Intel Corporation.
+
+  Licensed under the Apache License, Version 2.0 (the "License");
+  you may not use this file except in compliance with the License.
+  You may obtain a copy of the License at
+
+     http://www.apache.org/licenses/LICENSE-2.0
+
+  Unless required by applicable law or agreed to in writing, software
+  distributed under the License is distributed on an "AS IS" BASIS,
+  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+  See the License for the specific language governing permissions and
+  limitations under the License.
+
+  GOVERNMENT LICENSE RIGHTS-OPEN SOURCE SOFTWARE
+  The Government's rights to use, modify, reproduce, release, perform, display,
+  or disclose this software are subject to the terms of the Apache License as
+  provided in Contract No. B609815.
+  Any reproduction of computer software, computer software documentation, or
+  portions thereof marked with this legend must also reproduce the markings.
+'''
+
+from __future__ import print_function
+
+import sys
+import subprocess
+
+from apricot import TestWithoutServers
+
+sys.path.append('./util')
+
+# Can't all this import before setting sys.path
+# pylint: disable=wrong-import-position
+from cart_utils import CartUtils
+
+class SetOtherEnvVars(TestWithoutServers):
+    """
+    Runs basic test of cart_utils.py:set_other_env_vars
+
+    :avocado: recursive
+    """
+    def setUp(self):
+        """ Test setup """
+        print("Running setup\n")
+        self.utils = CartUtils()
+
+    def tearDown(self):
+        """ Tear down """
+        print("Running tearDown\n")
+
+    def test_cart_no_pmix(self):
+        """
+        Test set_other_env_vars
+
+        :avocado: tags=set_other_env_vars
+        """
+
+        print("Before set_other_env_vars\n")
+        print(subprocess.check_output("env", shell=True))
+        self.utils.set_other_env_vars(self)
+        print("After set_other_env_vars\n")
+        print(subprocess.check_output("env", shell=True))
+
+        print("Before unset_other_env_vars\n")
+        print(subprocess.check_output("env", shell=True))
+        self.utils.unset_other_env_vars(self)
+        print("After unset_other_env_vars\n")
+        print(subprocess.check_output("env", shell=True))
+
+if __name__ == "__main__":
+    main()

src/tests/ftest/cart/utils_test/set_other_env_vars.yaml

@@ -0,0 +1,13 @@
+defaultENV:
+  D_LOG_MASK: "WARN"
+  CRT_PHY_ADDR_STR: "ofi+sockets"
+  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
+  set_other_env_vars:
+    name: set_other_env_var