o DDin( @sfddlZddlmZddlmZddlmZmZmZmZm Z m Z m Z ddl m Z ddlZddlmZmZddlmZddlmZmZdd lmZmZdd lmZdd lmZed d GdddZGdddeZ d!dej!dej!de e"ddfddZ#dedee"gej$ffddZ%dedee"gej&ffddZ'GdddeZ(ed d Gdd d ej!Z)dS)"N) ExitStack) dataclass)AnyCallableIterableListOptionalSequenceType)patch) CodegenStatevisit_required)CSTNodeT)CSTTransformer CSTVisitorT) CodeRangePositionProvider)PositionProvidingCodegenState)UnitTestT)frozenc@s0eZdZUeejed<eed<eded<dS)_CSTCodegenPatchTargettypename.N old_codegenN) __name__ __module__ __qualname__r cstCSTNode__annotations__strrr"r"Q/var/www/Datamplify/venv/lib/python3.10/site-packages/libcst/_nodes/tests/base.pyrs rc@s eZdZdS) _NOOPVisitorN)rrrr"r"r"r#r$sr$abmsgreturncCs<||s|dur dnd|}td|d||dS)z+ For use with addTypeEqualityFunc. N z is not deeply equal to ) deep_equalsAssertionError)r%r&r'suffixr"r"r#_cst_node_equality_func s r.configc dtdtjffdd }|S)Ncoder(ctj|tjdidSNr/r")rparse_expressionPartialParserConfigr1r4r"r#inner,z"parse_expression_as..inner)r!rBaseExpressionr/r8r"r4r#parse_expression_as+r<c r0)Nr1r(cr2r3)rparse_statementr6r7r4r"r#r83r9z!parse_statement_as..inner)r!r BaseStatementr;r"r4r#parse_statement_as2r=r@c @s,eZdZdddZ  d dededeeegefdeeddf d d Z d ege j fd eddfd dZ d ege j fd eddfddZ d!de j dedeeddfddZde j ddfddZde j dee j fddZde j ddfddZde j ddfddZdedeege j fdeddfddZdS)" CSTNodeTestr(NcCsBtjD]}t|trt|tjr||tq|t tdSN) r__dict__values isinstancer issubclassraddTypeEqualityFuncr.DummyIndentedBlock)selfvr"r"r#setUp;s  zCSTNodeTest.setUpnoder1parserexpected_positioncCsr||||||dur||}||||}t|tr(|j}t|ts ||||||dSrB) validate_types_deep_CSTNodeTest__assert_codegen assertEqualrErHchild+_CSTNodeTest__assert_children_match_codegen*_CSTNodeTest__assert_children_match_fields+_CSTNodeTest__assert_visit_returns_identity)rIrLr1rMrN parsed_nodeunwrapped_noder"r"r# validate_nodeEs    zCSTNodeTest.validate_nodeget_node expected_recCs:|tj| |WddS1swYdSrB)assertRaisesRegexrCSTValidationErrorrIrYrZr"r"r#assert_invalid\s"zCSTNodeTest.assert_invalidcCs<|t||WddS1swYdSrB)r[ TypeErrorvalidate_types_shallowr]r"r"r#assert_invalid_typesbs "z CSTNodeTest.assert_invalid_typesexpectedcCs`tg}|||||dur.t}t|j|j|d}||||j ||dSdS)zN Verifies that the given node's `_codegen` method is correct. N)default_indentdefault_newlineprovider) rModulerQ code_for_noderrrcrd_codegen _computed)rIrLrbrNmodulerestater"r"r#__assert_codegenhs  zCSTNodeTest.__assert_codegencCs$|j}||}|j||dddS)NzThe list of children we got from `node.children` differs from the children that were visited by `node._codegen`. This is probably due to a mismatch between _visit_and_replace_children and _codegen_impl.r')children*_CSTNodeTest__derive_children_from_codegenassertSequenceEqual)rIrLrncodegen_childrenr"r"r#__assert_children_match_codegens  z+CSTNodeTest.__assert_children_match_codegencsddtjD}ggdtdtdffdd }t%}|D]}|td|jd ||q"t g |Wd S1sGwYS) a Patches all subclasses of `CSTNode` exported by the `cst` module to track which `_codegen` methods get called, generating a list of children. Because all children must be rendered out into lexical order, this should be equivalent to `node.children`. `node.children` uses `_visit_and_replace_children` under the hood, not `_codegen`, so this helps us verify that both of those two method's behaviors are in sync. cSs@g|]\}}t|trt|tjrt|drt|||jdqS)rh)rrr)rErrFrrhasattrrrh).0krJr"r"r# s z>CSTNodeTest.__derive_children_from_codegen..targetr(rcs&dtdtdtddffdd }|S)NrIargskwargsr(csld}tdksd|ur tdkr||d}j|g|Ri||r4dSdS)NFrT)lenappendrpop)rIrxry should_pop)rn codegen_stackrwr"r# _codegen_impls    z`CSTNodeTest.__derive_children_from_codegen.._get_codegen_override.._codegen_impl)rr)rwrrnr)rwr#_get_codegen_overrides"zICSTNodeTest.__derive_children_from_codegen.._get_codegen_overridezlibcst.z ._codegenN) rrCitemsrrr enter_contextr rrfrg)rIrL patch_targetsr patch_stacktr"rr#__derive_children_from_codegens( z*CSTNodeTest.__derive_children_from_codegencCsdd|jD}t|}t}|D]%}t||j}t|tjr(| t |qt|t r7| dd|Dq|j ||dddS)a We expect `node.children` to match everything we can extract from the node's fields, but maybe in a different order. This asserts that those things match. If you want to verify order as well, use `assert_children_ordered`. cSsh|]}t|qSr")id)rtrRr"r"r# sz=CSTNodeTest.__assert_children_match_fields..css$|] }t|tjrt|VqdSrB)rErrr)rtelr"r"r# s   z=CSTNodeTest.__assert_children_match_fields..zA`node.children` doesn't match what we found through introspectionrmN)rn dataclassesfieldssetgetattrrrErraddrrupdateassertSetEqual)rIrLnode_children_idsrfield_child_idsfvaluer"r"r#__assert_children_match_fieldss"      z*CSTNodeTest.__assert_children_match_fieldscCs|||tdS)z When visit is called with a visitor that acts as a no-op, the visit method should return the same node it started with. N)rQvisitr$)rIrLr"r"r#__assert_visit_returns_identitysz+CSTNodeTest.__assert_visit_returns_identityexpect_successcCsJ|s|tj ||WddS1swYdS||dSrB) assertRaisesrParserSyntaxError)rIr1rMrr"r"r# assert_parsess  " zCSTNodeTest.assert_parses)r(N)NNrB)rrrrKrr!rrrrXrrr^rarPrSr rorTrUboolrr"r"r"r#rA:sp         9 rAc@sHeZdZUdZeed<ejed<deddfddZ d e ddfd d Z dS) rHz A stripped-down version of cst.IndentedBlock that only sets/clears the indentation state for the purpose of testing cst.IndentWhitespace in isolation. rrRrkr(NcCsX||j|j||j|jd|j|Wdn1s!wY|dS)N) start_nodeend_node)increase_indentrrecord_syntactic_positionrRrhdecrease_indent)rIrkr"r"r#rs   z DummyIndentedBlock._codegen_implvisitorcCst|jt|d|j|dS)NrR)rrR)rHrr rR)rIrr"r"r#_visit_and_replace_childrensz.DummyIndentedBlock._visit_and_replace_children) rrr__doc__r!r rrr rrrr"r"r"r#rHs  rHrB)*r contextlibrrtypingrrrrrr r unittest.mockr libcstrlibcst._nodes.internalr r libcst._typesrlibcst._visitorsrrlibcst.metadatarr!libcst.metadata.position_providerrlibcst.testing.utilsrrr$rr!r.r:r<r?r@rArHr"r"r"r#s>  $      C