mirror of https://github.com/mitsuhiko/flask.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
2.1 KiB
71 lines
2.1 KiB
# Tester for the flaskext_migrate.py module located in flask/scripts/ |
|
# |
|
# Author: Keyan Pishdadian |
|
import pytest |
|
from redbaron import RedBaron |
|
import flaskext_migrate as migrate |
|
|
|
|
|
def test_simple_from_import(): |
|
red = RedBaron("from flask.ext import foo") |
|
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_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_tester(red) |
|
assert output == "from flask_foo import bar, foobar, something" |
|
|
|
|
|
def test_multiline_import(): |
|
red = RedBaron("from flask.ext.foo import \ |
|
bar,\ |
|
foobar,\ |
|
something") |
|
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_tester(red) |
|
assert output == "import flask_foo" |
|
|
|
|
|
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_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_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)" |
|
|
|
|
|
def test_nested_function_call_migration(): |
|
red = RedBaron("import flask.ext.foo\n\n" |
|
"flask.ext.foo.bar(var)") |
|
output = migrate.fix_tester(red) |
|
assert output == ("import flask_foo\n\n" |
|
"flask_foo.bar(var)")
|
|
|