selftest.target: Add tests for check.

diff --git a/selftest/target/__init__.py b/selftest/target/__init__.py
index ac2cc19a480cc18848e102710c52426918249ded..72e0770d92b21fa1ee7d74131b648cd567c996bd 100644 (file)
--- a/selftest/target/__init__.py
+++ b/selftest/target/__init__.py
@@ -128,11 +128,10 @@ class EnvironmentManager(object):
                 raise EnvironmentDown(running_env.get_log())
             return running_env
 
-
         env = self.target.setup_env(envname, prefix)
         if env is None:
             return None
 
-        self.running_envs[env] = env
+        self.running_envs[envname] = env
 
         return env

diff --git a/selftest/tests/test_target.py b/selftest/tests/test_target.py
index 343de930cbf67bfd27de74217eb559c6c3eff08c..0abc79b6bb558b7ee10198a88925e851362f4afa 100644 (file)
--- a/selftest/tests/test_target.py
+++ b/selftest/tests/test_target.py
@@ -63,3 +63,11 @@ class EnvironmentManagerTests(unittest.TestCase):
         env = self.mgr.setup_env("something", "prefix")
         self.assertEquals(env.name, "something")
         self.assertEquals(env.prefix, "prefix")
+
+    def test_check(self):
+        env = self.mgr.setup_env("something", "prefix")
+        self.assertTrue(env.check())
+        self.assertTrue(self.mgr.check_env("something"))
+        env.check_ret = False
+        self.assertFalse(env.check())
+        self.assertFalse(self.mgr.check_env("something"))