File: //usr/local/lib/python3.10/lib2to3/tests/__pycache__/test_refactor.cpython-310.pyc
o
�i*1 � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z m Z ddl
mZ ej�ej�e�d�Zej�ed�Zej�e� z e
�d�ZW ej�� nej�� w e
�d�ZG d d
� d
ej�ZdS ) z
Unit tests for refactor.py.
� N)�refactor�pygram�
fixer_base)�token�data�fixers�myfixes�
lib2to3.fixesc @ s e Zd Zdd� Zdd� Zdd� Zdedfdd �Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zeddd fd d �Zd!d"� Zd#d$� Zefd%d&�Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5d6� Z d7d8� Z d9d:� Z dS );�TestRefactoringToolc C s t j�t� d S �N)�sys�path�append� FIXER_DIR��self� r �8/usr/local/lib/python3.10/lib2to3/tests/test_refactor.py�setUp s zTestRefactoringTool.setUpc C s t j�� d S r )r r
�popr r r r �tearDown" � z TestRefactoringTool.tearDownc C s2 t ||�D ]\}}t||�s| �d| |� qd S )Nz%s are not instances of %s)�zip�
isinstance�fail)r Z instancesZclasses�inst�clsr r r �check_instances% s
��z#TestRefactoringTool.check_instancesNc C s t �|||�S r )r �RefactoringTool)r �optionsr �explicitr r r �rt* r zTestRefactoringTool.rtc C �4 | � ddi�}| �d|jj� | �d|jjj� d S )N�print_functionT�print�r! � assertNotInZgrammar�keywordsZdriver�r r! r r r �test_print_function_option- � z.TestRefactoringTool.test_print_function_optionc C r" )NZ
exec_functionT�execr% r( r r r �test_exec_function_option2 r* z-TestRefactoringTool.test_exec_function_optionc C s2 | � � }| �|j� | � ddi�}| �|j� d S )N�write_unchanged_filesT)r! Z assertFalser- �
assertTruer( r r r �!test_write_unchanged_files_option7 s z5TestRefactoringTool.test_write_unchanged_files_optionc C sd g d�}t �d�}t �dd�}t �d�}| �|dd� |D �� | �||� | �|dd� |D �� d S )N)r �firstZlastZparrotZpreorderr Fc S � g | ]}d | �qS )Zfix_r ��.0�namer r r �
<listcomp>B � zBTestRefactoringTool.test_fixer_loading_helpers.<locals>.<listcomp>c S r1 )z myfixes.fix_r r2 r r r r5 E r6 )r Zget_all_fix_names�get_fixers_from_package� assertEqual)r �contentsZ non_prefixedZprefixedZ
full_namesr r r �test_fixer_loading_helpers= s
�z.TestRefactoringTool.test_fixer_loading_helpersc C s~ t j}t}|� }| �|d�|� | �|d�|d�� | �|d�|d�� | �|d�|d�� d}| �||�|d �� d
}| �||�|d �� d }| �||�|d�� d
}| �||�|d �� d}| �||�|d�� d}| �||�|d �� d}|D ]
}| �||�|� qzd}| �||�|d�� d}| �||�|� d}| �||�|d�� d}| �||�|d�� d}| �||�|� d S )N� z%from __future__ import print_function)r# z!from __future__ import generators)�
generatorsz*from __future__ import generators, feature)r< Zfeaturez1from __future__ import generators, print_function)r< r# z1from __future__ import print_function, generators)r# r< z(from __future__ import (print_function,)z3from __future__ import (generators, print_function)z2from __future__ import (generators, nested_scopes))r< Z
nested_scopeszGfrom __future__ import generators
from __future__ import print_function)�fromzfrom 4zfrom xzfrom x 5z from x imz
from x importzfrom x import 4z1'docstring'
from __future__ import print_functionz9'docstring'
'somng'
from __future__ import print_functionz/# comment
from __future__ import print_functionz5# comment
'doc'
from __future__ import print_functionz3class x: pass
from __future__ import print_function)r Z_detect_future_features� frozensetr8 )r �run�fs�empty�inp�invalidr r r �test_detect_future_featuresG sL
�
�
�z/TestRefactoringTool.test_detect_future_featuresc C s� G dd� dt j�}G dd� dt j�}G dd� dt j�}|i g �}|i g �}|i g �}t�|||g�}|�tjj�}| �|||g� |�t j
�} | �| ||g� |� � D ] }
| �|
|g� qSd S )Nc @ s e Zd ZdS )z;TestRefactoringTool.test_get_headnode_dict.<locals>.NoneFixN)�__name__�
__module__� __qualname__r r r r �NoneFixu s rH c @ � e Zd ZdZdS )z@TestRefactoringTool.test_get_headnode_dict.<locals>.FileInputFixzfile_input< any * >N�rE rF rG ZPATTERNr r r r � FileInputFixx � rK c @ rI )z=TestRefactoringTool.test_get_headnode_dict.<locals>.SimpleFixz'name'NrJ r r r r � SimpleFix{ rL rM ) r ZBaseFixr Z_get_headnode_dictr r Zpython_symbolsZ
file_inputr8 r �NAME�values) r rH rK rM Zno_headZ with_head�simple�dZ top_fixesZ
name_fixes�fixesr r r �test_get_headnode_dictt s
�z*TestRefactoringTool.test_get_headnode_dictc C sh ddl m} ddlm} ddlm} ddlm} | �� }|� � \}}| �
||g� | �
||||g� d S )Nr )�FixFirst)�FixLast)� FixParrot)� FixPreorder) Zmyfixes.fix_firstrT Zmyfixes.fix_lastrU Zmyfixes.fix_parrotrV Zmyfixes.fix_preorderrW r! Z
get_fixersr )r rT rU rV rW r! �preZpostr r r �test_fixer_loading� s z&TestRefactoringTool.test_fixer_loadingc C sD | j t| jdgd� | j tj| jdgd� | j tj| jdgd� d S )NZnot_here�r Z no_fixer_clsZ bad_order)Z assertRaises� ImportErrorr! r Z
FixerErrorr r r r �test_naughty_fixers� s z'TestRefactoringTool.test_naughty_fixersc C sL | � � }d}|�|d�}| �t|�|� d}|�|d�}| �t|�|� d S )N�def parrot(): pass
�<test>zdef f(): pass
)r! Zrefactor_string�assertNotEqual�strr8 )r r! �input�treer r r �test_refactor_string� s z(TestRefactoringTool.test_refactor_stringc sh G � fdd�dt j�}g � |t�}tj}t�d�t_z |�� W |t_n|t_w g d�}| �� |� d S )Nc s e Zd Z� fdd�ZdS )z5TestRefactoringTool.test_refactor_stdin.<locals>.MyRTc s � � ||||g� d S r )�extend)r Zold_textZnew_text�filenameZequal��resultsr r � print_output� s zBTestRefactoringTool.test_refactor_stdin.<locals>.MyRT.print_outputN)rE rF rG rh r rf r r �MyRT� s ri r] )r] zdef cheese(): pass
z<stdin>F) r r �_DEFAULT_FIXERSr �stdin�io�StringIOZrefactor_stdinr8 )r ri r! �save�expectedr rf r �test_refactor_stdin� s
z'TestRefactoringTool.test_refactor_stdinTc C sv | � |�}| �|�}| j||d�}|r||_|�|� | �|| �|�� |s(d S |�|d� | �|�}| �||� |S )N)r r T)�init_test_file� read_filer! Z log_debug�
refactor_filer8 r_ ) r � test_filer r �mock_log_debug�actually_write� old_contentsr! � new_contentsr r r �check_file_refactoring� s
z*TestRefactoringTool.check_file_refactoringc C sL t jdd�}| �tj|� t�||� tj�|tj� |��}t�
|d� |S )N�2to3-test_refactor��prefixi� ) �tempfile�mkdtempZ
addCleanup�shutil�rmtree�copy�osr
�join�basename�chmod)r rt Ztmpdirr r r rq � s z"TestRefactoringTool.init_test_filec C s6 t |d�� }|�� W d � S 1 sw Y d S )N�rb)�open�read)r rt �fpr r r rr � s $�z TestRefactoringTool.read_filec C s> | � |�}| �|�}| j|d�}|�|d� | �|�}||fS )NrZ T)rq rr r! rs )r rt r rw r! rx r r r rs � s
z!TestRefactoringTool.refactor_filec C s t j�td�}| �|t� d S )N�parrot_example.py)r� r
r� r ry rj )r rt r r r �test_refactor_file� s z&TestRefactoringTool.test_refactor_filec s� t j�td�}g � � fdd�}| j|dddi|dd� d t�t jt j�|� � }� D ]}d
|v r:| � ||� d S q+| �
d |� f � d S ) Nr� c s � � | | � d S r �r )�msg�args�Zdebug_messagesr r �recording_log_debug� s zXTestRefactoringTool.test_refactor_file_write_unchanged_file.<locals>.recording_log_debugr r- TF)r r ru rv zNot writing changes to .*%szNot writing changesz%r not matched in %r) r� r
r� r ry �re�escape�sepr� Z assertRegexr )r rt r� Z
message_regex�messager r� r �'test_refactor_file_write_unchanged_file� s" �� �z;TestRefactoringTool.test_refactor_file_write_unchanged_filec sN � fdd�}|g g � g d�}dg}|||� dt j�dd�g}|||� d S )Nc s� �fdd�}t jj}|t j_��� }g �tjdd�� z+t�tj� � d�� | D ]}t
tj� � |�d�� � q'|� � � W |t j_t
�� � n
|t j_t
�� � w ���� fdd�|D �� d S ) Nc s � � |� d S r r� )r �fr� )�gotr r �mock_refactor_file� r zPTestRefactoringTool.test_refactor_dir.<locals>.check.<locals>.mock_refactor_filerz r{ �a_dir�wbc s g | ] }t j�� |��qS r �r� r
r� )r3 r
)�dirr r r5 s zHTestRefactoringTool.test_refactor_dir.<locals>.check.<locals>.<listcomp>)r r rs r! r} r~ r� �mkdirr
r� r� �closeZ refactor_dirr r� r8 )Z structurero r� Z save_funcr! �fnr )r� r� r �check� s$ � �z4TestRefactoringTool.test_refactor_dir.<locals>.check)Znothing�hi.pyz.dumbz .after.pyz notpy.npyZsappyr� r� zstuff.pyr� )r r� rb ro r r r �test_refactor_dir� s
�z%TestRefactoringTool.test_refactor_dirc C s t j�td�}| �|� d S )Nzdifferent_encoding.py�r� r
r� �
TEST_DATA_DIRry )r r� r r r �test_file_encoding � z&TestRefactoringTool.test_file_encodingc C s t j�td�}| �|�}d S )Nzfalse_encoding.pyr� �r r� r r r r �test_false_file_encoding r� z,TestRefactoringTool.test_false_file_encodingc C s. t j�td�}| �|�}| �|�tj�� d S )Nzbom.py) r� r
r� r� ry r. �
startswith�codecs�BOM_UTF8r� r r r �test_bom s
z TestRefactoringTool.test_bomc C sF t j}dt _zt j�td�}t�d�}| �||� W |t _d S |t _w )Nz
�crlf.pyr )r� �linesepr
r� r� r r7 ry )r Zold_sepr� rR r r r �test_crlf_newlines$ s
z&TestRefactoringTool.test_crlf_newlinesc C sD t j�td�}| �|�\}}| �d|� | �d|� | �d|� d S )Nr� s
s
)r� r
r� r� rs ZassertInr& )r r� �old�newr r r �test_crlf_unchanged. s
z'TestRefactoringTool.test_crlf_unchangedc C sD | � � }d}|�|d�}| �||� d}|�|d�}| �||� d S )Nz
>>> example()
42
r^ z&
>>> def parrot():
... return 43
)r! Zrefactor_docstringr8 r_ )r r! �doc�outr r r �test_refactor_docstring5 s z+TestRefactoringTool.test_refactor_docstringc C sd ddl m} | jdgd�}| �t|j�d� | jdgd�}|jD ]
}t||�r* d S q | �d� d S )Nr )� FixExplicitzmyfixes.fix_explicitrZ )r zexplicit fixer not loaded)Zmyfixes.fix_explicitr� r! r8 �lenZ
post_orderr r )r r� r! Zfixr r r �
test_explicitF s
�z!TestRefactoringTool.test_explicit)!rE rF rG r r r rj r! r) r, r/ r: rD rS rY r\ rc rp � _2TO3_FIXERSry rq rr rs r� r� r� r� r� r� r� r� r� r� r r r r r
s@
-
�
r
) �__doc__r r� r� rl r� r} r ZunittestZlib2to3r r r Z
lib2to3.pgen2r r
r� �dirname�__file__r� r r r7 rj r r� ZTestCaser
r r r r �<module> s&