mirror of
https://github.com/Instagram/LibCST.git
synced 2025-12-23 10:35:53 +00:00
4f810dbc13
* Simplify command specifier parsing * Allow running codemods without configuring in YAML This enables codemodding things by just plonking a CodemodCommand class into any old importable module and running `python -m libcst.tool codemod -x some_module.SomeClass ...`
65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
#
|
|
# This source code is licensed under the MIT license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
#
|
|
|
|
|
|
import platform
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
from unittest import skipIf
|
|
|
|
from libcst._parser.entrypoints import is_native
|
|
from libcst.testing.utils import UnitTest
|
|
|
|
|
|
class TestCodemodCLI(UnitTest):
|
|
# pyre-ignore - no idea why pyre is complaining about this
|
|
@skipIf(platform.system() == "Windows", "Windows")
|
|
def test_codemod_formatter_error_input(self) -> None:
|
|
rlt = subprocess.run(
|
|
[
|
|
sys.executable,
|
|
"-m",
|
|
"libcst.tool",
|
|
"codemod",
|
|
"remove_unused_imports.RemoveUnusedImportsCommand",
|
|
# `ArgumentParser.parse_known_args()`'s behavior dictates that options
|
|
# need to go after instead of before the codemod command identifier.
|
|
"--python-version",
|
|
"3.6",
|
|
str(Path(__file__).parent / "codemod_formatter_error_input.py.txt"),
|
|
],
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
)
|
|
if not is_native():
|
|
self.assertIn(
|
|
"ParserSyntaxError: Syntax Error @ 14:11.",
|
|
rlt.stderr.decode("utf-8"),
|
|
)
|
|
else:
|
|
self.assertIn(
|
|
"error: cannot format -: Cannot parse: 13:10: async with AsyncExitStack() as stack:",
|
|
rlt.stderr.decode("utf-8"),
|
|
)
|
|
|
|
def test_codemod_external(self) -> None:
|
|
# Test running the NOOP command as an "external command"
|
|
# against this very file.
|
|
output = subprocess.check_output(
|
|
[
|
|
sys.executable,
|
|
"-m",
|
|
"libcst.tool",
|
|
"codemod",
|
|
"-x", # external module
|
|
"libcst.codemod.commands.noop.NOOPCommand",
|
|
str(Path(__file__)),
|
|
],
|
|
encoding="utf-8",
|
|
stderr=subprocess.STDOUT,
|
|
)
|
|
assert "Finished codemodding 1 files!" in output
|