blob: f4174979bfa9688b02a5c341b512630a20c701c9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
import sys
import pytest
from snakeoil import modules
class TestModules:
@pytest.fixture(autouse=True)
def _setup(self, tmp_path):
# set up some test modules for our use
packdir = tmp_path / 'mod_testpack'
packdir.mkdir()
# create an empty file
(packdir / '__init__.py').touch()
for directory in (tmp_path, packdir):
for i in range(3):
(directory / f'mod_test{i}.py').write_text('def foo(): pass\n')
(directory / 'mod_horked.py').write_text('1/0\n')
# append them to path
sys.path.insert(0, str(tmp_path))
yield
# pop the test module dir from path
sys.path.pop(0)
# make sure we don't keep the sys.modules entries around
for i in range(3):
sys.modules.pop('mod_test%s' % i, None)
sys.modules.pop('mod_testpack.mod_test%s' % i, None)
sys.modules.pop('mod_testpack', None)
sys.modules.pop('mod_horked', None)
sys.modules.pop('mod_testpack.mod_horked', None)
def test_load_module(self):
# import an already-imported module
assert modules.load_module('snakeoil.modules') is modules
# and a system one, just for kicks
assert modules.load_module('sys') is sys
# non-existing module from an existing package
with pytest.raises(modules.FailedImport):
modules.load_module('snakeoil.__not_there')
# (hopefully :) non-existing top-level module/package
with pytest.raises(modules.FailedImport):
modules.load_module('__not_there')
# "Unable to import"
# pylint: disable=F0401
# unimported toplevel module
modtest1 = modules.load_module('mod_test1')
import mod_test1
assert mod_test1 is modtest1
# unimported in-package module
packtest2 = modules.load_module('mod_testpack.mod_test2')
from mod_testpack import mod_test2
assert mod_test2 is packtest2
def test_load_attribute(self):
# already imported
assert modules.load_attribute('sys.path') is sys.path
# unimported
myfoo = modules.load_attribute('mod_testpack.mod_test2.foo')
# "Unable to import"
# pylint: disable=F0401
from mod_testpack.mod_test2 import foo
assert foo is myfoo
# nonexisting attribute
with pytest.raises(modules.FailedImport):
modules.load_attribute('snakeoil.froznicator')
# nonexisting top-level
with pytest.raises(modules.FailedImport):
modules.load_attribute('spork_does_not_exist.foo')
# not an attr
with pytest.raises(modules.FailedImport):
modules.load_attribute('sys')
# not imported yet
with pytest.raises(modules.FailedImport):
modules.load_attribute('mod_testpack.mod_test3')
def test_load_any(self):
# import an already-imported module
assert modules.load_any('snakeoil.modules') is modules
# attribute of an already imported module
assert modules.load_any('sys.path') is sys.path
# already imported toplevel.
assert sys is modules.load_any('sys')
# unimported
myfoo = modules.load_any('mod_testpack.mod_test2.foo')
# "Unable to import"
# pylint: disable=F0401
from mod_testpack.mod_test2 import foo
assert foo is myfoo
# nonexisting attribute
with pytest.raises(modules.FailedImport):
modules.load_any('snakeoil.froznicator')
# nonexisting top-level
with pytest.raises(modules.FailedImport):
modules.load_any('spork_does_not_exist.foo')
with pytest.raises(modules.FailedImport):
modules.load_any('spork_does_not_exist')
# not imported yet
with pytest.raises(modules.FailedImport):
modules.load_any('mod_testpack.mod_test3')
def test_broken_module(self):
for func in [modules.load_module, modules.load_any]:
with pytest.raises(modules.FailedImport):
func('mod_testpack.mod_horked')
assert 'mod_testpack.mod_horked' not in sys.modules
|
GitWeb