o DDi@sddlmZddlmZmZmZddlmZmZddl m Z ddl m Z ddl mZddlmZGdd d ZGd d d eeZd S) )dedent)OptionalSequenceType) parse_modulePartialParserConfig)Codemod)CodemodContext)SkipFile)UnitTestc@seZdZUdZdZeeed<ede de fddZ de d e dd fd d Z d d d d dde de de de ede e de ee dede dd fddZd S) _CodemodTesta Mixin that can be added to a unit test framework in order to provide convenience features. This is provided as an internal-only feature so that CodemodTest can be used with other frameworks. This is necessary since we set a metaclass on our UnitTest implementation. . TRANSFORMdatareturncst|d}dtdtfddfdd|D}|r/|dd kr/|d d }|r/|dd ks!|rE|d d krE|d d }|rE|d d ks7d|}|dsS|dS|S) z Given a code string originting from a multi-line triple-quoted string, normalize the code using ``dedent`` and ensuring a trailing newline is present.  linercSst|dkr dS|S)Nr)lenstrip)rrP/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/codemod/_testing.py filter_line'sz3_CodemodTest.make_fixture_data..filter_linecsg|]}|qSrr).0rrrr -sz2_CodemodTest.make_fixture_data..rrN)rsplitstrjoinendswith)rlinescoderrrmake_fixture_datas    z_CodemodTest.make_fixture_dataexpectedactualNcCs|t|t|dS)a Given an expected and actual code string, makes sure they equal. This ensures that both the expected and actual are sanitized, so its safe to use this on strings that may have come from a triple-quoted multi-line string. N) assertEqual CodemodTestr#)selfr$r%rrrassertCodeEqual;s z_CodemodTest.assertCodeEqualF)context_overridepython_versionexpected_warnings expected_skipbeforeafterargsr*r+r,r-kwargsc Os|dur|nt} |j| g|Ri|} tt||dur#t|dntd} z| | } Wnty=|s9| } Ynw|rE|d| t|t| j |dur_| || j dSdS)a Given a before and after code string, and any args/kwargs that should be passed to the codemod constructor specified in :attr:`~CodemodTest.TRANSFORM`, validate that the codemod executes as expected. Verify that the codemod completes successfully, unless the ``expected_skip`` option is set to ``True``, in which case verify that the codemod skips. Optionally, a :class:`CodemodContext` can be provided. If none is specified, a default, empty context is created for you. Additionally, the python version for the code parser can be overridden to a valid python version string such as `"3.6"`. If none is specified, the version of the interpreter running your tests will be used. Also, a list of warning strings can be specified and :meth:`~CodemodTest.assertCodemod` will verify that the codemod generates those warnings in the order specified. If it is left out, warnings are not checked. N)r+)configz$Expected SkipFile but was not raised) r r rr'r#rtransform_moduler failr&r"assertSequenceEqualwarnings) r(r.r/r*r+r,r-r0r1contexttransform_instance input_tree output_treerrr assertCodemodIs0    z_CodemodTest.assertCodemod)__name__ __module__ __qualname____doc__r rr__annotations__ staticmethodrr#r)objectrr rboolr;rrrrr s:    r c@seZdZdZdS)r'aP Base test class for a :class:`Codemod` test. Provides facilities for auto-instantiating and executing a codemod, given the args/kwargs that should be passed to it. Set the :attr:`~CodemodTest.TRANSFORM` class attribute to the :class:`Codemod` class you wish to test and call :meth:`~CodemodTest.assertCodemod` inside your test method to verify it transforms various source code chunks correctly. Note that this is a subclass of ``UnitTest`` so any :class:`CodemodTest` can be executed using your favorite test runner such as the ``unittest`` module. N)r<r=r>r?rrrrr'sr'N)textwraprtypingrrrlibcstrrlibcst.codemod._codemodrlibcst.codemod._contextr libcst.codemod._runnerr libcst.testing.utilsr r r'rrrrs     p