This way, tests can get information about Qemu and the modules that were
installed by the 'build' test if needed.

Signed-off-by: Eduardo Habkost <[email protected]>
---
 client/tests/kvm/kvm/env.py      |   15 +++++++++++++++
 client/tests/kvm/kvm/env_test.py |   19 +++++++++++++++++++
 2 files changed, 34 insertions(+), 0 deletions(-)

diff --git a/client/tests/kvm/kvm/env.py b/client/tests/kvm/kvm/env.py
index 0a02ade..3870101 100644
--- a/client/tests/kvm/kvm/env.py
+++ b/client/tests/kvm/kvm/env.py
@@ -61,6 +61,21 @@ class KvmEnv(UserDict.IterableUserDict):
         """
         del self.data["vm__%s" % name]
 
+    def register_installer(self, installer):
+        """Register a installer that was just run
+
+        The installer will be available for other tests, so that
+        information about the installed KVM modules and qemu-kvm can be used by
+        them.
+        """
+        self['last_installer'] = installer
+
+    def previous_installer(self):
+        """Return the last installer that was registered
+        """
+        return self.get('last_installer')
+
+
 def _load_env(filename, version):
     """
     Load KVM test environment from an env file.
diff --git a/client/tests/kvm/kvm/env_test.py b/client/tests/kvm/kvm/env_test.py
index 41f0e54..5704fe3 100644
--- a/client/tests/kvm/kvm/env_test.py
+++ b/client/tests/kvm/kvm/env_test.py
@@ -77,3 +77,22 @@ class KvmEnvtest(unittest.TestCase):
 
         self.assertIs(e.get_vm('vm1'), vm1)
         self.assertIs(e.get_vm('vm2'), vm2)
+
+    def test_installer_register(self):
+        i1 = [1]
+        i2 = [2]
+
+        e = kvm.env.KvmEnv({})
+
+        e.register_installer(i1)
+        self.assertIs(e.previous_installer(), i1)
+
+        e.register_installer(i2)
+        self.assertIs(e.previous_installer(), i2)
+
+    def test_no_installer(self):
+        # in case there's no installe registered,
+        # KvmEnv should return None
+
+        e = kvm.env.KvmEnv({})
+        self.assertIs(e.previous_installer(), None)
-- 
1.7.3.2

_______________________________________________
Autotest mailing list
[email protected]
http://test.kernel.org/cgi-bin/mailman/listinfo/autotest

Reply via email to