tahoe-lafs/src/allmydata/test/test_import.py
Brian Warner fb4dd43310 Rename test_package_initialization.py to (much shorter) test_import.py .
The former name was making my 'ls' listings hard to read, by forcing them
down to just two columns.
2011-06-11 12:02:34 -07:00

26 lines
880 B
Python

from twisted.trial import unittest
import allmydata
import mock
real_import_func = __import__
class T(unittest.TestCase):
@mock.patch('__builtin__.__import__')
def test_report_import_error(self, mockimport):
def raiseIE_from_this_particular_func(name, *args):
if name == "foolscap":
marker = "wheeeyo"
raise ImportError(marker + " foolscap cant be imported")
else:
return real_import_func(name, *args)
mockimport.side_effect = raiseIE_from_this_particular_func
vers_and_locs = allmydata.get_package_versions_and_locations()
for (pkgname, stuff) in vers_and_locs:
if pkgname == 'foolscap':
self.failUnless('wheeeyo' in str(stuff[2]), stuff)
self.failUnless('raiseIE_from_this_particular_func' in str(stuff[2]), stuff)