File: //usr/local/lib/python3.7/lib2to3/tests/__pycache__/test_refactor.cpython-37.opt-1.pyc
B
��g�0 � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m
Z
m Z m Z ddl
mZ ej�ej�e�d�Zej�ed�Zej�e� ze
�d�ZW dej�� X e
�d�ZG d d
� d
ej�ZdS ) z
Unit tests for refactor.py.
� N)�refactor�pygram�
fixer_base)�token�data�fixers�myfixesz
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� Zeddd fd d �Zd d � Zd!d"� Zefd#d$�Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Z d3d4� Z d5d6� Z d7d8� Z dS )9�TestRefactoringToolc C s t j�t� d S )N)�sys�path�append� FIXER_DIR)�self� r �7/usr/local/lib/python3.7/lib2to3/tests/test_refactor.py�setUp s zTestRefactoringTool.setUpc C s t j�� d S )N)r
r �pop)r r r r �tearDown" s z TestRefactoringTool.tearDownc C s6 x0t ||�D ]"\}}t||�s | �d| |� q W d S )Nz%s are not instances of %s)�zip�
isinstance�fail)r Z instances�classes�inst�clsr r r �check_instances% s
z#TestRefactoringTool.check_instancesNc C s t �|||�S )N)r �RefactoringTool)r �optionsr �explicitr r r �rt* s zTestRefactoringTool.rtc C s4 | � ddi�}| �|jtj� | �|jjtj� d S )N�print_functionT)r ZassertIsZgrammarr Z!python_grammar_no_print_statementZdriver)r r r r r �test_print_function_option- s
z.TestRefactoringTool.test_print_function_optionc C s2 | � � }| �|j� | � ddi�}| �|j� d S )N�write_unchanged_filesT)r Z assertFalser! �
assertTrue)r r r r r �!test_write_unchanged_files_option3 s z5TestRefactoringTool.test_write_unchanged_files_optionc C sj dddddg}t �d�}t �dd�}t �d�}| �|dd � |D �� | �||� | �|d
d � |D �� d S ) Nr �firstZlastZparrotZpreorderr Fc S s g | ] }d | �qS )Zfix_r )�.0�namer r r �
<listcomp>> s zBTestRefactoringTool.test_fixer_loading_helpers.<locals>.<listcomp>c S s g | ] }d | �qS )z myfixes.fix_r )r% r& r r r r' A s )r Zget_all_fix_names�get_fixers_from_package� assertEqual)r �contentsZ non_prefixedZprefixedZ
full_namesr r r �test_fixer_loading_helpers9 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}x |D ]}| �||�|� q�W d}| �||�|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� frozensetr) )r �runZfs�emptyZinpZinvalidr r r �test_detect_future_featuresC sF
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� x |� � D ]}
| �|
|g� q�W d S )Nc @ s e Zd ZdS )z;TestRefactoringTool.test_get_headnode_dict.<locals>.NoneFixN)�__name__�
__module__� __qualname__r r r r �NoneFixq s r6 c @ s e Zd ZdZdS )z@TestRefactoringTool.test_get_headnode_dict.<locals>.FileInputFixzfile_input< any * >N)r3 r4 r5 �PATTERNr r r r � FileInputFixt s r8 c @ s e Zd ZdZdS )z=TestRefactoringTool.test_get_headnode_dict.<locals>.SimpleFixz'name'N)r3 r4 r5 r7 r r r r � SimpleFixw s r9 ) r ZBaseFixr Z_get_headnode_dictr r Zpython_symbolsZ
file_inputr) r �NAME�values) r r6 r8 r9 Zno_headZ with_headZsimple�dZ top_fixesZ
name_fixes�fixesr r r �test_get_headnode_dictp 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_firstr? Zmyfixes.fix_lastr@ Zmyfixes.fix_parrotrA Zmyfixes.fix_preorderrB r Z
get_fixersr )r r? r@ rA rB r ZpreZpostr 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
FixerError)r r r r �test_naughty_fixers� s z'TestRefactoringTool.test_naughty_fixersc C sL | � � }d}|�|d�}| �t|�|� d}|�|d�}| �t|�|� d S )Nzdef parrot(): pass
z<test>zdef f(): pass
)r Zrefactor_string�assertNotEqual�strr) )r r �input�treer r r �test_refactor_string� s z(TestRefactoringTool.test_refactor_stringc sf G � fdd�dt j�}g � |t�}tj}t�d�t_z |�� W d |t_X ddddg}| �� |� d S )Nc s e Zd Z� fdd�ZdS )z5TestRefactoringTool.test_refactor_stdin.<locals>.MyRTc s � � ||||g� d S )N)�extend)r Zold_textZnew_text�filenameZequal)�resultsr r � print_output� s zBTestRefactoringTool.test_refactor_stdin.<locals>.MyRT.print_outputN)r3 r4 r5 rN r )rM r r �MyRT� s rO zdef parrot(): pass
zdef cheese(): pass
z<stdin>F) r r �_DEFAULT_FIXERSr
�stdin�io�StringIOZrefactor_stdinr) )r rO r Zsave�expectedr )rM r �test_refactor_stdin� s z'TestRefactoringTool.test_refactor_stdinTc C sv | � |�}| �|�}| j||d�}|r,||_|�|� | �|| �|�� |sPd S |�|d� | �|�}| �||� |S )N)r r T)�init_test_file� read_filer Z log_debug�
refactor_filer) rF ) 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 )Nz2to3-test_refactor)�prefixi� ) �tempfile�mkdtempZ
addCleanup�shutil�rmtree�copy�osr �join�basename�chmod)r rY Ztmpdirr r r rV � s z"TestRefactoringTool.init_test_filec C s t |d��
}|�� S Q R X d S )N�rb)�open�read)r rY �fpr r r rW � s z TestRefactoringTool.read_filec C s> | � |�}| �|�}| j|d�}|�|d� | �|�}||fS )N)r T)rV rW r rX )r rY r r\ r r] r r r rX � s
z!TestRefactoringTool.refactor_filec C s t j�td�}| �|t� d S )Nzparrot_example.py)re r rf r
r^ rP )r rY 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�|� � }x4� D ]}d
|krX| � ||� P qXW | �
d |� f � d S ) Nzparrot_example.pyc s � � | | � d S )N)r )�msg�args)�debug_messagesr r �recording_log_debug� s zXTestRefactoringTool.test_refactor_file_write_unchanged_file.<locals>.recording_log_debugr r! TF)r r rZ r[ zNot writing changes to .*%szNot writing changesz%r not matched in %r) re r rf r
r^ �re�escape�seprg Z assertRegexr )r rY rq Z
message_regex�messager )rp r �'test_refactor_file_write_unchanged_file� s
z;TestRefactoringTool.test_refactor_file_write_unchanged_filec sV � fdd�}|g g � ddddddg}dg}|||� dt j�d d
�g}|||� d S ) Nc s� �fdd�}t jj}|t j_��� }g �tjdd�� zHt�tj� � d�� x$| D ] }t
tj� � |�d�� � qPW |� � � W d |t j_t
�� � X ���� fdd�|D �� d S ) Nc s � � |� d S )N)r )r �fro )�gotr r �mock_refactor_file� s zPTestRefactoringTool.test_refactor_dir.<locals>.check.<locals>.mock_refactor_filez2to3-test_refactor)r_ �a_dir�wbc s g | ]}t j�� |��qS r )re r rf )r% r )�dirr r r' s zHTestRefactoringTool.test_refactor_dir.<locals>.check.<locals>.<listcomp>)r r rX r r` ra re �mkdirr rf rj �closeZ refactor_dirrb rc r) )Z structurerT ry Z save_funcr �fn)r )r| rx r �check� s
z4TestRefactoringTool.test_refactor_dir.<locals>.checkZnothingzhi.pyz.dumbz .after.pyz notpy.npyZsappyrz zstuff.py)re r rf )r r� rI rT r )r r �test_refactor_dir� s
z%TestRefactoringTool.test_refactor_dirc C s t j�td�}| �|� d S )Nzdifferent_encoding.py)re r rf �
TEST_DATA_DIRr^ )r r r r r �test_file_encoding s z&TestRefactoringTool.test_file_encodingc C s t j�td�}| �|�}d S )Nzfalse_encoding.py)re r rf r� r^ )r r r r r r �test_false_file_encoding s z,TestRefactoringTool.test_false_file_encodingc C s. t j�td�}| �|�}| �|�tj�� d S )Nzbom.py) re r rf r� r^ r" �
startswith�codecs�BOM_UTF8)r r r r r r �test_bom s
z TestRefactoringTool.test_bomc C sB t j}dt _z(t j�td�}t�d�}| �||� W d |t _X d S )Nz
zcrlf.pyz
lib2to3.fixes)re �linesepr rf r� r r( r^ )r Zold_sepr r= r r r �test_crlf_newlines s
z&TestRefactoringTool.test_crlf_newlinesc C sD t j�td�}| �|�\}}| �d|� | �d|� | �d|� d S )Nzcrlf.pys
s
)re r rf r� rX ZassertInZ assertNotIn)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
z<test>z&
>>> def parrot():
... return 43
)r Zrefactor_docstringr) rF )r r �doc�outr r r �test_refactor_docstring1 s z+TestRefactoringTool.test_refactor_docstringc C sd ddl m} | jdgd�}| �t|j�d� | jdgd�}x$|jD ]}t||�rBP qBW | �d� d S )Nr )� FixExplicitzmyfixes.fix_explicit)r )r zexplicit fixer not loaded)Zmyfixes.fix_explicitr� r r) �lenZ
post_orderr r )r r� r Zfixr r r �
test_explicitB s
z!TestRefactoringTool.test_explicit) r3 r4 r5 r r r rP r r r# r+ r2 r> rC rE rJ rU � _2TO3_FIXERSr^ rV rW rX rm rv r� r� r� r� r� r� r� r� r r r r r s:
-
r ) �__doc__r
re r� rR rr r` rb ZunittestZlib2to3r r r Z
lib2to3.pgen2r r rf �dirname�__file__r� r
r r( rP r r� ZTestCaser r r r r �<module> s$