def test_decorator(self): test_case_cls = fixture_test_case() test_case_cls.test_method1 = save_screenshot_on_error(test_case_cls.test_method1) stream = StringIO() runner = unittest.TextTestRunner(stream=stream) result = runner.run(unittest.TestSuite(map(test_case_cls, ['test_method1']))) self.assertEqual(1, test_case_cls.driver.save_screenshot.call_count) self.assertEqual(1, len(result.errors))
def setUp(self): config.utils = {'save_screenshots': ''} test_case_cls = fixture_simple_test_case() test_case_cls.runTest = save_screenshot_on_error(test_case_cls.runTest) self.test_case = test_case_cls()