B
    ØÓY\$  ã               @   sâ   d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZ d dl	m
Z
 d dlmZ d dlmZ d dlmZmZ d dlmZmZmZmZmZmZmZmZmZ d dlmZ eZG d	d
„ d
eƒZG dd„ dƒZ G dd„ deƒZ!dS )é    N)ÚDistutilsErrorÚDistutilsOptionError)Úlog)Ú
TestLoader)Úsix)ÚmapÚfilter)	Úresource_listdirÚresource_existsÚnormalize_pathÚworking_setÚ_namespace_packagesÚevaluate_markerÚadd_activation_listenerÚrequireÚ
EntryPoint)ÚCommandc               @   s   e Zd Zdd„ Zddd„ZdS )ÚScanningLoaderc             C   s   t  | ¡ tƒ | _d S )N)r   Ú__init__ÚsetÚ_visited)Úself© r   ú6lib/python3.7/site-packages/setuptools/command/test.pyr      s    
zScanningLoader.__init__Nc             C   sæ   || j krdS | j  |¡ g }| t | |¡¡ t|dƒrH| | ¡ ¡ t|dƒrÄxpt|jdƒD ]`}| 	d¡rŽ|dkrŽ|jd |dd…  }n"t
|j|d	 ƒr`|jd | }nq`| |  |¡¡ q`W t|ƒd
krÚ|  |¡S |d S dS )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
   ZloadTestsFromNameÚlenZ
suiteClass)r   ÚmoduleÚpatternZtestsÚfileZ	submoduler   r   r   r"      s$    



z"ScanningLoader.loadTestsFromModule)N)r$   Ú
__module__Ú__qualname__r   r"   r   r   r   r   r      s   r   c               @   s   e Zd Zdd„ Zddd„ZdS )ÚNonDataPropertyc             C   s
   || _ d S )N)Úfget)r   r-   r   r   r   r   @   s    zNonDataProperty.__init__Nc             C   s   |d kr| S |   |¡S )N)r-   )r   ÚobjZobjtyper   r   r   Ú__get__C   s    zNonDataProperty.__get__)N)r$   r*   r+   r   r/   r   r   r   r   r,   ?   s   r,   c               @   s¢   e Zd ZdZdZdddgZdd„ Zdd	„ Zed
d„ ƒZ	dd„ Z
dd„ Zejg fdd„ƒZeejdd„ ƒƒZedd„ ƒZdd„ Zdd„ Zedd„ ƒZedd„ ƒZdS )Útestz.Command to run unit tests after in-place buildz#run unit tests after in-place build)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=ÚrzTest runner to usec             C   s   d | _ d | _d | _d | _d S )N)Ú
test_suiteÚtest_moduleÚtest_loaderÚtest_runner)r   r   r   r   Úinitialize_optionsU   s    ztest.initialize_optionsc             C   sŒ   | j r| jrd}t|ƒ‚| j d krD| jd kr8| jj | _ n| jd | _ | jd kr^t| jdd ƒ| _| jd krnd| _| jd krˆt| jdd ƒ| _d S )Nz1You may specify a module or a suite, but not bothz.test_suiter6   z&setuptools.command.test:ScanningLoaderr7   )r4   r5   r   Údistributionr6   Úgetattrr7   )r   Úmsgr   r   r   Úfinalize_options[   s    




ztest.finalize_optionsc             C   s   t |  ¡ ƒS )N)ÚlistÚ
_test_args)r   r   r   r   Ú	test_argsn   s    ztest.test_argsc             c   s4   | j stjdkrdV  | jr"dV  | j r0| j V  d S )N)é   é   Zdiscoverz	--verbose)r4   ÚsysÚversion_infoÚverbose)r   r   r   r   r>   r   s    ztest._test_argsc          	   C   s   |   ¡  |ƒ  W dQ R X dS )zI
        Backward compatibility for project_on_sys_path context.
        N)Úproject_on_sys_path)r   Úfuncr   r   r   Úwith_project_on_sys_pathz   s    
ztest.with_project_on_sys_pathc       	   
   c   sP  t jot| jddƒ}|rv| jddd |  d¡ |  d¡}t|jƒ}| jd|d |  d¡ | jddd |  d¡ n"|  d¡ | jdd	d |  d¡ |  d¡}t	j
d d … }t	j ¡ }zbt|jƒ}t	j
 d|¡ t ¡  td
d„ ƒ td|j|jf ƒ |  |g¡ d V  W d Q R X W d |t	j
d d …< t	j ¡  t	j |¡ t ¡  X d S )NÚuse_2to3FZbuild_pyr   )ZinplaceZegg_info)Úegg_baseZ	build_extr   c             S   s   |   ¡ S )N)Zactivate)Údistr   r   r   Ú<lambda>¥   s    z*test.project_on_sys_path.<locals>.<lambda>z%s==%s)r   ÚPY3r:   r9   Zreinitialize_commandZrun_commandZget_finalized_commandr   Z	build_librB   ÚpathÚmodulesÚcopyrI   Úinsertr   r   r   r   Zegg_nameZegg_versionÚpaths_on_pythonpathÚclearÚupdate)	r   Zinclude_distsZ	with_2to3Zbpy_cmdZ
build_pathZei_cmdZold_pathZold_modulesZproject_pathr   r   r   rE      s8    









ztest.project_on_sys_pathc          
   c   sŠ   t ƒ }tj d|¡}tj dd¡}z>tj | ¡}td||gƒ}tj |¡}|rX|tjd< dV  W d||krztj dd¡ n
|tjd< X dS )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.
        Z
PYTHONPATHr   N)ÚobjectÚosÚenvironÚgetÚpathsepÚjoinr   Úpop)ÚpathsZnothingZorig_pythonpathZcurrent_pythonpathÚprefixZto_joinÚnew_pathr   r   r   rQ   ¯   s    


ztest.paths_on_pythonpathc             C   sD   |   | j¡}|   | jpg ¡}|   dd„ | 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 ¡rt|dd… ƒr|V  qdS )ú:r   N)Ú
startswithr   )Ú.0ÚkÚvr   r   r   ú	<genexpr>Ò   s    z%test.install_dists.<locals>.<genexpr>)Zfetch_build_eggsZinstall_requiresZtests_requireZextras_requireÚitemsÚ	itertoolsÚchain)rJ   Zir_dZtr_dZer_dr   r   r   Úinstall_distsÉ   s
    ztest.install_distsc          
   C   s„   |   | j¡}d | j¡}| jr0|  d| ¡ d S |  d| ¡ tt d¡|ƒ}|  	|¡" |  
¡  |  ¡  W d Q R X W d Q R X d S )Nú zskipping "%s" (dry run)zrunning "%s"Úlocation)rg   r9   rY   Ú_argvZdry_runÚannouncer   ÚoperatorÚ
attrgetterrQ   rE   Ú	run_tests)r   Zinstalled_distsÚcmdr[   r   r   r   Úrun×   s    
ztest.runc             C   sØ   t jr‚t| jddƒr‚| j d¡d }|tkr‚g }|tjkrD| 	|¡ |d7 }x"tjD ]}| 
|¡rT| 	|¡ qTW tttjj|ƒƒ tjd d | j|  | j¡|  | j¡dd}|j ¡ sÔd|j }|  |tj¡ t|ƒ‚d S )NrH   Fr   r   )Z
testLoaderZ
testRunnerÚexitzTest failed: %s)r   rL   r:   r9   r4   Úsplitr   rB   rN   r!   r_   r=   r   Ú__delitem__ÚunittestÚmainrj   Ú_resolve_as_epr6   r7   ÚresultZwasSuccessfulrk   r   ZERRORr   )r   r'   Zdel_modulesÚnamer0   r;   r   r   r   rn   æ   s(    






ztest.run_testsc             C   s   dg| j  S )Nrt   )r?   )r   r   r   r   rj     s    z
test._argvc             C   s$   | dkrdS t  d|  ¡}| ¡ ƒ S )zu
        Load the indicated attribute value, called, as a as if it were
        specified as an entry point.
        Nzx=)r   ÚparseZresolve)ÚvalZparsedr   r   r   rv     s    ztest._resolve_as_epN)r$   r*   r+   Ú__doc__ZdescriptionZuser_optionsr8   r<   r,   r?   r>   rG   Ú
contextlibÚcontextmanagerrE   ÚstaticmethodrQ   rg   rp   rn   Úpropertyrj   rv   r   r   r   r   r0   I   s&   -r0   )"rU   rl   rB   r|   re   rt   Zdistutils.errorsr   r   Z	distutilsr   r   Zsetuptools.externr   Zsetuptools.extern.six.movesr   r   Zpkg_resourcesr	   r
   r   r   r   r   r   r   r   Z
setuptoolsr   ÚtypeZ__metaclass__r   r,   r0   r   r   r   r   Ú<module>   s    ,)
