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"