D7net
Home
Console
Upload
information
Create File
Create Folder
About
Tools
:
/
proc
/
self
/
root
/
proc
/
thread-self
/
root
/
lib64
/
python3.6
/
site-packages
/
psutil
/
tests
/
Filename :
__main__.py
back
Copy
# Copyright (c) 2009, Giampaolo Rodola'. All rights reserved. # Use of this source code is governed by a BSD-style license that can be # found in the LICENSE file. """ Run unit tests. This is invoked by: $ python -m psutil.tests """ import contextlib import optparse import os import ssl import sys import tempfile try: from urllib.request import urlopen # py3 except ImportError: from urllib2 import urlopen from psutil.tests import PYTHON_EXE from psutil.tests import run_suite HERE = os.path.abspath(os.path.dirname(__file__)) GET_PIP_URL = "https://bootstrap.pypa.io/get-pip.py" TEST_DEPS = [] if sys.version_info[:2] == (2, 6): TEST_DEPS.extend(["ipaddress", "unittest2", "argparse", "mock==1.0.1"]) elif sys.version_info[:2] == (2, 7) or sys.version_info[:2] <= (3, 2): TEST_DEPS.extend(["ipaddress", "mock"]) elif sys.version_info[:2] == (3, 3): TEST_DEPS.extend(["ipaddress"]) def install_pip(): try: import pip # NOQA except ImportError: f = tempfile.NamedTemporaryFile(suffix='.py') with contextlib.closing(f): print("downloading %s to %s" % (GET_PIP_URL, f.name)) if hasattr(ssl, '_create_unverified_context'): ctx = ssl._create_unverified_context() else: ctx = None kwargs = dict(context=ctx) if ctx else {} req = urlopen(GET_PIP_URL, **kwargs) data = req.read() f.write(data) f.flush() print("installing pip") code = os.system('%s %s --user' % (PYTHON_EXE, f.name)) return code def install_test_deps(deps=None): """Install test dependencies via pip.""" if deps is None: deps = TEST_DEPS deps = set(deps) if deps: is_venv = hasattr(sys, 'real_prefix') opts = "--user" if not is_venv else "" install_pip() code = os.system('%s -m pip install %s --upgrade %s' % ( PYTHON_EXE, opts, " ".join(deps))) return code def main(): usage = "%s -m psutil.tests [opts]" % PYTHON_EXE parser = optparse.OptionParser(usage=usage, description="run unit tests") parser.add_option("-i", "--install-deps", action="store_true", default=False, help="don't print status messages to stdout") opts, args = parser.parse_args() if opts.install_deps: install_pip() install_test_deps() else: for dep in TEST_DEPS: try: __import__(dep.split("==")[0]) except ImportError: sys.exit("%r lib is not installed; run %s -m psutil.tests " "--install-deps" % (dep, PYTHON_EXE)) run_suite() main()