U
|
Þ=®d¶) ã @ sî d dl mZmZ d dlmZmZ zd dlmZ W n ek
|
rP d dl mZ Y nX zd dl
|
m Z m Z m Z W n( ek
|
r d dlm Z m Z m Z Y nX ddlmZmZ ddlmZ dd lmZ G d
|
d d eZG d d d eZedkrêe dS )é )ÚStringIOÚ TextIOWrapper)ÚTestCaseÚmain)Ú ExitStack)Ú MagicMockÚMockÚpatché )Ú AnsiToWin32Ú StreamWrapper©Ú"ENABLE_VIRTUAL_TERMINAL_PROCESSINGé )Úosnamec @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )ÚStreamWrapperTestc C s& t }t|d }| |j|jk¡ d S ©N)r r Ú
|
assertTrueZ random_attr)ÚselfÚ
|
mockStreamÚwrapper© r úVd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\colorama/tests/ansitowin32_test.pyÚ testIsAProxy s
|
z StreamWrapperTest.testIsAProxyc C s8 t }t }t||}| d¡ | |jjdi f¡ d S )NÚhello)r )r r Úwriter Ú call_args)r r Ú mockConverterr r r r ÚtestDelegatesWrite s
|
|
|
z$StreamWrapperTest.testDelegatesWritec C s< t }t }t||}| d¡ W 5 Q R X | |j¡ d S ©Nr )r r r r r Úclosed)r r ÚsÚfpr r r ÚtestDelegatesContext" s
|
z&StreamWrapperTest.testDelegatesContextc
|
C sP t }t |j_t }| t¡&}t||}| d¡ W 5 Q R X W 5 Q R X d S r )r ÚAttributeErrorÚ __enter__Z side_effectr Ú assertRaisesr r )r r r Úexcinfor r r r ÚtestProxyNoContextManager) s
|
z+StreamWrapperTest.testProxyNoContextManagerc C s* t }| ¡ t|d }| |jd¡ d S ©NT)r Úcloser Ú assertEqualr ©r Ústreamr r r r Ú+test_closed_shouldnt_raise_on_closed_stream1 s
|
z=StreamWrapperTest.test_closed_shouldnt_raise_on_closed_streamc C s. t t }| ¡ t|d }| |jd¡ d S r) )r r Údetachr r+ r r, r r r Ú-test_closed_shouldnt_raise_on_detached_stream7 s
|
|
z?StreamWrapperTest.test_closed_shouldnt_raise_on_detached_streamN) Ú__name__Ú
|
__module__Ú __qualname__r r r# r( r. r0 r r r r r s r c @ s¶ e Zd Zdd Zeddeddd dd Zd
|
d Zd d Zdd Zd+ddZ dd Z
|
dd Z dd Z dd Z dd Zd d Zd d Zd!d" Zd#d$ Zd%d& Zd'd( Zd)d* ZdS ),ÚAnsiToWin32Testc C s8 t }t }t||d}| |j|¡ | |j|¡ d S )N)Ú autoreset)r r r+ Úwrappedr5 )r Ú
|
mockStdoutÚautor- r r r ÚtestInit? s
|
zAnsiToWin32Test.testInitú colorama.ansitowin32.wintermNú colorama.ansitowin32.winapi_testc G s dS r) r ©Ú_r r r Ú<lambda>G ó zAnsiToWin32Test.<lambda>c C s2 t d t }t|}| |j¡ W 5 Q R X d S )NÚnt)r r r r Ústrip©r r7 r- r r r ÚtestStripIsTrueOnWindowsF s
|
z(AnsiToWin32Test.testStripIsTrueOnWindowsc C s6 t d$ tdd}t|}| |j¡ W 5 Q R X d S )NÚposixF)r )r r r Ú assertFalserA rB r r r ÚtestStripIsFalseOffWindowsN s
|
|
z*AnsiToWin32Test.testStripIsFalseOffWindowsc C sV t }t|}t |_t |_d|_| d¡ | |jjj¡ | |jj di f¡ d S )NTÚabc©rG )
|
r r r6 Úwrite_and_convertrA r rE Úcalledr+ r rB r r r ÚtestWriteStripsAnsiT s
|
z#AnsiToWin32Test.testWriteStripsAnsic C s\ t }t|}t |_t |_d|_d|_| d¡ | |jj¡ | |jjj
|
di f¡ d S )NFrG rH ) r r r6 rI rA Úconvertr rE rJ r+ r rB r r r ÚtestWriteDoesNotStripAnsi` s
|
z)AnsiToWin32Test.testWriteDoesNotStripAnsiTc C sD t t }||_t |_||_t |_| d¡ | |jj|¡ d S )NrG ) r r rL Ú reset_allr5 Úwintermr r+ rJ )r rL r5 r- r r r Úassert_autoresetsm s
|
|
z!AnsiToWin32Test.assert_autoresetsc C s8 | j dd | j dd | j ddd | j ddd d S )NT©rL F)rL r5 )rP )r r r r ÚtestWriteAutoresetsx s z#AnsiToWin32Test.testWriteAutoresetsc C s. t t }| d¡ | |jjjdi f¡ d S )NrG rH )r r rI r+ r6 r r ©r r- r r r Ú"testWriteAndConvertWritesPlainText~ s
|
|
z2AnsiToWin32Test.testWriteAndConvertWritesPlainTextc C st t t }t |_ddddddddd d
|
d d d g }|D ]:}|jj ¡ | |¡ | dd |jjjD ddg¡ q4d S )Nz abc[mdefz
|
abc[0mdefz
|
abc[2mdefz abc[02mdefz abc[002mdefz abc[40mdefz abc[040mdefz abc[0;1mdefzabc[40;50mdefzabc[50;30;40mdefz abc[Adefz
|
abc[0Gdefúabc[1;20;128Hdefc S s g | ] }|d qS ©r r )Ú.0Úargsr r r Ú
|
<listcomp> s zIAnsiToWin32Test.testWriteAndConvertStripsAllValidAnsi.<locals>.<listcomp>rH )Údef) r r Ú
|
call_win32r6 r Ú
|
reset_mockrI r+ Úcall_args_list)r r- ÚdataÚdatumr r r Ú%testWriteAndConvertStripsAllValidAnsi s.
|
ó
|
þz5AnsiToWin32Test.testWriteAndConvertStripsAllValidAnsic C s0 t t }t |_| d¡ | |jjj¡ d S )Nz
|
[40m[41m)r r r[ rI rE r6 r rJ rS r r r Ú%testWriteAndConvertSkipsEmptySnippets s
|
|
z5AnsiToWin32Test.testWriteAndConvertSkipsEmptySnippetsc C sr t t }d|_t |_tdd|_dddddd }| ¡ D ]0\}}|j ¡ | |¡ | |jj d
|
|¡ q<d S ) NTÚparams)Ú return_value)Úarb )Úbrb )Úcrb )ÚGrb )ÚHrb )z abc[adefz abc[;;bdefz
|
abc[0cdefzabc[;;0;;GdefrU r )
|
r r rL r[ Úextract_paramsÚitemsr\ rI r+ r ©r r- r^ r_ Úexpectedr r r Ú1testWriteAndConvertCallsWin32WithParamsAndCommand£ s
|
û
|
|
zAAnsiToWin32Test.testWriteAndConvertCallsWin32WithParamsAndCommandc C s" t }t|}| ¡ | ¡ d S r )r r r* rN ©r r- Ú converterr r r Ú3test_reset_all_shouldnt_raise_on_closed_orig_stdout´ s zCAnsiToWin32Test.test_reset_all_shouldnt_raise_on_closed_orig_stdoutc
|
C sb t }| ¡ tdd( tddd t|}W 5 Q R X W 5 Q R X | |j¡ | |j¡ d S )Nú colorama.ansitowin32.os.namer@ r; c S s dS r) r r r r r r> À r? zPAnsiToWin32Test.test_wrap_shouldnt_raise_on_closed_orig_stdout.<locals>.<lambda>)r r* r r r rA rE rL rn r r r Ú.test_wrap_shouldnt_raise_on_closed_orig_stdout» s ÿ þ z>AnsiToWin32Test.test_wrap_shouldnt_raise_on_closed_orig_stdoutc
|
C sV t dd* t ddd tt }W 5 Q R X W 5 Q R X | |j¡ | |j¡ d S )Nrq r@ r; c S s dS r) r r r r r r> È r? zQAnsiToWin32Test.test_wrap_shouldnt_raise_on_missing_closed_attr.<locals>.<lambda>)r r Úobjectr rA rE rL )r ro r r r Ú/test_wrap_shouldnt_raise_on_missing_closed_attrÅ s ÿ þ z?AnsiToWin32Test.test_wrap_shouldnt_raise_on_missing_closed_attrc C sH t t }dddddddd}| ¡ D ] \}}| | d|¡|¡ q&d S )NrV )r
|
)r r )é iÈ )é é é! é, é7 )Ú z;;Ú2z;;002;;z0;1z ;;003;;456;;z11;22;33;44;55Úm)r r rj r+ ri rk r r r ÚtestExtractParamsÍ s
|
ù z!AnsiToWin32Test.testExtractParamsc sh t t } fddf fddf fddfd|_| dd¡ | dd jD d
|
d d g¡ d S ) Nc s dS )Nrv r ©r= Ú__©Zlistenerr r r> ß r? z9AnsiToWin32Test.testCallWin32UsesLookup.<locals>.<lambda>c s dS )Nrw r r r r r r> à r? c s dS )Nrx r r r r r r> á r? )r r
|
ru r} )ru r éc r
|
c S s g | ]}|d d qS rV r )rW rd r r r rY å s z;AnsiToWin32Test.testCallWin32UsesLookup.<locals>.<listcomp>rx rv rw )r r Ú win32_callsr[ r+ r] rS r r r ÚtestCallWin32UsesLookupÛ s ý þz'AnsiToWin32Test.testCallWin32UsesLookupc C s^ t }t|dd}td:}dddddd g}|D ]}| |¡ q0| |jjd
|
¡ W 5 Q R X d S ) NTrQ r: z]0z]0;fooz]0;colorama_test_titlez]1;colorama_test_titlez]2;colorama_test_titlezC];;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;r
|
)r r r r r+ Ú set_titleZ
|
call_count)r r7 r- rO r^ Úcoder r r Útest_osc_codesè s
|
ú z AnsiToWin32Test.test_osc_codesc s. t fdd}|dd |ddd |dd d |d
|
d |d d d |ddd t }|d| t }d|_d|j_d|j_t|}| dt¡ | |j
|
¡ | |j ¡ | | ¡ ¡ |ddd t }|d| t|}| dt¡ | |j
|
¡ | |j ¡ | | ¡ ¡ W 5 Q R X d S )Nc s t| |dd¡ d S )NT)Úcreate)Ú enter_contextr )rd re ©Ústackr r Úpú s z3AnsiToWin32Test.test_native_windows_ansi.<locals>.prq r@ r; c S s dS r) r r r r r r> þ r? z:AnsiToWin32Test.test_native_windows_ansi.<locals>.<lambda>zcolorama.win32.winapi_testc S s dS r) r r r r r r> ÿ r? z colorama.winterm.win32.windllznon-Nonez colorama.winterm.get_osfhandlec S s dS )NéÒ r r< r r r r> r? z%colorama.winterm.win32.GetConsoleModec S s t S r r r< r r r r> r? z%colorama.winterm.win32.SetConsoleModeFTr r c S s dS )Nr r r< r r r r> r? )r r r Úisattyrc Úfilenor Zassert_called_withr rE rA rL Ú should_wrapr )r r ZSetConsoleModeÚstdoutr- r r r Útest_native_windows_ansiø s<
|
|
|
þ
|
|
z(AnsiToWin32Test.test_native_windows_ansi)T)r1 r2 r3 r9 r rC rF rK rM rP rR rT r` ra rm rp rr rt r~ r r r r r r r r4 = s(
|
|
r4 Ú__main__N)Úior r Zunittestr r Ú
|
contextlibr Ú ImportErrorÚ contextlib2Z unittest.mockr r r ZmockÚ ansitowin32r r Úwin32r Úutilsr r r4 r1 r r r r Ú<module> s" ) i
|
