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 unittest2argparsez 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_contextcontextzinstalling pipz %s %s --user)pip ImportErrorssltempfileNamedTemporaryFile
contextlibclosingprint GET_PIP_URLnamehasattrr dictr readwriteflushossystemr )r
r fctxkwargsZreqdatacode 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_DEPSsetr sysr% r r r join)ZdepsZis_venvoptsr" 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-depsaction
store_truedefaultFhelpz%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 pathabspathdirname__file__ZHEREr r( version_infoextendr% r- r; r# r# r# r$ <module> s,
2