def test_find_related_module(self): with patch('importlib.import_module') as imp: with patch('imp.find_module') as find: imp.return_value = Mock() imp.return_value.__path__ = 'foo' base.find_related_module(base, 'tasks') find.side_effect = ImportError() base.find_related_module(base, 'tasks')
def test_find_related_module(self): with patch('importlib.import_module') as imp: imp.return_value = Mock() imp.return_value.__path__ = 'foo' assert base.find_related_module('bar', 'tasks').__path__ == 'foo' imp.assert_any_call('bar') imp.assert_any_call('bar.tasks') imp.reset_mock() assert base.find_related_module('bar', None).__path__ == 'foo' imp.assert_called_once_with('bar') imp.side_effect = ImportError() with pytest.raises(ImportError): base.find_related_module('bar', 'tasks') assert base.find_related_module('bar.foo', 'tasks') is None
def test_find_related_module(self): with patch("importlib.import_module") as imp: with patch("imp.find_module") as find: imp.return_value = Mock() imp.return_value.__path__ = "foo" base.find_related_module(base, "tasks") imp.side_effect = AttributeError() base.find_related_module(base, "tasks") imp.side_effect = None find.side_effect = ImportError() base.find_related_module(base, "tasks")