summaryrefslogtreecommitdiff
path: root/test/unittest_pyreverse_writer.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/unittest_pyreverse_writer.py')
-rw-r--r--test/unittest_pyreverse_writer.py136
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()