B u9aã@stdZddlZddlZddlmZddlmZddlmZddlm Z e   d¡Z Gdd „d ej ƒZ ed krpe ¡dS) zCTest InteractiveConsole and InteractiveInterpreter from code moduleéN)Údedent)Ú ExitStack)Úmock)ÚsupportÚ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 ú&/usr/lib/python3.7/test_code_module.pyÚsetUps zTestInteractiveConsole.setUpcCs”tƒ}| |j¡| tjddd¡|_| t d¡¡|_| t d¡¡|_tjdt j t j d}| |¡|_ t j t j kr„|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Ú assertEqualrr)r r r r Útest_ps1"s    zTestInteractiveConsole.test_ps1cCsLtdƒ|j_|j ¡| |jjd¡d|j_|j ¡| |jjd¡dS)Nrz... z custom2> ) rrrrrr rrr)r r r r Útest_ps2*s    zTestInteractiveConsole.test_ps2cCsRddtdƒg|j_|j ¡x0t|jjƒD]}dd |d¡kr*Pq*Wt dƒ‚dS)Nz 'antioch'ÚrZantiochézno console stdout) rrrrrÚlistrÚ method_callsÚjoinÚAssertionError)r Úcallr r r Útest_console_stderr2s  z*TestInteractiveConsole.test_console_stderrcCsLdtdƒg|j_|j ¡x,|jjD]}dd |d¡kr$Pq$Wtdƒ‚dS)NZ undefinedrÚ NameErrorr#r$zNo syntax error from console) rrrrrrr&r'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) rrrrZMockrrrrZ assertTrueZcalled)r Úhookr r r Útest_sysexcepthookDs  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#é) rrrrrr Úlenrr&Ú reset_mock)r Z banner_callr r r Ú test_bannerLs    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#)r/r2r$z"now exiting InteractiveConsole... r1)r/Zexitmsgu bye! ζжÚ ) rrrrrr r3rr&r4)r Zerr_msgÚexpectedÚmessager r r Ú test_exit_msgZs$       z$TestInteractiveConsole.test_exit_msgcCsJdtdƒg|j_|j ¡d dd„|jjDƒ¡}tdƒ}|  ||¡dS)Nz(raise ValueError('') from AttributeErrorrr#css|]}d |d¡VqdS)r#r$N)r')Ú.0r)r r r ú zsz7TestInteractiveConsole.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 ) rrrrrr'rr&rÚassertIn)r Úoutputr7r r r Ú test_cause_tbvs z$TestInteractiveConsole.test_cause_tbcCsJdtdƒg|j_|j ¡d dd„|jjDƒ¡}tdƒ}|  ||¡dS)Nztry: ham except: eggs rr#css|]}d |d¡VqdS)r#r$N)r')r:r)r r r r;Šsz9TestInteractiveConsole.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 ) rrrrrr'rr&rr<)r r=r7r r r Útest_context_tb†s  z&TestInteractiveConsole.test_context_tbN)Ú__name__Ú __module__Ú __qualname__r r r!r"r*r,r.r5r9r>r?r r r r r s  rÚ__main__)Ú__doc__rZunittestÚtextwraprÚ contextlibrrZtestrÚ import_modulerZTestCaserr@Úmainr r r r Ús