tests: rearrange test modules again, they're used in multiple places

They're back on sys.path.
pull/595/head
David Wilson 6 years ago
parent 2f68a5a660
commit 2f29f3e8e6

@ -7,9 +7,7 @@ import unittest2
import mitogen.master import mitogen.master
import testlib import testlib
from testlib import MODS_DIR
MODS_PATH = testlib.data_path('module_finder')
sys.path.append(MODS_PATH)
class ConstructorTest(testlib.TestCase): class ConstructorTest(testlib.TestCase):
@ -89,14 +87,14 @@ class PkgutilMethodTest(testlib.TestCase):
def test_empty_source_pkg(self): def test_empty_source_pkg(self):
path, src, is_pkg = self.call('module_finder_testmod') path, src, is_pkg = self.call('module_finder_testmod')
self.assertEquals(path, self.assertEquals(path,
os.path.join(MODS_PATH, 'module_finder_testmod/__init__.py')) os.path.join(MODS_DIR, 'module_finder_testmod/__init__.py'))
self.assertEquals(mitogen.core.b(''), src) self.assertEquals(mitogen.core.b(''), src)
self.assertTrue(is_pkg) self.assertTrue(is_pkg)
def test_empty_source_module(self): def test_empty_source_module(self):
path, src, is_pkg = self.call('module_finder_testmod.empty_mod') path, src, is_pkg = self.call('module_finder_testmod.empty_mod')
self.assertEquals(path, self.assertEquals(path,
os.path.join(MODS_PATH, 'module_finder_testmod/empty_mod.py')) os.path.join(MODS_DIR, 'module_finder_testmod/empty_mod.py'))
self.assertEquals(mitogen.core.b(''), src) self.assertEquals(mitogen.core.b(''), src)
self.assertFalse(is_pkg) self.assertFalse(is_pkg)
@ -104,7 +102,7 @@ class PkgutilMethodTest(testlib.TestCase):
from module_finder_testmod import regular_mod from module_finder_testmod import regular_mod
path, src, is_pkg = self.call('module_finder_testmod.regular_mod') path, src, is_pkg = self.call('module_finder_testmod.regular_mod')
self.assertEquals(path, self.assertEquals(path,
os.path.join(MODS_PATH, 'module_finder_testmod/regular_mod.py')) os.path.join(MODS_DIR, 'module_finder_testmod/regular_mod.py'))
self.assertEquals(mitogen.core.to_text(src), self.assertEquals(mitogen.core.to_text(src),
inspect.getsource(regular_mod)) inspect.getsource(regular_mod))
self.assertFalse(is_pkg) self.assertFalse(is_pkg)
@ -166,7 +164,7 @@ class GetModuleViaParentEnumerationTest(testlib.TestCase):
# plumbum has been eating too many rainbow-colored pills # plumbum has been eating too many rainbow-colored pills
import pkg_like_plumbum.colors import pkg_like_plumbum.colors
path, src, is_pkg = self.call('pkg_like_plumbum.colors') path, src, is_pkg = self.call('pkg_like_plumbum.colors')
modpath = os.path.join(MODS_PATH, 'pkg_like_plumbum/colors.py') modpath = os.path.join(MODS_DIR, 'pkg_like_plumbum/colors.py')
self.assertEquals(path, modpath) self.assertEquals(path, modpath)
self.assertEquals(src, open(modpath, 'rb').read()) self.assertEquals(src, open(modpath, 'rb').read())
@ -182,7 +180,7 @@ class GetModuleViaParentEnumerationTest(testlib.TestCase):
) )
path, src, is_pkg = self.call('pkg_like_ansible.module_utils.distro') path, src, is_pkg = self.call('pkg_like_ansible.module_utils.distro')
modpath = os.path.join(MODS_PATH, modpath = os.path.join(MODS_DIR,
'pkg_like_ansible/module_utils/distro/__init__.py') 'pkg_like_ansible/module_utils/distro/__init__.py')
self.assertEquals(path, modpath) self.assertEquals(path, modpath)
self.assertEquals(src, open(modpath, 'rb').read()) self.assertEquals(src, open(modpath, 'rb').read())
@ -259,7 +257,7 @@ class FindRelatedTest(testlib.TestCase):
if sys.version_info > (2, 6): if sys.version_info > (2, 6):
class DjangoMixin(object): class DjangoMixin(object):
WEBPROJECT_PATH = os.path.join(MODS_PATH, 'webproject') WEBPROJECT_PATH = os.path.join(MODS_DIR, 'webproject')
# TODO: rip out Django and replace with a static tree of weird imports # 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 # that don't depend on .. Django! The hack below is because the version

@ -41,6 +41,11 @@ except NameError:
LOG = logging.getLogger(__name__) LOG = logging.getLogger(__name__)
DATA_DIR = os.path.join(os.path.dirname(__file__), 'data') DATA_DIR = os.path.join(os.path.dirname(__file__), 'data')
MODS_DIR = os.path.join(DATA_DIR, 'importer')
sys.path.append(DATA_DIR)
sys.path.append(MODS_DIR)
if mitogen.is_master: if mitogen.is_master:
mitogen.utils.log_to_file() mitogen.utils.log_to_file()

Loading…
Cancel
Save