U
|
Þ=®dU ã @ sÄ d dl Z d dlmZmZmZ zd dlmZmZ W n$ ek
|
rT d dl mZmZ Y nX ddl
|
m Z ddl m Z mZmZ ddlmZmZ e jZe jZG d d
|
d
|
eZG d d d eZed krÀe dS )é N)ÚTestCaseÚmainÚ
|
skipUnless)ÚpatchÚMocké )Ú StreamWrapper)ÚinitÚjust_fix_windows_consoleÚ _wipe_internal_state_for_testsé )ÚosnameÚ
|
replace_byc @ sæ e Zd Zeej ¡ ddd Zdd Zdd Z dd Z
|
e d
|
e d d d e ddd dd Z e d
|
e d dd dd Z dd Zdd Zdd Zdd Zd d Ze d e d d!d" Ze d d#d$ Ze d%d&d' Zd(S ))ÚInitTestzsys.stdout is not a ttyc C s | ¡ d S ©N)ÚassertNotWrapped©Úself© r úUd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\colorama/tests/initialise_test.pyÚsetUp s zInitTest.setUpc C s t tt_tt_d S r ©r Ú orig_stdoutÚsysÚstdoutÚ orig_stderrÚstderrr r r r ÚtearDown s zInitTest.tearDownc C sL | tjtd¡ | tjtd¡ | ttjtd¡ | ttjtd¡ d S )Nzstdout should be wrappedzstderr should be wrappedzbad stdout wrapperzbad stderr wrapper) Ú assertIsNotr r r r r Ú
|
assertTrueÚ
|
isinstancer r r r r Ú assertWrapped s ÿÿzInitTest.assertWrappedc C s$ | tjtd¡ | tjtd¡ d S )Nz stdout should not be wrappedz stderr should not be wrapped)ÚassertIsr r r r r r r r r r & s zInitTest.assertNotWrappedz colorama.initialise.reset_allú colorama.ansitowin32.winapi_testc G s dS ©NTr ©Ú_r r r Ú<lambda>+ ó zInitTest.<lambda>ú)colorama.ansitowin32.enable_vt_processingc G s dS ©NFr r% r r r r' , r( c C s& t d t | ¡ W 5 Q R X d S ©NÚnt©r r r! ©r r& r r r ÚtestInitWrapsOnWindows* s
|
z InitTest.testInitWrapsOnWindowsc G s dS r* r r% r r r r' 3 r( c C s& t d t | ¡ W 5 Q R X d S r+ ©r r r r. r r r Ú#testInitDoesntWrapOnEmulatedWindows2 s
|
z,InitTest.testInitDoesntWrapOnEmulatedWindowsc C s& t d t | ¡ W 5 Q R X d S )NÚposixr0 r r r r Ú testInitDoesntWrapOnNonWindows9 s
|
z'InitTest.testInitDoesntWrapOnNonWindowsc C s6 t d $ t | tj¡ | tj¡ W 5 Q R X d S r )r r Z assertIsNoner r r r r r r ÚtestInitDoesntWrapIfNone> s
|
z!InitTest.testInitDoesntWrapIfNonec C s* t d tdd | ¡ W 5 Q R X d S )Nr2 T©Ú autoresetr- r r r r Ú&testInitAutoresetOnWrapsOnAllPlatformsF s
|
|
z/InitTest.testInitAutoresetOnWrapsOnAllPlatformsc C s* t d tdd | ¡ W 5 Q R X d S )Nr, F)Úwrapr0 r r r r Ú"testInitWrapOffDoesntWrapOnWindowsK s
|
|
z+InitTest.testInitWrapOffDoesntWrapOnWindowsc C s | tdd ¡ d S )Nc S s t dddS )NTF)r6 r8 )r r r r r r' Q r( zEInitTest.testInitWrapOffIncompatibleWithAutoresetOn.<locals>.<lambda>)Ú assertRaisesÚ
|
ValueErrorr r r r Ú*testInitWrapOffIncompatibleWithAutoresetOnP s z3InitTest.testInitWrapOffIncompatibleWithAutoresetOnz&colorama.win32.SetConsoleTextAttributez colorama.initialise.AnsiToWin32c C sh t dV tdd | t|jd¡ | |jd d d d¡ | |jd d d d¡ W 5 Q R X d S )Nr, Tr5 r r r6 r ©r r Ú assertEqualÚlenZcall_args_list)r Ú mockATW32r& r r r ÚtestAutoResetPassedOnS s
|
|
|
z InitTest.testAutoResetPassedOnc C sº t d¨ t tdd | t|jd¡ | |jd d d d¡ | |jd d d d¡ t | t|jd ¡ | |jd d d d
|
¡ | |jd d d d
|
¡ W 5 Q R X d S ) Nr, Tr5 é r r r6 é é Fé r= )r r@ r r r ÚtestAutoResetChangeable\ s
|
|
ÿ ÿz InitTest.testAutoResetChangeablez#colorama.initialise.atexit.registerc C s0 t | |j¡ | ¡ t | |j¡ d S r )r r ÚcalledZ
|
reset_mockZ assertFalse)r Z mockRegisterr r r Ú testAtexitRegisteredOnlyOncen s
|
z%InitTest.testAtexitRegisteredOnlyOnceN)Ú__name__Ú
|
__module__Ú __qualname__r r r Úisattyr r r! r r r/ r1 r3 r4 r7 r9 r< rA rF rH r r r r r s0
|
|
r c @ s2 e Zd Zdd Zdd Zeddd dd Zd
|
S ) ÚJustFixWindowsConsoleTestc C s t tt_tt_d S r r r r r r Ú_resetx s z JustFixWindowsConsoleTest._resetc C s | ¡ d S r )rN r r r r r } s z"JustFixWindowsConsoleTest.tearDownr# c C s dS r$ r r r r r r' r( z"JustFixWindowsConsoleTest.<lambda>c
|
s" t jdkr.t | t jt¡ | t jt¡ nðdd }dD ]â td fddÆ | ¡ | t j}t j}t | t j|¡ r| t j|¡ n|
|
t j|¡ t j}t j}t | t j|¡ | t j|¡ | ¡ | t t j}t j}t | |t j¡ | |t j¡ W 5 Q R X q:d S )NÚwin32c S sH t } d| _d| j_d| j_| t_t }d|_d|j_d|j_|t_d S )NFr Tr )r ÚclosedrL Z return_valueÚfilenor r r )r r r r r Úfake_std s zEJustFixWindowsConsoleTest.testJustFixWindowsConsole.<locals>.fake_std)FTr) c s S r r r% ©Z native_ansir r r' r( zEJustFixWindowsConsoleTest.testJustFixWindowsConsole.<locals>.<lambda>) r Úplatformr
|
r" r r r r r rN r r )r rR Z prev_stdoutZ prev_stderrr rS r ÚtestJustFixWindowsConsole s@
|
|
þz3JustFixWindowsConsoleTest.testJustFixWindowsConsoleN)rI rJ rK rN r r rU r r r r rM w s rM Ú__main__)r Zunittestr r r Z unittest.mockr r Ú ImportErrorZmockÚ ansitowin32r Ú
|
initialiser r
|
r Úutilsr r r r r r r rM rI r r r r Ú<module> s eE
|
