Насмешка ImportError в Python
Я пробую это в течение почти двух часов теперь без любой удачи.
У меня есть модуль, который похож на это:
try:
from zope.component import queryUtility # and things like this
except ImportError:
# do some fallback operations <-- how to test this?
Позже в коде:
try:
queryUtility(foo)
except NameError:
# do some fallback actions <-- this one is easy with mocking
# zope.component.queryUtility to raise a NameError
Какие-либо идеи?
Править:
Предложение Alex, кажется, не работает:
>>> import __builtin__
>>> realimport = __builtin__.__import__
>>> def fakeimport(name, *args, **kw):
... if name == 'zope.component':
... raise ImportError
... realimport(name, *args, **kw)
...
>>> __builtin__.__import__ = fakeimport
При запущении тестов:
aatiis@aiur ~/work/ao.shorturl $ ./bin/test --coverage .
Running zope.testing.testrunner.layer.UnitTests tests:
Set up zope.testing.testrunner.layer.UnitTests in 0.000 seconds.
Error in test /home/aatiis/work/ao.shorturl/src/ao/shorturl/shorturl.txt
Traceback (most recent call last):
File "/usr/lib64/python2.5/unittest.py", line 260, in run
testMethod()
File "/usr/lib64/python2.5/doctest.py", line 2123, in runTest
test, out=new.write, clear_globs=False)
File "/usr/lib64/python2.5/doctest.py", line 1361, in run
return self.__run(test, compileflags, out)
File "/usr/lib64/python2.5/doctest.py", line 1282, in __run
exc_info)
File "/usr/lib64/python2.5/doctest.py", line 1148, in report_unexpected_exception
'Exception raised:\n' + _indent(_exception_traceback(exc_info)))
File "/usr/lib64/python2.5/doctest.py", line 1163, in _failure_header
out.append(_indent(source))
File "/usr/lib64/python2.5/doctest.py", line 224, in _indent
return re.sub('(?m)^(?!$)', indent*' ', s)
File "/usr/lib64/python2.5/re.py", line 150, in sub
return _compile(pattern, 0).sub(repl, string, count)
File "/usr/lib64/python2.5/re.py", line 239, in _compile
p = sre_compile.compile(pattern, flags)
File "/usr/lib64/python2.5/sre_compile.py", line 507, in compile
p = sre_parse.parse(p, flags)
AttributeError: 'NoneType' object has no attribute 'parse'
Error in test BaseShortUrlHandler (ao.shorturl)
Traceback (most recent call last):
File "/usr/lib64/python2.5/unittest.py", line 260, in run
testMethod()
File "/usr/lib64/python2.5/doctest.py", line 2123, in runTest
test, out=new.write, clear_globs=False)
File "/usr/lib64/python2.5/doctest.py", line 1351, in run
self.debugger = _OutputRedirectingPdb(save_stdout)
File "/usr/lib64/python2.5/doctest.py", line 324, in __init__
pdb.Pdb.__init__(self, stdout=out)
File "/usr/lib64/python2.5/pdb.py", line 57, in __init__
cmd.Cmd.__init__(self, completekey, stdin, stdout)
File "/usr/lib64/python2.5/cmd.py", line 90, in __init__
import sys
File "<doctest shorturl.txt[10]>", line 4, in fakeimport
NameError: global name 'realimport' is not defined
Однако это действительно работает, когда я выполняю тот же код из Python интерактивная консоль.
БОЛЬШЕ РЕДАКТИРОВАНИЯ:
Я использую zope.testing и тестовый файл, shorturl.txt это имеет все тесты, характерные для этой части моего модуля. Сначала я импортирую модуль с zope.component доступный, чтобы продемонстрировать и протестировать обычное использование. Отсутствие zope.* пакеты считают пограничным случаем, таким образом, я тестирую его позже. Таким образом я имею к reload() мой модуль, после создания zope.* недоступный, так или иначе.
До сих пор я даже попытался использовать tempfile.mktempdir() и пустой zope/__init__.py и zope/component/__init__.py файлы в tempdir, затем вставляя tempdir к sys.path[0], и удаление старого zope.* пакеты от sys.modules.
Не работал также.
ЕЩЕ БОЛЬШЕ РЕДАКТИРОВАНИЯ:
Тем временем я попробовал это:
>>> class NoZope(object):
... def find_module(self, fullname, path):
... if fullname.startswith('zope'):
... raise ImportError
...
>>> import sys
>>> sys.path.insert(0, NoZope())
И это работает хорошо на пространство имен набора тестов (= для всего импорта в shorturl.txt), но это не выполняется в моем основном модуле, ao.shorturl. Даже, когда я reload() это. Какая-либо идея, почему?
>>> import zope # ok, this raises an ImportError
>>> reload(ao.shorturl) <module ...>
Импорт zope.interfaces повышения ImportError, таким образом, это не добирается до части, куда я импортирую zope.component, и это остается в ao.shorturl пространстве имен. Почему?!
>>> ao.shorturl.zope.component # why?!
<module ...>
2 ответа
Просто добавьте в builtins свою собственную версию __import__ - она может вызывать все, что вы хотите, когда распознает, что ее вызывают на определенных модулях, для которых вы хотите имитировать ошибки. Подробную информацию смотрите в документации. Примерно так:
try:
import builtins
except ImportError:
import __builtin__ as builtins
realimport = builtins.__import__
def myimport(name, globals, locals, fromlist, level):
if ...:
raise ImportError
return realimport(name, globals, locals, fromlist, level)
builtins.__import__ = myimport
Вместо ... , вы можете жестко указать name == 'zope.component', или организовать все более гибко с помощью собственного обратного вызова, который может заставить импорт подниматься по требованию в различных случаях, в зависимости от ваших конкретных потребностей тестирования, не требуя от вас кода множества __import__-подобных функций;-).
Заметьте также, что если вы используете вместо import zope.component или from zope. component import something, будет from zope import component, то name будет 'zope', а 'component' будет единственным элементом в fromlist.
Edit: в документации к функции __import__ сказано, что имя для импорта - builtin (как в Python 3), но на самом деле вам нужно __builtins__ -- я отредактировал код выше, чтобы он работал в любом случае.
Это то, что я вычислил в своих юнит-тестах.
Он использует PEP-302 «Новые ловушки импорта» . (Предупреждение: документ PEP-302 и более краткие примечания к выпуску, на которые я ссылался, не совсем точны .)
Я использую meta_path , потому что он как можно раньше при импорте последовательность.
Если модуль уже был импортирован (как в моем случае, потому что предыдущие юнит-тесты имитируют его), то необходимо удалить его из sys.modules перед выполнением перезагрузки зависимого модуля.
Ensure we fallback to using ~/.pif if XDG doesn't exist.
>>> import sys
>>> class _():
... def __init__(self, modules):
... self.modules = modules
...
... def find_module(self, fullname, path=None):
... if fullname in self.modules:
... raise ImportError('Debug import failure for %s' % fullname)
>>> fail_loader = _(['xdg.BaseDirectory'])
>>> sys.meta_path.append(fail_loader)
>>> del sys.modules['xdg.BaseDirectory']
>>> reload(pif.index) #doctest: +ELLIPSIS
<module 'pif.index' from '...'>
>>> pif.index.CONFIG_DIR == os.path.expanduser('~/.pif')
True
>>> sys.meta_path.remove(fail_loader)
Где код внутри pif.index выглядит так:
try:
import xdg.BaseDirectory
CONFIG_DIR = os.path.join(xdg.BaseDirectory.xdg_data_home, 'pif')
except ImportError:
CONFIG_DIR = os.path.expanduser('~/.pif')
Чтобы ответить на вопрос о том, почему недавно перезагруженный модуль имеет свойства старой и новой нагрузок, вот два примера файлов.
Первый - это модуль y со случаем сбоя импорта.
# y.py
try:
import sys
_loaded_with = 'sys'
except ImportError:
import os
_loaded_with = 'os'
Второй - x , который демонстрирует, как оставление дескрипторов для модуля может повлиять на его свойства при перезагрузке.
# x.py
import sys
import y
assert y._loaded_with == 'sys'
assert y.sys
class _():
def __init__(self, modules):
self.modules = modules
def find_module(self, fullname, path=None):
if fullname in self.modules:
raise ImportError('Debug import failure for %s' % fullname)
# Importing sys will not raise an ImportError.
fail_loader = _(['sys'])
sys.meta_path.append(fail_loader)
# Demonstrate that reloading doesn't work if the module is already in the
# cache.
reload(y)
assert y._loaded_with == 'sys'
assert y.sys
# Now we remove sys from the modules cache, and try again.
del sys.modules['sys']
reload(y)
assert y._loaded_with == 'os'
assert y.sys
assert y.os
# Now we remove the handles to the old y so it can get garbage-collected.
del sys.modules['y']
del y
import y
assert y._loaded_with == 'os'
try:
assert y.sys
except AttributeError:
pass
assert y.os