Python save_screenshot_on_error Examples

Example #1

File: utils_screenshot_tests.py Project: tkurnosova/contesto

    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))

Example #2

File: utils_screenshot_tests.py Project: tkurnosova/contesto

 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()