class TestModuleFinder(unittest.TestCase):
def setUp(self):
self.finder = ModuleFinder(mock_decorator)
def tearDown(self):
self.finder.uninstall()
def test_module_found(self):
self.finder.install()
import typesafety.tests.mockmodule
self.assertTrue(self.finder.installed)
self.assertTrue(
isdecorated(typesafety.tests.mockmodule.function)
)
self.assertTrue(
isdecorated(typesafety.tests.mockmodule.ModuleClass.method)
)
def test_module_found_but_filtered_out(self):
self.finder.set_filter(lambda name: False)
self.finder.install()
import typesafety.tests.mockmodule
self.assertFalse(
isdecorated(typesafety.tests.mockmodule.function)
)
self.assertFalse(
isdecorated(typesafety.tests.mockmodule.ModuleClass.method)
)
def setUp(self):
self.finder = ModuleFinder(mock_decorator)
