Cs[� � @ sq d Z d d l Z d d l Z d d l Z d d l Z d d l Z y d d l m Z Wn" e k rz d d l m Z Yn Xd d l m Z d d l m Z e j j e j j e � � Z d Z g Z e j d d � d k r� e j d d d d g � nE e j d d � d k s,e j d d � d k r?e j d d g � d d � Z d d d � Z d d � Z e � d S)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�argparsez mock==1.0.1� � Zmockc C s y d d l } Wn� t k r d d l } t j d d � } t j | � �� t d t | j f � t | d � r� | j � } n d } | r� t d | � n i } t t | � } | j � } | j | � | j � t d � t j d t | j f � } | SWd QRXYn Xd S) Nr �suffixz.pyzdownloading %s to %s�_create_unverified_context�contextzinstalling pipz %s %s --user)�pip� ImportError�ssl�tempfile�NamedTemporaryFile� contextlib�closing�print� GET_PIP_URL�name�hasattrr �dictr �read�write�flush�os�systemr )r r �f�ctx�kwargsZreq�data�code� r# � /__main__.py� install_pip$ s$ r% c C sy | d k r t } t | � } | ru t t d � } | s? d n d } t � t j d t | d j | � f � } | Sd S)z"Install test dependencies via pip.NZ real_prefixz--user� z!%s -m pip install %s --upgrade %s� ) � TEST_DEPS�setr �sysr% r r r �join)ZdepsZis_venv�optsr" r# r# r$ �install_test_deps; s r- c C s� d t } t j d | d d � } | j d d d d d d d d �| j � \ } } | j rp t � t � n_ xU t D]M } y t | j d � d � Wqw t k r� t j d | t f � Yqw Xqw Wt � d S)Nz%s -m psutil.tests [opts]�usage� descriptionzrun unit testsz-iz--install-deps�action� store_true�defaultF�helpz%don't print status messages to stdoutz==r z>%r lib is not installed; run %s -m psutil.tests --install-deps)r �optparseZ OptionParserZ add_option� parse_argsZ install_depsr% r- r( � __import__�splitr r* �exitr )r. �parserr, �argsZdepr# r# r$ �mainI s r; )r r )r r )r r )�__doc__r r4 r r* r Zurllib.requestr r �urllib2Z psutil.testsr r �path�abspath�dirname�__file__ZHEREr r( � version_info�extendr% r- r; r# r# r# r$ �<module> s, 2