497 lines
17 KiB
Python
497 lines
17 KiB
Python
import inspect
|
|
import os
|
|
import sys
|
|
|
|
import unittest2
|
|
|
|
import mitogen.master
|
|
from mitogen.core import b
|
|
|
|
import testlib
|
|
from testlib import MODS_DIR
|
|
|
|
|
|
class ConstructorTest(testlib.TestCase):
|
|
klass = mitogen.master.ModuleFinder
|
|
|
|
def test_simple(self):
|
|
self.klass()
|
|
|
|
|
|
class ReprTest(testlib.TestCase):
|
|
klass = mitogen.master.ModuleFinder
|
|
|
|
def test_simple(self):
|
|
self.assertEquals('ModuleFinder()', repr(self.klass()))
|
|
|
|
|
|
class IsStdlibNameTest(testlib.TestCase):
|
|
func = staticmethod(mitogen.master.is_stdlib_name)
|
|
|
|
def test_builtin(self):
|
|
import sys
|
|
self.assertTrue(self.func('sys'))
|
|
|
|
def test_stdlib_1(self):
|
|
import logging
|
|
self.assertTrue(self.func('logging'))
|
|
|
|
def test_stdlib_2(self):
|
|
# virtualenv only symlinks some paths to its local site-packages
|
|
# directory. Ensure both halves of the search path return the correct
|
|
# result.
|
|
import email
|
|
self.assertTrue(self.func('email'))
|
|
|
|
def test_mitogen_core(self):
|
|
import mitogen.core
|
|
self.assertFalse(self.func('mitogen.core'))
|
|
|
|
def test_mitogen_fakessh(self):
|
|
import mitogen.fakessh
|
|
self.assertFalse(self.func('mitogen.fakessh'))
|
|
|
|
|
|
class GetMainModuleDefectivePython3x(testlib.TestCase):
|
|
klass = mitogen.master.DefectivePython3xMainMethod
|
|
|
|
def call(self, fullname):
|
|
return self.klass().find(fullname)
|
|
|
|
def test_builtin(self):
|
|
self.assertEquals(None, self.call('sys'))
|
|
|
|
def test_not_main(self):
|
|
self.assertEquals(None, self.call('mitogen'))
|
|
|
|
def test_main(self):
|
|
import __main__
|
|
|
|
path, source, is_pkg = self.call('__main__')
|
|
self.assertTrue(path is not None)
|
|
self.assertTrue(os.path.exists(path))
|
|
self.assertEquals(path, __main__.__file__)
|
|
fp = open(path, 'rb')
|
|
try:
|
|
self.assertEquals(source, fp.read())
|
|
finally:
|
|
fp.close()
|
|
self.assertFalse(is_pkg)
|
|
|
|
|
|
class PkgutilMethodTest(testlib.TestCase):
|
|
klass = mitogen.master.PkgutilMethod
|
|
|
|
def call(self, fullname):
|
|
return self.klass().find(fullname)
|
|
|
|
def test_empty_source_pkg(self):
|
|
path, src, is_pkg = self.call('module_finder_testmod')
|
|
self.assertEquals(path,
|
|
os.path.join(MODS_DIR, 'module_finder_testmod/__init__.py'))
|
|
self.assertEquals(mitogen.core.b(''), src)
|
|
self.assertTrue(is_pkg)
|
|
|
|
def test_empty_source_module(self):
|
|
path, src, is_pkg = self.call('module_finder_testmod.empty_mod')
|
|
self.assertEquals(path,
|
|
os.path.join(MODS_DIR, 'module_finder_testmod/empty_mod.py'))
|
|
self.assertEquals(mitogen.core.b(''), src)
|
|
self.assertFalse(is_pkg)
|
|
|
|
def test_regular_mod(self):
|
|
from module_finder_testmod import regular_mod
|
|
path, src, is_pkg = self.call('module_finder_testmod.regular_mod')
|
|
self.assertEquals(path,
|
|
os.path.join(MODS_DIR, 'module_finder_testmod/regular_mod.py'))
|
|
self.assertEquals(mitogen.core.to_text(src),
|
|
inspect.getsource(regular_mod))
|
|
self.assertFalse(is_pkg)
|
|
|
|
|
|
class SysModulesMethodTest(testlib.TestCase):
|
|
klass = mitogen.master.SysModulesMethod
|
|
|
|
def call(self, fullname):
|
|
return self.klass().find(fullname)
|
|
|
|
def test_main(self):
|
|
import __main__
|
|
path, src, is_pkg = self.call('__main__')
|
|
self.assertEquals(path, __main__.__file__)
|
|
|
|
# linecache adds a line ending to the final line if one is missing.
|
|
actual_src = open(path, 'rb').read()
|
|
if actual_src[-1:] != b('\n'):
|
|
actual_src += b('\n')
|
|
|
|
self.assertEquals(src, actual_src)
|
|
self.assertFalse(is_pkg)
|
|
|
|
def test_dylib_fails(self):
|
|
# _socket comes from a .so
|
|
import _socket
|
|
tup = self.call('_socket')
|
|
self.assertIsNone(tup)
|
|
|
|
def test_builtin_fails(self):
|
|
# sys is built-in
|
|
tup = self.call('sys')
|
|
self.assertIsNone(tup)
|
|
|
|
|
|
class GetModuleViaParentEnumerationTest(testlib.TestCase):
|
|
klass = mitogen.master.ParentEnumerationMethod
|
|
|
|
def call(self, fullname):
|
|
return self.klass().find(fullname)
|
|
|
|
def test_main_fails(self):
|
|
import __main__
|
|
self.assertIsNone(self.call('__main__'))
|
|
|
|
def test_dylib_fails(self):
|
|
# _socket comes from a .so
|
|
import _socket
|
|
tup = self.call('_socket')
|
|
self.assertIsNone(tup)
|
|
|
|
def test_builtin_fails(self):
|
|
# sys is built-in
|
|
tup = self.call('sys')
|
|
self.assertIsNone(tup)
|
|
|
|
def test_plumbum_colors_like_pkg_succeeds(self):
|
|
# plumbum has been eating too many rainbow-colored pills
|
|
import pkg_like_plumbum.colors
|
|
path, src, is_pkg = self.call('pkg_like_plumbum.colors')
|
|
modpath = os.path.join(MODS_DIR, 'pkg_like_plumbum/colors.py')
|
|
self.assertEquals(path, modpath)
|
|
|
|
self.assertEquals(src, open(modpath, 'rb').read())
|
|
self.assertFalse(is_pkg)
|
|
|
|
def test_ansible_module_utils_distro_succeeds(self):
|
|
# #590: a package that turns itself into a module.
|
|
import pkg_like_ansible.module_utils.distro as d
|
|
self.assertEquals(d.I_AM, "the module that replaced the package")
|
|
self.assertEquals(
|
|
sys.modules['pkg_like_ansible.module_utils.distro'].__name__,
|
|
'pkg_like_ansible.module_utils.distro._distro'
|
|
)
|
|
|
|
# ensure we can resolve the subpackage.
|
|
path, src, is_pkg = self.call('pkg_like_ansible.module_utils.distro')
|
|
modpath = os.path.join(MODS_DIR,
|
|
'pkg_like_ansible/module_utils/distro/__init__.py')
|
|
self.assertEquals(path, modpath)
|
|
self.assertEquals(src, open(modpath, 'rb').read())
|
|
self.assertEquals(is_pkg, True)
|
|
|
|
# ensure we can resolve a child of the subpackage.
|
|
path, src, is_pkg = self.call(
|
|
'pkg_like_ansible.module_utils.distro._distro'
|
|
)
|
|
modpath = os.path.join(MODS_DIR,
|
|
'pkg_like_ansible/module_utils/distro/_distro.py')
|
|
self.assertEquals(path, modpath)
|
|
self.assertEquals(src, open(modpath, 'rb').read())
|
|
self.assertEquals(is_pkg, False)
|
|
|
|
def test_ansible_module_utils_system_distro_succeeds(self):
|
|
# #590: a package that turns itself into a module.
|
|
# #590: a package that turns itself into a module.
|
|
import pkg_like_ansible.module_utils.sys_distro as d
|
|
self.assertEquals(d.I_AM, "the system module that replaced the subpackage")
|
|
self.assertEquals(
|
|
sys.modules['pkg_like_ansible.module_utils.sys_distro'].__name__,
|
|
'system_distro'
|
|
)
|
|
|
|
# ensure we can resolve the subpackage.
|
|
path, src, is_pkg = self.call('pkg_like_ansible.module_utils.sys_distro')
|
|
modpath = os.path.join(MODS_DIR,
|
|
'pkg_like_ansible/module_utils/sys_distro/__init__.py')
|
|
self.assertEquals(path, modpath)
|
|
self.assertEquals(src, open(modpath, 'rb').read())
|
|
self.assertEquals(is_pkg, True)
|
|
|
|
# ensure we can resolve a child of the subpackage.
|
|
path, src, is_pkg = self.call(
|
|
'pkg_like_ansible.module_utils.sys_distro._distro'
|
|
)
|
|
modpath = os.path.join(MODS_DIR,
|
|
'pkg_like_ansible/module_utils/sys_distro/_distro.py')
|
|
self.assertEquals(path, modpath)
|
|
self.assertEquals(src, open(modpath, 'rb').read())
|
|
self.assertEquals(is_pkg, False)
|
|
|
|
|
|
class ResolveRelPathTest(testlib.TestCase):
|
|
klass = mitogen.master.ModuleFinder
|
|
|
|
def call(self, fullname, level):
|
|
return self.klass().resolve_relpath(fullname, level)
|
|
|
|
def test_empty(self):
|
|
self.assertEquals('', self.call('', 0))
|
|
self.assertEquals('', self.call('', 1))
|
|
self.assertEquals('', self.call('', 2))
|
|
|
|
def test_absolute(self):
|
|
self.assertEquals('', self.call('email.utils', 0))
|
|
|
|
def test_rel1(self):
|
|
self.assertEquals('email.', self.call('email.utils', 1))
|
|
|
|
def test_rel2(self):
|
|
self.assertEquals('', self.call('email.utils', 2))
|
|
|
|
def test_rel_overflow(self):
|
|
self.assertEquals('', self.call('email.utils', 3))
|
|
|
|
|
|
class FakeSshTest(testlib.TestCase):
|
|
klass = mitogen.master.ModuleFinder
|
|
|
|
def call(self, fullname):
|
|
return self.klass().find_related_imports(fullname)
|
|
|
|
def test_simple(self):
|
|
import mitogen.fakessh
|
|
related = self.call('mitogen.fakessh')
|
|
self.assertEquals(related, [
|
|
'mitogen',
|
|
'mitogen.core',
|
|
'mitogen.master',
|
|
'mitogen.parent',
|
|
])
|
|
|
|
|
|
class FindRelatedTest(testlib.TestCase):
|
|
klass = mitogen.master.ModuleFinder
|
|
|
|
def call(self, fullname):
|
|
return self.klass().find_related(fullname)
|
|
|
|
SIMPLE_EXPECT = set([
|
|
u'mitogen',
|
|
u'mitogen.core',
|
|
u'mitogen.master',
|
|
u'mitogen.minify',
|
|
u'mitogen.parent',
|
|
])
|
|
|
|
if sys.version_info < (2, 7):
|
|
SIMPLE_EXPECT.add('mitogen.compat')
|
|
SIMPLE_EXPECT.add('mitogen.compat.tokenize')
|
|
if sys.version_info < (2, 6):
|
|
SIMPLE_EXPECT.add('mitogen.compat')
|
|
SIMPLE_EXPECT.add('mitogen.compat.pkgutil')
|
|
|
|
def test_simple(self):
|
|
import mitogen.fakessh
|
|
related = self.call('mitogen.fakessh')
|
|
self.assertEquals(set(related), self.SIMPLE_EXPECT)
|
|
|
|
|
|
if sys.version_info > (2, 6):
|
|
class DjangoMixin(object):
|
|
WEBPROJECT_PATH = os.path.join(MODS_DIR, 'webproject')
|
|
|
|
# TODO: rip out Django and replace with a static tree of weird imports
|
|
# that don't depend on .. Django! The hack below is because the version
|
|
# of Django we need to test against 2.6 doesn't actually run on 3.6.
|
|
# But we don't care, we just need to be able to import it.
|
|
#
|
|
# File "django/utils/html_parser.py", line 12, in <module>
|
|
# AttributeError: module 'html.parser' has no attribute
|
|
# 'HTMLParseError'
|
|
#
|
|
from django.utils.six.moves import html_parser as _html_parser
|
|
_html_parser.HTMLParseError = Exception
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
super(DjangoMixin, cls).setUpClass()
|
|
sys.path.append(cls.WEBPROJECT_PATH)
|
|
os.environ['DJANGO_SETTINGS_MODULE'] = 'webproject.settings'
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
sys.path.remove(cls.WEBPROJECT_PATH)
|
|
del os.environ['DJANGO_SETTINGS_MODULE']
|
|
super(DjangoMixin, cls).tearDownClass()
|
|
|
|
|
|
class FindRelatedImportsTest(DjangoMixin, testlib.TestCase):
|
|
klass = mitogen.master.ModuleFinder
|
|
|
|
def call(self, fullname):
|
|
return self.klass().find_related_imports(fullname)
|
|
|
|
def test_django_db(self):
|
|
import django.db
|
|
related = self.call('django.db')
|
|
self.assertEquals(related, [
|
|
'django',
|
|
'django.core',
|
|
'django.core.signals',
|
|
'django.db.utils',
|
|
'django.utils.functional',
|
|
])
|
|
|
|
def test_django_db_models(self):
|
|
import django.db.models
|
|
related = self.call('django.db.models')
|
|
self.maxDiff=None
|
|
self.assertEquals(related, [
|
|
u'django',
|
|
u'django.core.exceptions',
|
|
u'django.db',
|
|
u'django.db.models',
|
|
u'django.db.models.aggregates',
|
|
u'django.db.models.base',
|
|
u'django.db.models.deletion',
|
|
u'django.db.models.expressions',
|
|
u'django.db.models.fields',
|
|
u'django.db.models.fields.files',
|
|
u'django.db.models.fields.related',
|
|
u'django.db.models.fields.subclassing',
|
|
u'django.db.models.loading',
|
|
u'django.db.models.manager',
|
|
u'django.db.models.query',
|
|
u'django.db.models.signals',
|
|
])
|
|
|
|
|
|
class DjangoFindRelatedTest(DjangoMixin, testlib.TestCase):
|
|
klass = mitogen.master.ModuleFinder
|
|
maxDiff = None
|
|
|
|
def call(self, fullname):
|
|
return self.klass().find_related(fullname)
|
|
|
|
def test_django_db(self):
|
|
import django.db
|
|
related = self.call('django.db')
|
|
self.assertEquals(related, [
|
|
u'django',
|
|
u'django.conf',
|
|
u'django.conf.global_settings',
|
|
u'django.core',
|
|
u'django.core.exceptions',
|
|
u'django.core.signals',
|
|
u'django.db.utils',
|
|
u'django.dispatch',
|
|
u'django.dispatch.dispatcher',
|
|
u'django.dispatch.saferef',
|
|
u'django.utils',
|
|
u'django.utils._os',
|
|
u'django.utils.encoding',
|
|
u'django.utils.functional',
|
|
u'django.utils.importlib',
|
|
u'django.utils.module_loading',
|
|
u'django.utils.six',
|
|
])
|
|
|
|
@unittest2.skipIf(
|
|
condition=(sys.version_info >= (3, 0)),
|
|
reason='broken due to ancient vendored six.py'
|
|
)
|
|
def test_django_db_models(self):
|
|
import django.db.models
|
|
related = self.call('django.db.models')
|
|
self.assertEquals(related, [
|
|
u'django',
|
|
u'django.conf',
|
|
u'django.conf.global_settings',
|
|
u'django.core',
|
|
u'django.core.exceptions',
|
|
u'django.core.files',
|
|
u'django.core.files.base',
|
|
u'django.core.files.images',
|
|
u'django.core.files.locks',
|
|
u'django.core.files.move',
|
|
u'django.core.files.storage',
|
|
u'django.core.files.utils',
|
|
u'django.core.signals',
|
|
u'django.core.validators',
|
|
u'django.db',
|
|
u'django.db.backends',
|
|
u'django.db.backends.signals',
|
|
u'django.db.backends.util',
|
|
u'django.db.models.aggregates',
|
|
u'django.db.models.base',
|
|
u'django.db.models.constants',
|
|
u'django.db.models.deletion',
|
|
u'django.db.models.expressions',
|
|
u'django.db.models.fields',
|
|
u'django.db.models.fields.files',
|
|
u'django.db.models.fields.proxy',
|
|
u'django.db.models.fields.related',
|
|
u'django.db.models.fields.subclassing',
|
|
u'django.db.models.loading',
|
|
u'django.db.models.manager',
|
|
u'django.db.models.options',
|
|
u'django.db.models.query',
|
|
u'django.db.models.query_utils',
|
|
u'django.db.models.related',
|
|
u'django.db.models.signals',
|
|
u'django.db.models.sql',
|
|
u'django.db.models.sql.aggregates',
|
|
u'django.db.models.sql.constants',
|
|
u'django.db.models.sql.datastructures',
|
|
u'django.db.models.sql.expressions',
|
|
u'django.db.models.sql.query',
|
|
u'django.db.models.sql.subqueries',
|
|
u'django.db.models.sql.where',
|
|
u'django.db.transaction',
|
|
u'django.db.utils',
|
|
u'django.dispatch',
|
|
u'django.dispatch.dispatcher',
|
|
u'django.dispatch.saferef',
|
|
u'django.forms',
|
|
u'django.forms.fields',
|
|
u'django.forms.forms',
|
|
u'django.forms.formsets',
|
|
u'django.forms.models',
|
|
u'django.forms.util',
|
|
u'django.forms.widgets',
|
|
u'django.utils',
|
|
u'django.utils._os',
|
|
u'django.utils.crypto',
|
|
u'django.utils.datastructures',
|
|
u'django.utils.dateformat',
|
|
u'django.utils.dateparse',
|
|
u'django.utils.dates',
|
|
u'django.utils.datetime_safe',
|
|
u'django.utils.decorators',
|
|
u'django.utils.deprecation',
|
|
u'django.utils.encoding',
|
|
u'django.utils.formats',
|
|
u'django.utils.functional',
|
|
u'django.utils.html',
|
|
u'django.utils.html_parser',
|
|
u'django.utils.importlib',
|
|
u'django.utils.ipv6',
|
|
u'django.utils.itercompat',
|
|
u'django.utils.module_loading',
|
|
u'django.utils.numberformat',
|
|
u'django.utils.safestring',
|
|
u'django.utils.six',
|
|
u'django.utils.text',
|
|
u'django.utils.timezone',
|
|
u'django.utils.translation',
|
|
u'django.utils.tree',
|
|
u'django.utils.tzinfo',
|
|
u'pytz',
|
|
u'pytz.exceptions',
|
|
u'pytz.lazy',
|
|
u'pytz.tzfile',
|
|
u'pytz.tzinfo',
|
|
])
|
|
|
|
if __name__ == '__main__':
|
|
unittest2.main()
|