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

Add support for function call fixing, add tests

Browse source 
Addresses #1135, some code cleanup and refactoring. Changes wrapper function which handles testing, further modularized code, added test to cover function call fixing, and fixed duplicate test function name.
This commit is contained in:
Keyan Pishdadian 2015-02-13 15:27:29 -05:00
parent 9cbe83ef0d
commit 4cb311b945
2 changed files with 56 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

30
scripts/test_import_migration.py
View file

@ -8,19 +8,19 @@ import flaskext_migrate as migrate
def test_simple_from_import():
red = RedBaron("from flask.ext import foo")
output = migrate.fix(red)
output = migrate.fix_tester(red)
assert output == "import flask_foo as foo"
def test_from_to_from_import():
red = RedBaron("from flask.ext.foo import bar")
output = migrate.fix(red)
output = migrate.fix_tester(red)
assert output == "from flask_foo import bar as bar"
def test_multiple_import():
red = RedBaron("from flask.ext.foo import bar, foobar, something")
output = migrate.fix(red)
output = migrate.fix_tester(red)
assert output == "from flask_foo import bar, foobar, something"
@ -29,23 +29,35 @@ def test_multiline_import():
bar,\
foobar,\
something")
output = migrate.fix(red)
output = migrate.fix_tester(red)
assert output == "from flask_foo import bar, foobar, something"
def test_module_import():
red = RedBaron("import flask.ext.foo")
output = migrate.fix(red)
assert output == "import flask_foo"
output = migrate.fix_tester(red)
assert output == "import flask_foo as foo"
def test_module_import():
def test_named_module_import():
red = RedBaron("import flask.ext.foo as foobar")
output = migrate.fix_tester(red)
assert output == "import flask_foo as foobar"
def test__named_from_import():
red = RedBaron("from flask.ext.foo import bar as baz")
output = migrate.fix(red)
output = migrate.fix_tester(red)
assert output == "from flask_foo import bar as baz"
def test_parens_import():
red = RedBaron("from flask.ext.foo import (bar, foo, foobar)")
output = migrate.fix(red)
output = migrate.fix_tester(red)
assert output == "from flask_foo import (bar, foo, foobar)"
def test_function_call_migration():
red = RedBaron("flask.ext.foo(var)")
output = migrate.fix_tester(red)
assert output == "flask_foo(var)"