diff options
Diffstat (limited to 'test/unittest_pyreverse_writer.py')
-rw-r--r-- | test/unittest_pyreverse_writer.py | 136 |
1 files changed, 0 insertions, 136 deletions
diff --git a/test/unittest_pyreverse_writer.py b/test/unittest_pyreverse_writer.py deleted file mode 100644 index 19d94ab..0000000 --- a/test/unittest_pyreverse_writer.py +++ /dev/null @@ -1,136 +0,0 @@ -# Copyright (c) 2000-2013 LOGILAB S.A. (Paris, FRANCE). -# 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., -# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. -""" -unittest for visitors.diadefs and extensions.diadefslib modules -""" - - -import os -import sys -import codecs -from os.path import join, dirname, abspath -from difflib import unified_diff - -from logilab.common.testlib import TestCase, unittest_main - -from astroid import MANAGER -from astroid.inspector import Linker - -from pylint.pyreverse.diadefslib import DefaultDiadefGenerator, DiadefsHandler -from pylint.pyreverse.writer import DotWriter -from pylint.pyreverse.utils import get_visibility - - -_DEFAULTS = { - 'all_ancestors': None, 'show_associated': None, - 'module_names': None, - 'output_format': 'dot', 'diadefs_file': None, 'quiet': 0, - 'show_ancestors': None, 'classes': (), 'all_associated': None, - 'mode': 'PUB_ONLY', 'show_builtin': False, 'only_classnames': False - } - -class Config(object): - """config object for tests""" - def __init__(self): - for attr, value in _DEFAULTS.items(): - setattr(self, attr, value) - - -def _file_lines(path): - # we don't care about the actual encoding, but python3 forces us to pick one - lines = [line.strip() for line in codecs.open(path, encoding='latin1').readlines() - if (line.find('squeleton generated by ') == -1 and - not line.startswith('__revision__ = "$Id:'))] - return [line for line in lines if line] - -def get_project(module, name=None): - """return a astroid project representation""" - # flush cache - MANAGER._modules_by_name = {} - def _astroid_wrapper(func, modname): - return func(modname) - return MANAGER.project_from_files([module], _astroid_wrapper, - project_name=name) - -CONFIG = Config() - -class DotWriterTC(TestCase): - - @classmethod - def setUpClass(cls): - project = get_project(cls.datadir) - linker = Linker(project) - handler = DiadefsHandler(CONFIG) - dd = DefaultDiadefGenerator(linker, handler).visit(project) - for diagram in dd: - diagram.extract_relationships() - writer = DotWriter(CONFIG) - writer.write(dd) - - @classmethod - def tearDownClass(cls): - for fname in ('packages_No_Name.dot', 'classes_No_Name.dot',): - try: - os.remove(fname) - except: - continue - - def _test_same_file(self, generated_file): - expected_file = self.datapath(generated_file) - generated = _file_lines(generated_file) - expected = _file_lines(expected_file) - generated = '\n'.join(generated) - expected = '\n'.join(expected) - files = "\n *** expected : %s, generated : %s \n" % ( - expected_file, generated_file) - self.assertEqual(expected, generated, '%s%s' % ( - files, '\n'.join(line for line in unified_diff( - expected.splitlines(), generated.splitlines() ))) ) - os.remove(generated_file) - - def test_package_diagram(self): - self._test_same_file('packages_No_Name.dot') - - def test_class_diagram(self): - self._test_same_file('classes_No_Name.dot') - - - -class GetVisibilityTC(TestCase): - - def test_special(self): - for name in ["__reduce_ex__", "__setattr__"]: - self.assertEqual(get_visibility(name), 'special') - - def test_private(self): - for name in ["__g_", "____dsf", "__23_9"]: - got = get_visibility(name) - self.assertEqual(got, 'private', - 'got %s instead of private for value %s' % (got, name)) - - def test_public(self): - self.assertEqual(get_visibility('simple'), 'public') - - def test_protected(self): - for name in ["_","__", "___", "____", "_____", "___e__", - "_nextsimple", "_filter_it_"]: - got = get_visibility(name) - self.assertEqual(got, 'protected', - 'got %s instead of protected for value %s' % (got, name)) - - -if __name__ == '__main__': - unittest_main() |