1 files changed, 136 insertions, 0 deletions
diff --git a/pylint/test/unittest_pyreverse_writer.py b/pylint/test/unittest_pyreverse_writer.py
new file mode 100644
index 0000000..b4da835
--- /dev/null
+++ b/pylint/test/unittest_pyreverse_writer.py
@@ -0,0 +1,136 @@
+# 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 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
+    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=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(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()