root
/
flask
mirror of https://github.com/pallets/flask.git
1
0
Fork 0
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Added more tests for the import hook

This commit is contained in:
Armin Ronacher 2011-09-18 18:16:50 +02:00
parent 933d203828
commit 9c82840f52
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
flask/testsuite/ext.py
View File

@ -17,7 +17,9 @@ class ExtImportHookTestCase(FlaskTestCase):
def setup(self):
for entry, value in sys.modules.items():
if entry.startswith('flask.ext.') and value is not None:
if (entry.startswith('flask.ext.') or
entry.startswith('flask_') or
entry.startswith('flaskext.')) and value is not None:
sys.modules.pop(entry, None)
from flask import ext
reload(ext)
@ -54,6 +56,10 @@ class ExtImportHookTestCase(FlaskTestCase):
self.assert_equal(newext_package.ext_id, 'newext_package')
self.assert_equal(newext_package.__name__, 'flask_newext_package')
def test_flaskext_new_package_import_submodule_function(self):
from flask.ext.newext_package.submodule import test_function
self.assert_equal(test_function(), 42)
def test_flaskext_new_package_import_submodule(self):
from flask.ext.newext_package import submodule
self.assert_equal(submodule.__name__, 'flask_newext_package.submodule')
@ -82,6 +88,10 @@ class ExtImportHookTestCase(FlaskTestCase):
self.assert_equal(submodule.__name__, 'flaskext.oldext_package.submodule')
self.assert_equal(submodule.test_function(), 42)
def test_flaskext_old_package_import_submodule_function(self):
from flask.ext.oldext_package.submodule import test_function
self.assert_equal(test_function(), 42)
def suite():
suite = unittest.TestSuite()