diff --git a/tests/__init__.py b/tests/__init__.py index 7c00f68..e69de29 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -1,5 +0,0 @@ -# The COPYRIGHT file at the top level of this repository contains the full -# copyright notices and license terms. -from .test_quality_control_stress_test import suite - -__all__ = ['suite'] diff --git a/tests/test_quality_control_stress_test.py b/tests/test_module.py similarity index 93% rename from tests/test_quality_control_stress_test.py rename to tests/test_module.py index a311e63..e677bc4 100644 --- a/tests/test_quality_control_stress_test.py +++ b/tests/test_module.py @@ -1,17 +1,18 @@ -# The COPYRIGHT file at the top level of this repository contains the full -# copyright notices and license terms. -import unittest + +# This file is part of Tryton. The COPYRIGHT file at the top level of +# this repository contains the full copyright notices and license terms. + from decimal import Decimal -import trytond.tests.test_tryton from trytond.pool import Pool from trytond.tests.test_tryton import ModuleTestCase, with_transaction -from trytond.modules.company.tests import create_company, set_company +from trytond.modules.company.tests import (CompanyTestMixin, create_company, + set_company) -class TestCase(ModuleTestCase): - 'Test module' +class QualityControlStressTestTestCase(CompanyTestMixin, ModuleTestCase): + 'Test QualityControlStressTest module' module = 'quality_control_stress_test' @with_transaction() @@ -151,7 +152,4 @@ class TestCase(ModuleTestCase): self.assertEqual(qt_line.stress_test, low_stress_test) -def suite(): - suite = trytond.tests.test_tryton.suite() - suite.addTests(unittest.TestLoader().loadTestsFromTestCase(TestCase)) - return suite +del ModuleTestCase diff --git a/tests/test_scenario.py b/tests/test_scenario.py new file mode 100644 index 0000000..b5e24ad --- /dev/null +++ b/tests/test_scenario.py @@ -0,0 +1,23 @@ + +# This file is part of Tryton. The COPYRIGHT file at the top level of +# this repository contains the full copyright notices and license terms. + +import doctest +import glob +import os + +from trytond.tests.test_tryton import doctest_checker, doctest_teardown + + +def load_tests(loader, tests, pattern): + cwd = os.getcwd() + try: + os.chdir(os.path.dirname(__file__)) + for scenario in glob.glob('*.rst'): + tests.addTests(doctest.DocFileSuite( + scenario, tearDown=doctest_teardown, encoding='utf-8', + checker=doctest_checker, + optionflags=doctest.REPORT_ONLY_FIRST_FAILURE)) + finally: + os.chdir(cwd) + return tests \ No newline at end of file