def get_running_env(self, name):
envname = name.split(":")[0]
- return self.running_envs[envname]
+ if envname == "none":
+ return NoneEnvironment()
+ return self.running_envs.get(envname)
def getlog_env(self, envname):
env = self.get_running_env(envname)
return env.get_log()
- def check_env(envname):
+ def check_env(self, envname):
"""Check if an environment is still up.
:param envname: Environment to check
del self.running_envs[envname]
def teardown_all(self):
+ """Teardown all environments."""
for env in self.running_envs.iterkeys():
self.teardown_env(env)
def setup_env(self, envname, prefix):
- if envname == "none":
- return NoneEnvironment()
-
running_env = self.get_running_env(envname)
if running_env is not None:
if not running_env.check():
Target,
)
-import os
import unittest
class DummyTarget(Target):
- def get_target(self, name, prefix):
- return DummyTarget(name, prefix)
+ def setup_env(self, name, prefix):
+ return DummyEnvironment(name, prefix)
class EnvironmentManagerTests(unittest.TestCase):
self.mgr = EnvironmentManager(DummyTarget())
def test_none(self):
- self.assertIs(NoneEnvironment, type(self.mgr.setup_env("none", "prefix")))
+ self.assertIs(
+ NoneEnvironment, type(self.mgr.setup_env("none", "prefix")))
+
+ def test_setup(self):
+ env = self.mgr.setup_env("something", "prefix")
+ self.assertEquals(env.name, "something")
+ self.assertEquals(env.prefix, "prefix")