U
    /e	                     @   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 d dl	Z	d dl
Z	dd Zdd Ze dd Zd	d
 Zejdddd Zejdddd Zdd Ze dd Zdd ZdS )    N)	CliRunnerc                  C   sB   t  } | tjjdg}|jdks&t|jdtj dks>td S )Nz	--versionr   zcli, version 
)r   invokedaskcli	exit_codeAssertionErroroutput__version__)runnerresult r   7/tmp/pip-unpacked-wheel-dbjnr7gq/dask/tests/test_cli.pytest_version   s    r   c                  C   s   t  } | tjj}|jdks"tt|j	}|d d
dd tjd d D ksXt|d tjksjt|d t jkstzdd	lm} W n tk
r   d }Y nX |d
 |kstd S )Nr   Python.c                 s   s   | ]}t |V  qd S N)str).0xr   r   r   	<genexpr>   s     z%test_info_versions.<locals>.<genexpr>   r   Platform)r
   distributed)r   r   r   r   versionsr   r   jsonloadsr	   joinsysversion_infor
   platformunamesystemr   ImportError)r   r   tableZdistributed_versionr   r   r   test_info_versions   s    *
r%   c                   C   s   d S r   r   r   r   r   r   	dummy_cli+   s    r&   c                   C   s   d S r   r   r   r   r   r   bad_command0   s    r'   good)namec                   C   s   d S r   r   r   r   r   r   good_command4   s    r*   c                   C   s   d S r   r   r   r   r   r   good_command_29   s    r+   c               	   C   s   ddl m}  tjjdddd}tjjdddd}tjtd	d
 | t| W 5 Q R X | t| dtj	kslt
tj	d tks~t
d S )Nr   _register_command_epbadzdask.tests.test_cli:bad_commanddask_clir)   valuegroupr(    dask.tests.test_cli:good_commandzmust be instances ofmatch)dask.clir-   	importlibmetadata
EntryPointpytestwarnsUserWarningr&   commandsr   r*   )r-   Zbad_epZgood_epr   r   r   test_register_command_ep>   s     
r>   c                   C   s   d S r   r   r   r   r   r   dummy_cli_2U   s    r?   c               	   C   sb   ddl m}  tjjdddd}tjjdddd}| t| tjtd	d
 | t| W 5 Q R X d S )Nr   r,   oner3   r/   r0   twoz"dask.tests.test_cli:good_command_2z'While registering the command with namer4   )	r6   r-   r7   r8   r9   r?   r:   r;   r<   )r-   r@   rA   r   r   r   $test_repeated_name_registration_warnZ   s    
rB   )importlib.metadatar7   r   r    r   Zclickr:   Zclick.testingr   r   r6   r   r%   r2   r&   r'   commandr*   r+   r>   r?   rB   r   r   r   r   <module>   s*   





