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_single_file_input_runs_without_failures(runner, tmpdir):
import_line = 'from flask.ext.foo import bar'
temp_file = tmpdir.join('temp.py')
temp_file.write(import_line)
result = runner.invoke(startup, [str(temp_file)])
assert result.exit_code == 0
assert temp_file.read() == migrate.fix_tester(import_line)
def test_single_file_run_modifies_file_properly(runner, tmpdir):
import_line = 'from flask.ext.foo import bar'
temp_file = tmpdir.join('temp.py')
temp_file.write(import_line)
result = runner.invoke(startup, [str(temp_file)])
assert result.exit_code == 0
assert temp_file.read() == migrate.fix_tester(import_line)
def test_recursive_run_modifies_files_properly(runner, tmpdir):
import_line = 'from flask.ext.foo import bar'
expected_output = migrate.fix_tester(import_line)
temp_files = []
for x in range(2):
temp_files.append(tmpdir.join('temp{}.py'.format(x)))
for filepath in temp_files:
filepath.write(import_line)
result = runner.invoke(startup, [str(tmpdir)])
assert result.exit_code == 0
for filepath in temp_files:
assert filepath.read() == expected_output
def test_named_module_import():
output = migrate.fix_tester("import flask.ext.foo as foobar")
assert output == "import flask_foo as foobar"
def test_from_subpackages_named_import():
output = migrate.fix_tester(
"from flask.ext.foo.bar import foobar as foobaz")
assert output == "from flask_foo.bar import foobar as foobaz"
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)")
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_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():
output = migrate.fix_tester("flask.ext.foo(var)")
assert output == "flask_foo(var)"
def test_invalid_import_doesnt_raise():
try:
migrate.fix_tester("import adjfsjdn")
except Exception as e:
pytest.fail(e)
def test_base_non_flask_from_import_unchanged():
output = migrate.fix_tester("from foo import bar")
assert output == "from foo import bar"
def test_base_non_flask_from_import_doesnt_raise():
try:
migrate.fix_tester("from foo import bar")
except Exception as e:
pytest.fail(e)
def test_base_flask_from_import_unchanged():
output = migrate.fix_tester("from flask import Flask")
assert output == "from flask import Flask"
def test_base_flask_import_unchanged():
output = migrate.fix_tester("import flask")
assert output == "import flask"
def test_nested_function_call_migration():
output = migrate.fix_tester("import flask.ext.foo\n\n"
"flask.ext.foo.bar(var)")
assert output == ("import flask_foo\n\n" "flask_foo.bar(var)")
def test_named_from_import():
output = migrate.fix_tester("from flask.ext.foo import bar as baz")
assert output == "from flask_foo import bar as baz"
def test_parens_import():
output = migrate.fix_tester("from flask.ext.foo import (bar, foo, foobar)")
assert output == "from flask_foo import (bar, foo, foobar)"
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_invalid_import_unchanged():
output = migrate.fix_tester("import adjfsjdn")
assert output == "import adjfsjdn"
def test_multiline_from_subpackages_import():
output = migrate.fix_tester("from flask.ext.foo.bar import (foobar,\
foobarz,\
foobarred)")
assert output == "from flask_foo.bar import (foobar, foobarz, foobarred)"
def test_simple_from_import():
output = migrate.fix_tester("from flask.ext import foo")
assert output == "import flask_foo as foo"
def test_from_to_from_samename_import():
output = migrate.fix_tester("from flask.ext.foo import bar as bar")
assert output == "from flask_foo import bar"
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_from_to_from_samename_subpackages_import():
output = migrate.fix_tester("from flask.ext.foo.bar import baz as baz")
assert output == "from flask_foo.bar import baz"
def test_multiline_from_subpackages_import():
red = RedBaron("from flask.ext.foo.bar import (foobar,\
foobarz,\
foobarred)")
output = migrate.fix_tester(red)
assert output == "from flask_foo.bar import (foobar, foobarz, foobarred)"
def test_multiline_import():
output = migrate.fix_tester("from flask.ext.foo import \
bar,\
foobar,\
something")
assert output == "from flask_foo import bar, foobar, something"
def test_function_call_migration():
red = RedBaron("flask.ext.foo(var)")
output = migrate.fix_tester(red)
assert output == "flask_foo(var)"
def test_module_import():
output = migrate.fix_tester("import flask.ext.foo")
assert output == "import flask_foo"
def test_no_change_to_import():
red = RedBaron("from flask import Flask")
output = migrate.fix_tester(red)
assert output == "from flask import Flask"
def test_non_flask_import_unchanged():
output = migrate.fix_tester("import requests")
assert output == "import requests"