mirror of
https://fuchsia.googlesource.com/third_party/github.com/pylint-dev/pylint
synced 2024-09-21 07:58:04 +00:00
176 lines
6.2 KiB
Python
176 lines
6.2 KiB
Python
# 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"),
|
|
]
|