sentry/tests/tools/mypy_helpers/test_sort_stronger_modules.py

from tools.mypy_helpers.sort_stronger_modules import main


def test_sort_stronger_modules(tmp_path):
    src = """\
# before

# begin: stronger typing
[[tool.mypy.overrides]]
module = [
    "mod2",
    "mod1",
    "mod3",
]
some_setting = true
# end: stronger typing

# after
"""
    expected = """\
# before

# begin: stronger typing
[[tool.mypy.overrides]]
module = [
    "mod1",
    "mod2",
    "mod3",
]
some_setting = true
# end: stronger typing

# after
"""

    f = tmp_path.joinpath("f")
    f.write_text(src)

    assert main((str(f),)) == 1

    assert f.read_text() == expected

    assert main((str(f),)) == 0


def test_removes_duplicates(tmp_path):
    src = """\
# begin: stronger typing
[[tool.mypy.overrides]]
module = [
    "mod1",
    "mod1",
]
some_setting = true
# end: stronger typing
"""
    expected = """\
# begin: stronger typing
[[tool.mypy.overrides]]
module = [
    "mod1",
]
some_setting = true
# end: stronger typing
"""
    f = tmp_path.joinpath("f")
    f.write_text(src)

    assert main((str(f),)) == 1

    assert f.read_text() == expected