maya-chen/flask
1
0
Fork 0
forked from orbit-oss/flask
Code Pull requests Activity

Add test for naming module and fix logic to cover

Browse source
This commit is contained in:
Keyan Pishdadian 2015-02-12 11:58:38 -05:00
parent 37217e85b8
commit b759aa2b95
2 changed files with 32 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/test_import_migration.py
View file

@ -1,7 +1,8 @@
# Tester for the flaskext_migrate.py module located in flask/scripts/
#
# Author: Keyan Pishdadian
import sys
sys.path.append('scripts')
import pytest
from redbaron import RedBaron
import flaskext_migrate as migrate
@ -16,7 +17,7 @@ def test_simple_from_import():
def test_from_to_from_import():
red = RedBaron("from flask.ext.foo import bar")
output = migrate.fix(red)
assert output == "from flask_foo import bar"
assert output == "from flask_foo import bar as bar"
def test_multiple_import():
@ -38,3 +39,9 @@ def test_module_import():
red = RedBaron("import flask.ext.foo")
output = migrate.fix(red)
assert output == "import flask_foo"
def test_module_import():
red = RedBaron("from flask.ext.foo import bar as baz")
output = migrate.fix(red)
assert output == "from flask_foo import bar as baz"