Format: application/x-python-code
1 o
2
3 1'hU � @ s� d dl Z d dlmZmZmZ z
4 d dlmZmZ W n ey) d dl mZmZ Y nw ddl
5 m Z ddl m
6 Z
7 mZmZ ddlmZmZ e jZe jZG d d
8 � d
9 e�ZG d d � d e�Zed
10 krae� dS dS )� N)�TestCase�main�
11 skipUnless)�patch�Mock� )�
12 StreamWrapper)�init�just_fix_windows_console� _wipe_internal_state_for_tests� )�osname�
13 replace_byc @ s� e Zd Zeej�� d�dd� �Zdd� Zdd� Z dd � Z
14 e d
15 �e d d d
16 � �e ddd
17 � �dd� ���Z e d
18 �e d dd
19 � �dd� ��Z
20 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 � | � � d S �N)�assertNotWrapped��self� r �TC:\Users\shubh\Videos\agent\nova\Lib\site-packages\colorama/tests/initialise_test.py�setUp s zInitTest.setUpc C � t � tt_tt_d S r �r � orig_stdout�sys�stdout� orig_stderr�stderrr r r r �tearDown �
21 zInitTest.tearDownc C sL | � tjtd� | � tjtd� | �ttjt�d� | �ttjt�d� d S )Nzstdout should be wrappedzstderr should be wrappedzbad stdout wrapperzbad stderr wrapper) � assertIsNotr r r r r �
22 assertTrue�
23 isinstancer r r r r �
24 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 � dS �NTr ��_r r r �<lambda>+ � zInitTest.<lambda>�)colorama.ansitowin32.enable_vt_processingc G r' �NFr r) r r r r+ , r, c C �< t d�� t� | �� W d � d S 1 sw Y d S �N�nt�r
25 r r$ �r r* r r r �testInitWrapsOnWindows* s
26
27 "�z InitTest.testInitWrapsOnWindowsc G r' r. r r) r r r r+ 3 r, c C r/ r0 �r
28 r r r3 r r r �#testInitDoesntWrapOnEmulatedWindows2 s
29
30 "�z,InitTest.testInitDoesntWrapOnEmulatedWindowsc C r/ )N�posixr5 r r r r � testInitDoesntWrapOnNonWindows9 s
31
32 "�z'InitTest.testInitDoesntWrapOnNonWindowsc C sL t d �� t� | �tj� | �tj� W d � d S 1 s w Y d S r )r r Z assertIsNoner r r r r r r �testInitDoesntWrapIfNone> s
33
34 "�z!InitTest.testInitDoesntWrapIfNonec C �@ t d�� tdd� | �� W d � d S 1 sw Y d S )Nr7 T�� autoresetr2 r r r r �&testInitAutoresetOnWrapsOnAllPlatformsF �
35
36
37 "�z/InitTest.testInitAutoresetOnWrapsOnAllPlatformsc C r: )Nr1 F)�wrapr5 r r r r �"testInitWrapOffDoesntWrapOnWindowsK r> z+InitTest.testInitWrapOffDoesntWrapOnWindowsc C s | � tdd� � d S )Nc S s t ddd�S )NTF)r< r? )r r r r r r+ Q s zEInitTest.testInitWrapOffIncompatibleWithAutoresetOn.<locals>.<lambda>)� assertRaises�
38 ValueErrorr r r r �*testInitWrapOffIncompatibleWithAutoresetOnP s z3InitTest.testInitWrapOffIncompatibleWithAutoresetOnz&colorama.win32.SetConsoleTextAttributez colorama.initialise.AnsiToWin32c C s~ t d��1 tdd� | �t|j�d� | �|jd d d d� | �|jd d d d� W d � d S 1 s8w Y d S )Nr1 Tr; r r r< r �r
39 r � assertEqual�lenZcall_args_list)r � mockATW32r* r r r �testAutoResetPassedOnS s
40
41 "�z InitTest.testAutoResetPassedOnc C s� t d��Z t� tdd� | �t|j�d� | �|jd d d d� | �|jd d d d� t� | �t|j�d � | �|jd d d d
42 � | �|jd d d d
43 � W d � d S 1 saw Y d S ) Nr1 Tr; � r r r< � � F� rD )r rG r r r �testAutoResetChangeable\ s
44
45 ��"�z InitTest.testAutoResetChangeablez#colorama.initialise.atexit.registerc C s0 t � | �|j� |�� t � | �|j� d S r )r r" �calledZ
46 reset_mockZ assertFalse)r Z mockRegisterr r r � testAtexitRegisteredOnlyOncen s
47 z%InitTest.testAtexitRegisteredOnlyOnceN)�__name__�
48 __module__� __qualname__r r r �isattyr r r$ r r r4 r6 r8 r9 r= r@ rC rH rM rO r r r r r s2
49
50 r c @ s2 e Zd Zdd� Zdd� Zeddd� �dd � �Zd
51 S ) �JustFixWindowsConsoleTestc C r r r r r r r �_resetx r z JustFixWindowsConsoleTest._resetc C r r )rU r r r r r } s z"JustFixWindowsConsoleTest.tearDownr& c C r' r( r r r r r r+ � r, z"JustFixWindowsConsoleTest.<lambda>c s8 t jdkrt� | �t jt� | �t jt� d S dd� }dD ]{� td� fdd���h | � � |� t j}t j}t� | �t j|� � rK| �t j|� n| �
52 t j|� t j}t j}t� | �t j|� | �t j|� | � � |� t � t j}t j}t� | �|t j� | �|t j� W d � n1 s�w Y 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 �closedrS � return_value�filenor r r )r r r r r �fake_std� s
53 zEJustFixWindowsConsoleTest.testJustFixWindowsConsole.<locals>.fake_std)FTr- c s � S r r r) �Z native_ansir r r+ � r, zEJustFixWindowsConsoleTest.testJustFixWindowsConsole.<locals>.<lambda>) r �platformr
54 r% r r r r r rU r! r )r rZ Z prev_stdoutZ prev_stderrr r[ r �testJustFixWindowsConsole� sF
55
56 � ���z3JustFixWindowsConsoleTest.testJustFixWindowsConsoleN)rP rQ rR rU r r r] r r r r rT w s
57 rT �__main__)r Zunittestr r r Z
58 unittest.mockr r � ImportErrorZmockZ ansitowin32r Z
59 initialiser r
60 r �utilsr
61 r r r r r r rT rP r r r r �<module> s" � eE
62 �