selftest.target: Add Environment.get_vars.

1 files changed, 11 insertions, 0 deletions

diff --git a/selftest/target/__init__.py b/selftest/target/__init__.py
index 72e0770d92..b6a297ce02 100644
--- a/selftest/target/__init__.py
+++ b/selftest/target/__init__.py
@@ -28,6 +28,7 @@ class EnvironmentDown(Exception):
     def __init__(self, msg):
         super(EnvironmentDown, self).__init__("environment went down: %s" % msg)
 
+
 class Target(object):
     """A target for Samba tests."""
 
@@ -67,6 +68,13 @@ class Environment(object):
         """
         raise NotImplementedError(self.teardown)
 
+    def get_vars(self):
+        """Retrieve the environment variables for this environment.
+
+        :return: Dictionary with string -> string values
+        """
+        raise NotImplementedError(self.get_vars)
+
 
 class NoneEnvironment(Environment):
     """Empty environment.
@@ -81,6 +89,9 @@ class NoneEnvironment(Environment):
     def teardown(self):
         return
 
+    def get_vars(self):
+        return {}
+
 
 class EnvironmentManager(object):
     """Manager of environments."""