Your IP : 216.73.216.213
3
� fl � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl m
Z
ej j ej j
e�d�Zej j ed�ZG dd� dej�Zedkr�ej
� dS )� N)�main�datazpy2_test_grammar.pyc @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�TestMainc C s
d | _ d S )N)�temp_dir)�self� r �</opt/alt/python36/lib64/python3.6/lib2to3/tests/test_main.py�setUp s zTestMain.setUpc C s$ t jjd d �= | jr tj| j� d S )N)�logging�rootZhandlersr �shutilZrmtree)r r r r �tearDown s zTestMain.tearDownc C sJ t j}t j}t j}|t _|t _|t _z tjd|�S |t _|t _|t _X d S )Nz
lib2to3.fixes)�sys�stdin�stdout�stderrr )r �argsZ
in_captureZ out_captureZ err_captureZ
save_stdinZ save_stdoutZ save_stderrr r r �run_2to3_capture s zTestMain.run_2to3_capturec C st t jd�}t j� }tjd�|�}t j� }| jdg|||�}| j|d� |j� jd�}| j d|� | j d|j� � d S )Nu print 'nothing'
print u'über'
�ascii�-r z-print 'nothing'z9WARNING: couldn't encode <stdin>'s diff for your terminal)
�io�StringIO�BytesIO�codecs� getwriterr � assertEqual�getvalue�decode�assertIn)r Z input_stream�outZout_enc�err�ret�outputr r r �test_unencodable_diff, s
z TestMain.test_unencodable_diffc C s� t j� | _tjj| jd�| _tjj| jd�| _tj| j� tj| j� g | _ t
tjj| jd�d�j � | j j d� t
jt| j� | j j tjjt�� tjj| jd�| _tjj| jd�| _t
| jd��}|jd� W dQ R X | j j d� dS )z5Setup a test source tree and output destination tree.Zpython2_projectZpython3_projectz __init__.py�wz
trivial.pyz#print 'I need a simple conversion.'N)�tempfileZmkdtempr �os�path�join� py2_src_dir� py3_dest_dir�mkdir� setup_files�open�close�appendr �copy�PY2_TEST_MODULE�basename�trivial_py2_file�
init_py2_file�write)r Ztrivialr r r �setup_test_source_trees8 s
z TestMain.setup_test_source_treesc s" | j � tj� }tj� }d� | jdd� ddd| j| jgtjd�||�}| j|d� |j� }| jd |� | jd
| j| jf |� | jt � fd d �| j
D ��t t j | j��� x>| j
D ]4}| jd
t j
j| j|�t j
j| j|� �f |� q�W tjt j�}| j|djd|�� | j|djd|�� dS )z92to3 a single directory with a new output dir and suffix.ZTESTz-nz --add-suffixz--write-unchanged-filesz
--no-diffsz --output-dir� r z implies -w.z6Output in %r will mirror the input directory %r layoutc 3 s | ]}|� V qd S )Nr )�.0�name)�suffixr r � <genexpr>\ s zGTestMain.test_filename_changing_on_output_single_dir.<locals>.<genexpr>zWriting converted %s to %sz No changes to .*/__init__\.py�/z No changes to .*/trivial\.pyN)r6 r r r r* r) r r r �setr, r& �listdirr' r( �re�escape�sepZ assertRegex�replaceZassertNotRegex)r r r r! r r9 rA r )r: r �+test_filename_changing_on_output_single_dirK s4
z4TestMain.test_filename_changing_on_output_single_dirc C s� | j � tj� }| j| jg}tdd� |D ��}| jddddd| jg| tjd�tj� |�}| j|d � |j � }| j
d
| j| j f |� | j|tt j
| j��� d S ) z62to3 two files in one directory with a new output dir.c s s | ]}t jj|�V qd S )N)r& r' r2 )r8 r9 r r r r; m s zFTestMain.test_filename_changing_on_output_two_files.<locals>.<genexpr>z-nz-wz--write-unchanged-filesz
--no-diffsz --output-dirr7 r z6Output in %r will mirror the input directory %r layoutN)r6 r r r3 r4 r= r r* r r r r) r& r> )r r Z py2_filesZexpected_filesr! r r r r �*test_filename_changing_on_output_two_filesh s z3TestMain.test_filename_changing_on_output_two_filesc C s� | j � tj� }| jdddd| j| jgtjd�tj� |�}| j|d� |j� }| jd| j| j f |� | jt
t j j
| j�g�t
t j| j��� dS ) z)2to3 a single file with a new output dir.z-nz-wz
--no-diffsz --output-dirr7 r z6Output in %r will mirror the input directory %r layoutN)r6 r r r r* r3 r r r r) r= r& r' r2 r> )r r r! r r r r �,test_filename_changing_on_output_single_filey s z5TestMain.test_filename_changing_on_output_single_fileN) �__name__�
__module__� __qualname__r r
r r# r6 rC rD rE r r r r r s r �__main__)r r r
r& r? r r r% ZunittestZlib2to3r r' r( �dirname�__file__Z
TEST_DATA_DIRr1 ZTestCaser rF r r r r �<module> s w