mirror of
https://gitlab.com/flectra-community/mis-builder.git
synced 2024-11-16 19:22:04 +00:00
81 lines
2.2 KiB
Python
81 lines
2.2 KiB
Python
|
# Copyright 2017 ACSONE SA/NV (<http://acsone.eu>)
|
||
|
|
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
|
||
|
|
|
||
|
|
import doctest
|
||
|
|
|
||
|
|
from flectra.tests import BaseCase, tagged
|
||
|
|
|
||
|
|
|
||
|
|
def setup_test_model(env, model_cls):
|
||
|
|
model_cls._build_model(env.registry, env.cr)
|
||
|
|
env.registry.setup_models(env.cr)
|
||
|
|
env.registry.init_models(
|
||
|
|
env.cr, [model_cls._name], dict(env.context, update_custom_fields=True)
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def teardown_test_model(env, model_cls):
|
||
|
|
del env.registry.models[model_cls._name]
|
||
|
|
env.registry.setup_models(env.cr)
|
||
|
|
|
||
|
|
|
||
|
|
def _zip(iter1, iter2):
|
||
|
|
i = 0
|
||
|
|
iter1 = iter(iter1)
|
||
|
|
iter2 = iter(iter2)
|
||
|
|
while True:
|
||
|
|
i1 = next(iter1, None)
|
||
|
|
i2 = next(iter2, None)
|
||
|
|
if i1 is None and i2 is None:
|
||
|
|
return
|
||
|
|
yield i, i1, i2
|
||
|
|
i += 1
|
||
|
|
|
||
|
|
|
||
|
|
def assert_matrix(matrix, expected):
|
||
|
|
for i, row, expected_row in _zip(matrix.iter_rows(), expected):
|
||
|
|
if row is None and expected_row is not None:
|
||
|
|
raise AssertionError("not enough rows")
|
||
|
|
if row is not None and expected_row is None:
|
||
|
|
raise AssertionError("too many rows")
|
||
|
|
for j, cell, expected_val in _zip(row.iter_cells(), expected_row):
|
||
|
|
assert (
|
||
|
|
cell and cell.val
|
||
|
|
) == expected_val, "{} != {} in row {} col {}".format(
|
||
|
|
cell and cell.val, expected_val, i, j
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@tagged("doctest")
|
||
|
|
class FlectraDocTestCase(BaseCase):
|
||
|
|
"""
|
||
|
|
We need a custom DocTestCase class in order to:
|
||
|
|
- define test_tags to run as part of standard tests
|
||
|
|
- output a more meaningful test name than default "DocTestCase.runTest"
|
||
|
|
"""
|
||
|
|
|
||
|
|
__qualname__ = "doctests for "
|
||
|
|
|
||
|
|
def __init__(self, test):
|
||
|
|
self.__test = test
|
||
|
|
self.__name = test._dt_test.name
|
||
|
|
super().__init__(self.__name)
|
||
|
|
|
||
|
|
def __getattr__(self, item):
|
||
|
|
if item == self.__name:
|
||
|
|
return self.__test
|
||
|
|
|
||
|
|
|
||
|
|
def load_doctests(module):
|
||
|
|
"""
|
||
|
|
Generates a tests loading method for the doctests of the given module
|
||
|
|
https://docs.python.org/3/library/unittest.html#load-tests-protocol
|
||
|
|
"""
|
||
|
|
|
||
|
|
def load_tests(loader, tests, ignore):
|
||
|
|
for test in doctest.DocTestSuite(module):
|
||
|
|
tests.addTest(FlectraDocTestCase(test))
|
||
|
|
return tests
|
||
|
|
|
||
|
|
return load_tests
|