o g#@sv ddlZddlZddlmZddlmZddlmZddlmZe dZ Gdddej Z e d kr9edSdS) N)dedent) ExitStack)mock) import_helpercodec@sdeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ dS)TestInteractiveConsolecCst|_|dS)N)rZInteractiveConsoleconsolemock_sysselfr K/Users/admin/.pyenv/versions/3.10.0/lib/python3.10/test/test_code_module.pysetUps  zTestInteractiveConsole.setUpcCs t}||j|tjddd|_|td|_|td|_tjdt j t j d}|||_ t j t j urC|j j |j _ |j `|j `dS)Nz code.inputT)Zcreatezcode.sys.stdoutzcode.sys.stderrzcode.sys)wrapsspec)rZ addCleanupclose enter_contextrZpatchinfuncstdoutstderrrsyssysmod excepthook__excepthook__ps1ps2)r stackZprepatchr r r r s       zTestInteractiveConsole.mock_syscCsLtd|j_|j||jjdd|j_|j||jjddS)NFinishedz>>> z custom1> )EOFErrorr side_effectrinteract assertEqualrrr r r r test_ps1#   zTestInteractiveConsole.test_ps1cCsLtd|j_|j||jjdd|j_|j||jjddS)Nrz... z custom2> ) rrrrr r!rrrr r r r test_ps2+r#zTestInteractiveConsole.test_ps2cCsNddtdg|j_|jt|jjD]}dd|dvr"dSqt d)Nz 'antioch'rZantiochzno console stdout) rrrrr listr method_callsjoinAssertionErrorr callr r r test_console_stderr3s z*TestInteractiveConsole.test_console_stderrcCsHdtdg|j_|j|jjD]}dd|dvrdSqtd)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:dtdg|j_t}||j_|j| |j dS)Nzraise ValueError('')r) rrrrZMockrrrr Z assertTrueZcalled)r hookr r r test_sysexcepthookEs z)TestInteractiveConsole.test_sysexcepthookcCstd|j_|jjdd|t|jjd|jjd}||ddig|j td|j_|jjdd|t|jjd dS) NrZFoobannerrwrite)zFoo r% rrrrr r!lenrr(Z reset_mock)r Z banner_callr r r test_bannerMs    z"TestInteractiveConsole.test_bannercCstd|j_|jjdd|t|jjd|jjd}d}||d|fig|j td|j_|jjddd|t|jjd|j d }td|j_|jjd|d|t|jjd|jjd}|d }||d|figdS) Nrr%r2r6r&z"now exiting InteractiveConsole... r5)r3Zexitmsgu bye! ζж r7)r err_msgexpectedmessager r r test_exit_msg[s&       z$TestInteractiveConsole.test_exit_msgcCJdtdg|j_|jddd|jjD}td}| ||dS)Nz(raise ValueError('') from AttributeErrorrr%cs|] }d|dVqdSr%r&Nr).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(rZassertInr outputr<r r r test_cause_tbws  z$TestInteractiveConsole.test_cause_tbcCr?)Nztry: ham except: eggs rr%csr@rArBrCr r r rErFz9TestInteractiveConsole.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_tbs  z&TestInteractiveConsole.test_context_tbN)__name__ __module__ __qualname__rr r"r$r-r/r1r9r>rJrKr r r r r s   r__main__)rZunittesttextwrapr contextlibrrZ test.supportr import_modulerZTestCaserrLmainr r r r s