3 ��JZ� � @s:dZddlZddlZddlZddlZddlZddlZyddlmZWn e k rdddl mZYnXddlmZddlm Z ejjejje��ZdZgZejdd�dkr�ejd d ddg�nVejdd�dks�ejdd�dk�r�ejd dg�n ejdd�dk�rejd g�dd�Zddd�Zdd�Ze�dS)z? Run unit tests. This is invoked by: $ python -m psutil.tests �N)�urlopen)� PYTHON_EXE)� run_suitez$https://bootstrap.pypa.io/get-pip.py��Z ipaddressZ unittest2�argparsezmock==1.0.1��ZmockcCs�yddl}Wn�tk r�tjdd�}tj|��tdt|jf�t t d�rZt j�}nd}|rlt|d�ni}t tf|�}|j�}|j|�|j�td�tjdt|jf�}|SQRXYnXdS) Nrz.py)�suffixzdownloading %s to %s�_create_unverified_context)�contextzinstalling pipz%s %s --user)�pip�ImportError�tempfileZNamedTemporaryFile� contextlib�closing�print�GET_PIP_URL�name�hasattr�sslr�dictr�read�write�flush�os�systemr)r �fZctx�kwargsZreq�data�code�r!� /usr/lib64/python3.6/__main__.py�install_pip&s" r#cCsV|dkrt}t|�}|rRttd�}|s*dnd}t�tjdt|dj|�f�}|SdS)z"Install test dependencies via pip.NZreal_prefixz--user�z!%s -m pip install %s --upgrade %s� ) � TEST_DEPS�setr�sysr#rrr�join)ZdepsZis_venv�optsr r!r!r"�install_test_deps<s r+cCs�dt}tj|dd�}|jdddddd �|j�\}}|jrJt�t�nRxJtD]B}yt |j d �d�WqPtk r�tj d|tf�YqPXqPWt�dS) Nz%s -m psutil.tests [opts]zrun unit tests)�usage�descriptionz-iz--install-deps� store_trueFz%don't print status messages to stdout)�action�default�helpz==rz>%r lib is not installed; run %s -m psutil.tests --install-deps)r�optparseZOptionParserZ add_option� parse_argsZinstall_depsr#r+r&� __import__�splitrr(�exitr)r,�parserr*�argsZdepr!r!r"�mainJs r9)rr)rr)r r)r r )N)�__doc__rr2rrr(rZurllib.requestrrZurllib2Zpsutil.testsrr�path�abspath�dirname�__file__ZHERErr&�version_info�extendr#r+r9r!r!r!r"� s2&