File: //usr/local/lib/python3.7/unittest/test/__pycache__/test_discovery.cpython-37.opt-2.pyc
B
"��g�� � @ s� d dl Zd dl mZ d dlZd dlZd dlZd dlZd dlmZ d dl Zd dl
Z
d dl Z
d dl Z
G dd� de
j
�ZG dd� de
j�Zedkr�e
�� dS ) � N)�abspath)�supportc @ s< e Zd ZdZdZd Z Z ZZdZ dZ
d Z Z dd� Z
dS )�TestableTestProgramNT� � c C s d S )N� )�selfr r �8/usr/local/lib/python3.7/unittest/test/test_discovery.py�__init__ s z TestableTestProgram.__init__)�__name__�
__module__� __qualname__�module�exitZ defaultTest�failfast�
catchbreak�buffer� verbosityZprogNameZ
testRunner�
testLoaderr
r r r r r s r c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7S )8�
TestDiscoveryc C s2 t �� }d|_|�d�}| �|d� d S Q R X d S )Nz/fooz/foo/bar/baz.pyzbar.bazz /bar/baz.py)�unittest�
TestLoader�_top_level_dirZ_get_name_from_path� assertEqual� assertRaises�AssertionError)r �loader�namer r r �test_get_name_from_path s
z%TestDiscovery.test_get_name_from_pathc s* t �� }tj��fdd�}tjj��fdd�}tjj��fdd�}ddd d
d d d
gddgg��fdd�t_| �|� dd� }|tj_| �|� dd� }|tj_| �|� dd� |_|j � d!� fdd� }||_ dd� |_
tj� d�}||_ t
|�|d��} d d � d D �}
|
�d d � d D �� | �| |
� d S )"Nc s
� t _d S )N)�os�listdirr )�original_listdirr r �restore_listdir/ s z6TestDiscovery.test_find_tests.<locals>.restore_listdirc s � t j_d S )N)r �path�isfiler )�original_isfiler r �restore_isfile2 s z5TestDiscovery.test_find_tests.<locals>.restore_isfilec s � t j_d S )N)r r# �isdirr )�original_isdirr r �
restore_isdir5 s z4TestDiscovery.test_find_tests.<locals>.restore_isdirztest2.pyztest1.pyz
not_a_test.py�test_dirztest.fooztest-not-a-module.py� another_dirztest4.pyztest3.pyc s
� � d�S )Nr )�pop)r# )�
path_listsr r �<lambda>; � z/TestDiscovery.test_find_tests.<locals>.<lambda>c S s
| � d�S )N�dir)�endswith)r# r r r r' > s z,TestDiscovery.test_find_tests.<locals>.isdirc S s | � d� od| kS )Nr0 r+ )r1 )r# r r r r$ C s z-TestDiscovery.test_find_tests.<locals>.isfilec S s | d S )Nz moduler )r# r r r r. I r/ c s � | |d�}|| d g S )N)�patternz testsr )r r2 �base)�orig_load_testsr r �loadTestsFromModuleK s z:TestDiscovery.test_find_tests.<locals>.loadTestsFromModulec S s | S )Nr )�thingr r r r. P r/ z/fooztest*.pyc S s g | ]}|d g�qS )z
module testsr )�.0r r r r �
<listcomp>X s z1TestDiscovery.test_find_tests.<locals>.<listcomp>)Ztest1Ztest2r* c S s g | ]}d | d g�qS )z test_dir.%sz
module testsr )r7 r r r r r8 Z s )Ztest3Ztest4)N)r r r r r# r$ r' �
addCleanup�_get_module_from_namer5 �
suiteClassr r �list� _find_tests�extendr ) r r r" r& r) r' r$ r5 � top_level�suiteZexpectedr )r4 r( r% r! r- r �test_find_tests+ s>
z TestDiscovery.test_find_testsc s� t �� }tj��fdd�}tjj��fdd�}tjj��fdd�}dgg��fdd �t_| �|� d
d � tj_| �|� d d � tj_| �|� d d � |_|j � d� fd
d� }||_ dd � |_
tj� d�}||_ t
|�|d��}| �|g � d S )Nc s
� t _d S )N)r r r )r! r r r" d s z=TestDiscovery.test_find_tests_socket.<locals>.restore_listdirc s � t j_d S )N)r r# r$ r )r% r r r& g s z<TestDiscovery.test_find_tests_socket.<locals>.restore_isfilec s � t j_d S )N)r r# r' r )r( r r r) j s z;TestDiscovery.test_find_tests_socket.<locals>.restore_isdirZsocketc s
� � d�S )Nr )r, )r# )r- r r r. n r/ z6TestDiscovery.test_find_tests_socket.<locals>.<lambda>c S s dS )NFr )r# r r r r. q r/ c S s dS )NFr )r# r r r r. t r/ c S s | d S )Nz moduler )r# r r r r. w r/ c s � | |d�}|| d g S )N)r2 z testsr )r r2 r3 )r4 r r r5 y s zATestDiscovery.test_find_tests_socket.<locals>.loadTestsFromModulec S s | S )Nr )r6 r r r r. ~ r/ z/fooztest*.py)N)r r r r r# r$ r' r9 r: r5 r; r r r< r= r )r r r" r& r) r5 r? r@ r )r4 r( r% r! r- r �test_find_tests_socket^ s.
z$TestDiscovery.test_find_tests_socketc s@ t �� }tj��fdd�}tjj��fdd�}tjj��fdd�}ddd g��g g g g��fd
d �t_| �|� d d � tj_| �|� �fd
d �tj_| �|� G dd� dt�� � fdd �|_ |j
�d�fdd� }||_
dd � |_ d|_ t
|�dd��}| �|dgddgdgg� | �� jddd g� | �� j|g dfg� d S )Nc s
� t _d S )N)r r r )r! r r r" � s zCTestDiscovery.test_find_tests_with_package.<locals>.restore_listdirc s � t j_d S )N)r r# r$ r )r% r r r& � s zBTestDiscovery.test_find_tests_with_package.<locals>.restore_isfilec s � t j_d S )N)r r# r' r )r( r r r) � s zATestDiscovery.test_find_tests_with_package.<locals>.restore_isdir� a_directory�test_directory�test_directory2c s
� � d�S )Nr )r, )r# )r- r r r. � r/ z<TestDiscovery.test_find_tests_with_package.<locals>.<lambda>c S s dS )NTr )r# r r r r. � r/ c s t j�| �� kS )N)r r# �basename)r# )� directoriesr r r. � r/ c @ s$ e Zd Zg Zg Zdd� Zdd� ZdS )z:TestDiscovery.test_find_tests_with_package.<locals>.Modulec s8 |� _ � j�|� tj �|�dkr4� fdd�}|� _d S )NrD c s � j �| ||f� � jd gS )Nz load_tests)�load_tests_args�appendr# )r �testsr2 )r r r �
load_tests� s zWTestDiscovery.test_find_tests_with_package.<locals>.Module.__init__.<locals>.load_tests)r# �pathsrI r rF rK )r r# rK r )r r r
� s
zCTestDiscovery.test_find_tests_with_package.<locals>.Module.__init__c S s | j |j kS )N)r# )r �otherr r r �__eq__� s zATestDiscovery.test_find_tests_with_package.<locals>.Module.__eq__N)r r r
rL rH r
rN r r r r �Module� s rO c s � | �S )Nr )r )rO r r r. � r/ c s � | |d�}|| j d g S )N)r2 z
module tests)r# )r r2 r3 )r4 r r r5 � s zGTestDiscovery.test_find_tests_with_package.<locals>.loadTestsFromModulec S s | S )Nr )r6 r r r r. � r/ z/fooztest*za_directory module testsztest_directory load_testsztest_directory module testsz test_directory2 module tests)N)r r r r r# r$ r' r9 �objectr: r5 r; r r< r= r rL rH )r r r" r& r) r5 r@ r )rO rG r4 r( r% r! r- r � test_find_tests_with_package� s@
z*TestDiscovery.test_find_tests_with_packagec s@ t �� }tj��fdd�}tjj��fdd�}tjj��fdd�}ddd g��g g g g��fd
d �t_| �|� d d � tj_| �|� �fd
d �tj_| �|� G dd� dt�� � fdd �|_ |j
�d�fdd� }||_
dd � |_ d|_ t
|�dd��}| �|dgddgdgg� | �� jddd g� | �� j|g dfg� d S )Nc s
� t _d S )N)r r r )r! r r r" � s zWTestDiscovery.test_find_tests_default_calls_package_load_tests.<locals>.restore_listdirc s � t j_d S )N)r r# r$ r )r% r r r& � s zVTestDiscovery.test_find_tests_default_calls_package_load_tests.<locals>.restore_isfilec s � t j_d S )N)r r# r' r )r( r r r) � s zUTestDiscovery.test_find_tests_default_calls_package_load_tests.<locals>.restore_isdirrC rD rE c s
� � d�S )Nr )r, )r# )r- r r r. � r/ zPTestDiscovery.test_find_tests_default_calls_package_load_tests.<locals>.<lambda>c S s dS )NTr )r# r r r r. � r/ c s t j�| �� kS )N)r r# rF )r# )rG r r r. � r/ c @ s$ e Zd Zg Zg Zdd� Zdd� ZdS )zNTestDiscovery.test_find_tests_default_calls_package_load_tests.<locals>.Modulec s8 |� _ � j�|� tj �|�dkr4� fdd�}|� _d S )NrD c s � j �| ||f� � jd gS )Nz load_tests)rH rI r# )r rJ r2 )r r r rK � s zkTestDiscovery.test_find_tests_default_calls_package_load_tests.<locals>.Module.__init__.<locals>.load_tests)r# rL rI r rF rK )r r# rK r )r r r
� s
zWTestDiscovery.test_find_tests_default_calls_package_load_tests.<locals>.Module.__init__c S s | j |j kS )N)r# )r rM r r r rN � s zUTestDiscovery.test_find_tests_default_calls_package_load_tests.<locals>.Module.__eq__N)r r r
rL rH r
rN r r r r rO � s rO c s � | �S )Nr )r )rO r r r. � r/ c s � | |d�}|| j d g S )N)r2 z
module tests)r# )r r2 r3 )r4 r r r5 � s z[TestDiscovery.test_find_tests_default_calls_package_load_tests.<locals>.loadTestsFromModulec S s | S )Nr )r6 r r r r. � r/ z/fooztest*.pyza_directory module testsztest_directory load_testsztest_directory module testsz test_directory2 module tests)N)r r r r r# r$ r' r9 rP r: r5 r; r r< r= r rL rH )r r r" r& r) r5 r@ r )rO rG r4 r( r% r! r- r �0test_find_tests_default_calls_package_load_tests� s@
z>TestDiscovery.test_find_tests_default_calls_package_load_testsc sH t j��fdd�}| �|� t jj��fdd�}| �|� t jj��fdd�}| �|� | �tjjtd�� td�dgtd �d
d gi��fd d
�}|t _dd� t j_dd� t j_G dd� dt �� t
� � }� fdd�|_ dd� |_
td�|_t|�td�d��}| �|dg dggg� | �� jddg� | �� j|g df|g dfg� d S )Nc s
� t _d S )N)r r r )r! r r r" s zTTestDiscovery.test_find_tests_customize_via_package_pattern.<locals>.restore_listdirc s � t j_d S )N)r r# r$ r )r% r r r& ! s zSTestDiscovery.test_find_tests_customize_via_package_pattern.<locals>.restore_isfilec s � t j_d S )N)r r# r' r )r( r r r) % s zRTestDiscovery.test_find_tests_customize_via_package_pattern.<locals>.restore_isdirz/foo�
my_packagez/foo/my_packagez __init__.pyztest_module.pyc s t � | �S )N)r< )r# )�vfsr r �list_dir7 s zMTestDiscovery.test_find_tests_customize_via_package_pattern.<locals>.list_dirc S s | � d� S )Nz.py)r1 )r# r r r r. : r/ zMTestDiscovery.test_find_tests_customize_via_package_pattern.<locals>.<lambda>c S s
| � d�S )Nz.py)r1 )r# r r r r. ; r/ c @ s$ e Zd Zg Zg Zdd� Zdd� ZdS )zKTestDiscovery.test_find_tests_customize_via_package_pattern.<locals>.Modulec s@ |� _ � j�|� |�d�r*� fdd�}n � fdd�}|� _d S )NZ test_modulec s � j �| ||f� � jd gS )Nz load_tests)rH rI r# )r rJ r2 )r r r rK E s zhTestDiscovery.test_find_tests_customize_via_package_pattern.<locals>.Module.__init__.<locals>.load_testsc sB � j �| ||f� d}tj�|�}| j||d�}� jd |g| S )Nz/foo/my_package/__init__.py)� start_dirr2 z load_tests)rH rI r r# �dirname�discover)r rJ r2 �__file__Zthis_dirZ pkg_tests)r r r rK I s
)r# rL rI r1 rK )r r# rK r )r r r
A s
zTTestDiscovery.test_find_tests_customize_via_package_pattern.<locals>.Module.__init__c S s | j |j kS )N)r# )r rM r r r rN T s zRTestDiscovery.test_find_tests_customize_via_package_pattern.<locals>.Module.__eq__N)r r r
rL rH r
rN r r r r rO = s rO c s � | �S )Nr )r )rO r r r. X r/ c S s | S )Nr )r6 r r r r. Y r/ ztest*.pyzmy_package load_testsz!my_package.test_module load_testszmy_package.test_module)r r r9 r# r$ r' �sys�remover rP r r r: r; r r< r= r rL rH )r r" r& r) rU r r@ r )rO r( r% r! rT r �-test_find_tests_customize_via_package_pattern s<
z;TestDiscovery.test_find_tests_customize_via_package_patternc
sb t �� }tjj�tjj��fdd�}dd� tj_| �|� tjd d � ��fdd�}| �|� tj�tj� d��}| �
t �� |j ddd � W d Q R X | �
|j|� | �|tj� d
d� tj_d d� tj_�fd d
�}| �|� g � d� fdd� }||_t|_|� ddd�}tj�d�}tj�d�} | �
|d� | �
|j|� | �
� | dfg� | �|tj� d S )Nc s � t j_d S )N)r r# r$ r )r% r r r& t s z3TestDiscovery.test_discover.<locals>.restore_isfilec S s dS )NFr )r# r r r r. w r/ z-TestDiscovery.test_discover.<locals>.<lambda>c s � t jd d �< d S )N)rZ r# r )�
orig_sys_pathr r � restore_path{ s z1TestDiscovery.test_discover.<locals>.restore_pathz/fooz/foo/bar)�
top_level_dirc S s dS )NTr )r# r r r r. � r/ c S s dS )NTr )r# r r r r. � r/ c s � t j_d S )N)r r# r' r )r( r r r) � s z2TestDiscovery.test_discover.<locals>.restore_isdirc s � � | |f� dgS )NrJ )rI )rV r2 � namespace)�_find_tests_argsr r r= � s z0TestDiscovery.test_discover.<locals>._find_testsz /foo/bar/bazr2 z ['tests'])N)r r r r# r$ r' r9 rZ r �normpathr � ImportErrorrX r r �assertInr= �strr; )
r r r& r^ � full_pathr) r= r@ r_ rV r )ra r] r( r% r �
test_discovero s:
zTestDiscovery.test_discoverc s� t d�dgt d�dgi��fdd�}| �ttdtj� |t_| �ttjdtjj� d d
� tj_| �ttjd tjj� d d
� tj_| �tjj t d�� G d
d� dt
�� t � � }� fdd
�|_
dd
� |_|jddd�}| �|dgg� d S )Nz /toplevelZstartdirz/toplevel/startdirz __init__.pyc s t � | �S )N)r< )r# )rT r r rU � s z[TestDiscovery.test_discover_start_dir_is_package_calls_package_load_tests.<locals>.list_dirr r$ c S s
| � d�S )Nz.py)r1 )r# r r r r. � r/ z[TestDiscovery.test_discover_start_dir_is_package_calls_package_load_tests.<locals>.<lambda>r' c S s | � d� S )Nz.py)r1 )r# r r r r. � r/ c @ s, e Zd Zg Zg Zdd� Zdd� Zdd� ZdS )zYTestDiscovery.test_discover_start_dir_is_package_calls_package_load_tests.<locals>.Modulec S s
|| _ d S )N)r# )r r# r r r r
� s zbTestDiscovery.test_discover_start_dir_is_package_calls_package_load_tests.<locals>.Module.__init__c S s d| j gS )Nzload_tests called )r# )r r rJ r2 r r r rK � s zdTestDiscovery.test_discover_start_dir_is_package_calls_package_load_tests.<locals>.Module.load_testsc S s | j |j kS )N)r# )r rM r r r rN � s z`TestDiscovery.test_discover_start_dir_is_package_calls_package_load_tests.<locals>.Module.__eq__N)r r r
rL rH r
rK rN r r r r rO � s
rO c s � | �S )Nr )r )rO r r r. � r/ c S s | S )Nr )r6 r r r r. � r/ )r_ zload_tests called startdir)r r9 �setattrr r r# r$ r' rZ r[ rP r r r: r; rX r )r rU r r@ r )rO rT r �;test_discover_start_dir_is_package_calls_package_load_tests� s"
zITestDiscovery.test_discover_start_dir_is_package_calls_package_load_testsc sT t j�� fdd�t _t jj�dd� t j_tjd d � ����fdd�}| �|� d S )Nc s � gS )Nr )�_)�fakefiler r r. � r/ z8TestDiscovery.setup_import_issue_tests.<locals>.<lambda>c S s dS )NTr )rj r r r r. � r/ c s � t j_�t _�tjd d �< d S )N)r r# r$ r rZ r )r$ r r] r r �restore� s z7TestDiscovery.setup_import_issue_tests.<locals>.restore)r r r# r$ rZ r9 )r rk rl r )rk r$ r r] r �setup_import_issue_tests� s z&TestDiscovery.setup_import_issue_testsc s� | � ttdtj� | � ttjdtjj� | � ttjdtjj� | � tjjt d �t
tj�� � fdd�}|t_dd� tj_dd� tj_d S ) Nr r$ r' c s t � | �S )N)r< )r# )rT r r rU � s z@TestDiscovery.setup_import_issue_package_tests.<locals>.list_dirc S s | � d� S )Nz.py)r1 )r# r r r r. � r/ z@TestDiscovery.setup_import_issue_package_tests.<locals>.<lambda>c S s
| � d�S )Nz.py)r1 )r# r r r r. � r/ ) r9 rh r r r# r$ r' rZ � __setitem__�slicer< )r rT rU r )rT r � setup_import_issue_package_tests� s z.TestDiscovery.setup_import_issue_package_testsc C s� t �� }| �d� |�d�}| �t�� tj� | � |�
� d� | � g |j � | � dt
|j �� |j d }| �d|kd| � tt|�d �d }| �t�� |�� W d Q R X d S )Nztest_this_does_not_exist.py�.r r z6Failed to import test module: test_this_does_not_existzmissing error string in %r)r r rm rX rd r �getcwdrZ r# r �countTestCases�assertNotEqual�errors�len�
assertTruer< r rc Ztest_this_does_not_exist)r r r@ �error�testr r r �.test_discover_with_modules_that_fail_to_import� s
z<TestDiscovery.test_discover_with_modules_that_fail_to_importc s t d�dgt d�ddgi}| �|� g � � fdd�}t�� }||_|�t d��}| �t d�tj� | � |�
� d� | � g |j � | � dt
|j �� |j d }| �d
|kd | � tt|�d �d }| �t�� |�� W d Q R X | � � dg� x*ttjd �D ]}t�t�||�� �q W d S ) Nz/foorS z/foo/my_packagez __init__.pyztest_module.pyc s � � | � td��d S )NzCannot import Name)rI rc )r )� import_callsr r r: � s
z`TestDiscovery.test_discover_with_init_modules_that_fail_to_import.<locals>._get_module_from_namer r z(Failed to import test module: my_packagezmissing error string in %r)r rp r r r: rX rd rZ r# r rs rt ru rv rw r< r rc rS �range�pickle�HIGHEST_PROTOCOL�loads�dumps)r rT r: r r@ rx ry �protor )r{ r �3test_discover_with_init_modules_that_fail_to_import� s,
zATestDiscovery.test_discover_with_init_modules_that_fail_to_importc C s� t jjst �d��t �� }dd� }||_| �d� |�d�}| �|� � d� t �
� }|� |� | �t |j
�d� x(ttjd �D ]}t�t�||�� q�W d S )NzSuite cleanup is disabledc S s t �d��d S )N� skipperoo)r �SkipTest)r r r r r: s zeTestDiscovery.test_discover_with_module_that_raises_SkipTest_on_import.<locals>._get_module_from_nameztest_skip_dummy.pyrq r )r �
BaseTestSuite�_cleanupr� r r: rm rX r rs �
TestResult�runrv �skippedr| r} r~ r r� )r r r: r@ �resultr� r r r �8test_discover_with_module_that_raises_SkipTest_on_import s
zFTestDiscovery.test_discover_with_module_that_raises_SkipTest_on_importc s� t jjst �d��td�dgtd�ddgi}| �|� g � � fdd�}t �� }||_|�td��}| � td�t
j � | � |�
� d � t �� }|�|� | � t|j�d � | � |jd � | � � dg� x(ttjd �D ]}t�t�||�� q�W d S )
NzSuite cleanup is disabledz/foorS z/foo/my_packagez __init__.pyztest_module.pyc s � � | � t�d��d S )Nr� )rI r r� )r )r{ r r r: 2 s
zjTestDiscovery.test_discover_with_init_module_that_raises_SkipTest_on_import.<locals>._get_module_from_namer )r r� r� r� r rp r r: rX rd rZ r# r rs r� r� rv r� ZtestsRunr| r} r~ r r� )r rT r: r r@ r� r� r )r{ r �=test_discover_with_init_module_that_raises_SkipTest_on_import* s&
zKTestDiscovery.test_discover_with_init_module_that_raises_SkipTest_on_importc C sb t � }g }|j|_|�ddg� | �|g g� g |d d �< |�ddddg� | �|ddgg� d S )N� somethingrX �foo�bar)r rI �
_do_discovery� parseArgsr )r �program�argsr r r �$test_command_line_handling_parseArgsE s z2TestDiscovery.test_command_line_handling_parseArgsc C sh t � }g }|j|_|�dg� | �|g g� | �|jd� | �|jd� | �|jd� | �|j d� d S )Nr� r F)
r rI r� r� r r �assertIsr r r )r r� r� r r r �.test_command_line_handling_discover_by_defaultQ s z<TestDiscovery.test_command_line_handling_discover_by_defaultc C sr t � }g }|j|_|�ddddddg� | �|g g� | �|jd� | �|jd� | �|jd� | �|j d� d S )Nr� z-vz-bz-cz-f� T)
r rI r� r� r r r� r r r )r r� r� r r r �;test_command_line_handling_discover_by_default_with_options] s zITestDiscovery.test_command_line_handling_discover_by_default_with_optionsc
C sl t � }d |_t�� �.}| �t��}|�ddddg� W d Q R X W d Q R X | �|jj d� | �
d|� � � d S )NZoneZtwoZthreeZfour)r� zusage:) r r r Zcaptured_stderrr �
SystemExitr� r � exceptionr� rd �getvalue)r r� �stderr�cmr r r �:test_command_line_handling_do_discovery_too_many_argumentsj s
&zHTestDiscovery.test_command_line_handling_do_discovery_too_many_argumentsc C sL t �tj�}|�� G dd� dt �}|� |_|�dg� | �|jdg� d S )Nc @ s e Zd Zg Zdd� ZdS )zYTestDiscovery.test_command_line_handling_do_discovery_uses_default_loader.<locals>.Loaderc S s | j �|||f� dS )NrJ )r� rI )r rV r2 r_ r r r rX | s zbTestDiscovery.test_command_line_handling_do_discovery_uses_default_loader.<locals>.Loader.discoverN)r r r
r� rX r r r r �Loaderz s r� z-v)rq ztest*.pyN) rP �__new__r � TestProgramZ_initArgParsersr r� r r� )r r� r� r r r �;test_command_line_handling_do_discovery_uses_default_loaderv s zITestDiscovery.test_command_line_handling_do_discovery_uses_default_loaderc C s� t � }G dd� dt�}|jdg|d� | �|jd� | �|jd� | �|jdg� g |_t � }|jdg|d� | �|jd� | �|jdg� g |_t � }|jg |d� | �|jd� | �|jdg� g |_t � }|jd g|d� | �|jd� | �|jd
g� g |_t � }|jd d g|d� | �|jd� | �|jd g� g |_t � }|jd d d
g|d� | �|jd� | �|jdg� g |_t � }|jdd g|d� | �|jd� | �|jd
g� g |_t � }|jdd g|d� | �|jd� | �|jdg� g |_t � }|jdd g|d� | �|jd� | �|jdg� | �|j� | �|j � g |_t � }|jdd dd dddg|d� | �|jd� | �|jd g� | �|jd� | �
|j� | �
|j � d S )Nc @ s e Zd Zg Zdd� ZdS )zRTestDiscovery.test_command_line_handling_do_discovery_calls_loader.<locals>.Loaderc S s | j �|||f� dS )NrJ )r� rI )r rV r2 r_ r r r rX � s z[TestDiscovery.test_command_line_handling_do_discovery_calls_loader.<locals>.Loader.discoverN)r r r
r� rX r r r r r� � s r� z-v)r� r� rJ )rq ztest*.pyNz --verbose�fish)r� ztest*.pyN�eggs)r� r� N�ham)r� r� r� z-sz-t)rq ztest*.pyr� z-p)rq r� Nz-fz-c) r rP r� r r ry r� Z assertFalser r rw )r r� r� r r r �4test_command_line_handling_do_discovery_calls_loader� sr zBTestDiscovery.test_command_line_handling_do_discovery_calls_loaderc s� G dd� dt �}|tjd< tj�d�� tj�tjj�tjj�� ���fdd�}| � |� dd� }dd � }d
d � }|t_|tj_|tj_� S ) Nc @ s e Zd ZdZdS )z0TestDiscovery.setup_module_clash.<locals>.Modulez
bar/foo.pyN)r r r
rY r r r r rO � s rO r� c s8 �t _�t j_�t j_tjd= � tjkr4tj�� � d S )Nr� )r r r# r$ r' rZ �modulesr[ r )rf r( r% r! r r �cleanup� s
z1TestDiscovery.setup_module_clash.<locals>.cleanupc S s dgS )Nzfoo.pyr )rj r r r r � s z1TestDiscovery.setup_module_clash.<locals>.listdirc S s dS )NTr )rj r r r r$ � s z0TestDiscovery.setup_module_clash.<locals>.isfilec S s dS )NTr )rj r r r r' � s z/TestDiscovery.setup_module_clash.<locals>.isdir)
rP rZ r� r r# r r r$ r' r9 )r rO r� r r$ r' r )rf r( r% r! r �setup_module_clash� s
z TestDiscovery.setup_module_clashc C sj | � � }t�� }tj�d�}tj�d�}t�d||f �}| jt d| |j
ddd� | � t jd |� d S )Nr� r� zZ'foo' module incorrectly imported from %r. Expected %r. Is this module globally installed?z^%s$zfoo.py)rV r2 r )
r� r r r r# r �re�escapeZassertRaisesRegexrc rX r rZ )r rf r �mod_dir� expected_dir�msgr r r �test_detect_module_clash� s
z&TestDiscovery.test_detect_module_clashc sn | � � }tjj�tj�d��tj�d�� �fdd�}| �|� � �fdd�}|tj_t�� }|jddd� d S ) Nr� r� c s � t j_d S )N)r r# �realpathr )�original_realpathr r r� s z5TestDiscovery.test_module_symlink_ok.<locals>.cleanupc s$ | t j��d�kr t j�� d�S | S )Nzfoo.py)r r# �join)r# )r� r� r r r� s z6TestDiscovery.test_module_symlink_ok.<locals>.realpathzfoo.py)rV r2 ) r� r r# r� r r9 r r rX )r rf r� r� r r )r� r� r� r �test_module_symlink_ok� s
z$TestDiscovery.test_module_symlink_okc sl t �� }�g�tj�tj�t jj��� d�_d� ��fdd� }||_ |�
d�}�� �j� �� |j
�� d S )NFc s d�_ ��| � � �S )NT)�wasRunr )rV r2 r` )� expectedPathr rJ r r r= s zBTestDiscovery.test_discovery_from_dotted_path.<locals>._find_testsz
unittest.test)N)r r r r# r rW ry rY r� r= rX rw r Z_tests)r r r= r@ r )r� r rJ r � test_discovery_from_dotted_path
s
z-TestDiscovery.test_discovery_from_dotted_pathc s� t �� }tj�dd� t_tjj�tjj� dd� tj_tjd d � �� ���fdd�}| �|� | � t
��}|� d� W d Q R X | � t
|j�d� d S )Nc S s dgS )Nztest_this_does_not_exist.pyr )rj r r r r. # r/ zOTestDiscovery.test_discovery_from_dotted_path_builtin_modules.<locals>.<lambda>c S s dS )NFr )rj r r r r. & r/ c s( �t j_� t j_�t _�tjd d �< d S )N)r r# r$ r' r rZ r )r' r$ r r] r r rl ( s zNTestDiscovery.test_discovery_from_dotted_path_builtin_modules.<locals>.restorerZ z2Can not use builtin modules as dotted module names)r r r r r# r$ r' rZ r9 r � TypeErrorrX r re r� )r r rl r� r )r' r$ r r] r �/test_discovery_from_dotted_path_builtin_modules s
z=TestDiscovery.test_discovery_from_dotted_path_builtin_modulesc s� t �� }t�d��ddg�_tjd ddgd��_�fdd�}g � d � fdd� }||_t|_ t j
� d |��> t �
� �* tjj�d�� |�d�}W d Q R X W d Q R X W d Q R X | �|d
d g� d S )
N�packagez/az/b)r �submodule_search_locationsc s � t j| < � S )N)rZ r� )� packagenamer� �kwargs)r� r r �_import? s
zLTestDiscovery.test_discovery_from_dotted_namespace_packages.<locals>._importc s � � | |f� d| gS )Nz%s/tests)rI )rV r2 r` )ra r r r= D s zPTestDiscovery.test_discovery_from_dotted_namespace_packages.<locals>._find_testszbuiltins.__import__z/a/testsz/b/tests)N)r r �types�
ModuleType�__path__�SimpleNamespace�__spec__r= r< r; �mock�patchr �
DirsOnSysPathry �test_importlib�util�uncacherX r )r r r� r= r@ r )ra r� r �-test_discovery_from_dotted_namespace_packages5 s
(z;TestDiscovery.test_discovery_from_dotted_namespace_packagesc s� t �� }t�d�� � fdd�}t j�d|��l t�� �X tj j
� d��> | � t
��}|�d� W d Q R X | �t|j�d�� �� W d Q R X W d Q R X W d Q R X d S )Nr� c s � t j| < � S )N)rZ r� )r� r� r� )r� r r r� X s
z>TestDiscovery.test_discovery_failed_discovery.<locals>._importzbuiltins.__import__z$don't know how to discover from {!r})r r r� r� r� r� r r� ry r� r� r� r r� rX r re r� �format)r r r� r� r )r� r � test_discovery_failed_discoveryT s
z-TestDiscovery.test_discovery_failed_discoveryN) r r r
r rA rB rQ rR r\ rg ri rm rp rz r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r s6
3(JGX.-
J r �__main__)Zos.pathr r r� rZ r� r} ry r Ztest.test_importlib.utilr Z
unittest.mockZ
unittest.testr� r ZTestCaser r �mainr r r r �<module> s( S