File: //opt/cppython/lib/python3.8/test/__pycache__/test_tabnanny.cpython-38.opt-2.pyc
U
.D�g�5 � @ s� d dl mZmZ d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
m Z m Z m
Z
mZ ddddd d
d �ZG d d
� d
�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZdS )� )�TestCase�mock)r N)�captured_stderr�captured_stdout�
script_helper�findfile�unlinkzCfruits = [
"Apple",
"Orange",
"Banana",
print(fruits)
zLif True:
print("hello")
print("world")
else:
print("else called")
zIif True:
print("hello")
print("world")
else:
print("else called")
zNif True:
print("hello")
print("world")
else:
print("else called")
zGdef my_func():
print("hello world")
if True:
print("If called")zLdef my_func():
print("Hello world")
if True:
print("If called"))�incomplete_expression�wrong_indented�nannynag_errored�
error_free�tab_space_errored_1�tab_space_errored_2c @ s& e Zd Zd dd�Zdd� Zdd� ZdS )
�TemporaryPyFile� Nc C s || _ || _d S �N)� source_code�dir)�selfr � directory� r �1/opt/cppython/lib/python3.8/test/test_tabnanny.py�__init__B s zTemporaryPyFile.__init__c C s: t jd| jddd��}|�| j� W 5 Q R X |j| _| jS )N�wz.pyF)�moder �suffix�delete)�tempfile�NamedTemporaryFiler �writer �name� file_path)r �fr r r � __enter__F s �zTemporaryPyFile.__enter__c C s t | j� d S r )r r! )r �exc_type� exc_value�
exc_tracebackr r r �__exit__N s zTemporaryPyFile.__exit__)r N)�__name__�
__module__� __qualname__r r# r' r r r r r ? s
r c @ s e Zd Zdd� ZdS )�TestFormatWitnessesc
C sJ ddddg}|D ]4\}}| j ||d�� | �t�|�|� W 5 Q R X qd S )N)ZTestzat tab sizes T, e, s, t)r z at tab size )�tz
at tab size t)z t zat tab sizes , , t, , )�words�expected)�subTest� assertEqual�tabnannyZformat_witnesses)r �testsr- r. r r r �test_format_witnessesU s � z)TestFormatWitnesses.test_format_witnessesN)r( r) r* r3 r r r r r+ R s r+ c @ s e Zd Zdd� ZdS )� TestErrPrintc C s� ddgdfdgdfdddgdfg d fg}|D ]N\}}| j ||d
��2 t� �}tj|� W 5 Q R X | �|�� |� W 5 Q R X q,d S ) N�first�secondz
first second
zfirst
� � � z1 2 3
�
)Z argumentsr. )r/ r r1 Zerrprintr0 �getvalue)r r2 �argsr. �stderrr r r �
test_errprintf s
� zTestErrPrint.test_errprintN)r( r) r* r> r r r r r4 c s r4 c @ s e Zd Zdd� ZdS )� TestNannyNagc
C s� t �ddd�dddd�ft �ddd�dddd�fg}|D ]�\}}|�� }|�� }|�� }| j||d d �� | �|d |� W 5 Q R X | j||d
d �� | �|d
|� W 5 Q R X | j||d d
�� | �|d |� W 5 Q R X q8d S )Nr ZfooZbar)�lineno�msg�line� ZtestmsgZtestliner@ )� line_numberr. rA )rA r. rB )rB r. )r1 �NannyNagZ
get_linenoZget_msgZget_liner/ r0 )r r2 Znannyr. rD rA rB r r r �test_all_methodsw s(
�
��
�z TestNannyNag.test_all_methodsN)r( r) r* rF r r r r r? v s r? c @ sf e Zd Zdd� Zddd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dS )� TestCheckc C s | � ttdtj� dt_d S )N�verboser )Z
addCleanup�setattrr1 rH )r r r r �setUp� s zTestCheck.setUpr c
C sR t � �"}t� �}t�|� W 5 Q R X W 5 Q R X | �|�� |� | �|�� |� d S r )r r r1 �checkr0 r; )r Z dir_or_file�out�err�stdoutr= r r r �verify_tabnanny_check� s z TestCheck.verify_tabnanny_checkc C s& t td ��}| �|� W 5 Q R X d S �Nr �r � SOURCE_CODESrO �r r! r r r �test_correct_file� s zTestCheck.test_correct_filec
C s� t �� ��}|�d�g}ttd |d�}ttd |d�}|��}|��}||fD ]}|�|�d�� qJdt_t� �"}t� �} t� |� W 5 Q R X W 5 Q R X |�
� }|D ](}
| j |
d�� | � |
|� W 5 Q R X q�| �
| �
� d� W 5 Q R X W 5 Q R X W 5 Q R X d S )Nz: listing directory
r �r z: Clean bill of health.
r7 )rB r )r �TemporaryDirectoryr rR �appendr1 rH r r rK r; r/ �assertInr0 ) r �tmp_dir�linesZfile1Zfile2Z
file1_pathZ
file2_pathr! rN r= rB r r r � test_correct_directory_verbose� s
z(TestCheck.test_correct_directory_verbosec
C s> t �� �,}ttd |d�� | �|� W 5 Q R X W 5 Q R X d S )Nr rU �r rV r rR rO )r rY r r r �test_correct_directory� s
z TestCheck.test_correct_directoryc C s< t td ��&}d}|�d|� �}| j||d� W 5 Q R X d S )Nr
�Iunindent does not match any outer indentation level (<tokenize>, line 3)
�: Indentation Error: �rM rQ �r r! rM r r r �test_when_wrong_indented� s z"TestCheck.test_when_wrong_indentedc C s< t td ��&}d}|�d|� �}| j||d� W 5 Q R X d S )Nr z(('EOF in multi-line statement', (7, 0))
z: Token Error: r` rQ ra r r r � test_when_tokenize_tokenerror� s z'TestCheck.test_when_tokenize_tokenerrorc C sJ t td ��4}|�d�}|d7 }|d7 }dt_| j||d� W 5 Q R X d S )Nr z': *** Line 3: trouble in tab city! ***
z%offending line: '\tprint("world")\n'
z$indent not equal e.g. at tab size 1
r7 �rL )r rR r1 rH rO �r r! rL r r r � test_when_nannynag_error_verbose� s
z*TestCheck.test_when_nannynag_error_verbosec C s4 t td �� }|� d�}| j||d� W 5 Q R X d S )Nr z 3 '\tprint("world")\n'
rd rQ re r r r �test_when_nannynag_error� s
z"TestCheck.test_when_nannynag_errorc C s< d}|�dt j� dt�t j�� d|�d�}| j||d� d S )Nz
no_file.pyz: I/O Error: [Errno z] z: r: r` )�errno�ENOENT�os�strerrorrO )r �pathrM r r r �test_when_no_file� s &zTestCheck.test_when_no_filec C sx t �� �f}ttd |d�}ttd |d�}|�6}|�&}d}|�d|� �}| j||d� W 5 Q R X W 5 Q R X W 5 Q R X d S )Nr
rU r r^ r_ r` r\ )r rY Z
error_fileZ code_fileZe_fileZc_filerM r r r �test_errored_directory� s
� � z TestCheck.test_errored_directoryN)r r )r( r) r* rJ rO rT r[ r] rb rc rf rg rm rn r r r r rG � s
rG c @ s& e Zd Ze�d�dd� �Zdd� ZdS )�TestProcessTokensztabnanny.NannyNagc
C sN t td ��8}t|��}t�t�|j�� W 5 Q R X | �|j � W 5 Q R X d S rP )
r rR �openr1 �process_tokens�tokenize�generate_tokens�readlineZ assertFalseZcalled)r Z MockNannyNagr! r" r r r �test_with_correct_code s
z(TestProcessTokens.test_with_correct_codec
C s� dD ]v}| j |d��` tt| ��H}t|��4}t�|j�}| �tj �� t�
|� W 5 Q R X W 5 Q R X W 5 Q R X W 5 Q R X qd S )N)r
r )�key) r/ r rR rp rr rs rt Z assertRaisesr1 rE rq )r rv r! r" �tokensr r r � test_with_errored_codes_samples
s
z1TestProcessTokens.test_with_errored_codes_samplesN)r( r) r* r Zpatchru rx r r r r ro s
ro c @ sN e Zd Zdddd�dd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Zdd� Z dS )�TestCommandLiner F)rN r= �partialc G s� t jd|�� \}}}|�d�}|�d�}|r�||f||ffD ]D\}} | �� }
|�� D ]*} | j| |
d�� | �| |
� W 5 Q R X qTq<n(| �|�� |�� � | �|�� |�� � d S )N�-mr1 �ascii)�std�output)r{ r1 )r Zassert_python_ok�decode�
splitlinesr/ rX ZassertListEqual) r rN r= rz r< �_rL rM r} r~ Z_outputZ_stdr r r � validate_cmd" s
z TestCommandLine.validate_cmdc C s< t td ��&}|�d�}|d7 }| j||d� W 5 Q R X d S )Nr
r_ zHunindent does not match any outer indentation level (<tokenize>, line 3)�r= �r rR r� )r r! r= r r r �test_with_errored_file3 s
z&TestCommandLine.test_with_errored_filec C s& t td ��}| �|� W 5 Q R X d S rP r� rS r r r �test_with_error_free_file; s z)TestCommandLine.test_with_error_free_filec C s$ t d�}d|� d�}| j|d� d S )Nz tabnanny.pyzUsage: z [-v] file_or_directory ...r� )r r� )r rl r= r r r �test_command_usage@ s z"TestCommandLine.test_command_usagec C s6 t td �� }|� d�}| jd||d� W 5 Q R X d S )Nr r: z-q)rN r� )r r! rN r r r �test_quiet_flagF s
z TestCommandLine.test_quiet_flagc C s< t td ��&}t�d��� }| jd||dd� W 5 Q R X d S )Nr �$offending line: '\tprint("world")\n'z-vT�rN rz �r rR �textwrap�dedent�stripr� �r rl rN r r r �test_verbose_modeL s
�z!TestCommandLine.test_verbose_modec C s< t td ��&}t�d��� }| jd||dd� W 5 Q R X d S )Nr r� z-vvTr� r� r� r r r �test_double_verbose_modeT s
�z(TestCommandLine.test_double_verbose_modeN)
r( r) r* r� r� r� r� r� r� r� r r r r ry s ry )Zunittestr r rh rj r1 rr r r� Z test.supportr r r r r rR r r+ r4 r? rG ro ry r r r r �<module> s, �-p