2013-12-20 14:12:50 +00:00
|
|
|
# Copyright (c) 2000-2013 LOGILAB S.A. (Paris, FRANCE).
|
2006-04-26 10:48:09 +00:00
|
|
|
# http://www.logilab.fr/ -- mailto:contact@logilab.fr
|
|
|
|
#
|
|
|
|
# This program is free software; you can redistribute it and/or modify it under
|
|
|
|
# the terms of the GNU General Public License as published by the Free Software
|
|
|
|
# Foundation; either version 2 of the License, or (at your option) any later
|
|
|
|
# version.
|
|
|
|
#
|
|
|
|
# This program is distributed in the hope that it will be useful, but WITHOUT
|
|
|
|
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
|
|
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
|
|
|
|
#
|
|
|
|
# You should have received a copy of the GNU General Public License along with
|
|
|
|
# this program; if not, write to the Free Software Foundation, Inc.,
|
2014-02-23 23:02:47 +00:00
|
|
|
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
2006-04-26 10:48:09 +00:00
|
|
|
"""
|
|
|
|
unittest for the extensions.diadefslib modules
|
|
|
|
"""
|
|
|
|
|
|
|
|
import unittest
|
|
|
|
import sys
|
|
|
|
|
2013-06-17 13:06:48 +00:00
|
|
|
import astroid
|
|
|
|
from astroid import MANAGER
|
|
|
|
from astroid.inspector import Linker
|
2009-03-18 09:32:53 +00:00
|
|
|
|
|
|
|
from pylint.pyreverse.diadefslib import *
|
|
|
|
|
2013-12-20 14:12:50 +00:00
|
|
|
from unittest_pyreverse_writer import Config, get_project
|
2006-04-26 10:48:09 +00:00
|
|
|
|
2013-12-20 14:12:50 +00:00
|
|
|
PROJECT = get_project('data')
|
|
|
|
HANDLER = DiadefsHandler(Config())
|
2006-04-26 10:48:09 +00:00
|
|
|
|
|
|
|
def _process_classes(classes):
|
2009-03-17 14:48:24 +00:00
|
|
|
"""extract class names of a list"""
|
2013-06-17 13:06:48 +00:00
|
|
|
return sorted([(isinstance(c.node, astroid.Class), c.title) for c in classes])
|
2006-04-26 10:48:09 +00:00
|
|
|
|
2009-03-17 14:48:24 +00:00
|
|
|
def _process_relations(relations):
|
|
|
|
"""extract relation indices from a relation list"""
|
|
|
|
result = []
|
2013-12-20 14:12:50 +00:00
|
|
|
for rel_type, rels in relations.iteritems():
|
2009-03-17 14:48:24 +00:00
|
|
|
for rel in rels:
|
|
|
|
result.append( (rel_type, rel.from_object.title,
|
|
|
|
rel.to_object.title) )
|
|
|
|
result.sort()
|
|
|
|
return result
|
|
|
|
|
|
|
|
|
2008-08-27 10:32:55 +00:00
|
|
|
class DiaDefGeneratorTC(unittest.TestCase):
|
2008-09-08 15:03:54 +00:00
|
|
|
def test_option_values(self):
|
|
|
|
"""test for ancestor, associated and module options"""
|
2013-12-20 14:12:50 +00:00
|
|
|
handler = DiadefsHandler(Config())
|
2009-03-17 13:21:23 +00:00
|
|
|
df_h = DiaDefGenerator(Linker(PROJECT), handler)
|
2008-08-27 10:32:55 +00:00
|
|
|
cl_config = Config()
|
|
|
|
cl_config.classes = ['Specialization']
|
2009-03-17 13:21:23 +00:00
|
|
|
cl_h = DiaDefGenerator(Linker(PROJECT), DiadefsHandler(cl_config) )
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual( (0, 0), df_h._get_levels())
|
|
|
|
self.assertEqual( False, df_h.module_names)
|
|
|
|
self.assertEqual( (-1, -1), cl_h._get_levels())
|
|
|
|
self.assertEqual( True, cl_h.module_names)
|
2008-08-27 10:32:55 +00:00
|
|
|
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()
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual( (-1, -1), hndl._get_levels())
|
|
|
|
self.assertEqual( True, hndl.module_names)
|
2008-08-27 10:32:55 +00:00
|
|
|
handler = DiadefsHandler( Config())
|
2009-03-17 13:21:23 +00:00
|
|
|
df_h = DiaDefGenerator(Linker(PROJECT), handler)
|
2008-08-27 10:32:55 +00:00
|
|
|
cl_config = Config()
|
|
|
|
cl_config.classes = ['Specialization']
|
2009-03-17 13:21:23 +00:00
|
|
|
cl_h = DiaDefGenerator(Linker(PROJECT), DiadefsHandler(cl_config) )
|
2008-08-27 10:32:55 +00:00
|
|
|
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()
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual( (2, 1), hndl._get_levels())
|
|
|
|
self.assertEqual( False, hndl.module_names)
|
2009-03-17 13:21:23 +00:00
|
|
|
#def test_default_values(self):
|
|
|
|
"""test efault values for package or class diagrams"""
|
|
|
|
# TODO : should test difference between default values for package
|
|
|
|
# or class diagrams
|
2008-08-27 10:32:55 +00:00
|
|
|
|
|
|
|
class DefaultDiadefGeneratorTC(unittest.TestCase):
|
2006-04-26 10:48:09 +00:00
|
|
|
def test_known_values1(self):
|
2009-03-17 13:21:23 +00:00
|
|
|
dd = DefaultDiadefGenerator(Linker(PROJECT), HANDLER).visit(PROJECT)
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual(len(dd), 2)
|
2006-04-26 10:48:09 +00:00
|
|
|
keys = [d.TYPE for d in dd]
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual(keys, ['package', 'class'])
|
2006-04-26 10:48:09 +00:00
|
|
|
pd = dd[0]
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual(pd.title, 'packages No Name')
|
2013-06-17 13:06:48 +00:00
|
|
|
modules = sorted([(isinstance(m.node, astroid.Module), m.title)
|
2010-10-28 15:38:15 +00:00
|
|
|
for m in pd.objects])
|
|
|
|
self.assertEqual(modules, [(True, 'data'),
|
|
|
|
(True, 'data.clientmodule_test'),
|
|
|
|
(True, 'data.suppliermodule_test')])
|
2006-04-26 10:48:09 +00:00
|
|
|
cd = dd[1]
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual(cd.title, 'classes No Name')
|
2006-04-26 10:48:09 +00:00
|
|
|
classes = _process_classes(cd.objects)
|
2010-10-28 15:38:15 +00:00
|
|
|
self.assertEqual(classes, [(True, 'Ancestor'),
|
|
|
|
(True, 'DoNothing'),
|
|
|
|
(True, 'Interface'),
|
|
|
|
(True, 'Specialization')]
|
2006-04-26 10:48:09 +00:00
|
|
|
)
|
2009-03-17 13:21:23 +00:00
|
|
|
|
2010-04-21 15:06:33 +00:00
|
|
|
_should_rels = [('association', 'DoNothing', 'Ancestor'),
|
|
|
|
('association', 'DoNothing', 'Specialization'),
|
|
|
|
('implements', 'Ancestor', 'Interface'),
|
|
|
|
('specialization', 'Specialization', 'Ancestor')]
|
2009-03-17 14:48:24 +00:00
|
|
|
def test_exctract_relations(self):
|
|
|
|
"""test extract_relations between classes"""
|
|
|
|
cd = DefaultDiadefGenerator(Linker(PROJECT), HANDLER).visit(PROJECT)[1]
|
|
|
|
cd.extract_relationships()
|
|
|
|
relations = _process_relations(cd.relationships)
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual(relations, self._should_rels)
|
2009-03-17 14:48:24 +00:00
|
|
|
|
|
|
|
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
|
2013-12-20 14:12:50 +00:00
|
|
|
project = get_project('data')
|
|
|
|
handler = DiadefsHandler(Config())
|
2009-03-17 14:48:24 +00:00
|
|
|
diadefs = handler.get_diadefs(project, Linker(project, tag=True) )
|
|
|
|
cd = diadefs[1]
|
|
|
|
relations = _process_relations(cd.relationships)
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual(relations, self._should_rels)
|
2009-03-17 14:48:24 +00:00
|
|
|
|
2006-04-26 10:48:09 +00:00
|
|
|
def test_known_values2(self):
|
2013-12-20 14:12:50 +00:00
|
|
|
project = get_project('data.clientmodule_test')
|
2009-03-17 13:21:23 +00:00
|
|
|
dd = DefaultDiadefGenerator(Linker(project), HANDLER).visit(project)
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual(len(dd), 1)
|
2006-04-26 10:48:09 +00:00
|
|
|
keys = [d.TYPE for d in dd]
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual(keys, ['class'])
|
2006-04-26 10:48:09 +00:00
|
|
|
cd = dd[0]
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual(cd.title, 'classes No Name')
|
2006-04-26 10:48:09 +00:00
|
|
|
classes = _process_classes(cd.objects)
|
2010-10-28 15:38:15 +00:00
|
|
|
self.assertEqual(classes, [(True, 'Ancestor'),
|
|
|
|
(True, 'DoNothing'),
|
|
|
|
(True, 'Specialization')]
|
2006-04-26 10:48:09 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
class ClassDiadefGeneratorTC(unittest.TestCase):
|
|
|
|
def test_known_values1(self):
|
2009-03-17 13:21:23 +00:00
|
|
|
HANDLER.config.classes = ['Specialization']
|
|
|
|
cdg = ClassDiadefGenerator(Linker(PROJECT), HANDLER)
|
|
|
|
special = 'data.clientmodule_test.Specialization'
|
|
|
|
cd = cdg.class_diagram(PROJECT, special)
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual(cd.title, special)
|
2006-04-26 10:48:09 +00:00
|
|
|
classes = _process_classes(cd.objects)
|
2010-10-28 15:38:15 +00:00
|
|
|
self.assertEqual(classes, [(True, 'data.clientmodule_test.Ancestor'),
|
|
|
|
(True, special),
|
|
|
|
(True, 'data.suppliermodule_test.DoNothing'),
|
|
|
|
])
|
2013-12-20 14:12:50 +00:00
|
|
|
|
2006-04-26 10:48:09 +00:00
|
|
|
def test_known_values2(self):
|
2009-03-17 13:21:23 +00:00
|
|
|
HANDLER.config.module_names = False
|
|
|
|
cd = ClassDiadefGenerator(Linker(PROJECT), HANDLER).class_diagram(PROJECT, 'data.clientmodule_test.Specialization')
|
2010-09-28 13:02:35 +00:00
|
|
|
self.assertEqual(cd.title, 'data.clientmodule_test.Specialization')
|
2006-04-26 10:48:09 +00:00
|
|
|
classes = _process_classes(cd.objects)
|
2010-10-28 15:38:15 +00:00
|
|
|
self.assertEqual(classes, [(True, 'Ancestor'),
|
|
|
|
(True, 'DoNothing'),
|
|
|
|
(True, 'Specialization')
|
|
|
|
])
|
2006-04-26 10:48:09 +00:00
|
|
|
|
2008-09-02 10:06:01 +00:00
|
|
|
|
2006-04-26 10:48:09 +00:00
|
|
|
if __name__ == '__main__':
|
|
|
|
unittest.main()
|