diff options
Diffstat (limited to 'tests/unittest_pyreverse_diadefs.py')
-rw-r--r-- | tests/unittest_pyreverse_diadefs.py | 175 |
1 files changed, 175 insertions, 0 deletions
diff --git a/tests/unittest_pyreverse_diadefs.py b/tests/unittest_pyreverse_diadefs.py new file mode 100644 index 000000000..0659a5b11 --- /dev/null +++ b/tests/unittest_pyreverse_diadefs.py @@ -0,0 +1,175 @@ +# Copyright (c) 2008-2010, 2013 LOGILAB S.A. (Paris, FRANCE) <contact@logilab.fr> +# Copyright (c) 2014 Google, Inc. +# Copyright (c) 2014 Arun Persaud <arun@nubati.net> +# Copyright (c) 2015-2017 Claudiu Popa <pcmanticore@gmail.com> +# Copyright (c) 2015 Ionel Cristian Maries <contact@ionelmc.ro> +# Copyright (c) 2016 Derek Gustafson <degustaf@gmail.com> +# Copyright (c) 2018 Sushobhit <31987769+sushobhit27@users.noreply.github.com> + +# Licensed under the GPL: https://www.gnu.org/licenses/old-licenses/gpl-2.0.html +# For details: https://github.com/PyCQA/pylint/blob/master/COPYING + +""" +unit test for the extensions.diadefslib modules +""" + +import astroid +import pytest + +from pylint.pyreverse.diadefslib import * +from pylint.pyreverse.inspector import Linker +from unittest_pyreverse_writer import Config, get_project + + +def _process_classes(classes): + """extract class names of a list""" + return sorted([(isinstance(c.node, astroid.ClassDef), c.title) for c in classes]) + + +def _process_relations(relations): + """extract relation indices from a relation list""" + result = [] + for rel_type, rels in relations.items(): + for rel in rels: + result.append((rel_type, rel.from_object.title, rel.to_object.title)) + result.sort() + return result + + +@pytest.fixture +def HANDLER(): + return DiadefsHandler(Config()) + + +@pytest.fixture(scope="module") +def PROJECT(): + return get_project("data") + + +def test_option_values(HANDLER, PROJECT): + """test for ancestor, associated and module options""" + df_h = DiaDefGenerator(Linker(PROJECT), HANDLER) + cl_config = Config() + cl_config.classes = ["Specialization"] + cl_h = DiaDefGenerator(Linker(PROJECT), DiadefsHandler(cl_config)) + assert (0, 0) == df_h._get_levels() + assert False == df_h.module_names + assert (-1, -1) == cl_h._get_levels() + assert True == cl_h.module_names + for hndl in [df_h, cl_h]: + hndl.config.all_ancestors = True + hndl.config.all_associated = True + hndl.config.module_names = True + hndl._set_default_options() + assert (-1, -1) == hndl._get_levels() + assert True == hndl.module_names + handler = DiadefsHandler(Config()) + df_h = DiaDefGenerator(Linker(PROJECT), handler) + cl_config = Config() + cl_config.classes = ["Specialization"] + cl_h = DiaDefGenerator(Linker(PROJECT), DiadefsHandler(cl_config)) + for hndl in [df_h, cl_h]: + hndl.config.show_ancestors = 2 + hndl.config.show_associated = 1 + hndl.config.module_names = False + hndl._set_default_options() + assert (2, 1) == hndl._get_levels() + assert False == hndl.module_names + + # def test_default_values(): + """test efault values for package or class diagrams""" + # TODO : should test difference between default values for package + # or class diagrams + + +class TestDefaultDiadefGenerator(object): + def test_known_values1(self, HANDLER, PROJECT): + dd = DefaultDiadefGenerator(Linker(PROJECT), HANDLER).visit(PROJECT) + assert len(dd) == 2 + keys = [d.TYPE for d in dd] + assert keys == ["package", "class"] + pd = dd[0] + assert pd.title == "packages No Name" + modules = sorted( + [(isinstance(m.node, astroid.Module), m.title) for m in pd.objects] + ) + assert modules == [ + (True, "data"), + (True, "data.clientmodule_test"), + (True, "data.suppliermodule_test"), + ] + cd = dd[1] + assert cd.title == "classes No Name" + classes = _process_classes(cd.objects) + assert classes == [ + (True, "Ancestor"), + (True, "DoNothing"), + (True, "Interface"), + (True, "Specialization"), + ] + + _should_rels = [ + ("association", "DoNothing", "Ancestor"), + ("association", "DoNothing", "Specialization"), + ("implements", "Ancestor", "Interface"), + ("specialization", "Specialization", "Ancestor"), + ] + + def test_exctract_relations(self, HANDLER, PROJECT): + """test extract_relations between classes""" + cd = DefaultDiadefGenerator(Linker(PROJECT), HANDLER).visit(PROJECT)[1] + cd.extract_relationships() + relations = _process_relations(cd.relationships) + assert relations == self._should_rels + + def test_functional_relation_extraction(self): + """functional test of relations extraction; + different classes possibly in different modules""" + # XXX should be catching pyreverse environnement problem but doesn't + # pyreverse doesn't extracts the relations but this test ok + project = get_project("data") + handler = DiadefsHandler(Config()) + diadefs = handler.get_diadefs(project, Linker(project, tag=True)) + cd = diadefs[1] + relations = _process_relations(cd.relationships) + assert relations == self._should_rels + + def test_known_values2(self, HANDLER): + project = get_project("data.clientmodule_test") + dd = DefaultDiadefGenerator(Linker(project), HANDLER).visit(project) + assert len(dd) == 1 + keys = [d.TYPE for d in dd] + assert keys == ["class"] + cd = dd[0] + assert cd.title == "classes No Name" + classes = _process_classes(cd.objects) + assert classes == [(True, "Ancestor"), (True, "Specialization")] + + +def test_known_values1(HANDLER, PROJECT): + HANDLER.config.classes = ["Specialization"] + cdg = ClassDiadefGenerator(Linker(PROJECT), HANDLER) + special = "data.clientmodule_test.Specialization" + cd = cdg.class_diagram(PROJECT, special) + assert cd.title == special + classes = _process_classes(cd.objects) + assert classes == [ + (True, "data.clientmodule_test.Ancestor"), + (True, special), + (True, "data.suppliermodule_test.DoNothing"), + ] + + +def test_known_values2(HANDLER, PROJECT): + HANDLER.config.classes = ["Specialization"] + HANDLER.config.module_names = False + cd = ClassDiadefGenerator(Linker(PROJECT), HANDLER).class_diagram( + PROJECT, "data.clientmodule_test.Specialization" + ) + assert cd.title == "data.clientmodule_test.Specialization" + classes = _process_classes(cd.objects) + assert classes == [ + (True, "Ancestor"), + (True, "DoNothing"), + (True, "Specialization"), + ] |