djavu-asus/elpa/elpy-20230803.1455/elpy/tests/test_pydocutils.py

   1 import os
   2 import unittest
   3 import shutil
   4 import sys
   5 import tempfile
   6
   7 try:
   8     from unittest import mock
   9 except ImportError:
  10     import mock
  11
  12 import elpy.pydocutils
  13
  14
  15 class TestGetPydocCompletions(unittest.TestCase):
  16     def test_should_return_top_level_modules(self):
  17         modules = elpy.pydocutils.get_pydoc_completions("")
  18         self.assertIn('sys', modules)
  19         self.assertIn('json', modules)
  20
  21     def test_should_return_submodules(self):
  22         modules = elpy.pydocutils.get_pydoc_completions("elpy")
  23         self.assertIn("elpy.rpc", modules)
  24         self.assertIn("elpy.server", modules)
  25         modules = elpy.pydocutils.get_pydoc_completions("os")
  26         self.assertIn("os.path", modules)
  27
  28     def test_should_find_objects_in_module(self):
  29         self.assertIn("elpy.tests.test_pydocutils.TestGetPydocCompletions",
  30                       elpy.pydocutils.get_pydoc_completions
  31                       ("elpy.tests.test_pydocutils"))
  32
  33     def test_should_find_attributes_of_objects(self):
  34         attribs = elpy.pydocutils.get_pydoc_completions(
  35             "elpy.tests.test_pydocutils.TestGetPydocCompletions")
  36         self.assertIn("elpy.tests.test_pydocutils.TestGetPydocCompletions."
  37                       "test_should_find_attributes_of_objects",
  38                       attribs)
  39
  40     def test_should_return_none_for_inexisting_module(self):
  41         self.assertEqual([],
  42                          elpy.pydocutils.get_pydoc_completions
  43                          ("does_not_exist"))
  44
  45     def test_should_work_for_unicode_strings(self):
  46         self.assertIsNotNone(elpy.pydocutils.get_pydoc_completions
  47                              (u"sys"))
  48
  49     def test_should_find_partial_completions(self):
  50         self.assertIn("multiprocessing",
  51                       elpy.pydocutils.get_pydoc_completions
  52                       ("multiprocess"))
  53         self.assertIn("multiprocessing.util",
  54                       elpy.pydocutils.get_pydoc_completions
  55                       ("multiprocessing.ut"))
  56
  57     def test_should_ignore_trailing_dot(self):
  58         self.assertIn("elpy.pydocutils",
  59                       elpy.pydocutils.get_pydoc_completions
  60                       ("elpy."))
  61
  62
  63 class TestGetModules(unittest.TestCase):
  64     def test_should_return_top_level_modules(self):
  65         modules = elpy.pydocutils.get_modules()
  66         self.assertIn('sys', modules)
  67         self.assertIn('json', modules)
  68
  69     def test_should_return_submodules(self):
  70         modules = elpy.pydocutils.get_modules("elpy")
  71         self.assertIn("rpc", modules)
  72         self.assertIn("server", modules)
  73
  74     @mock.patch.object(elpy.pydocutils, 'safeimport')
  75     def test_should_catch_import_errors(self, safeimport):
  76         def raise_function(message):
  77             raise elpy.pydocutils.ErrorDuringImport(message,
  78                                                     (None, None, None))
  79         safeimport.side_effect = raise_function
  80         self.assertEqual([], elpy.pydocutils.get_modules("foo.bar"))
  81
  82     def test_should_not_fail_for_permission_denied(self):
  83         tmpdir = tempfile.mkdtemp(prefix="test-elpy-get-modules-")
  84         sys.path.append(tmpdir)
  85         os.chmod(tmpdir, 0o000)
  86         try:
  87             elpy.pydocutils.get_modules()
  88         finally:
  89             os.chmod(tmpdir, 0o755)
  90             shutil.rmtree(tmpdir)
  91             sys.path.remove(tmpdir)