You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
560 lines
18 KiB
560 lines
18 KiB
"""A library of helper functions for the CherryPy test suite."""
|
|
|
|
import datetime
|
|
import io
|
|
import logging
|
|
import os
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
import unittest
|
|
import warnings
|
|
|
|
import nose
|
|
import six
|
|
|
|
import cherrypy
|
|
from cherrypy._cpcompat import text_or_bytes, copyitems, HTTPSConnection, ntob
|
|
from cherrypy.lib import httputil
|
|
from cherrypy.lib import gctools
|
|
from cherrypy.lib.reprconf import unrepr
|
|
from cherrypy.test import webtest
|
|
|
|
_testconfig = None
|
|
log = logging.getLogger(__name__)
|
|
thisdir = os.path.abspath(os.path.dirname(__file__))
|
|
serverpem = os.path.join(os.getcwd(), thisdir, 'test.pem')
|
|
|
|
|
|
def get_tst_config(overconf={}):
|
|
global _testconfig
|
|
if _testconfig is None:
|
|
conf = {
|
|
'scheme': 'http',
|
|
'protocol': 'HTTP/1.1',
|
|
'port': 54583,
|
|
'host': '127.0.0.1',
|
|
'validate': False,
|
|
'server': 'wsgi',
|
|
}
|
|
try:
|
|
import testconfig
|
|
_conf = testconfig.config.get('supervisor', None)
|
|
if _conf is not None:
|
|
for k, v in _conf.items():
|
|
if isinstance(v, text_or_bytes):
|
|
_conf[k] = unrepr(v)
|
|
conf.update(_conf)
|
|
except ImportError:
|
|
pass
|
|
_testconfig = conf
|
|
conf = _testconfig.copy()
|
|
conf.update(overconf)
|
|
|
|
return conf
|
|
|
|
|
|
class Supervisor(object):
|
|
|
|
"""Base class for modeling and controlling servers during testing."""
|
|
|
|
def __init__(self, **kwargs):
|
|
for k, v in kwargs.items():
|
|
if k == 'port':
|
|
setattr(self, k, int(v))
|
|
setattr(self, k, v)
|
|
|
|
|
|
log_to_stderr = lambda msg, level: sys.stderr.write(msg + os.linesep)
|
|
|
|
|
|
class LocalSupervisor(Supervisor):
|
|
|
|
"""Base class for modeling/controlling servers which run in the same
|
|
process.
|
|
|
|
When the server side runs in a different process, start/stop can dump all
|
|
state between each test module easily. When the server side runs in the
|
|
same process as the client, however, we have to do a bit more work to
|
|
ensure config and mounted apps are reset between tests.
|
|
"""
|
|
|
|
using_apache = False
|
|
using_wsgi = False
|
|
|
|
def __init__(self, **kwargs):
|
|
for k, v in kwargs.items():
|
|
setattr(self, k, v)
|
|
|
|
cherrypy.server.httpserver = self.httpserver_class
|
|
|
|
# This is perhaps the wrong place for this call but this is the only
|
|
# place that i've found so far that I KNOW is early enough to set this.
|
|
cherrypy.config.update({'log.screen': False})
|
|
engine = cherrypy.engine
|
|
if hasattr(engine, 'signal_handler'):
|
|
engine.signal_handler.subscribe()
|
|
if hasattr(engine, 'console_control_handler'):
|
|
engine.console_control_handler.subscribe()
|
|
|
|
def start(self, modulename=None):
|
|
"""Load and start the HTTP server."""
|
|
if modulename:
|
|
# Unhook httpserver so cherrypy.server.start() creates a new
|
|
# one (with config from setup_server, if declared).
|
|
cherrypy.server.httpserver = None
|
|
|
|
cherrypy.engine.start()
|
|
|
|
self.sync_apps()
|
|
|
|
def sync_apps(self):
|
|
"""Tell the server about any apps which the setup functions mounted."""
|
|
pass
|
|
|
|
def stop(self):
|
|
td = getattr(self, 'teardown', None)
|
|
if td:
|
|
td()
|
|
|
|
cherrypy.engine.exit()
|
|
|
|
for name, server in copyitems(getattr(cherrypy, 'servers', {})):
|
|
server.unsubscribe()
|
|
del cherrypy.servers[name]
|
|
|
|
|
|
class NativeServerSupervisor(LocalSupervisor):
|
|
|
|
"""Server supervisor for the builtin HTTP server."""
|
|
|
|
httpserver_class = 'cherrypy._cpnative_server.CPHTTPServer'
|
|
using_apache = False
|
|
using_wsgi = False
|
|
|
|
def __str__(self):
|
|
return 'Builtin HTTP Server on %s:%s' % (self.host, self.port)
|
|
|
|
|
|
class LocalWSGISupervisor(LocalSupervisor):
|
|
|
|
"""Server supervisor for the builtin WSGI server."""
|
|
|
|
httpserver_class = 'cherrypy._cpwsgi_server.CPWSGIServer'
|
|
using_apache = False
|
|
using_wsgi = True
|
|
|
|
def __str__(self):
|
|
return 'Builtin WSGI Server on %s:%s' % (self.host, self.port)
|
|
|
|
def sync_apps(self):
|
|
"""Hook a new WSGI app into the origin server."""
|
|
cherrypy.server.httpserver.wsgi_app = self.get_app()
|
|
|
|
def get_app(self, app=None):
|
|
"""Obtain a new (decorated) WSGI app to hook into the origin server."""
|
|
if app is None:
|
|
app = cherrypy.tree
|
|
|
|
if self.validate:
|
|
try:
|
|
from wsgiref import validate
|
|
except ImportError:
|
|
warnings.warn(
|
|
'Error importing wsgiref. The validator will not run.')
|
|
else:
|
|
# wraps the app in the validator
|
|
app = validate.validator(app)
|
|
|
|
return app
|
|
|
|
|
|
def get_cpmodpy_supervisor(**options):
|
|
from cherrypy.test import modpy
|
|
sup = modpy.ModPythonSupervisor(**options)
|
|
sup.template = modpy.conf_cpmodpy
|
|
return sup
|
|
|
|
|
|
def get_modpygw_supervisor(**options):
|
|
from cherrypy.test import modpy
|
|
sup = modpy.ModPythonSupervisor(**options)
|
|
sup.template = modpy.conf_modpython_gateway
|
|
sup.using_wsgi = True
|
|
return sup
|
|
|
|
|
|
def get_modwsgi_supervisor(**options):
|
|
from cherrypy.test import modwsgi
|
|
return modwsgi.ModWSGISupervisor(**options)
|
|
|
|
|
|
def get_modfcgid_supervisor(**options):
|
|
from cherrypy.test import modfcgid
|
|
return modfcgid.ModFCGISupervisor(**options)
|
|
|
|
|
|
def get_modfastcgi_supervisor(**options):
|
|
from cherrypy.test import modfastcgi
|
|
return modfastcgi.ModFCGISupervisor(**options)
|
|
|
|
|
|
def get_wsgi_u_supervisor(**options):
|
|
cherrypy.server.wsgi_version = ('u', 0)
|
|
return LocalWSGISupervisor(**options)
|
|
|
|
|
|
class CPWebCase(webtest.WebCase):
|
|
|
|
script_name = ''
|
|
scheme = 'http'
|
|
|
|
available_servers = {'wsgi': LocalWSGISupervisor,
|
|
'wsgi_u': get_wsgi_u_supervisor,
|
|
'native': NativeServerSupervisor,
|
|
'cpmodpy': get_cpmodpy_supervisor,
|
|
'modpygw': get_modpygw_supervisor,
|
|
'modwsgi': get_modwsgi_supervisor,
|
|
'modfcgid': get_modfcgid_supervisor,
|
|
'modfastcgi': get_modfastcgi_supervisor,
|
|
}
|
|
default_server = 'wsgi'
|
|
|
|
@classmethod
|
|
def _setup_server(cls, supervisor, conf):
|
|
v = sys.version.split()[0]
|
|
log.info('Python version used to run this test script: %s' % v)
|
|
log.info('CherryPy version: %s' % cherrypy.__version__)
|
|
if supervisor.scheme == 'https':
|
|
ssl = ' (ssl)'
|
|
else:
|
|
ssl = ''
|
|
log.info('HTTP server version: %s%s' % (supervisor.protocol, ssl))
|
|
log.info('PID: %s' % os.getpid())
|
|
|
|
cherrypy.server.using_apache = supervisor.using_apache
|
|
cherrypy.server.using_wsgi = supervisor.using_wsgi
|
|
|
|
if sys.platform[:4] == 'java':
|
|
cherrypy.config.update({'server.nodelay': False})
|
|
|
|
if isinstance(conf, text_or_bytes):
|
|
parser = cherrypy.lib.reprconf.Parser()
|
|
conf = parser.dict_from_file(conf).get('global', {})
|
|
else:
|
|
conf = conf or {}
|
|
baseconf = conf.copy()
|
|
baseconf.update({'server.socket_host': supervisor.host,
|
|
'server.socket_port': supervisor.port,
|
|
'server.protocol_version': supervisor.protocol,
|
|
'environment': 'test_suite',
|
|
})
|
|
if supervisor.scheme == 'https':
|
|
# baseconf['server.ssl_module'] = 'builtin'
|
|
baseconf['server.ssl_certificate'] = serverpem
|
|
baseconf['server.ssl_private_key'] = serverpem
|
|
|
|
# helper must be imported lazily so the coverage tool
|
|
# can run against module-level statements within cherrypy.
|
|
# Also, we have to do "from cherrypy.test import helper",
|
|
# exactly like each test module does, because a relative import
|
|
# would stick a second instance of webtest in sys.modules,
|
|
# and we wouldn't be able to globally override the port anymore.
|
|
if supervisor.scheme == 'https':
|
|
webtest.WebCase.HTTP_CONN = HTTPSConnection
|
|
return baseconf
|
|
|
|
@classmethod
|
|
def setup_class(cls):
|
|
''
|
|
# Creates a server
|
|
conf = get_tst_config()
|
|
supervisor_factory = cls.available_servers.get(
|
|
conf.get('server', 'wsgi'))
|
|
if supervisor_factory is None:
|
|
raise RuntimeError('Unknown server in config: %s' % conf['server'])
|
|
supervisor = supervisor_factory(**conf)
|
|
|
|
# Copied from "run_test_suite"
|
|
cherrypy.config.reset()
|
|
baseconf = cls._setup_server(supervisor, conf)
|
|
cherrypy.config.update(baseconf)
|
|
setup_client()
|
|
|
|
if hasattr(cls, 'setup_server'):
|
|
# Clear the cherrypy tree and clear the wsgi server so that
|
|
# it can be updated with the new root
|
|
cherrypy.tree = cherrypy._cptree.Tree()
|
|
cherrypy.server.httpserver = None
|
|
cls.setup_server()
|
|
# Add a resource for verifying there are no refleaks
|
|
# to *every* test class.
|
|
cherrypy.tree.mount(gctools.GCRoot(), '/gc')
|
|
cls.do_gc_test = True
|
|
supervisor.start(cls.__module__)
|
|
|
|
cls.supervisor = supervisor
|
|
|
|
@classmethod
|
|
def teardown_class(cls):
|
|
''
|
|
if hasattr(cls, 'setup_server'):
|
|
cls.supervisor.stop()
|
|
|
|
do_gc_test = False
|
|
|
|
def test_gc(self):
|
|
if not self.do_gc_test:
|
|
return
|
|
|
|
self.getPage('/gc/stats')
|
|
try:
|
|
self.assertBody('Statistics:')
|
|
except Exception:
|
|
'Failures occur intermittently. See #1420'
|
|
|
|
def prefix(self):
|
|
return self.script_name.rstrip('/')
|
|
|
|
def base(self):
|
|
if ((self.scheme == 'http' and self.PORT == 80) or
|
|
(self.scheme == 'https' and self.PORT == 443)):
|
|
port = ''
|
|
else:
|
|
port = ':%s' % self.PORT
|
|
|
|
return '%s://%s%s%s' % (self.scheme, self.HOST, port,
|
|
self.script_name.rstrip('/'))
|
|
|
|
def exit(self):
|
|
sys.exit()
|
|
|
|
def getPage(self, url, headers=None, method='GET', body=None,
|
|
protocol=None, raise_subcls=None):
|
|
"""Open the url. Return status, headers, body.
|
|
|
|
`raise_subcls` must be a tuple with the exceptions classes
|
|
or a single exception class that are not going to be considered
|
|
a socket.error regardless that they were are subclass of a
|
|
socket.error and therefore not considered for a connection retry.
|
|
"""
|
|
if self.script_name:
|
|
url = httputil.urljoin(self.script_name, url)
|
|
return webtest.WebCase.getPage(self, url, headers, method, body,
|
|
protocol, raise_subcls)
|
|
|
|
def skip(self, msg='skipped '):
|
|
raise nose.SkipTest(msg)
|
|
|
|
def assertErrorPage(self, status, message=None, pattern=''):
|
|
"""Compare the response body with a built in error page.
|
|
|
|
The function will optionally look for the regexp pattern,
|
|
within the exception embedded in the error page."""
|
|
|
|
# This will never contain a traceback
|
|
page = cherrypy._cperror.get_error_page(status, message=message)
|
|
|
|
# First, test the response body without checking the traceback.
|
|
# Stick a match-all group (.*) in to grab the traceback.
|
|
def esc(text):
|
|
return re.escape(ntob(text))
|
|
epage = re.escape(page)
|
|
epage = epage.replace(
|
|
esc('<pre id="traceback"></pre>'),
|
|
esc('<pre id="traceback">') + ntob('(.*)') + esc('</pre>'))
|
|
m = re.match(epage, self.body, re.DOTALL)
|
|
if not m:
|
|
self._handlewebError(
|
|
'Error page does not match; expected:\n' + page)
|
|
return
|
|
|
|
# Now test the pattern against the traceback
|
|
if pattern is None:
|
|
# Special-case None to mean that there should be *no* traceback.
|
|
if m and m.group(1):
|
|
self._handlewebError('Error page contains traceback')
|
|
else:
|
|
if (m is None) or (
|
|
not re.search(ntob(re.escape(pattern), self.encoding),
|
|
m.group(1))):
|
|
msg = 'Error page does not contain %s in traceback'
|
|
self._handlewebError(msg % repr(pattern))
|
|
|
|
date_tolerance = 2
|
|
|
|
def assertEqualDates(self, dt1, dt2, seconds=None):
|
|
"""Assert abs(dt1 - dt2) is within Y seconds."""
|
|
if seconds is None:
|
|
seconds = self.date_tolerance
|
|
|
|
if dt1 > dt2:
|
|
diff = dt1 - dt2
|
|
else:
|
|
diff = dt2 - dt1
|
|
if not diff < datetime.timedelta(seconds=seconds):
|
|
raise AssertionError('%r and %r are not within %r seconds.' %
|
|
(dt1, dt2, seconds))
|
|
|
|
|
|
def _test_method_sorter(_, x, y):
|
|
"""Monkeypatch the test sorter to always run test_gc last in each suite."""
|
|
if x == 'test_gc':
|
|
return 1
|
|
if y == 'test_gc':
|
|
return -1
|
|
if x > y:
|
|
return 1
|
|
if x < y:
|
|
return -1
|
|
return 0
|
|
unittest.TestLoader.sortTestMethodsUsing = _test_method_sorter
|
|
|
|
|
|
def setup_client():
|
|
"""Set up the WebCase classes to match the server's socket settings."""
|
|
webtest.WebCase.PORT = cherrypy.server.socket_port
|
|
webtest.WebCase.HOST = cherrypy.server.socket_host
|
|
if cherrypy.server.ssl_certificate:
|
|
CPWebCase.scheme = 'https'
|
|
|
|
# --------------------------- Spawning helpers --------------------------- #
|
|
|
|
|
|
class CPProcess(object):
|
|
|
|
pid_file = os.path.join(thisdir, 'test.pid')
|
|
config_file = os.path.join(thisdir, 'test.conf')
|
|
config_template = """[global]
|
|
server.socket_host: '%(host)s'
|
|
server.socket_port: %(port)s
|
|
checker.on: False
|
|
log.screen: False
|
|
log.error_file: r'%(error_log)s'
|
|
log.access_file: r'%(access_log)s'
|
|
%(ssl)s
|
|
%(extra)s
|
|
"""
|
|
error_log = os.path.join(thisdir, 'test.error.log')
|
|
access_log = os.path.join(thisdir, 'test.access.log')
|
|
|
|
def __init__(self, wait=False, daemonize=False, ssl=False,
|
|
socket_host=None, socket_port=None):
|
|
self.wait = wait
|
|
self.daemonize = daemonize
|
|
self.ssl = ssl
|
|
self.host = socket_host or cherrypy.server.socket_host
|
|
self.port = socket_port or cherrypy.server.socket_port
|
|
|
|
def write_conf(self, extra=''):
|
|
if self.ssl:
|
|
serverpem = os.path.join(thisdir, 'test.pem')
|
|
ssl = """
|
|
server.ssl_certificate: r'%s'
|
|
server.ssl_private_key: r'%s'
|
|
""" % (serverpem, serverpem)
|
|
else:
|
|
ssl = ''
|
|
|
|
conf = self.config_template % {
|
|
'host': self.host,
|
|
'port': self.port,
|
|
'error_log': self.error_log,
|
|
'access_log': self.access_log,
|
|
'ssl': ssl,
|
|
'extra': extra,
|
|
}
|
|
with io.open(self.config_file, 'w', encoding='utf-8') as f:
|
|
f.write(six.text_type(conf))
|
|
|
|
def start(self, imports=None):
|
|
"""Start cherryd in a subprocess."""
|
|
cherrypy._cpserver.wait_for_free_port(self.host, self.port)
|
|
|
|
args = [
|
|
'-m',
|
|
'cherrypy.__main__', # __main__ is needed for `-m` in Python 2.6
|
|
'-c', self.config_file,
|
|
'-p', self.pid_file,
|
|
]
|
|
"""
|
|
Command for running cherryd server with autoreload enabled
|
|
|
|
Using
|
|
|
|
```
|
|
['-c',
|
|
"__requires__ = 'CherryPy'; \
|
|
import pkg_resources, re, sys; \
|
|
sys.argv[0] = re.sub(r'(-script\.pyw?|\.exe)?$', '', sys.argv[0]); \
|
|
sys.exit(\
|
|
pkg_resources.load_entry_point(\
|
|
'CherryPy', 'console_scripts', 'cherryd')())"]
|
|
```
|
|
|
|
doesn't work as it's impossible to reconstruct the `-c`'s contents.
|
|
Ref: https://github.com/cherrypy/cherrypy/issues/1545
|
|
"""
|
|
|
|
if not isinstance(imports, (list, tuple)):
|
|
imports = [imports]
|
|
for i in imports:
|
|
if i:
|
|
args.append('-i')
|
|
args.append(i)
|
|
|
|
if self.daemonize:
|
|
args.append('-d')
|
|
|
|
env = os.environ.copy()
|
|
# Make sure we import the cherrypy package in which this module is
|
|
# defined.
|
|
grandparentdir = os.path.abspath(os.path.join(thisdir, '..', '..'))
|
|
if env.get('PYTHONPATH', ''):
|
|
env['PYTHONPATH'] = os.pathsep.join(
|
|
(grandparentdir, env['PYTHONPATH']))
|
|
else:
|
|
env['PYTHONPATH'] = grandparentdir
|
|
self._proc = subprocess.Popen([sys.executable] + args, env=env)
|
|
if self.wait:
|
|
self.exit_code = self._proc.wait()
|
|
else:
|
|
cherrypy._cpserver.wait_for_occupied_port(self.host, self.port)
|
|
|
|
# Give the engine a wee bit more time to finish STARTING
|
|
if self.daemonize:
|
|
time.sleep(2)
|
|
else:
|
|
time.sleep(1)
|
|
|
|
def get_pid(self):
|
|
if self.daemonize:
|
|
return int(open(self.pid_file, 'rb').read())
|
|
return self._proc.pid
|
|
|
|
def join(self):
|
|
"""Wait for the process to exit."""
|
|
if self.daemonize:
|
|
return self._join_daemon()
|
|
self._proc.wait()
|
|
|
|
def _join_daemon(self):
|
|
try:
|
|
try:
|
|
# Mac, UNIX
|
|
os.wait()
|
|
except AttributeError:
|
|
# Windows
|
|
try:
|
|
pid = self.get_pid()
|
|
except IOError:
|
|
# Assume the subprocess deleted the pidfile on shutdown.
|
|
pass
|
|
else:
|
|
os.waitpid(pid, 0)
|
|
except OSError:
|
|
x = sys.exc_info()[1]
|
|
if x.args != (10, 'No child processes'):
|
|
raise
|