From c57ab7ba9d7e25ac18d1e9883d3fc86a688fa7be Mon Sep 17 00:00:00 2001 From: Jelmer Vernooij Date: Sun, 4 Mar 2012 02:28:06 +0100 Subject: selftest.target: Add tests for setup. --- selftest/tests/test_target.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) (limited to 'selftest/tests/test_target.py') diff --git a/selftest/tests/test_target.py b/selftest/tests/test_target.py index 9b3d37b270..343de930cb 100644 --- a/selftest/tests/test_target.py +++ b/selftest/tests/test_target.py @@ -26,7 +26,6 @@ from selftest.target import ( Target, ) -import os import unittest @@ -47,8 +46,8 @@ class DummyEnvironment(Environment): 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): @@ -57,4 +56,10 @@ 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") -- cgit