selftest.target: Add Environment.get_vars.
authorJelmer Vernooij <jelmer@samba.org>
Sun, 4 Mar 2012 01:38:15 +0000 (02:38 +0100)
committerJelmer Vernooij <jelmer@samba.org>
Sun, 4 Mar 2012 17:02:06 +0000 (18:02 +0100)
selftest/target/__init__.py

index 72e0770..b6a297c 100644 (file)
@@ -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."""