o GXîh#ã@sxdZddlZddlZddlmZddlmZddlmZddlm Z e   d¡Z Gdd „d ej ƒZ ed kr:e ¡dSdS) zCTest InteractiveConsole and InteractiveInterpreter from code moduleéN)Údedent)Ú ExitStack)Úmock)Ú import_helperÚcodec@sdeZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dd„Z dS)ÚTestInteractiveConsolecCst ¡|_| ¡dS)N)rZInteractiveConsoleÚconsoleÚmock_sys©Úself©r ú?/opt/python-3.10.19/usr/lib/python3.10/test/test_code_module.pyÚsetUps  zTestInteractiveConsole.setUpcCs”tƒ}| |j¡| tjddd¡|_| t d¡¡|_| t d¡¡|_tjdt j t j d}| |¡|_ t j t j urB|j j |j _ |j `|j `dS) z.Mock system environment for InteractiveConsolez code.inputT)Zcreatezcode.sys.stdoutzcode.sys.stderrzcode.sys)ÚwrapsÚspecN)rZ addCleanupÚcloseÚ enter_contextrZpatchÚinfuncÚstdoutÚstderrrÚsysÚsysmodÚ excepthookÚ__excepthook__Úps1Úps2)r ÚstackZprepatchr r r r s   ÿ    zTestInteractiveConsole.mock_syscCsLtdƒ|j_|j ¡| |jjd¡d|j_|j ¡| |jjd¡dS)NÚFinishedz>>> z custom1> )ÚEOFErrorrÚ side_effectrÚinteractÚ assertEqualrrr r r r Útest_ps1#ó   zTestInteractiveConsole.test_ps1cCsLtdƒ|j_|j ¡| |jjd¡d|j_|j ¡| |jjd¡dS)Nrz... z custom2> ) rrrrr r!rrrr r r r Útest_ps2+r#zTestInteractiveConsole.test_ps2cCsNddtdƒg|j_|j ¡t|jjƒD]}dd |d¡vr"dSqt dƒ‚)Nz 'antioch'ÚrZantiochézno console stdout) rrrrr ÚlistrÚ method_callsÚjoinÚAssertionError©r Úcallr r r Útest_console_stderr3s ÿz*TestInteractiveConsole.test_console_stderrcCsHdtdƒg|j_|j ¡|jjD]}dd |d¡vrdSqtdƒ‚)NZ undefinedrÚ NameErrorr%r&zNo syntax error from console) rrrrr rr(r)r*r+r r r Útest_syntax_error<s  ÿz(TestInteractiveConsole.test_syntax_errorcCs:dtdƒg|j_t ¡}||j_|j ¡|  |j ¡dS)Nzraise ValueError('')r) rrrrZMockrrrr Z assertTrueZcalled)r Úhookr r r Útest_sysexcepthookEsÿ z)TestInteractiveConsole.test_sysexcepthookcCsˆtdƒ|j_|jjdd| t|jjƒd¡|jjd}| |ddig¡|j  ¡tdƒ|j_|jjdd| t|jjƒd ¡dS) NrZFoo©ÚbannerérÚwrite)zFoo r%é© rrrrr r!Úlenrr(Z reset_mock)r Z banner_callr r r Ú test_bannerMs    z"TestInteractiveConsole.test_bannercCsötdƒ|j_|jjdd| t|jjƒd¡|jjd}d}| |d|fig¡|j  ¡tdƒ|j_|jjddd| t|jjƒd¡|j  ¡d }tdƒ|j_|jjd|d| t|jjƒd¡|jjd}|d }| |d|fig¡dS) Nrr%r2r6r&z"now exiting InteractiveConsole... r5)r3Zexitmsgu bye! ζжÚ r7)r Úerr_msgÚexpectedÚmessager r r Ú test_exit_msg[s&     ÿ  z$TestInteractiveConsole.test_exit_msgcCóJdtdƒg|j_|j ¡d dd„|jjDƒ¡}tdƒ}|  ||¡dS)Nz(raise ValueError('') from AttributeErrorrr%csó|] }d |d¡VqdS©r%r&N©r)©Ú.0r,r r r Ú {ó€z7TestInteractiveConsole.test_cause_tb..zÝ AttributeError The above exception was the direct cause of the following exception: Traceback (most recent call last): File "", line 1, in ValueError © rrrrr r)rr(rZassertIn©r Úoutputr<r r r Ú test_cause_tbwsÿ  z$TestInteractiveConsole.test_cause_tbcCr?)Nztry: ham except: eggs rr%csr@rArBrCr r r rE‹rFz9TestInteractiveConsole.test_context_tb..ah Traceback (most recent call last): File "", line 1, in NameError: name 'ham' is not defined During handling of the above exception, another exception occurred: Traceback (most recent call last): File "", line 2, in NameError: name 'eggs' is not defined rGrHr r r Útest_context_tb‡sÿ  z&TestInteractiveConsole.test_context_tbN)Ú__name__Ú __module__Ú __qualname__rr r"r$r-r/r1r9r>rJrKr r r r r s   rÚ__main__)Ú__doc__rZunittestÚtextwraprÚ contextlibrrZ test.supportrÚ import_modulerZTestCaserrLÚmainr r r r Ús      ÿ