src/allmydata/test/test_import.py

   1
   2 from twisted.trial import unittest
   3
   4 import allmydata
   5 import mock
   6
   7 real_import_func = __import__
   8
   9 class T(unittest.TestCase):
  10     @mock.patch('__builtin__.__import__')
  11     def test_report_import_error(self, mockimport):
  12         def raiseIE_from_this_particular_func(name, *args):
  13             if name == "foolscap":
  14                 marker = "wheeeyo"
  15                 raise ImportError(marker + " foolscap cant be imported")
  16             else:
  17                 return real_import_func(name, *args)
  18
  19         mockimport.side_effect = raiseIE_from_this_particular_func
  20
  21         vers_and_locs =  allmydata.get_package_versions_and_locations()
  22         for (pkgname, stuff) in vers_and_locs:
  23             if pkgname == 'foolscap':
  24                 self.failUnless('wheeeyo' in str(stuff[2]), stuff)
  25                 self.failUnless('raiseIE_from_this_particular_func' in str(stuff[2]), stuff)