diff options
author | Robert Collins <robertc@robertcollins.net> | 2009-03-08 14:10:56 +1100 |
---|---|---|
committer | Robert Collins <robertc@robertcollins.net> | 2009-03-08 14:10:56 +1100 |
commit | ef3a10eaef1ac7140beaa154cf0bf9dbc0e76306 (patch) | |
tree | f2b2505e6272d668c5bc38ddd1c3af8ec6e961bc /lib | |
parent | b3d45b91f898653843a4444567733c5bdb71a3f7 (diff) | |
download | testscenarios-ef3a10eaef1ac7140beaa154cf0bf9dbc0e76306.tar.gz |
TestWithScenarios.run now sets attributes too, and includes some docstring docs.
Diffstat (limited to 'lib')
-rw-r--r-- | lib/testscenarios/testcase.py | 11 | ||||
-rw-r--r-- | lib/testscenarios/tests/test_testcase.py | 31 |
2 files changed, 41 insertions, 1 deletions
diff --git a/lib/testscenarios/testcase.py b/lib/testscenarios/testcase.py index 0b8898c..ec27e39 100644 --- a/lib/testscenarios/testcase.py +++ b/lib/testscenarios/testcase.py @@ -26,7 +26,14 @@ import unittest from testtools.testcase import clone_test_with_new_id class TestWithScenarios(unittest.TestCase): - """A TestCase with support for scenarios via a scenarios attribute.""" + """A TestCase with support for scenarios via a scenarios attribute. + + When a test object which is an instance of TestWithScenarios is run, + and there is a non-empty scenarios attribute on the object, the test is + multiplied by the run method into one test per scenario. For this to work + reliably the TestWithScenarios.run method must not be overriden in a + subclass (or overridden compatibly with TestWithScenarios). + """ def run(self, result=None): scenarios = getattr(self, 'scenarios', None) @@ -35,6 +42,8 @@ class TestWithScenarios(unittest.TestCase): newtest = clone_test_with_new_id(self, self.id() + '(' + name + ')') newtest.scenarios = None + for key, value in parameters.iteritems(): + setattr(newtest, key, value) newtest.run(result) return else: diff --git a/lib/testscenarios/tests/test_testcase.py b/lib/testscenarios/tests/test_testcase.py index 9cfd724..5b86e24 100644 --- a/lib/testscenarios/tests/test_testcase.py +++ b/lib/testscenarios/tests/test_testcase.py @@ -67,3 +67,34 @@ class TestTestWithScenarios(unittest.TestCase): self.assertEqual( 'testscenarios.tests.test_testcase.ReferenceTest.test_pass(2)', log[4][1].id()) + + def test_attributes_set(self): + class ReferenceTest(testscenarios.TestWithScenarios): + scenarios = [ + ('1', {'foo': 1, 'bar': 2}), + ('2', {'foo': 2, 'bar': 4})] + def test_check_foo(self): + self.assertEqual(self.foo * 2, self.bar) + test = ReferenceTest("test_check_foo") + log = [] + result = LoggingResult(log) + test.run(result) + self.assertTrue(result.wasSuccessful()) + self.assertEqual(2, result.testsRun) + + def test_scenarios_attribute_cleared(self): + class ReferenceTest(testscenarios.TestWithScenarios): + scenarios = [ + ('1', {'foo': 1, 'bar': 2}), + ('2', {'foo': 2, 'bar': 4})] + def test_check_foo(self): + self.assertEqual(self.foo * 2, self.bar) + test = ReferenceTest("test_check_foo") + log = [] + result = LoggingResult(log) + test.run(result) + self.assertTrue(result.wasSuccessful()) + self.assertEqual(2, result.testsRun) + self.assertNotEqual(None, test.scenarios) + self.assertEqual(None, log[0][1].scenarios) + self.assertEqual(None, log[4][1].scenarios) |