# 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 import unittest from astroid import MANAGER from pylint.pyreverse.inspector import Linker, project_from_files 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 with codecs.open(path, encoding='latin1') as stream: lines = [line.strip() for line in stream.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="No Name"): """return a astroid project representation""" def _astroid_wrapper(func, modname): return func(modname) return project_from_files([module], _astroid_wrapper, project_name=name) CONFIG = Config() class DotWriterTC(unittest.TestCase): @classmethod def setUpClass(cls): project = get_project(os.path.join(os.path.dirname(__file__), 'data')) 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 = os.path.join(os.path.dirname(__file__), 'data', 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(unittest.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()