Your IP : 216.73.216.182
U
,�]�M � @ s� d dl mZmZ d dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
m Z d dl m
Z
mZmZmZmZ d dlmZmZmZ er�d dlZdd� Zdd � ZG d
d � d e�ZG d d
� d
e�ZG dd� de�ZG dd� dej �Z e� ed�Z dd� Z dd� Z!dd� Z"dd� Z#e$ej d��s(ej j%ej _&d$d d �Z'e$ej d ��sNe
ej d e'� G d d � d e(�Z)G d d!� d!e)�Z*d%d"d#�Z+e$ej d#��s�e
ej d#e+� dS )&� )�print_function�absolute_importN)�dedent)� bind_method�PY26�PY3�PY2�PY27)� check_output�STDOUT�CalledProcessErrorc C s | � d�r| dd� } t| �S )z,
Removes any leading
and dedents.
�
� N)�
startswithr )�code� r �3/usr/lib/python3/dist-packages/future/tests/base.py�
reformat_code s
r c sd | � d�� dd� t� �D �}dd� t� �D �}dd� t� �D �}| �� | ksTtd��dd� }d d
� }||�||�ks|td ��t� fd d�|D ��}tt||��}t� fd
d�|D ��}tt||��} t� fdd�|D ��}
tt||
��} g } tt� ��D ]`}
|
|k�r| � ||
� q�|
|k�r0| � | |
� q�|
|k�rJ| � | |
� q�| � � |
� q�d�
| �S )a
Returns the code block with any ``__future__`` import lines sorted, and
then any ``future`` import lines sorted, then any ``builtins`` import lines
sorted.
This only sorts the lines within the expected blocks.
See test_order_future_lines() for an example.
r
c S s g | ]\}}|� d �r|�qS )�from __future__ import �r ��.0�i�liner r r �
<listcomp>* s
�z&order_future_lines.<locals>.<listcomp>c S s( g | ] \}}|� d �s |� d�r|�qS )z from futurez from pastr r r r r r - s
�c S s g | ]\}}|� d �r|�qS )z
from builtinsr r r r r r 1 s
�zIinternal usage error: dedent the code before calling order_future_lines()c S s t | �dkrt| �S dS )Nr )�len�max�Znumbersr r r �mymax7 s z!order_future_lines.<locals>.mymaxc S s t | �dkrt| �S td�S )Nr �inf)r �min�floatr r r r �mymin: s z!order_future_lines.<locals>.myminz2the __future__ and future imports are out of orderc s g | ] }� | �qS r r �r r ��linesr r r C s c s g | ] }� | �qS r r r# r$ r r r F s c s g | ] }� | �qS r r r# r$ r r r I s ) �split� enumerate�lstrip�AssertionError�sorted�dict�zip�ranger �append�join)r Zuufuture_line_numbersZfuture_line_numbersZbuiltins_line_numbersr r" ZuulZsorted_uufuture_linesZflZsorted_future_linesZblZsorted_builtins_linesZ new_linesr r r$ r �order_future_lines s4
�
r0 c @ s" e Zd ZdZddd�Zdd� ZdS )�VerboseCalledProcessErrorz�
Like CalledProcessError, but it displays more information (message and
script output) for diagnosing test failures etc.
Nc C s || _ || _|| _|| _d S �N)�msg�
returncode�cmd�output)�selfr3 r4 r5 r6 r r r �__init__` s z"VerboseCalledProcessError.__init__c C s d| j | j| j| jf S )Nz>Command '%s' failed with exit status %d
Message: %s
Output: %s)r5 r4 r3 r6 )r7 r r r �__str__f s �z!VerboseCalledProcessError.__str__)N)�__name__�
__module__� __qualname__�__doc__r8 r9 r r r r r1 [ s
r1 c @ s e Zd ZdS )�
FuturizeErrorN�r: r; r< r r r r r> j s r> c @ s e Zd ZdS )�PasteurizeErrorNr? r r r r r@ m s r@ c @ st e Zd ZdZdd� Zddd�Zd d d
�Zd d � Zd d
d�Zdd� Z d dd�Z
d dd�Z d dd�Z de
jfdd�ZdS )!� CodeHandlerzt
Handy mixin for test classes for writing / reading / futurizing /
running .py files in the test suite.
c C sj t d�| _t d�| _tjg| _t�� tj j
| _ t� d�}|rXdt�
� tj | i| _ndt�
� i| _dS )zi
The outputs from the various futurize stages should have the
following headers:
z�
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
a4
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals
from future import standard_library
standard_library.install_aliases()
from builtins import *
�
PYTHONPATHN)r �headers1�headers2�sys�
executable� interpreters�tempfileZmkdtemp�os�path�sep�tempdir�getenv�getcwd�pathsep�env)r7 Zpypathr r r �setUpv s
zCodeHandler.setUp�r � FTc C sP |r t |�}| �|� | j||||d� | �� }|rL| jD ]} | j| d�}
q:|S )a�
Converts the code block using ``futurize`` and returns the
resulting code.
Passing stages=[1] or stages=[2] passes the flag ``--stage1`` or
``stage2`` to ``futurize``. Passing both stages runs ``futurize``
with both stages by default.
If from3 is False, runs ``futurize``, converting from Python 2 to
both 2 and 3. If from3 is True, runs ``pasteurize`` to convert
from Python 3 to both 2 and 3.
Optionally reformats the code block first using the reformat() function.
If run is True, runs the resulting code under all Python
interpreters in self.interpreters.
)�stages� all_imports�from3� conservative)� interpreter)r �_write_test_script�_futurize_test_script�_read_test_scriptrG �_run_test_script) r7 r rT rU rV Zreformat�runrW r6 rX �_r r r �convert� s
�
zCodeHandler.convertc C sp |r| � |�}| � |�}t|t�r6t|t�s6|�d�}t|t�rTt|t�sT|�d�}| �t|�� �|�� � dS )a�
Compares whether the code blocks are equal. If not, raises an
exception so the test fails. Ignores any trailing whitespace like
blank lines.
If ignore_imports is True, passes the code blocks into the
strip_future_imports method.
If one code block is a unicode string and the other a
byte-string, it assumes the byte-string is encoded as utf-8.
�utf-8N)�strip_future_imports�
isinstance�bytes�decodeZ assertEqualr0 �rstrip)r7 r6 �expected�ignore_importsr r r �compare� s
�zCodeHandler.comparec C s` g }|� d�D ]F}|�d�s|�d�s|�d�sd|ksd|ks|�d�s|�|� qd�|�S )a
Strips any of these import lines:
from __future__ import <anything>
from future <anything>
from future.<anything>
from builtins <anything>
or any line containing:
install_hooks()
or:
install_aliases()
Limitation: doesn't handle imports split across multiple lines like
this:
from __future__ import (absolute_import, division, print_function,
unicode_literals)
r
r z from future zfrom builtins zinstall_hooks()zinstall_aliases()z from future.)r& r r. r/ )r7 r r6 r r r r ra � s
����� z CodeHandler.strip_future_importsc C s` | j ||||||d�} |r0d|kr(| jn| j}
nd}
t|�} |
| krHd}
| j| |
| |d� dS )a<
Convenience method that calls convert() and compare().
Reformats the code blocks automatically using the reformat_code()
function.
If all_imports is passed, we add the appropriate import headers
for the stage(s) selected to the ``expected`` code-block, so they
needn't appear repeatedly in the test code.
If ignore_imports is True, ignores the presence of any lines
beginning:
from __future__ import ...
from future import ...
for the purpose of the comparison.
)rT rU rV r] rW rS � )rg N)r_ rD rC r rh ) r7 Zbeforerf rT rU rg rV r] rW r6 ZheadersZ reformattedr r r �
convert_check� s
� �zCodeHandler.convert_checkc K s | j ||f|� dS )ze
Convenience method to ensure the code is unchanged by the
futurize process.
N)rj )r7 r �kwargsr r r � unchanged s zCodeHandler.unchanged�mytestscript.pyc C sH t |t�r|�d�}tj| j| ddd��}|�t|�� W 5 Q R X dS )z�
Dedents the given code (a multiline string) and writes it out to
a file in a temporary folder like /tmp/tmpUDCn7x/mytestscript.py.
r` Zwt��encodingN)rb rc rd �io�openrL �writer )r7 r �filename�fr r r rY ! s
z CodeHandler._write_test_scriptc C s. t j| j| ddd��}|�� }W 5 Q R X |S )NZrtr` rn )rp rq rL �read)r7 rs rt Z newsourcer r r r[ , s z CodeHandler._read_test_scriptc C s@ g }t |�}|r|�d� |r$d}nNd}|dgkr>|�d� n&|dgkrT|�d� n|ddgksdt�|rr|�d� | j| }tj|g| d |g } zt| t| jd
�}
W n� t k
�r: } ztt
|��*} d d � | �d
| j |d| � � f }
W 5 Q R X d|kr�t
nt}t| d��sd | _||
| j| j| jd��W 5 d } ~ X Y nX |
S )Nz
--all-importsz
pasteurize.pyz futurize.pyr z--stage1rS z--stage2z--conservativez-w)�stderrrP �8Error running the command %s
%s
Contents of file %s:
%s� �env=%s� ----
%s
----Zfuturizer6 �r6 )�listr. r) rL rE rF r
r rP r rq r/ ru r> r@ �hasattrr6 r4 r5 )r7 rs rT rU rV rW ZparamsZscript�fnZ call_argsr6 �ert r3 Z
ErrorClassr r r rZ 1 s@
�� (z!CodeHandler._futurize_test_scriptc C s� | j | }zt||g| jtd�}W n� tk
r� } zft|��.}dd�||g�d| j |d|�� f }W 5 Q R X t|d�s�d |_ t
||j |j |j d��W 5 d }~X Y nX |S )N)rP rv rw rx ry rz r6 r{ )
rL r
rP r r rq r/ ru r} r6 r1 r4 r5 )r7 rs rX r~ r6 r rt r3 r r r r\ ` s&
�
��
(z CodeHandler._run_test_scriptN)rR FFTTF)T)rR FTFTF)rm )rm )rm rR FFF)r: r; r< r= rQ r_ rh ra rj rl rY r[ rZ rE rF r\ r r r r rA q s4 + �
" �
#
�
/�rA z#this test is known to fail on Py2.6c C s t s| S t�| �S r2 )r �unittest�expectedFailure��funcr r r �expectedFailurePY3 s r� c C s t s| S t�| �S r2 )r r� r� r� r r r �expectedFailurePY26� s r� c C s t s| S t�| �S r2 )r r� r� r� r r r �expectedFailurePY27� s r� c C s t s| S t�| �S r2 )r r� r� r� r r r �expectedFailurePY2� s r� �assertRaisesRegexc C sT t |ttf�r$|std��t�|�}|�|�sP|p4d}d||j|f }| �|��dS )z=Fail the test unless the text matches the regular expression.z!expected_regex must not be empty.zRegex didn't matchz%s: %r not found in %rN) rb �strZunicoder) �re�compile�search�pattern�failureException)r7 �text�expected_regexr3 r r r � assertRegex� s
r� c @ s& e Zd Zddd�Zdd� Zdd� ZdS ) �_AssertRaisesBaseContextNc C sp || _ || _|d k rBz |j| _W qH tk
r> t|�| _Y qHX nd | _t|ttf�r`t� |�}|| _
d | _ d S r2 ) rf � test_caser: �obj_name�AttributeErrorr� rb rc r� r� r� r3 )r7 rf r� � callable_objr� r r r r8 � s
z!_AssertRaisesBaseContext.__init__c C s | j �| j|�}| j �|��d S r2 )r� Z_formatMessager3 r� )r7 Z standardMsgr3 r r r �
_raiseFailure� s z&_AssertRaisesBaseContext._raiseFailurec C s8 |dkr|� dd�| _| S | � |||� W 5 Q R X dS )z�
If callable_obj is None, assertRaises/Warns is being used as a
context manager, so check for a 'msg' kwarg and return self.
If callable_obj is not None, call it passing args and kwargs.
Nr3 )�popr3 )r7 �namer� �argsrk r r r �handle� s
z _AssertRaisesBaseContext.handle)NN)r: r; r< r8 r� r� r r r r r� � s
�
r� c @ s e Zd ZdZdd� Zdd� ZdS )�_AssertWarnsContextzBA context manager used to implement TestCase.assertWarns* methods.c C sN t j�� D ]}t|dd �r
i |_q
tjdd�| _| j�� | _t� d| j
� | S )N�__warningregistry__T)�record�always) rE �modules�values�getattrr� �warnings�catch_warnings�warnings_manager� __enter__� simplefilterrf )r7 �vr r r r� � s z _AssertWarnsContext.__enter__c C s | j �|||� |d k r d S z | jj}W n tk
rF t| j�}Y nX d }| jD ]Z}|j}t|| j�sjqR|d krv|}| j d k r�| j �
t|��s�qR|| _ |j | _ |j
| _
d S |d k r�| �d�| j jt|��� | jr�| �d�|| j�� n| �d�|�� d S )Nz"{}" does not match "{}"z{} not triggered by {}z{} not triggered)r� �__exit__rf r: r� r� r� �messagerb r� r� Zwarningrs �linenor� �formatr� r� )r7 �exc_type� exc_value�tbZexc_nameZfirst_matching�m�wr r r r� � s@
� �
�z _AssertWarnsContext.__exit__N)r: r; r< r= r� r� r r r r r� � s r� c O s t || |�}|�d|||�S )a� Fail unless a warning of class warnClass is triggered
by callable_obj when invoked with arguments args and keyword
arguments kwargs. If a different type of warning is
triggered, it will not be handled: depending on the other
warning filtering rules in effect, it might be silenced, printed
out, or raised as an exception.
If called with callable_obj omitted or None, will return a
context object used like this::
with self.assertWarns(SomeWarning):
do_something()
An optional keyword argument 'msg' can be provided when assertWarns
is used as a context object.
The context manager keeps a reference to the first matching
warning as the 'warning' attribute; similarly, the 'filename'
and 'lineno' attributes give you information about the line
of Python code from which the warning was triggered.
This allows you to inspect the warning after the assertion::
with self.assertWarns(SomeWarning) as cm:
do_something()
the_warning = cm.warning
self.assertEqual(the_warning.some_attribute, 147)
� assertWarns)r� r� )r7 Zexpected_warningr� r� rk �contextr r r r� � s r� )N)N),Z
__future__r r rI rH r� rE r� r� rp �textwrapr Z future.utilsr r r r r Zfuture.moves.subprocessr
r r Z unittest2r r0 r1 r> r@ ZTestCaserA ZskipIfZskip26r� r� r� r� r} ZassertRaisesRegexpr� r� �objectr� r� r� r r r r �<module> sF @
"1