1 # test_target.py -- The tests for selftest target code
2 # Copyright (C) 2012 Jelmer Vernooij <jelmer@samba.org>
4 # This program is free software; you can redistribute it and/or
5 # modify it under the terms of the GNU General Public License
6 # as published by the Free Software Foundation; version 3
7 # of the License or (at your option) any later version of
10 # This program is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 # GNU General Public License for more details.
15 # You should have received a copy of the GNU General Public License
16 # along with this program; if not, write to the Free Software
17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
20 """Tests for selftest.target."""
22 from selftest.target import (
33 class DummyEnvironment(Environment):
35 def __init__(self, name, prefix):
48 class DummyTarget(Target):
50 def setup_env(self, name, prefix):
51 return DummyEnvironment(name, prefix)
54 class NoneEnvironmentTests(unittest.TestCase):
57 super(NoneEnvironmentTests, self).setUp()
58 self.env = NoneEnvironment()
60 def test_get_vars(self):
61 self.assertEquals({}, self.env.get_vars())
64 self.assertEquals(True, self.env.check())
66 def test_get_log(self):
67 self.assertEquals("", self.env.get_log())
70 class EnvironmentManagerTests(unittest.TestCase):
73 super(EnvironmentManagerTests, self).setUp()
74 self.mgr = EnvironmentManager(DummyTarget())
78 NoneEnvironment, type(self.mgr.setup_env("none", "prefix")))
81 env = self.mgr.setup_env("something", "prefix")
82 self.assertEquals(env.name, "something")
83 self.assertEquals(env.prefix, "prefix")
85 def test_setup_reuses(self):
86 env1 = self.mgr.setup_env("something", "prefix")
87 env2 = self.mgr.setup_env("something", "prefix")
88 self.assertIs(env1, env2)
90 def test_setup_down(self):
91 env1 = self.mgr.setup_env("something", "prefix")
92 env1.check_ret = False
93 self.assertRaises(EnvironmentDown, self.mgr.setup_env, "something", "")
96 env = self.mgr.setup_env("something", "prefix")
97 self.assertTrue(env.check())
98 self.assertTrue(self.mgr.check_env("something"))
100 self.assertFalse(env.check())
101 self.assertFalse(self.mgr.check_env("something"))
103 def test_get_log(self):
104 env = self.mgr.setup_env("something", "prefix")
105 self.assertEquals("", env.get_log())
106 self.assertEquals("", self.mgr.getlog_env("something"))
108 self.assertEquals('bla', env.get_log())
109 self.assertEquals('bla', self.mgr.getlog_env("something"))
111 def test_get_running_env(self):
112 env = self.mgr.setup_env("something", "prefix")
113 self.assertIs(env, self.mgr.get_running_env("something"))
115 def test_get_running_env_nonexistant(self):
116 self.assertIs(None, self.mgr.get_running_env("something"))