Format: application/x-python-code
1 o
2
3 1'h�) � @ s� d dl mZmZ d dlmZmZ zd dlmZ W n ey' d dl mZ Y nw z d dl
4 m Z m Z m
5 Z
6 W n eyG d dlm Z m Z m
7 Z
8 Y nw ddlmZmZ ddlmZ dd lmZ G d
9 d � d e�ZG d d
10 � d
11 e�Zedkrue� dS dS )� )�StringIO�
12 TextIOWrapper)�TestCase�main)� ExitStack)� MagicMock�Mock�patch� )� AnsiToWin32�
13 StreamWrapper��"ENABLE_VIRTUAL_TERMINAL_PROCESSING� )�osnamec @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
14 � Zd d � Zd
15 S )�StreamWrapperTestc C s& t � }t|d �}| �|j|ju � d S �N)r r �
16 assertTrueZ random_attr)�self�
17 mockStream�wrapper� r �UC:\Users\shubh\Videos\agent\nova\Lib\site-packages\colorama/tests/ansitowin32_test.py� testIsAProxy s
18 z StreamWrapperTest.testIsAProxyc C s8 t � }t � }t||�}|�d� | �|jjdi f� d S )N�hello)r )r r �writer � call_args)r r �
19 mockConverterr r r r �testDelegatesWrite s
20
21
22 z$StreamWrapperTest.testDelegatesWritec C sP t � }t� }t||��
23 }|�d� W d � n1 sw Y | �|j� d S �Nr )r r r r r �closed)r r �s�fpr r r �testDelegatesContext" s �z&StreamWrapperTest.testDelegatesContextc C s� t � }t� |j_t� }| �t��+}t||��
24 }|�d� W d � n1 s&w Y W d � d S W d � d S 1 s>w Y d S r )r �AttributeError� __enter__Z side_effectr � assertRaisesr r )r r r �excinfor r r r �testProxyNoContextManager) s
25 ��"�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
26 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
27
28 z?StreamWrapperTest.test_closed_shouldnt_raise_on_detached_streamN) �__name__�
29 __module__� __qualname__r r r# r( r. r0 r r r r r s r c @ s� e Zd Zdd� Zedd�eddd� �dd � ��Zd
30 d � Zd d
31 � Zdd� Zd+dd�Z dd� Z
32 dd� Z dd� Z dd� Z
33 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 �
34 mockStdout�autor- r r r �testInit? s
35 zAnsiToWin32Test.testInit� colorama.ansitowin32.wintermN� colorama.ansitowin32.winapi_testc G � dS r) r ��_r r r �<lambda>G � zAnsiToWin32Test.<lambda>c C sH t d�� t� }t|�}| �|j� W d � d S 1 s w Y d S )N�nt)r r r r �strip�r r7 r- r r r �testStripIsTrueOnWindowsF s
36
37 "�z(AnsiToWin32Test.testStripIsTrueOnWindowsc C sL t d�� tdd�}t|�}| �|j� W d � d S 1 s w Y d S )N�posixF)r )r r r � assertFalserB rC r r r �testStripIsFalseOffWindowsN s
38
39
40 "�z*AnsiToWin32Test.testStripIsFalseOffWindowsc C sV t � }t|�}t � |_t � |_d|_|�d� | �|jjj� | �|jj di f� d S )NT�abc�rH )
41 r r r6 �write_and_convertrB r rF �calledr+ r rC r r r �testWriteStripsAnsiT s
42 z#AnsiToWin32Test.testWriteStripsAnsic C s\ t � }t|�}t � |_t � |_d|_d|_|�d� | �|jj� | � |jjj
43 di f� d S )NFrH rI ) r r r6 rJ rB �convertr rF rK r+ r rC r r r �testWriteDoesNotStripAnsi` s
44 z)AnsiToWin32Test.testWriteDoesNotStripAnsiTc C sD t t� �}||_t� |_||_t� |_|�d� | �|jj|� d S )NrH ) r r rM � reset_allr5 �wintermr r+ rK )r rM r5 r- r r r �assert_autoresetsm s
45
46 z!AnsiToWin32Test.assert_autoresetsc C s8 | j dd� | j dd� | j ddd� | j ddd� d S )NT�rM F)rM r5 )rQ )r r r r �testWriteAutoresetsx s z#AnsiToWin32Test.testWriteAutoresetsc C s. t t� �}|�d� | �|jjjdi f� d S )NrH rI )r r rJ r+ r6 r r �r r- r r r �"testWriteAndConvertWritesPlainText~ s
47
48 z2AnsiToWin32Test.testWriteAndConvertWritesPlainTextc C s^ t t� �}t� |_g d�}|D ] }|jj�� |�|� | �dd� |jjjD �ddg� qd S )N)
49 z abc[mdefz
50 abc[0mdefz
51 abc[2mdefz abc[02mdefz abc[002mdefz abc[40mdefz abc[040mdefz abc[0;1mdefzabc[40;50mdefzabc[50;30;40mdefz abc[Adefz
52 abc[0Gdef�abc[1;20;128Hdefc S s g | ]}|d �qS �r r )�.0�argsr r r �
53 <listcomp>� s zIAnsiToWin32Test.testWriteAndConvertStripsAllValidAnsi.<locals>.<listcomp>rI )�def) r r �
54 call_win32r6 r �
55 reset_mockrJ r+ �call_args_list)r r- �data�datumr r r �%testWriteAndConvertStripsAllValidAnsi� s
56
57 ��z5AnsiToWin32Test.testWriteAndConvertStripsAllValidAnsic C s0 t t� �}t� |_|�d� | �|jjj� d S )Nz
58 [40m[41m)r r r\ rJ rF r6 r rK rT r r r �%testWriteAndConvertSkipsEmptySnippets� s
59
60 z5AnsiToWin32Test.testWriteAndConvertSkipsEmptySnippetsc C sr t t� �}d|_t� |_tdd�|_dddddd �}|�� D ]\}}|j�� |�|� | �|jj d
61 |� q d S ) NT�params)� return_value)�arc )�brc )�crc )�Grc )�Hrc )z abc[adefz abc[;;bdefz
62 abc[0cdefzabc[;;0;;GdefrV r )
63 r r rM r\ �extract_params�itemsr] rJ r+ r �r r- r_ r` �expectedr r r �1testWriteAndConvertCallsWin32WithParamsAndCommand� s
64 �
65
66 �zAAnsiToWin32Test.testWriteAndConvertCallsWin32WithParamsAndCommandc C s" t � }t|�}|�� |�� d S r )r r r* rO �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 s� t � }|�� tdd��# tddd� �� t|�}W d � n1 s#w Y W d � n1 s2w Y | �|j� | �|j� d S )N� colorama.ansitowin32.os.namerA r; c S r< r) r r r r r r? � r@ zPAnsiToWin32Test.test_wrap_shouldnt_raise_on_closed_orig_stdout.<locals>.<lambda>)r r* r r r rB rF rM ro r r r �.test_wrap_shouldnt_raise_on_closed_orig_stdout� s � �
67 �� z>AnsiToWin32Test.test_wrap_shouldnt_raise_on_closed_orig_stdoutc C s~ t dd��$ t ddd� ��
68 tt� �}W d � n1 s w Y W d � n1 s,w Y | �|j� | �|j� d S )Nrr rA r; c S r< r) r r r r r r? � r@ zQAnsiToWin32Test.test_wrap_shouldnt_raise_on_missing_closed_attr.<locals>.<lambda>)r r �objectr rB rF rM )r rp 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|�|� qd S )NrW )r
69 )r r )� i� )� � �! �, �7 )� z;;�2z;;002;;z0;1z ;;003;;456;;z11;22;33;44;55�m)r r rk r+ rj rl r r r �testExtractParams� s
70 � �z!AnsiToWin32Test.testExtractParamsc sf t � � t� �}� fdd�f� fdd�f� fdd�fd�|_|�dd� | �dd � � jD �g d
71 �� d S ) Nc � � d�S )Nrw r �r> �__��listenerr r r? � � z9AnsiToWin32Test.testCallWin32UsesLookup.<locals>.<lambda>c r� )Nrx r r� r� r r r? � r� c r� )Nry r r� r� r r r? � r� )r r
72 rv r~ )rv r �c r
73 c S s g | ]}|d d �qS rW r )rX re r r r rZ � s z;AnsiToWin32Test.testCallWin32UsesLookup.<locals>.<listcomp>)ry rw rx )r r Z win32_callsr\ r+ r^ rT r r� r �testCallWin32UsesLookup� s � �z'AnsiToWin32Test.testCallWin32UsesLookupc C sl t � }t|dd�}td�� }g d�}|D ]}|�|� q| �|jjd� W d � d S 1 s/w Y d S )NTrR r: )z]0z]0;fooz]0;colorama_test_titlez]1;colorama_test_titlez]2;colorama_test_titlezC];;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;r
74 )r r r r r+ � set_titleZ
75 call_count)r r7 r- rP r_ �coder r r �test_osc_codes� s
76 "�z AnsiToWin32Test.test_osc_codesc sB t � ��� � fdd�}|dd� |ddd� � |dd d� � |d
77 d � |d d
78 d� � |ddd� � t� }|d|� t� }d|_d|j_d|j_t|�}|�dt� | � |j
79 � | � |j � | � |� � � |ddd� � t� }|d|� t|�}|�dt� | �
80 |j
81 � | �
82 |j � | �
83 |� � � W d � d S 1 s�w Y d S )Nc s � � t| |dd�� d S )NT)�create)�
84 enter_contextr )re rf ��stackr r �p� s z3AnsiToWin32Test.test_native_windows_ansi.<locals>.prr rA r; c S r< r) r r r r r r? � r@ z:AnsiToWin32Test.test_native_windows_ansi.<locals>.<lambda>zcolorama.win32.winapi_testc S r< r) r r r r r r? � r@ z colorama.winterm.win32.windllznon-Nonez colorama.winterm.get_osfhandlec S r< )N�� r r= r r r r? r@ z%colorama.winterm.win32.GetConsoleModec S s t S r r
85 r= r r r r? r@ z%colorama.winterm.win32.SetConsoleModeFTr r� c S r< )Nr r r= r r r r? r@ )r r r �isattyrd �filenor Zassert_called_withr rF rB rM Z should_wrapr )r r� ZSetConsoleMode�stdoutr- r r� r �test_native_windows_ansi� s>
86
87 �
88
89 "�z(AnsiToWin32Test.test_native_windows_ansi)T)r1 r2 r3 r9 r rD rG rL rN rQ rS rU ra rb rn rq rs ru r r� r� r� r r r r r4 = s*
90
91
92
93 r4 �__main__N)�ior r Zunittestr r �
94 contextlibr � ImportErrorZ contextlib2Z
95 unittest.mockr r r ZmockZ ansitowin32r r �win32r �utilsr r r4 r1 r r r r �<module> s* � � ) i
96 �