123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164 |
- # 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>
-
- # 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 six
-
- import pytest
-
- import astroid
-
- from pylint.pyreverse.inspector import Linker
- from pylint.pyreverse.diadefslib import *
-
- 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 six.iteritems(relations):
- 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')]
|