U
+ �a% � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl m
Z
d dlm Z d dl m
Z
mZmZmZmZmZmZmZmZ d dlmZ d dlmZ G dd � d e �ZG d
d � d �ZG d d
� d
e�Z dS )� N)�DistutilsError�DistutilsOptionError)�log)�
TestLoader) �resource_listdir�resource_exists�normalize_path� working_set�_namespace_packages�evaluate_marker�add_activation_listener�require�
EntryPoint)�Command)�unique_everseenc @ s e Zd Zdd� Zddd�ZdS )�ScanningLoaderc C s t �| � t� | _d S �N)r �__init__�set�_visited��self� r �?/tmp/pip-install-1bd_1mtk/setuptools/setuptools/command/test.pyr s
zScanningLoader.__init__Nc C s� || j krdS | j �|� g }|�t�| |�� t|d�rH|�|�� � t|d�r�t|jd�D ]`}|� d�r�|dkr�|jd |dd� }n"t
|j|d �r^|jd | }nq^|�| � |�� q^t |�d
kr�| �
|�S |d S dS ) a Return a suite of all tests cases contained in the given module
If the module is a package, load tests from all the modules in it.
If the module has an ``additional_tests`` function, call it and add
the return value to the tests.
N�additional_tests�__path__� z.pyz __init__.py�.���z /__init__.py� r )r �add�appendr �loadTestsFromModule�hasattrr r �__name__�endswithr ZloadTestsFromName�lenZ
suiteClass)r �module�pattern�tests�file� submoduler r r r" s$
z"ScanningLoader.loadTestsFromModule)N)r$ �
__module__� __qualname__r r" r r r r r s r c @ s e Zd Zdd� Zddd�ZdS )�NonDataPropertyc C s
|| _ d S r ��fget)r r0 r r r r < s zNonDataProperty.__init__Nc C s |d kr | S | � |�S r r/ )r �objZobjtyper r r �__get__? s zNonDataProperty.__get__)N)r$ r, r- r r2 r r r r r. ; s r. c @ s� e Zd ZdZdZdddgZdd� Zdd � Zed
d � �Z d d
� Z
dd� Z e j
g fdd��Zee j
dd� ��Zedd� �Zdd� Zdd� Zedd� �Zed d � �Zd S ) �testz.Command to run unit tests after in-place buildz0run unit tests after in-place build (deprecated))z test-module=�mz$Run 'test_suite' in specified module)z test-suite=�sz9Run single test, case or suite (e.g. 'module.test_suite'))z test-runner=�rzTest runner to usec C s d | _ d | _d | _d | _d S r )�
test_suite� test_module� test_loader� test_runnerr r r r �initialize_optionsQ s ztest.initialize_optionsc C s� | j r| jrd}t|��| j d krD| jd kr8| jj | _ n | jd | _ | jd kr^t| jdd �| _| jd krnd| _| jd kr�t| jdd �| _d S )Nz1You may specify a module or a suite, but not bothz .test_suiter9 z&setuptools.command.test:ScanningLoaderr: )r7 r8 r � distributionr9 �getattrr: )r �msgr r r �finalize_optionsW s
ztest.finalize_optionsc C s t | �� �S r )�list�
_test_argsr r r r � test_argsj s ztest.test_argsc c s4 | j stjdkrdV | jr"dV | j r0| j V d S )N)� � Zdiscoverz --verbose)r7 �sys� version_info�verboser r r r rA n s ztest._test_argsc C s | � � � |� W 5 Q R X dS )zI
Backward compatibility for project_on_sys_path context.
N)�project_on_sys_path)r �funcr r r �with_project_on_sys_pathv s
z test.with_project_on_sys_pathc
c sJ t | jdd�}|rp| jddd� | �d� | �d�}t|j�}| jd|d� | �d� | jddd� | �d� n"| �d� | jdd d� | �d� | �d�}tjd d � }tj �
� }zbt|j�}tj�d|� t
�� td
d � � td |j|jf � | �|g�� d V W 5 Q R X W 5 |tjd d �< tj � � tj � |� t
�� X d S )
N�use_2to3F�build_pyr )Zinplace�egg_info)�egg_base� build_extr c S s | � � S r )�activate)�distr r r �<lambda>� � z*test.project_on_sys_path.<locals>.<lambda>z%s==%s)r= r< �reinitialize_command� run_command�get_finalized_commandr � build_librE �path�modules�copy�clear�updater r rN �insertr r
�egg_nameZ egg_version�paths_on_pythonpath) r Z
include_distsZ with_2to3Zbpy_cmdZ
build_pathZei_cmd�old_pathZ old_modulesZ project_pathr r r rH } s8
ztest.project_on_sys_pathc
c s� t � }tj�d|�}tj�dd�}zBtj�t| ��}td||g�}tj�|�}|r\|tjd< dV W 5 ||kr~tj�dd� n
|tjd< X dS )z�
Add the indicated paths to the head of the PYTHONPATH environment
variable so that subprocesses will also see the packages at
these paths.
Do this in a context that restores the value on exit.
�
PYTHONPATHr N) �object�os�environ�get�pop�pathsep�joinr �filter)�pathsZnothingZorig_pythonpathZcurrent_pythonpath�prefixZto_join�new_pathr r r r_ � s
ztest.paths_on_pythonpathc C sD | � | j�}| � | jpg �}| � dd� | j�� D ��}t�|||�S )z�
Install the requirements indicated by self.distribution and
return an iterable of the dists that were built.
c s s0 | ](\}}|� d �rt|dd� �r|V qdS )�:r N)�
startswithr )�.0�k�vr r r � <genexpr>� s
�z%test.install_dists.<locals>.<genexpr>)Zfetch_build_eggsZinstall_requiresZ
tests_requireZextras_require�items� itertools�chain)rQ Zir_dZtr_dZer_dr r r �
install_dists� s
�ztest.install_distsc
C s� | � dtj� | �| j�}d�| j�}| jr>| � d| � d S | � d| � tt �
d�|�}| � |��" | � � � | �
� W 5 Q R X W 5 Q R X d S )Nz�WARNING: Testing via this command is deprecated and will be removed in a future version. Users looking for a generic test entry point independent of test runner are encouraged to use tox.� zskipping "%s" (dry run)z running "%s"�location)�announcer �WARNrv r< rh �_argv�dry_run�map�operator�
attrgetterr_ rH � run_tests)r Zinstalled_dists�cmdrj r r r �run� s �
ztest.runc C s� t | jdd�rx| j�d�d }|tkrxg }|tjkr>|�|� |d7 }tjD ]}|�|�rL|�|� qLt t
tjj |�� t j
d d | j| �| j�| �| j�dd�}|j�� s�d|j }| �|tj� t|��d S )NrK Fr r )Z
testLoaderZ
testRunner�exitzTest failed: %s)r= r< r7 �splitr
rE rY r! rn r@ r} � __delitem__�unittest�mainr{ �_resolve_as_epr9 r: �resultZ
wasSuccessfulry r �ERRORr )r r' Z del_modules�namer3 r> r r r r� � s.
�
ztest.run_testsc C s dg| j S )Nr� )rB r r r r r{ s z
test._argvc C s$ | dkr dS t �d| �}|�� � S )zu
Load the indicated attribute value, called, as a as if it were
specified as an entry point.
Nzx=)r �parse�resolve)�val�parsedr r r r� s ztest._resolve_as_epN)r$ r, r- �__doc__� description� user_optionsr; r? r. rB rA rJ �
contextlib�contextmanagerrH � staticmethodr_ rv r� r� �propertyr{ r� r r r r r3 E s2 �
-
r3 ) rc r~ rE r� rt r� �distutils.errorsr r � distutilsr r �
pkg_resourcesr r r r r
r r r
r �
setuptoolsr Z setuptools.extern.more_itertoolsr r r. r3 r r r r �<module> s , )