Format: application/x-python-code
1 o
2
3 1'h} � @ s� d dl Z d dlmZmZmZ z
4 d dlmZmZ W n ey) d dl mZmZ Y nw ddl
5 m Z m Z m
6 Z
7 G dd� de�ZedkrEe� dS dS ) � N)�TestCase�main�
8 skipUnless)�Mock�patch� )�WinColor�WinStyle�WinTermc @ s� e Zd Zed�dd� �Zeej�d�d�dd� �Z ed�dd � �Z
9 eej�d�d�d
10 d � �Z eej�d�d�d d
11 � �Z eej�d�d�dd� �Z
12 ed�dd� �Zed�dd� �ZdS )� WinTermTestzcolorama.winterm.win32c C sH t � }d|_||j_t� }| �|jd� | �|jd� | �|jd� d S )N�o � � � ) r � wAttributes�GetConsoleScreenBufferInfo� return_valuer
13 � assertEqual�_fore�_back�_style��selfZ mockWin32ZmockAttr�term� r �QC:\Users\shubh\Videos\agent\nova\Lib\site-packages\colorama/tests/winterm_test.py�testInit s zWinTermTest.testInit�winzrequires Windowsc C s� t � }d|_d|_d|_| �|�� d� tj|_| �|�� tj� tj|_| �|�� tjtjd � t j
14 |_| �|�� tjtjd t j
15 � d S )Nr � ) r
16 r r r r � get_attrsr ZYELLOWZMAGENTAr ZBRIGHT�r r r r r � testGetAttrs s" ��zWinTermTest.testGetAttrsc C sz t � }d|_||j_t� }t � |_d|_d|_d|_|� � | �
17 |jd� | �
18 |jd� | �
19 |jd� | �
20 |jj d� d S )N�) ������ r r T) r r r r r
21 � set_consoler r r Z reset_allr �calledr r r r � testResetAll/ s zWinTermTest.testResetAllc C �@ t � }t� |_d|_|�d� | �|jd� | �|jjd� d S �Nr � T)r
22 r r% r �forer r& r r r r �testForeB �
23 zWinTermTest.testForec C r( r) )r
24 r r% r �backr r& r r r r �testBackM r- zWinTermTest.testBackc C r( )Nr � T)r
25 r r% r �styler r& r r r r � testStyleX r- zWinTermTest.testStylec C sL t � }d|_||j_t� }t � |_|�� | �|jj |j
26 |� � fi f� d S )Nr ) r r r r r
27 �windllr% r �SetConsoleTextAttribute� call_args�STDOUTr r r r r �testSetConsolec s �zWinTermTest.testSetConsolec C sP t � }d|_||j_t� }t � |_|jdd� | �|jj |j
28 |� � fi f� d S )Nr T)Z on_stderr) r r r r r
29 r3 r% r r4 r5 ZSTDERRr r r r r �testSetConsoleOnStderrr s �z"WinTermTest.testSetConsoleOnStderrN)�__name__�
30 __module__� __qualname__r r r �sys�platform�
31 startswithr! r' r, r/ r2 r7 r8 r r r r r
32 s"
33
34
35
36
37
38
39
40
41
42 r �__main__)r< Zunittestr r r Z
43 unittest.mockr r � ImportErrorZmockZwintermr r r
44 r r9 r r r r �<module> s �u
45 �