#!/usr/bin/env python
# PySys System Test Framework, Copyright (C) 2006-2020 M.B. Grieve
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
# Lesser General Public License for more details.
# You should have received a copy of the GNU Lesser General Public
# License along with this library; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
"""
Contains `pysys.process.user.ProcessUser` which supports using processes from PySys, and provides the
shared functionality of subclasses `pysys.basetest.BaseTest` and `pysys.baserunner.BaseRunner`.
"""
import time, collections, inspect, locale, fnmatch, sys
import threading
import shutil
import contextlib
from pysys import log, process_lock
from pysys.constants import *
from pysys.exceptions import *
from pysys.utils.filegrep import getmatches
from pysys.utils.logutils import BaseLogFormatter, stripANSIEscapeCodes
from pysys.xml.project import Project
from pysys.process.helper import ProcessWrapper
from pysys.utils.allocport import TCPPortOwner
from pysys.utils.fileutils import mkdir, deletedir, pathexists, toLongPathSafe, fromLongPathSafe
from pysys.utils.pycompat import *
import pysys.internal.safe_eval
if IS_WINDOWS:
import win32api, win32con
else:
import fcntl
STDOUTERR_TUPLE = collections.namedtuple('stdouterr', ['stdout', 'stderr'])
"""
Returned by `ProcessUser.allocateUniqueStdOutErr` to hold a pair of ``(stdout,stderr)`` names.
"""
[docs]class ProcessUser(object):
"""
ProcessUser provides support for safely using processes in PySys, including starting processes (with an
appropriate output directory), waiting for them to do what they're supposed to do, and finally ensuring all
processes and ports are cleaned up when each test (or the overall runner) terminates.
As the common base class of both `pysys.basetest.BaseTest` and `pysys.baserunner.BaseRunner`, ProcessUser also
holds shared functionality such as dynamic port allocation, copying files, creating directories etc, and
keeps track of the "outcome" generated by some methods (although the outcome value is only used when
it is subclassed by BaseTest).
Each ProcessUser instance is responsible for managing the lifetime of the processes started using its
`startProcess` method, and ensuring they are all terminated when the `cleanup` method is invoked. Additional
functions to be executed during cleanup can be registered using `addCleanupFunction`, for example to delete large
output directories, or terminate non-process resources such as docker containers or remote servers.
Apart from the `addOutcome` method this class is not thread-safe, so if
you need to access it from multiple threads be sure to add your own locking
around use of its fields and methods, including any cleanup functions.
:ivar str ~.input: Full path to the directory containing input files (e.g. ``testdir/Input``)
:ivar str ~.output: Full path to the directory that output files should be written to (e.g. ``testdir/Output/<platformname>``)
:ivar logging.Logger ~.log: The Python ``Logger`` instance that should be used to record progress and status information.
:ivar pysys.xml.project.Project ~.project: A reference to the singleton project instance containing the
configuration of this PySys test project as defined by ``pysysproject.xml``.
The project can be used to access information such as the project properties which are shared across all tests
(e.g. for hosts and credentials).
:ivar bool ~.disableCoverage: Set to True to disable all code coverage collection for processes
started from this instance. This is automatically set for any tests marked with the "disableCoverage" group
in their pysystest.xml file.
The built-in Python code coverage functionality in L{startPython} checks this
flag. It is recommended that any other languages supporting code coverage
also check the self.disableCoverage flag.
Additional variables that affect only the behaviour of a single method are documented in the associated method.
"""
def __init__(self):
"""Default constructor.
"""
self.log = log
"""The logger instance that should be used to log from this class. """
self.project = Project.getInstance()
"""The L{pysys.xml.project.Project} instance containing settings for this PySys project."""
if self.project is None:
assert 'doctest' in sys.argv[0], 'Project was not loaded yet' # allow it only during doctest-ing
else:
self.defaultAbortOnError = self.project.getProperty('defaultAbortOnError', True)
self.defaultIgnoreExitStatus = self.project.getProperty('defaultIgnoreExitStatus', False)
self.input = self.project.testRootDir
self.output = None # must be set by subclass
self.processList = []
self.processCount = {}
self.__cleanupFunctions = [] # (fn, ignoreErrors)
self.outcome = [] # internal, do NOT use directly
self.__outcomeReason = ''
self.__outcomeLocation = (None,None)
self.__uniqueProcessKeys = {}
self.__pythonCoverageFile = 0
self.disableCoverage = False
self.lock = threading.RLock()
"""
A recursive lock that can be used for protecting the fields of this instance
from access by background threads, as needed.
"""
# variables affecting a specific method (documented there rather than above)
self.logFileContentsDefaultExcludes = []
[docs] def allocateUniqueStdOutErr(self, processKey):
"""Allocate unique filenames of the form ``processKey[.n].out/.err``
which can be used for the `startProcess` ``stdouterr`` parameter.
The first time this is called it will return names like
``('myprocess.out', 'myprocess.err')``, the second time it will return
``('myprocess.1.out', 'myprocess.1.err')``, then
``('myprocess.2.out', 'myprocess.2.err')`` etc.
:param str processKey: A user-defined identifier that will form the prefix onto which ``[.n].out`` is appended
:return: A STDOUTERR_TUPLE named tuple of (stdout, stderr)
:rtype: STDOUTERR_TUPLE
"""
newval = self.__uniqueProcessKeys.get(processKey, -1)+1
self.__uniqueProcessKeys[processKey] = newval
suffix = '.%d'%(newval) if newval > 0 else ''
return STDOUTERR_TUPLE(
os.path.join(self.output, processKey+suffix+'.out'),
os.path.join(self.output, processKey+suffix+'.err'),
)
[docs] def getInstanceCount(self, displayName):
"""(Deprecated) Return the number of processes started within the testcase matching the supplied displayName.
:deprecated: The recommended way to allocate unique names is now L{allocateUniqueStdOutErr}
The ProcessUser class maintains a reference count of processes started within the class instance
via the L{startProcess()} method. The reference count is maintained against a logical name for
the process, which is the C{displayName} used in the method call to L{startProcess()}, or the
basename of the command if no displayName was supplied. The method returns the number of
processes started with the supplied logical name, or 0 if no processes have been started.
:param displayName: The process display name
:return: The number of processes started matching the command basename
:rtype: int
"""
if displayName in self.processCount:
return self.processCount[displayName]
else:
return 0
def setKeywordArgs(self, xargs):
"""Set the xargs as data attributes of the class. For internal use by BaseTest/BaseRunner only.
:meta private:
Values in the xargs dictionary are set as data attributes using the builtin C{setattr} method.
Thus an xargs dictionary of the form C{{'foo': 'bar'}} will result in a data attribute of the
form C{self.foo} with C{value bar}. This is used so that subclasses can define default values of
data attributes, which can be overriden on instantiation e.g. using the -X options to the
runTest.py launch executable.
If an existing attribute is present on this test class (typically a
static class variable) and it has a type of bool, int or float, then
any -X options will be automatically converted from string to that type.
This facilitates providing default values for parameters such as
iteration count or timeouts as static class variables with the
possibility of overriding on the command line, for example `-Xiterations=123`.
:param xargs: A dictionary of the user defined extra arguments
"""
pysys.utils.misc.setInstanceVariablesFromDict(self, xargs)
[docs] def getBoolProperty(self, propertyName, default=False):
"""
Get a True/False indicating whether the specified attribute is set
on this object (typically as a result of specifying -X on the command
line), or else from the project configuration.
See also `pysys.baserunner.getXArg()` and `pysys.xml.project.Project.getProperty()`.
:param propertyName: The name of a property set on the command line
or project configuration.
"""
val = getattr(self, propertyName, None)
if val is None: val = getattr(self.project, propertyName, None)
if val is None: return default
if val is True or val is False: return val
return val.lower()=='true'
[docs] def startPython(self, arguments, disableCoverage=False, **kwargs):
"""
Start a Python process with the specified arguments.
Uses the same Python process the tests are running under.
If PySys was run with the argument ``-XcodeCoverage`` or ``-XpythonCoverage`` then
`startPython` will add the necessary arguments to enable generation of
code coverage. Note that this requried the coverage.py library to be
installed. If a project property called `pythonCoverageArgs` exists
then its value will be added as (space-delimited) arguments to the
coverage tool.
:param arguments: The arguments to pass to the Python executable.
Typically the first one be either the name of a Python script
to execute, or ``-m`` followed by a module name.
:param kwargs: See L{startProcess} for detail on available arguments.
:param disableCoverage: Disables code coverage for this specific
process. Coverage can also be disabled by setting
``self.disableCoverage==True`` on this test instance.
:return: The process handle of the process.
:rtype: pysys.process.Process
"""
args = arguments
if 'environs' in kwargs:
environs = kwargs['environs']
else:
environs = kwargs.setdefault('environs', self.getDefaultEnvirons(command=sys.executable))
coverageWriter = [writer for writer in getattr(self, 'runner', self).writers if isinstance(writer, pysys.writer.PythonCoverageWriter)]
if coverageWriter and not disableCoverage and not self.disableCoverage:
args = ['-m', 'coverage', 'run']+coverageWriter[0].getCoverageArgsList()+args
if 'COVERAGE_FILE' not in environs:
kwargs['environs'] = dict(environs)
with self.lock:
self.__pythonCoverageFile += 1
kwargs['environs']['COVERAGE_FILE'] = self.output+'/.coverage.python.%02d'%(self.__pythonCoverageFile)
return self.startProcess(sys.executable, arguments=args, **kwargs)
[docs] def startProcess(self, command, arguments, environs=None, workingDir=None, state=None,
timeout=TIMEOUTS['WaitForProcess'], stdout=None, stderr=None, displayName=None,
abortOnError=None, expectedExitStatus='==0', ignoreExitStatus=None, onError=None, quiet=False, stdouterr=None,
background=False, info={}):
"""Start a process running in the foreground or background, and return
the `pysys.process.Process` object.
Typical use is::
myexecutable = self.startProcess('path_to_my_executable',
arguments=['myoperation', 'arg1','arg2'],
environs=self.createEnvirons(addToLibPath=['my_ld_lib_path']), # if a customized environment is needed
stdouterr=self.allocateUniqueStdOutErr('myoperation'), # for stdout/err files, pick a suitable logical name for what it's doing
background=True # or remove for default behaviour of executing in foreground
)
The method allows spawning of new processes in a platform independent way. The command, arguments,
environment and working directory to run the process in can all be specified in the arguments to the
method, along with the filenames used for capturing the stdout and stderr of the process. Processes may
be started in the foreground, in which case the method does not return until the process has completed
or a time out occurs, or in the background in which case the method returns immediately to the caller
returning a handle to the process to allow manipulation at a later stage, typically with L{waitProcess}.
All processes started in the background are automatically killed on completion of the test via the L{cleanup()}
destructor. To wait for background processes to complete during `execute` and/or before `verify` commences,
call `waitForBackgroundProcesses()`. This is especially useful when you have a test that needs to execute
lots of processes asynchronously, since having them all execute concurrently in the background and then calling
`waitForBackgroundProcesses()` will be a lot quicker than executing them serially in the foreground.
When starting a process that will listen on a server socket, use `getNextAvailableTCPPort`
to allocate a free port before calling this method.
.. versionchanged:: 1.6.0
Added onError parameter and default behaviour of logging stderr/out when there's a failure.
Added info parameter.
:param str command: The path to the executable to be launched (should include the full path)
:param list[str] arguments: A list of arguments to pass to the command
:param dict(str,str) environs: A dictionary specifying the environment to run the process in.
If a None or empty dictionary is passed, L{getDefaultEnvirons} will be invoked to
produce a suitable clean default environment for this `command`, containing a minimal set of variables.
If you wish to specify a customized environment, L{createEnvirons()} is a great way to create it.
:param str workingDir: The working directory for the process to run in (defaults to the testcase output subdirectory)
:param bool background: Set to True to start the process in the background. By default processes are started
in the foreground, meaning execution of the test will continue only once the process has terminated.
:param state: Alternative way to set ``background=True``.
Run the process either in the C{FOREGROUND} or C{BACKGROUND} (defaults to C{FOREGROUND}). Setting
state=BACKGROUND is equivalent to setting background=True; in new tests using background=True is the
preferred way to do this.
:param int timeout: The number of seconds after which to terminate processes running in the foreground. For processes
that complete in a few seconds or less, it is best to avoid overriding this and stick with the default.
However for long-running foreground processes it will be necessary to set a larger number, for example
if running a soak test where the process needs to run for up to 2 hours you could set ``timeout=2*60*60``.
:param str stdouterr: The filename prefix to use for the stdout and stderr of the process
(`.out`/`.err` will be appended), or a tuple of (stdout,stderr) as returned from
L{allocateUniqueStdOutErr}.
The stdouterr prefix is also used to form a default display name for the process if none is explicitly provided.
The files are created relative to the test output directory.
The filenames can be accessed from the returned process object using ``.stdout/err`` from
`pysys.process.Process`.
:param str stdout: The filename used to capture the stdout of the process. It is usually simpler to use `stdouterr` instead of this.
:param str stderr: The filename used to capture the stderr of the process. It is usually simpler to use `stdouterr` instead of this.
:param str displayName: Logical name of the process used for display in log messages, and the str(...)
representation of the returned process object
(defaults to a string generated from the stdouterr and/or the command).
:param bool abortOnError: If true abort the test on any error outcome (defaults to the defaultAbortOnError
project setting)
:param str expectedExitStatus: The condition string used to determine whether the exit status/code
returned by the process is correct. The default is '==0', as an exit code of zero usually indicates success, but if you
are expecting a non-zero exit status (for example because you are testing correct handling of
a failure condition) this could be set to '!=0' or a specific value such as '==5'.
:param bool ignoreExitStatus: If False, a BLOCKED outcome is added if the process terminates with an
exit code that doesn't match expectedExitStatus (or if the command cannot be run at all).
This can be set to True in cases where you do not care whether the command succeeds or fails, or wish to handle the
exit status separately with more complicated logic.
The default value of ignoreExitStatus=None means the value will
be taken from the project property defaultIgnoreExitStatus, which can be configured in the project XML
(the recommended default property value is defaultIgnoreExitStatus=False), or is set to True for
compatibility with older PySys releases if no project property is set.
:param Callable[pysys.process.Process]->str onError: A function that will be called
if the process times out or returns an unexpected exit status (unless ignoreExitStatus=True), before
any abort exception is raised. This provides a convenient place to add logging of
diagnostic information (perhaps using the stdout/err of the process) and/or extracting and returning an
error message from the output, for example: ``onError=lambda process: self.logFileContents(process.stderr, tail=True) or self.logFileContents(process.stdout, tail=True)``.
If a string value is returned from it will be added to the failure reason, e.g. ``onError=lambda process: self.logFileContents(process.stderr, tail=True) and self.getExprFromFile(process.stderr, 'Error: (.*)')``.
If no onError function is specified, the default is to log the last few lines of stderr (or if empty, stdout)
when a process fails and abortOnError=True.
The ``self.logFileContentsDefaultExcludes`` variable can be used to add regular expressions to exclude
unimportant lines of output such as standard startup lines (see `logFileContents`).
:param bool quiet: If True, this method will not do any INFO or WARN level logging
(only DEBUG level), unless a failure outcome is appended. This parameter can be
useful to avoid filling up the log where it is necessary to repeatedly execute a
command check for completion of some operation until it succeeds; in such cases
you should usually set ignoreExitStatus=True as well since both success and
failure exit statuses are valid.
:param dict[str,obj] info: A dictionary of user-defined information about this process that will be set as
a field on the returned Process instance. This is useful for keeping track of things like server port
numbers and log file paths.
:return: The process wrapper object.
:rtype: pysys.process.Process
"""
if state is None: state = FOREGROUND
if background: state = BACKGROUND
if ignoreExitStatus == None: ignoreExitStatus = self.defaultIgnoreExitStatus
workingDir = os.path.join(self.output, workingDir or '')
if abortOnError == None: abortOnError = self.defaultAbortOnError
if stdouterr:
if stdout or stderr: raise Exception('Cannot specify both stdouterr and stdout/stderr')
if isstring(stdouterr):
stdout = stdouterr+'.out'
stderr = stdouterr+'.err'
else:
stdout, stderr = stdouterr
if not displayName:
# Heuristically the name selected by the user for stdout/err usually represents the
# logical purpose of the process so makes a great display name.
# Also add the command (unless they're the same).
# NB: We do not do this if stdout/stderr are used since that could break
# behaviour for old tests using getInstanceCount.
displayName = os.path.basename(stdout.replace('.out',''))
if os.path.basename(command) not in displayName and displayName not in command:
displayName = '%s<%s>'%(os.path.basename(command), displayName)
# in case stdout/err were given as non-absolute paths, make sure they go to the output dir not the cwd
if stdout: stdout = os.path.join(self.output, stdout)
if stderr: stderr = os.path.join(self.output, stderr)
if not displayName: displayName = os.path.basename(command)
if not environs: # a truly empty env isn't really usable, so populate it with a minimal default environment instead
environs = self.getDefaultEnvirons(command=command)
startTime = time.time()
process = ProcessWrapper(command, arguments, environs, workingDir, state, timeout, stdout, stderr,
displayName=displayName, expectedExitStatus=expectedExitStatus, info=info)
def handleErrorAndGetOutcomeSuffix(process):
if onError:
try:
suffix = onError(process)
except Exception as ex:
self.log.debug('Ignoring exception from onError handler: ', exc_info=True)
else:
if suffix and isstring(suffix): return ' - '+suffix.strip()
elif abortOnError:
self.logFileContents(process.stderr, tail=True) or self.logFileContents(process.stdout, tail=True)
return ''
try:
process.start()
if state == FOREGROUND:
correctExitStatus = pysys.internal.safe_eval.safe_eval('%d %s'%(process.exitStatus, expectedExitStatus), extraNamespace={'self':self})
logmethod = log.info if correctExitStatus else log.warn
if quiet: logmethod = log.debug
logmethod("Executed %s, exit status %d%s", displayName, process.exitStatus,
", duration %d secs" % (time.time()-startTime) if (int(time.time()-startTime)) > 10 else "")
if not ignoreExitStatus and not correctExitStatus:
if not stderr and not quiet: log.warn('Process %s has no stdouterr= specified; providing this parameter will allow PySys to capture the process output that shows why it failed', process)
self.addOutcome(BLOCKED,
(
('%s returned non-zero exit code %d'%(process, process.exitStatus))
if expectedExitStatus=='==0' else
('%s returned exit code %d (expected %s)'%(process, process.exitStatus, expectedExitStatus))
)+handleErrorAndGetOutcomeSuffix(process),
abortOnError=abortOnError)
elif state == BACKGROUND:
(log.info if not quiet else log.debug)("Started %s with process id %d", displayName, process.pid)
hasFailed = False
except ProcessError as e:
if not ignoreExitStatus:
self.addOutcome(BLOCKED, 'Could not start %s process: %s'%(displayName, e), abortOnError=abortOnError)
else: # this wouldn't happen during a polling-until-success use case so is always worth logging even in quiet mode
log.info("%s", sys.exc_info()[1], exc_info=0)
except ProcessTimeout:
(log.warn if not quiet else log.debug)("Process %r timed out after %d seconds, stopping process", process, timeout, extra=BaseLogFormatter.tag(LOG_TIMEOUTS))
process.stop()
self.addOutcome(TIMEDOUT, '%s timed out after %d seconds%s'%(process, timeout, handleErrorAndGetOutcomeSuffix(process)), printReason=False, abortOnError=abortOnError)
else:
with self.lock:
self.processList.append(process)
if displayName in self.processCount:
self.processCount[displayName] = self.processCount[displayName] + 1
else:
self.processCount[displayName] = 1
return process
[docs] def getDefaultEnvirons(self, command=None, **kwargs):
"""
Create a new dictionary of environment variables, suitable for passing to
L{startProcess()}, with a minimal clean set of environment variables
for this platform, unaffected (as much as possible) by the
environment that the tests are being run under.
This environment contains a minimal PATH/LD_LIBRARY_PATH but does not
attempt to replicate the full set of default environment variables
on each OS, and in particular it does not include any that identify
the the current username or home area. Additional environment
variables can be added as needed with L{createEnvirons} overrides. If
you don't care about minimizing the risk of your local environment
affecting the test processes you start, just use C{environs=os.environ}
to allow child processes to inherit the entire parent environment.
The L{createEnvirons()} and L{startProcess()} methods use this as the
basis for creating a new set of default environment variables.
If needed this method can be overridden in subclasses to add common
environment variables for every process invoked by startProcess, for
example to enable options such as code coverage for Java/Python/etc.
This is also a good place to customize behaviour for different
operating systems.
Some features of this method can be configured by setting project
properties:
- ``defaultEnvirons.ENV_KEY``: if any properties with this prefix are
defined, an environment variable with the ENV_KEY is set by this method
(unless the property value is empty). For example, to set a default
JVM heap size for all processes with the ``JAVA_TOOL_OPTIONS`` environment
variable you could set ``defaultEnvirons.JAVA_TOOL_OPTIONS = -Xmx512M``.
- ``defaultEnvironsDefaultLang``: if set to a value such as ``en_US.UTF-8``
the specified value is set for the LANG= variable on Unix; otherwise,
the LANG variable is not set (which might result in use of the
legacy POSIX/C encoding).
- ``defaultEnvironsTempDir``: if set the expression will be passed to
Python ``eval()`` and used to set the OS-specific temp directory
environment variables. A typical value is `self.output`.
- ``defaultEnvironsLegacyMode``: set to true to enable compatibility
mode which keeps the behaviour the same as PySys v1.1, 1.2 and 1.3,
namely using a completely empty default environment on Unix, and
a copy of the entire parent environment on Windows. This is not
recommended unless you have a lot of legacy tests that cannot
easily be changed to only set minimal required environment
variables using `createEnvirons()`.
:param command: If known, the full path of the executable for which
a default environment is being created (when called from `startProcess`
this is always set). This allows default environment variables to be
customized for different process types e.g. Java, Python, etc.
When using ``command=sys.executable`` to launch another copy of the
current Python executable, extra items from this process's path
environment variables are added to the returned dictionary so that it
can start correctly. On Unix-based systems this includes copying all of
the load library path environment variable from the parent process.
:param kwargs: Overrides of this method should pass any additional
kwargs down to the super implementation, to allow for future extensions.
:return: A new dictionary containing the environment variables.
"""
assert not kwargs, 'Unknown keyword arguments: %s'%kwargs.keys()
# this feature is a workaround to maintain compatibility for a bug in PySys v1.1-1.3
# (see https://github.com/pysys-test/pysys-test/issues/9 for details)
if getattr(self.project, 'defaultEnvironsLegacyMode','').lower()=='true':
if IS_WINDOWS:
return dict(os.environ)
else:
return {}
e = {}
for k, v in self.project.properties.items():
if k.startswith('defaultEnvirons.') and v:
e[k[k.find('.'):]] = v
# allows setting TEMP to output dir to avoid contamination/filling up of system location; set to blank to do nothing
if self.project.getProperty('defaultEnvironsTempDir',''):
tempDir = pysys.internal.safe_eval.safe_eval(self.project.defaultEnvironsTempDir, extraNamespace={'self':self})
self.mkdir(tempDir)
if IS_WINDOWS: # pragma: no cover
e['TEMP'] = e['TMP'] = os.path.normpath(tempDir)
else:
e['TMPDIR'] = os.path.normpath(tempDir)
inherited = []
# env vars where it is safe and useful to inherit parent values
# avoid anything user-specific or that might cause tests to store data
# outside the test outpuot directory
if IS_WINDOWS:
# for windows there are lots; as a matter of policy we set this to a small
# minimal set used by a lot of programs. Keeping up with every single env
# var Microsoft sets in every Windows OS release would be too painful,
# and better to make users explicitly opt-in to the env vars they want
inherited.extend(['ComSpec', 'OS', 'PATHEXT', 'SystemRoot', 'SystemDrive', 'windir',
'NUMBER_OF_PROCESSORS', 'PROCESSOR_ARCHITECTURE',
'COMMONPROGRAMFILES', 'COMMONPROGRAMFILES(X86)', 'PROGRAMFILES', 'PROGRAMFILES(X86)',
'SYSTEM', 'SYSTEM32'])
for k in inherited:
if k in os.environ: e[k] = os.environ[k]
# always set PATH/LD_LIB_PATH to clean values from constants.py
# note that if someone is using an OS with different defaults they won't
# be able to edit constants.py but will be able to provide a custom
# implementation of this method
e['PATH'] = PATH
if LD_LIBRARY_PATH:
e['LD_LIBRARY_PATH'] = LD_LIBRARY_PATH
if DYLD_LIBRARY_PATH:
e['DYLD_LIBRARY_PATH'] = DYLD_LIBRARY_PATH
if not IS_WINDOWS:
if getattr(self.project, 'defaultEnvironsDefaultLang',''):
e['LANG'] = self.project.defaultEnvironsDefaultLang
if command == sys.executable:
# Ensure it's possible to run another instance of this Python, by adding it to the start of the path env vars
# (but only if full path to the Python executable exactly matches).
# Keep it as clean as possible by not passing sys.path/PYTHONPATH
# - but it seems we do need to copy the LD_LIBRARY_PATH from the parent process to ensure the required libraries are present.
# Do not set PYTHONHOME here, as doesn't work well in virtualenv, and messes up grandchildren
# processes that need a different Python version
e['PATH'] = os.path.dirname(sys.executable)+os.pathsep+e['PATH']
if LIBRARY_PATH_ENV_VAR != 'PATH': # if it's an os with something like LD_LIBRARY_PATH
# It's a shame it's necessary to copy parent environment, but there's no sane way to unpick which libraries are
# actually required on Unix. Make sure we don't set this env var to an empty string just in case that
# doesn't anything weird.
newlibpath = (os.getenv(LIBRARY_PATH_ENV_VAR,'')+os.pathsep+e.get(LIBRARY_PATH_ENV_VAR,'')).strip(os.pathsep)
if newlibpath:
e[LIBRARY_PATH_ENV_VAR] = newlibpath
self.log.debug('getDefaultEnvirons was called with a command matching this Python executable; adding required path environment variables from parent environment, including %s=%s', LIBRARY_PATH_ENV_VAR, os.getenv(LIBRARY_PATH_ENV_VAR,''))
else:
self.log.debug('getDefaultEnvirons was called with a command matching this Python executable; adding required path environment variables from parent environment')
return e
[docs] def createEnvirons(self, overrides=None, addToLibPath=[], addToExePath=[], command=None, **kwargs):
"""
Create a new customized dictionary of environment variables suitable
for passing to L{startProcess()}'s ``environs=`` argument.
As a starting point, this method uses the value returned by
L{getDefaultEnvirons()} for this `command`. See the documentation on
that method for more details. If you don't care about minimizing the
risk of your local environment affecting the test processes you start,
just use C{environs=os.environ} to allow child processes to inherit the
entire parent environment instead of using this method.
:param overrides: A dictionary of environment variables whose
values will be used instead of any existing values.
You can use `os.getenv('VARNAME','')` if you need to pass selected
variables from the current process as part of the overrides list.
If the value is set to None then any variable of this name will be
deleted. Use unicode strings if possible (byte strings will be
converted depending on the platform).
A list of dictionaries can be specified, in which case the latest
will override the earlier if there are any conflicts.
:param addToLibPath: A path or list of paths to be prepended to the
default value for the environment variable used to load libraries
(or the value specified in overrides, if any),
i.e. ``[DY]LD_LIBRARY_PATH`` on Unix or ``PATH`` on Windows. This is usually
more convenient than adding it directly to `overrides`.
:param addToExePath: A path or list of paths to be prepended to the
default value for the environment variable used to locate executables
(or the value specified in overrides, if any),
i.e. ``PATH`` on both Unix and Windows. This is usually
more convenient than adding it directly to ``overrides``.
:param command: If known, the full path of the executable for which
a default environment is being created (passed to L{getDefaultEnvirons}).
:param kwargs: Overrides of this method should pass any additional
kwargs down to the super implementation, to allow for future extensions.
:return: A new dictionary containing the environment variables.
"""
assert not kwargs, 'Unknown keyword arguments: %s'%kwargs.keys()
e = self.getDefaultEnvirons(command=command)
if overrides:
if not isinstance(overrides, list): overrides = [overrides]
for d in overrides:
if d:
for k in d:
if k.upper() in ['PATH']: k = k.upper() # normalize common ones to avoid chance of duplicates
if d[k] is None:
e.pop(k, None) # remove
else:
e[k] = d[k]
def preparepath(path):
if isstring(path):
if os.pathsep not in path: path = os.path.normpath(path)
else:
path = os.pathsep.join([os.path.normpath(p) for p in path if p])
return path
if addToLibPath:
e[LIBRARY_PATH_ENV_VAR] = preparepath(addToLibPath)+os.pathsep+e[LIBRARY_PATH_ENV_VAR]
if addToExePath:
e['PATH'] = preparepath(addToExePath)+os.pathsep+e['PATH']
return e
[docs] def stopProcess(self, process, abortOnError=None):
"""Stops the specified process, if it is currently running.
Does nothing if the process is not running.
This is equivalent to calling `pysys.process.Process.stop()`, except it also
logs an info message when the process is stopped.
:param process: The process handle returned from the L{startProcess} method
:param abortOnError: If True abort the test on any error outcome (defaults to the defaultAbortOnError
project setting), if False a failure to stop the process will just be logged as a warning.
"""
if abortOnError == None: abortOnError = self.defaultAbortOnError
if process.running():
try:
process.stop()
log.info("Stopped process %r", process)
except ProcessError as e:
if not abortOnError:
log.warn("Ignoring failure to stop process %r due to: %s", process, e)
else:
self.abort(BLOCKED, 'Unable to stop process %r'%(process), self.__callRecord())
[docs] def signalProcess(self, process, signal, abortOnError=None):
"""Send a signal to a running process.
This method uses the `pysys.process.Process.signal` method to send a signal to a
running process.
Should the request to send the signal to the running process fail, a C{BLOCKED} outcome will be added to the
outcome list.
:param process: The process handle returned from the L{startProcess} method
:param signal: The integer value of the signal to send
:param abortOnError: If True aborts the test with an exception on any error, if False just log it as a warning.
(defaults to the defaultAbortOnError project setting)
"""
if abortOnError == None: abortOnError = self.defaultAbortOnError
if process.running():
try:
process.signal(signal)
log.info("Sent %d signal to process %r", signal, process)
except ProcessError as e:
if not abortOnError:
log.warn("Ignoring failure to signal process %r due to: %s", process, e)
else:
self.abort(BLOCKED, 'Unable to signal process %r'%(process), self.__callRecord())
[docs] def waitProcess(self, process, timeout=TIMEOUTS['WaitForProcess'], abortOnError=None, checkExitStatus=False):
"""Wait for a background process to terminate.
For a convenient way to wait for all remaining background processes to complete, see `waitForBackgroundProcesses`.
Timeouts will result in an exception and TIMEDOUT outcome unless the project property ``defaultAbortOnError==False``
is set.
:param pysys.process.Process process: The process handle returned from the L{startProcess} method
:param int timeout: The timeout value in seconds to wait before returning, for example ``timeout=TIMEOUTS['WaitForProcess']``.
:param bool abortOnError: If True aborts the test with an exception on any error, if False just log it as a warning.
(defaults to the defaultAbortOnError project setting)
:param bool checkExitStatus: By default this method does not check the exit status, but set this argument to True
to check that the exit status matches the ``expectedExitStatus`` specified in the call to `startProcess`
(typically ``==0``).
Note that this argument is not affected by the ``defaultIgnoreExitStatus`` project property.
The last few lines of the stderr (or stdout) will be logged if the exit status is wrong.
:return: The process's ``exitStatus``. This will be None if the process timed out and abortOnError is disabled.
"""
if abortOnError == None: abortOnError = self.defaultAbortOnError
try:
log.info("Waiting up to %d secs for process %r", timeout, process)
t = time.time()
process.wait(timeout)
if (time.time()-t > 10) or process.exitStatus != 0:
log.info("Process %s terminated after %d secs with exit status %d", process, time.time()-t, process.exitStatus)
except ProcessTimeout:
if not abortOnError:
log.warn("Ignoring timeout waiting for process %r after %d secs (as abortOnError=False)", process, time.time() - t, extra=BaseLogFormatter.tag(LOG_TIMEOUTS))
else:
self.abort(TIMEDOUT, 'Timed out waiting for process %s after %d secs'%(process, timeout), self.__callRecord())
else:
if checkExitStatus:
if not pysys.internal.safe_eval.safe_eval('%d %s'%(process.exitStatus, process.expectedExitStatus), extraNamespace={'self':self}):
self.logFileContents(process.stderr, tail=True) or self.logFileContents(process.stdout, tail=True)
self.addOutcome(BLOCKED,
('%s returned exit status %d (expected %s)'%(process, process.exitStatus, process.expectedExitStatus)),
abortOnError=abortOnError)
return process.exitStatus
[docs] def waitForBackgroundProcesses(self, includes=[], excludes=[], timeout=TIMEOUTS['WaitForProcess'], abortOnError=None, checkExitStatus=True):
"""Wait for any running background processes to terminate, then check that all background processes
completed with the expected exit status.
This can be useful for speeding up a test that needs to run many processes, by executing all its subprocesses
in parallel in the background (and then waiting for completion) rather than one-by-one in the foreground.
Timeouts will result in a TIMEDOUT outcome and an exception unless the project property ``defaultAbortOnError==False``
is set.
:param list[pysys.process.Process] includes: A list of processes to wait for, each returned by `startProcess()`.
If none are specified, this method will wait for (and check the exit status of) all background processes.
:param list[pysys.process.Process] excludes: A list of processes which are not expected to have terminated yet
(this is only useful when not setting includes=[]).
:param int timeout: The total time in seconds to wait for all processes to have completed, for example ``timeout=TIMEOUTS['WaitForProcess']``.
:param bool abortOnError: If True aborts the test with an exception on any error, if False appends an outcome but does not
raise an exception.
:param bool checkExitStatus: By default this method not only waits for completion but also checks the exit status of
all (included) background processes (regardless of when they completed), but set this argument to False
to disable checking that the exit status of the processes matches the ``expectedExitStatus`` specified in
the call to `startProcess` (typically ``==0``).
The last few lines of the stderr (or stdout) will be logged if the exit status is wrong.
"""
assert timeout>0, 'timeout must be specified'
starttime = time.time()
includes = includes or self.processList
running = [p for p in includes if p.state==BACKGROUND and p.running() and p not in excludes]
self.log.info('Waiting up to %d secs for %d background process(es) to complete', timeout, len(running))
for p in running:
try:
thistimeout = starttime+timeout-time.time()
if thistimeout <= 0: # we've run out of time
raise ProcessTimeout('waitForBackgroundProcesses timed out')
if p.running():
p.wait(timeout=thistimeout) # may raise ProcessTimeout
except ProcessTimeout:
stillrunning = [p for p in running if p.state==BACKGROUND and p.running()]
if stillrunning:
self.addOutcome(TIMEDOUT, 'Timed out waiting for %d processes after %d secs: %s'%(len(stillrunning), timeout, ', '.join(map(str, stillrunning))), abortOnError=abortOnError)
return False
# now check exit statuses - for all background processes, not only those that happen to have completed recently;
# but only after we've confirmed (above) there were no timeouts
if checkExitStatus:
failures = []
for process in includes:
if process.state!=BACKGROUND or process in excludes: continue
if not pysys.internal.safe_eval.safe_eval('%s %s'%(process.exitStatus, process.expectedExitStatus), extraNamespace={'self':self}):
self.logFileContents(process.stderr, tail=True) or self.logFileContents(process.stdout, tail=True)
failures.append('%s returned exit status %s (expected %s)'%(process, process.exitStatus, process.expectedExitStatus))
if failures:
self.addOutcome(BLOCKED, ('%d processes failed: '%len(failures) if len(failures)>1 else 'Process ')+'; '.join(failures), abortOnError=abortOnError)
(self.log.info if (time.time()-starttime>10) else self.log.debug)('All processes completed, after waiting %d secs'%(time.time()-starttime))
[docs] def writeProcess(self, process, data, addNewLine=True):
"""Write binary data to the stdin of a process.
This method uses `pysys.process.Process.write` to write binary data to the stdin of a process. This
wrapper around the write method of the process helper only adds checking of the process running status prior
to the write being performed, and logging to the testcase run log to detail the write.
:param pysys.process.Process process: The process handle returned from the L{startProcess()} method
:param bytes data: The data to write to the process stdin.
As only binary data can be written to a process stdin,
if a character string rather than a byte object is passed as the data,
it will be automatically converted to a bytes object using the encoding
given by locale.getpreferredencoding().
:param bool addNewLine: True if a new line character is to be added to the end of the data string
"""
if process.running():
process.write(data, addNewLine)
log.info("Written to stdin of process %r", process)
log.debug(" %s" % data)
else:
raise Exception("Write to process %r stdin not performed as process is not running", process)
[docs] def waitForSocket(self, port, host='localhost', timeout=TIMEOUTS['WaitForSocket'], abortOnError=None, process=None,
socketAddressFamily=socket.AF_INET):
"""Wait until it is possible to establish a socket connection to a
server running on the specified local or remote port.
This method blocks until connection to a particular host:port pair can be established. This is useful for
test timing where a component under test creates a socket for client server interaction - calling of this
method ensures that on return of the method call the server process is running and a client is able to
create connections to it. If a connection cannot be made within the specified timeout interval, the method
returns to the caller, or aborts the test if abortOnError=True.
.. versionchanged:: 1.5.1
Added host and socketAddressFamily parameters.
:param port: The port value in the socket host:port pair
:param host: The host value in the socket host:port pair
:param timeout: The timeout in seconds to wait for connection to the socket
:param abortOnError: If true abort the test on any failure (defaults to the defaultAbortOnError
project setting)
:param process: If a handle to a process is specified, the wait will abort if
the process dies before the socket becomes available. It is recommended to set this wherever possible.
:param socketAddressFamily: The socket address family e.g. IPv4 vs IPv6. See Python's ``socket`` module for
details.
"""
if abortOnError == None: abortOnError = self.defaultAbortOnError
log.debug("Performing wait for socket creation %s:%s", host, port)
s = None
try:
startTime = time.time()
while True:
if s is None:
with process_lock:
s = socket.socket(socketAddressFamily, socket.SOCK_STREAM)
# the following lines are to prevent handles being inherited by
# other processes started while this test is runing
if IS_WINDOWS:
s.setsockopt(socket.SOL_SOCKET, socket.SO_LINGER, 0)
win32api.SetHandleInformation(s.fileno(), win32con.HANDLE_FLAG_INHERIT, 0)
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
else:
fcntl.fcntl(s.fileno(), fcntl.F_SETFD, 1)
try:
s.connect((host, port))
s.shutdown(socket.SHUT_RDWR)
log.debug("Wait for socket creation completed successfully")
if time.time()-startTime>10:
log.info("Wait for socket creation completed after %d secs", time.time()-startTime)
return True
except socket.error as ex:
if process and not process.running():
msg = "Waiting for socket connection aborted due to unexpected process %s termination"%(process)
if abortOnError:
self.abort(BLOCKED, msg, self.__callRecord())
else:
log.warn('%s', msg)
return False
if timeout:
currentTime = time.time()
if currentTime > startTime + timeout:
msg = "Timed out waiting for creation of socket after %d secs: %s"%(time.time()-startTime, ex)
if abortOnError:
self.abort(TIMEDOUT, msg, self.__callRecord())
else:
log.warn('%s', msg)
return False
if PLATFORM == 'darwin':
# MacOS gives an error if we try to connect to the same socket again after connection refused
s.close()
s = None
time.sleep(0.15)
finally:
if s is not None: s.close()
[docs] def waitForFile(self, file, filedir=None, timeout=TIMEOUTS['WaitForFile'], abortOnError=None):
"""Wait for a file to exist on disk.
This method blocks until a file is created on disk. This is useful for test timing where
a component under test creates a file (e.g. for logging) indicating it has performed all
initialisation actions and is ready for the test execution steps. If a file is not created
on disk within the specified timeout interval, the method returns to the caller.
:param file: The basename of the file used to wait to be created
:param filedir: The dirname of the file (defaults to the testcase output subdirectory)
:param timeout: The timeout in seconds to wait for the file to be created
:param abortOnError: If true abort the test on any failure (defaults to the defaultAbortOnError
project setting)
"""
if abortOnError == None: abortOnError = self.defaultAbortOnError
if filedir is None: filedir = self.output
f = os.path.join(filedir, file)
log.debug("Performing wait for file creation: %s", f)
startTime = time.time()
while True:
if timeout:
currentTime = time.time()
if currentTime > startTime + timeout:
msg = "Timed out waiting for creation of file %s after %d secs" % (file, time.time()-startTime)
if abortOnError:
self.abort(TIMEDOUT, msg, self.__callRecord())
else:
log.warn(msg)
break
time.sleep(0.01)
if pathexists(f):
log.debug("Wait for '%s' file creation completed successfully", file)
return
[docs] def waitForSignal(self, file, filedir=None, expr="", **waitForGrepArgs):
"""Old alias for `waitForGrep`; please use `waitForGrep` in new tests.
All parameters are the same, except that in waitForSignal the (rarely used) ``filedir`` argument can be
specified as the 2nd positional argument (after ``file`` and before ``expr``) whereas in waitForGrep it can
only be specified as a ``filedir=`` keyword argument.
"""
return self.waitForGrep(file, expr=expr, filedir=filedir, **waitForGrepArgs)
[docs] def waitForGrep(self, file, expr="", condition=">=1", timeout=TIMEOUTS['WaitForSignal'], poll=0.25,
ignores=[], process=None, errorExpr=[], errorIf=None, abortOnError=None, encoding=None, detailMessage='', filedir=None,
reFlags=0, mappers=[]):
"""Wait for a regular expression line to be seen (one or more times) in a text file in the output
directory (waitForGrep was formerly known as `waitForSignal`).
This method provides some parameters that give helpful fail-fast behaviour with a descriptive outcome reason;
use these whenever possible:
- ``process=`` to abort if success becomes impossible due to premature termination of the process that's
generating the output
- ``errorExpr=`` to abort if an error message/expression is written to the file being grepped
- ``errorIf=`` to abort if the specified lambda function returns an error string (which can be used if the
error messages go to a different file than that being grepped
This will generate much clearer outcome reasons, which makes test failures easy to triage,
and also avoids wasting time waiting for something that will never happen.
Example::
self.waitForGrep('myprocess.log', expr='INFO .*Started successfully', encoding='utf-8',
process=myprocess, errorExpr=[' (ERROR|FATAL) ', 'Failed to start'])
self.waitForGrep('myoutput.txt', expr='My message', encoding='utf-8',
process=myprocess, errorIf=lambda: self.getExprFromFile('myprocess.log', ' ERROR .*', returnNoneIfMissing=True))
Note that waitForGrep fails the test if the expression is not found (unless abortOnError was set to False,
which isn't recommended), so there is no need to add duplication with an
`assertGrep <pysys.basetest.BaseTest.assertGrep>` to check for the same expression in your validation logic.
You can extract information from the matched expression, optionally perform assertions on it, by
using one or more ``(?P<groupName>...)`` named groups in the expression. A common
pattern is to unpack the resulting dict using ``**kwargs`` syntax and pass to `BaseTest.assertThat`. For
example::
self.assertThat('username == expected', expected='myuser',
**self.waitForGrep('myserver.log', expr=r'Successfully authenticated user "(?P<username>[^"]*)"'))
.. versionadded:: 1.5.1
:param str file: The path of the file to be searched. Usually this is a name/path relative to the
``self.output`` directory, but alternatively an absolute path can be specified.
:param str expr: The regular expression to search for in the text file.
:param str condition: The condition to be met for the number of lines matching the regular expression; by default
we wait until there is at least one occurrence.
:param int timeout: The number of seconds to wait for the regular expression before giving up and aborting
the test with `pysys.constants.TIMEDOUT` (unless abortOnError=False in which case execution will continue).
:param pysys.process.Process process: The process that is generating the specified
file, to allow the wait to fail fast (instead of timing out) if the process dies before the expected signal
appears. Can be None if the process is not known or is expected to terminate itself during this period.
:param list[str] errorExpr: Optional list of regular expressions, which if found in the file will cause waiting
for the main expression to be aborted with a `pysys.constants.BLOCKED` outcome. This is useful to avoid waiting
a long time for the expected expression when an ERROR is logged that means it will never happen, and
also provides much clearer test failure messages in this case.
:param callable->str errorIf: A zero-arg function that returns False/None when there is no error, or a non-empty
string if an error is detected which should cause us to abort looking for the grep expression.
This function will be executed frequently (every ``poll`` seconds) so avoid
doing anything time-consuming here unless you set a large polling interval.
Added in PySys 1.6.0.
:param list[str] ignores: A list of regular expressions used to identify lines in the files which should be ignored
when matching both `expr` and `errorExpr`.
:param List[callable[str]->str] mappers: A list of filter functions that will be used to pre-process each
line from the file (returning None if the line is to be filtered out). This provides a very powerful
capability for filtering the file, for example `pysys.mappers.IncludeLinesBetween`
provides the ability to filter in/out sections of a file.
Do not share mapper instances across multiple tests or threads as this can cause race conditions.
Added in PySys 1.6.0.
:param float poll: The time in seconds between to poll the file looking for the regular expression and to check against the condition
:param bool abortOnError: If True abort the test on any error outcome (defaults to the defaultAbortOnError
project setting, which for a modern project will be True).
:param str encoding: The encoding to use to open the file and convert from bytes to characters.
The default value is None which indicates that the decision will be delegated
to the L{getDefaultFileEncoding()} method.
:param str detailMessage: An extra string to add to the message logged when waiting to provide extra
information about the wait condition. e.g. ``detailMessage='(downstream message received)'``.
Added in v1.5.1.
:param int reFlags: Zero or more flags controlling how the behaviour of regular expression matching,
combined together using the ``|`` operator, for example ``reFlags=re.VERBOSE | re.IGNORECASE``.
For details see the ``re`` module in the Python standard library. Note that ``re.MULTILINE`` cannot
be used because expressions are matched against one line at a time. Added in PySys 1.5.1.
:param str filedir: Can be used to provide a directory name to add to the beginning of the ``file`` parameter;
however usually it is clearer just to specify that directory in the ``file``.
:return list[re.Match]: Usually this returns a list of ``re.Match`` objects found for the ``expr``, or an
empty list if there was no match.
If the expr contains any ``(?P<groupName>...)`` named groups, and assuming the condition is still the
default of ">=1" (i.e. not trying to find multiple matches), then instead of a list, a dict is returned
containing ``dict(groupName: str, matchValue: str or None)`` (or an empty ``{}`` dict if there is no match)
which allows the result to be passed to `assertThat` for further checking of the matched groups (typically
unpacked using the ``**`` operator; see example above).
"""
assert expr, 'expr= argument must be specified'
assert '\n' not in expr, 'expr= cannot contain multiple lines'
if abortOnError == None: abortOnError = self.defaultAbortOnError
if filedir is None: filedir = self.output
f = os.path.join(filedir, file)
if errorExpr: assert not isstring(errorExpr), 'errorExpr must be a list of strings not a string'
matches = []
startTime = time.time()
msg = "Waiting for {expr} {condition}in {file}{detail}".format(
expr=quotestring(expr), # repr performs escaping of embedded quotes, newlines, etc
condition=condition.strip()+' ' if condition!='>=1' else '', # only include if non-default
file=os.path.basename(file),
detail=' '+detailMessage.strip(' ') if detailMessage else ''
)
log.debug("Performing wait for grep signal %s %s in file %s with ignores %s", expr, condition, f, ignores)
verboseWaitForSignal = self.getBoolProperty('verboseWaitForSignal', True) and self.getBoolProperty('verboseWaitForGrep', True)
if verboseWaitForSignal:
# if verbose, log when starting (which is very helpful for debugging hangs); non-verbose users get the message only when it's done
log.info('%s%s', msg, '; timeout=%ss'%timeout if timeout!=TIMEOUTS['WaitForSignal'] else '')
encoding = encoding or self.getDefaultFileEncoding(f)
# If condition was customized (typically to be more than 1) named groups mode isn't so useful since there would
# be multiple matches, so restrict it to 1 which is the common case anyway.
compiled = re.compile(expr, flags=reFlags)
namedGroupsMode = compiled.groupindex and condition.replace(' ','')=='>=1'
timetaken = time.time()
while 1:
if pathexists(f):
matches = getmatches(f, expr, encoding=encoding, ignores=ignores, flags=reFlags, mappers=mappers)
if pysys.internal.safe_eval.safe_eval("%d %s" % (len(matches), condition), extraNamespace={'self':self}):
timetaken = time.time()-timetaken
# Old-style/non-verbose behaviour is to log only after complete,
# new/verbose style does the main logging at INFO when starting, and only logs on completion if it took a long time
# (this helps people debug tests that sometimes timeout and sometimes "nearly" timeout)
if verboseWaitForSignal:
(log.info if timetaken > 30 else log.debug)(" ... found %d matches in %ss", len(matches), int(timetaken))
else:
# We use the phrase "grep signal" to avoid misleading anyone, whether people used waitForGrep or the older waitForSignal
log.info("Wait for grep signal in %s completed successfully", file)
break
if errorExpr:
for err in errorExpr:
errmatches = getmatches(f, err+'.*', encoding=encoding, ignores=ignores, flags=reFlags, mappers=mappers) # add .* to capture entire err msg for a better outcome reason
if errmatches:
err = errmatches[0].group(0).strip()
msg = '%s found while %s'%(quotestring(err), msg)
# always report outcome for this case; additionally abort if requested to
self.addOutcome(BLOCKED, outcomeReason=msg, abortOnError=abortOnError, callRecord=self.__callRecord())
return {} if namedGroupsMode else matches
currentTime = time.time()
if currentTime > startTime + timeout:
msg = "%s timed out after %d secs, %s"%(msg, timeout,
("with %d matches"%len(matches)) if pathexists(f) else 'file does not exist')
if abortOnError:
self.abort(TIMEDOUT, msg, self.__callRecord())
else:
log.warn(msg, extra=BaseLogFormatter.tag(LOG_TIMEOUTS))
break
if errorIf is not None:
errmsg = errorIf()
if errmsg:
msg = "%s aborted due to errorIf=%s"%(msg, errmsg)
if abortOnError:
self.abort(BLOCKED, msg, self.__callRecord())
else:
log.warn(msg)
break
if process and not process.running():
msg = "%s aborted due to process %s termination"%(msg, process)
if abortOnError:
self.abort(BLOCKED, msg, self.__callRecord())
else:
log.warn(msg)
break
time.sleep(poll)
if namedGroupsMode:
return {} if not matches else matches[0].groupdict()
return matches
[docs] def addCleanupFunction(self, fn, ignoreErrors=False):
""" Registers a function that will be called as part of the `cleanup` of this object.
Cleanup functions should have no arguments, and are invoked in reverse order with the most recently added first (LIFO), and
before the automatic termination of any remaining processes associated with this object.
e.g.::
self.addCleanupFunction(lambda: self.cleanlyShutdownMyProcess(params))
:param Callable[] fn: The cleanup function.
:param bool ignoreErrors: By default, errors from cleanup functions will result in a test failure; set this to
True to log them but not produce a test failure. This parameter was added in 1.6.0.
"""
with self.lock:
if fn and (fn,ignoreErrors) not in self.__cleanupFunctions:
self.__cleanupFunctions.append( (fn, ignoreErrors) )
[docs] def cleanup(self):
""" Tear down function that frees resources managed by this object.
Should be called exactly once by the owner of this object when is no longer needed.
Do not override this method, instead use `addCleanupFunction`.
"""
exceptions = []
try:
# although we don't yet state this method is thread-safe, make it
# as thread-safe as possible by using swap operations
with self.lock:
cleanupfunctions, self.__cleanupFunctions = self.__cleanupFunctions, []
if cleanupfunctions:
log.info('')
log.info('cleanup:')
for fn, ignoreErrors in reversed(cleanupfunctions):
try:
log.debug('Running registered cleanup function: %r'%fn)
fn()
except Exception as e:
(log.warn if ignoreErrors else log.error)('Error while running cleanup function%s: ', ' (ignoreErrors=True)' if ignoreErrors else '', exc_info=True)
if not ignoreErrors:
exceptions.append('Cleanup function failed: %s (%s)'%(e, type(e).__name__))
finally:
with self.lock:
processes, self.processList = self.processList, []
for process in processes:
try:
if process.running(): process.stop()
except Exception as e: # this is pretty unlikely to fail, but we'd like to know if it does
log.warning("caught %s: %s", sys.exc_info()[0], sys.exc_info()[1], exc_info=1)
exceptions.append('Failed to stop process %s: %s'%(process, e))
self.processCount = {}
log.debug('ProcessUser cleanup function done.')
if exceptions:
raise UserError('Cleanup failed%s: %s'%(' with %d errors'%len(exceptions), '; '.join(exceptions)))
[docs] def addOutcome(self, outcome, outcomeReason='', printReason=True, abortOnError=False, callRecord=None, override=False):
"""Add a validation outcome (and optionally a reason string) to the validation list.
The method provides the ability to add a validation outcome to the internal data structure
storing the list of validation outcomes. Multiple validations may be performed, the current
supported validation outcomes of which are described in :ref:`assertions-and-outcomes`.
The outcomes are considered to have a precedence order, as defined by the order of the outcomes listed
above. Thus a C{pysys.constants.BLOCKED} outcome has a higher precedence than a C{pysys.constants.PASSED}
outcome. The outcomes are defined in L{pysys.constants}.
This method is thread-safe.
Although this method exists on all subclasses of `pysys.process.user.ProcessUser`, in practice only
`pysys.basetest.BaseTest` subclasses actually do anything with the resulting outcome.
:param pysys.constants.Outcome outcome: The outcome to add, e.g. `pysys.constants.FAILED`.
:param str outcomeReason: A string summarizing the reason for the outcome,
for example "Grep on x.log contains 'ERROR: server failed'".
:param bool printReason: If True the specified outcomeReason will be printed
:param bool abortOnError: If true abort the test on any error outcome. This should usually be set to
False for assertions, or the configured `self.defaultAbortOnError` setting (typically True) for
operations that involve waiting.
:param list[str] callRecord: An array of strings of the form path:lineno indicating the call stack that lead
to this outcome. This will be appended to the log output for better test triage.
:param bool override: Remove any existing test outcomes when adding this one, ensuring
that this outcome is the one and only one reported even if an existing outcome
has higher precedence.
"""
assert outcome in OUTCOMES, outcome # ensure outcome type is known, and that numeric not string constant was specified!
with self.lock:
if abortOnError == None: abortOnError = self.defaultAbortOnError
if outcomeReason is None:
outcomeReason = ''
else:
if PY2 and isinstance(outcomeReason, str):
# The python2 logger is very unhappy about byte str objects containing
# non-ascii characters (specifically it will fail to log them and dump a
# traceback on stderr). Since it's pretty important that assertion
# messages and test outcome reasons don't get swallowed, add a
# workaround for this here. Not a problem in python 3.
outcomeReason = outcomeReason.decode('ascii', errors='replace')
outcomeReason = stripANSIEscapeCodes(outcomeReason).strip().replace(u'\t', u' ').replace('\r','').replace('\n', ' ; ')
if override:
log.debug('addOutcome is removing existing outcome(s): %s with reason "%s"', self.outcome, self.__outcomeReason)
del self.outcome[:]
self.__outcomeReason = None
old = self.getOutcome()
if (old == NOTVERIFIED and not self.__outcomeReason): old = None
self.outcome.append(outcome)
if callRecord==None: callRecord = self.__callRecord()
def parseLocation(cr):
if not cr or ':' not in cr: return None
return cr[:cr.rfind(':')], cr[cr.rfind(':')+1:]
#store the reason of the highest precedent outcome
# although we should print whatever is passed in, store a version with control characters stripped
# out so that it's easier to read (e.g. coloring codes from third party tools)
if self.getOutcome() != old:
self.__outcomeReason = re.sub(u'[\x00-\x08\x0b\x0c\x0e-\x1F]', '', outcomeReason)
self.__outcomeLocation = None, None
if callRecord:
locations = [parseLocation(loc) for loc in callRecord]
if len(locations)>1: # if possible, try to prioritize locations under the testDir above shared framework classes
locations = [loc for loc in locations if loc[0].lower().startswith(self.descriptor.testDir.lower()) ]
if len(locations) == 0: locations = [parseLocation(loc) for loc in callRecord]
loc = locations[0]
self.__outcomeLocation = (os.path.normpath(fromLongPathSafe(loc[0])), loc[1])
if outcome.isFailure() and abortOnError:
if callRecord==None: callRecord = self.__callRecord()
self.abort(outcome, outcomeReason, callRecord)
if outcomeReason and printReason:
if outcome.isFailure():
log.warn(u'%s ... %s %s', outcomeReason, str(outcome).lower(), u'[%s]'%','.join(
u'%s:%s'%(os.path.basename(loc[0]), loc[1]) for loc in map(parseLocation,callRecord)) if callRecord!=None else u'',
extra=BaseLogFormatter.tag(str(outcome).lower(),1))
else:
log.info(u'%s ... %s', outcomeReason, str(outcome).lower(), extra=BaseLogFormatter.tag(str(outcome).lower(),1))
[docs] def abort(self, outcome, outcomeReason, callRecord=None):
"""Raise an AbortException with the specified outcome and reason.
See also L{skipTest}.
:param outcome: The outcome, which will override any existing outcomes previously recorded.
:param outcomeReason: A string summarizing the reason for the outcome.
"""
raise AbortExecution(outcome, outcomeReason, callRecord)
[docs] def skipTest(self, outcomeReason, callRecord=None):
"""Raise an AbortException that will set the test outcome to SKIPPED and
ensure that the rest of the execute() and validate() methods do not execute.
This is useful when a test should not be executed in the current mode or platform.
:param outcomeReason: A string summarizing the reason the test is being skipped, for example
"Feature X is not supported on Windows".
"""
raise AbortExecution(SKIPPED, outcomeReason, callRecord)
[docs] def getOutcome(self):
"""Get the final outcome for this test, based on the precedence order defined in `pysys.constants.OUTCOMES`.
To find out whether this test has failed::
if self.getOutcome().isFailure():
...
:return pysys.constants.Outcome: The overall outcome. Use ``%s`` or ``str()`` to convert to a display name.
"""
with self.lock:
if len(self.outcome) == 0: return NOTVERIFIED
return sorted(self.outcome, key=lambda x: OUTCOMES.index(x))[0]
[docs] def getOutcomeReason(self):
"""Get the reason string for the current overall outcome (if specified).
:return: The overall test outcome reason or '' if not specified
:rtype: string
"""
with self.lock:
fails = len([o for o in self.outcome if o.isFailure()])
if self.__outcomeReason and (fails > 1): return u'%s (+%d other failures)'%(self.__outcomeReason, fails-1)
return self.__outcomeReason
[docs] def getOutcomeLocation(self):
"""Get the location in the Python source file where this location was added.
.. versionadded:: 1.6.0
:return (str,str): The absolute filename, and the line number. Returns (None,None) if not known.
"""
with self.lock:
return self.__outcomeLocation
[docs] def getNextAvailableTCPPort(self, hosts=['', 'localhost'], socketAddressFamily=socket.AF_INET):
"""Allocate a free TCP port which can be used for starting a server on this machine.
The port is taken from the pool of available server (non-ephemeral) ports on this machine, and will not
be available for use by any other code in the current PySys process until this object's `cleanup` method is
called to return it to the pool of available ports.
To allocate an IPv4 port for use only on this host::
port = self.getNextAvailableTCPPort(hosts=['localhost'])
.. versionchanged:: 1.5.1
Added hosts and socketAddressFamily parameters.
:param list(Str) hosts: A list of the host names or IP addresses to check when establishing that a potential
allocated port isn't already in use by a process outside the PySys framework.
By default we check ``""`` (which corresponds to ``INADDR_ANY`` and depending on the OS means
either one or all non-localhost IPv4 addresses) and also ``localhost``.
Many machines have multiple network cards each with its own host IP address, and typically you'll only be using
one of them in your test, most commonly ``localhost``. If you do know which host/IP you'll actually be using,
just specify that directly to save time, and avoid needlessly opening remote ports on hosts you're not using.
A list of available host addresses can be found from
``socket.getaddrinfo('', None)``.
:param socketAddressFamily: The socket address family e.g. IPv4 vs IPv6. See Python's ``socket`` module for
details.
"""
o = TCPPortOwner()
self.addCleanupFunction(lambda: o.cleanup())
return o.port
def __callRecord(self):
"""Retrieve a call record outside of this module, up to the execute or validate method of the test case.
"""
stack=[]
from pysys.basetest import BaseTest
if isinstance(self, BaseTest):
for record in inspect.stack():
info = inspect.getframeinfo(record[0])
if (self.__skipFrame(info.filename, ProcessUser) ): continue
if (self.__skipFrame(info.filename, BaseTest) ): continue
stack.append( '%s:%s' % (info.filename.strip(), info.lineno) )
if (os.path.splitext(info.filename)[0] == os.path.splitext(os.path.join(self.descriptor.testDir, self.descriptor.module))[0] and (info.function == 'execute' or info.function == 'validate')): return stack
return None
def __skipFrame(self, file, clazz):
"""Private method to check if a file is that for a particular class.
:param file: The filepatch to check
:param clazz: The class to check against
"""
return os.path.splitext(file)[0] == os.path.splitext(sys.modules[clazz.__module__].__file__)[0]
[docs] def getExprFromFile(self, path, expr, groups=[1], returnAll=False, returnNoneIfMissing=False, encoding=None, reFlags=0, mappers=[]):
""" Searches for a regular expression in the specified file, and returns it.
If the regex contains unnamed groups using ``(expr)`` syntax, the specified group is returned.
If the expression is not found, an exception is raised,
unless returnAll=True or returnNoneIfMissing=True. For example::
myKey = self.getExprFromFile('test.txt', r'myKey="(.*)"') # on a file containing 'myKey="foobar"' would return "foobar"
err = self.getExprFromFile('test.txt', r'ERROR .*') # on a file containing 'ERROR It went wrong' would return "that entire string"
If you have a complex expression with multiple values to extract, it is usually clearer to
use ``(?P<groupName>...)`` named groups rather than unnamed groups referenced by index. This produces a
dictionary
authInfo = self.getExprFromFile('myserver.log', expr=r'Successfully authenticated user "(?P<username>[^"]*)" in (?P<authSecs>[^ ]+) seconds\.'))
allAuthList = self.getExprFromFile('myserver.log', expr=r'Successfully authenticated user "(?P<username>[^"]*)" in (?P<authSecs>[^ ]+) seconds\.', returnAll=True))
See also `pysys.basetest.BaseTest.assertThatGrep` which should be used when instead of just finding out what's
in the file you want to assert that a specific expression is matched. The documentation for assertGrep also
provides some helpful examples of regular expressions that could also be applied to this method, and tips for
dealing with escaping in regular expressions.
.. versionchanged:: 1.6.0
Support for named groups was added in 1.6.0.
:param str path: file to search (located in the output dir unless an absolute path is specified)
:param str expr: the regular expression, optionally containing the regex group operator ``(...)``
Remember to escape regular expression special characters such as ``.``, ``(``, ``[``, ``{`` and ``\\`` if you want them to
be treated as literal values. If you have a string with regex backslashes, it's best to use a 'raw'
Python string so that you don't need to double-escape them, e.g. ``expr=r'function[(]"str", 123[.]4, (\d+), .*[)]'``.
:param List[int] groups: which numeric regex group numbers (as indicated by brackets in the regex) should be returned;
default is ``[1]`` meaning the first group.
If more than one group is specified, the result will be a tuple of group values, otherwise the
result will be the value of the group at the specified index as a str.
This parameter is ignored if the regular expression contains any ``(?P<groupName>...)`` named groups.
:param bool returnAll: returns a list containing all matching lines if True, the first matching line otherwise.
:param bool returnNoneIfMissing: True to return None instead of raising an exception
if the regex is not found in the file (not needed when returnAll is used).
:param str encoding: The encoding to use to open the file.
The default value is None which indicates that the decision will be delegated
to the L{getDefaultFileEncoding()} method.
:param List[callable[str]->str] mappers: A list of filter functions that will be used to pre-process each
line from the file (returning None if the line is to be filtered out). This provides a very powerful
capability for filtering the file, for example `pysys.mappers.IncludeLinesBetween`
provides the ability to filter in/out sections of a file.
Do not share mapper instances across multiple tests or threads as this can cause race conditions.
Added in PySys 1.6.0.
:param int reFlags: Zero or more flags controlling how the behaviour of regular expression matching,
combined together using the ``|`` operator, for example ``reFlags=re.VERBOSE | re.IGNORECASE``.
For details see the ``re`` module in the Python standard library. Note that ``re.MULTILINE`` cannot
be used because expressions are matched against one line at a time. Added in PySys 1.5.1.
:return: For a regular expression with one unnamed group, the match value is a str;
if there are multiple unnamed numeric groups it is List[str] (with values corresponding to the groups= argument);
if it contains any ``(?P<groupName>...)`` named groups a dict[str,str] is returned where the keys are the groupNames.
If returnAll=True, the return value is a list of all the match values, with types as above.
"""
namedGroupsMode = False
compiled = re.compile(expr, flags=reFlags)
namedGroupsMode = compiled.groupindex
path = os.path.join(self.output, path)
if None in mappers: mappers = [m for m in mappers if m]
with openfile(path, 'r', encoding=encoding or self.getDefaultFileEncoding(path)) as f:
matches = []
for l in f:
for mapper in mappers:
l = mapper(l)
if l is None: break
if l is None: continue
match = compiled.search(l)
if not match: continue
if namedGroupsMode:
val = match.groupdict()
elif match.groups():
val = match.group(*groups)
else:
val = match.group(0)
if returnAll:
matches.append(val)
else:
return val
if returnAll: return matches
if returnNoneIfMissing: return None
if os.path.getsize(path) == 0: # can happen due to race conditions in file system writing; maybe they need a waitForGrep
raise Exception('Could not find expression %s in %s because file is empty'%(quotestring(expr), os.path.basename(path)))
raise Exception('Could not find expression %s in %s'%(quotestring(expr), os.path.basename(path)))
[docs] def logFileContents(self, path, includes=None, excludes=None, maxLines=20, tail=False, encoding=None,
logFunction=None, reFlags=0, stripWhitespace=True):
""" Logs some or all of the lines from the specified file.
If the file does not exist or cannot be opened, does nothing. The method is useful for providing key
diagnostic information (e.g. error messages from tools executed by the test) directly in run.log, or
to make test failures easier to triage quickly.
:param str path: May be an absolute, or relative to the test output directory.
:param list[str] includes: Optional list of regex strings. If specified, only matches of these regexes will be logged.
:param list[str] excludes: Optional list of regex strings. If specified, no line containing these will be logged.
The variable ``self.logFileContentsDefaultExcludes`` (= ``[]`` by default) is used when this method
is called with the default argument of ``excludes=None``, and can be used to provide a global set of
default exclusion lines shared by all your tests, which is particularly useful if some processes always
log some unimportant text to stderr (or stdout) that would be distracting to log out.
Added in PySys 1.6.0.
:param int maxLines: Upper limit on the number of lines from the file that will be logged. Set to zero for unlimited
:param bool tail: Prints the _last_ 'maxLines' in the file rather than the first 'maxLines'.
:param str encoding: The encoding to use to open the file.
The default value is None which indicates that the decision will be delegated
to the L{getDefaultFileEncoding()} method.
:param Callable[[line],None] logFunction: The function that will be used to log individual lines from the file.
Usually this is ``self.log.info(u' %s', line, extra=BaseLogFormatter.tag(LOG_FILE_CONTENTS))``
but a custom implementation can be provided, for example to provide a different color using
`pysys.utils.logutils.BaseLogFormatter.tag`.
Added in PySys 1.5.1.
:param int reFlags: Zero or more flags controlling how the behaviour of regular expression matching,
combined together using the ``|`` operator, for example ``reFlags=re.VERBOSE | re.IGNORECASE``.
For details see the ``re`` module in the Python standard library. Note that ``re.MULTILINE`` cannot
be used because expressions are matched against one line at a time. Added in PySys 1.5.1.
:return: True if anything was logged, False if not.
"""
if excludes is None: excludes = self.logFileContentsDefaultExcludes
if not path: return False
actualpath= os.path.join(self.output, path)
try:
# always open with a specific encoding not in bytes mode, since otherwise we can't reliably pass the read lines to the logger
f = openfile(actualpath, 'r', encoding=encoding or self.getDefaultFileEncoding(actualpath) or locale.getpreferredencoding(), errors='replace')
except Exception as e:
self.log.debug('logFileContents cannot open file "%s": %s', actualpath, e)
return False
try:
lineno = 0
def matchesany(s, regexes):
assert not isstring(regexes), 'must be a list of strings not a string'
for x in regexes:
m = re.search(x, s, flags=reFlags)
if m: return m.group(0)
return None
tolog = []
for l in f:
l = l.rstrip()
if not l: continue
if includes:
l = matchesany(l, includes)
if not l: continue
if excludes and matchesany(l, excludes): continue
lineno +=1
tolog.append(l)
if maxLines:
if not tail and len(tolog) == maxLines:
tolog.append('...')
break
if tail and len(tolog)==maxLines+1:
del tolog[0]
finally:
f.close()
if not tolog:
return False
logextra = BaseLogFormatter.tag(LOG_FILE_CONTENTS)
if logFunction is None:
def logFunction(line):
self.log.info(u' %s', l, extra=logextra)
path = os.path.normpath(path)
if path.startswith(self.output): path = path[len(self.output)+1:]
self.log.info(u'Contents of %s%s: ', path, ' (filtered)' if includes or excludes else '', extra=logextra)
for l in tolog:
logFunction(l)
self.log.info(' -----', extra=logextra)
self.log.info('', extra=logextra)
return True
[docs] def mkdir(self, path):
"""
Create a directory, with recursive creation of any parent directories.
This function does nothing (does not raise an except) if the directory already exists.
:param path: The path to be created. This can be an absolute path or
relative to the testcase output directory.
:return: the absolute path of the new directory, to facilitate fluent-style method calling.
"""
path = os.path.join(self.output, path)
mkdir(path)
return path
def deletedir(self, path, **kwargs): return self.deleteDir(path, **kwargs)
[docs] def deleteDir(self, path, **kwargs):
"""
Recursively delete the specified directory.
Does nothing if it does not exist. Raises an exception if the deletion fails.
:param path: The path to be deleted. This can be an absolute path or
relative to the testcase output directory.
:param kwargs: Any additional arguments are passed to
L{pysys.utils.fileutils.deletedir()}.
"""
deletedir(os.path.join(self.output, path), **kwargs)
[docs] def getDefaultFileEncoding(self, file, **xargs):
"""
Specifies what encoding should be used to read or write the specified
text file.
This method is used to select the appropriate encoding whenever PySys
needs to open a file, for example to wait for a signal, for a
file-based assertion, or to write a file with replacements.
Many methods allow the encoding to be overridden for just that call,
but getDefaultFileEncoding exists to allow global defaults to be specified
based on the filename.
For example, this method could be overridden to specify that utf-8 encoding
is to be used for opening filenames ending in .xml, .json and .yaml.
The default implementation of this method uses pysysproject.xml
configuration rules such as::
<default-file-encoding pattern="*.xml" encoding="utf-8"/>
A return value of None indicates default behaviour, which on Python 3 is to
use the default encoding, as specified by python's
``locale.getpreferredencoding()``, and on Python 2 is to use binary ``str``
objects with no character encoding or decoding applied.
:param file: The filename to be read or written. This may be an
absolute path or a relative path.
:param xargs: Ensure that an ``**xargs`` argument is specified so that
additional information can be passed to this method in future releases.
:return: The encoding to use for this file, or None if default behaviour is
to be used. For example, ``utf-8`` or (for UTF-8 with a Byte Order Mark), ``utf-8-sig``.
"""
file = file.replace('\\','/').lower() # normalize slashes and ignore case
for e in self.project.defaultFileEncodings:
# first match wins
if fnmatch.fnmatchcase(file, e['pattern'].lower()) or fnmatch.fnmatchcase(os.path.basename(file), e['pattern'].lower()):
return e['encoding']
return None
[docs] @staticmethod
def compareVersions(v1, v2):
""" Compares two alphanumeric dotted version strings to see which is more recent.
Example usage::
if self.compareVersions(thisversion, '1.2.alpha-3') > 0:
... # thisversion is newer than 1.2.alpha-3
The comparison algorithm ignores case, and normalizes separators ./-/_
so that ``'1.alpha2'=='1Alpha2'``. Any string components are compared
lexicographically with other strings, and compared to numbers
strings are always considered greater.
>>> ProcessUser.compareVersions('10-alpha5.dev10', '10alpha-5-dEv_10') == 0 # normalization of case and separators
True
>>> ProcessUser.compareVersions(b'1....alpha.2', u'1Alpha2') == 0 # ascii byte and unicode strings both supported
True
>>> ProcessUser.compareVersions('1.2.0', '1.2')
0
>>> ProcessUser.compareVersions('1.02', '1.2')
0
>>> ProcessUser().compareVersions('1.2.3', '1.2') > 0
True
>>> ProcessUser.compareVersions('1.2', '1.2.3')
-1
>>> ProcessUser.compareVersions('10.2', '1.2')
1
>>> ProcessUser.compareVersions('1.2.text', '1.2.0') # letters are > numbers
1
>>> ProcessUser.compareVersions('1.2.text', '1.2') # letters are > numbers
1
>>> ProcessUser.compareVersions('10.2alpha1', '10.2alpha')
1
>>> ProcessUser.compareVersions('10.2dev', '10.2alpha') # letters are compared lexicographically
1
>>> ProcessUser.compareVersions('', '')
0
>>> ProcessUser.compareVersions('1', '')
1
:param v1: A string containing a version number, with any number of components.
:param v2: A string containing a version number, with any number of components.
:return: an integer > 0 if v1>v2,
an integer < 0 if v1<v2,
or 0 if they are semantically the same.
"""
return pysys.utils.misc.compareVersions(v1, v2)
[docs] def write_text(self, file, text, encoding=None):
"""
Writes the specified characters to a file in the output directory.
:param file: The path of the file to write, either an absolute path or
relative to the `self.output` directory.
:param text: The string to write to the file, with `\\n`
for newlines (do not use `os.linesep` as the file will be opened in
text mode so platform line separators will be added automatically).
On Python 3 this must be a character string.
On Python 2 this can be a character or byte string containing ASCII
characters. If non-ASCII characters are used, it must be a unicode
string if there is an encoding specified for this file/type, or
else a byte string.
:param encoding: The encoding to use to open the file.
The default value is None which indicates that the decision will be delegated
to the L{getDefaultFileEncoding()} method.
:return str: Return the absolute path of the generated file.
"""
# This method provides similar functionality to the Python3 pathlib write_text method.
out = os.path.join(self.output, file)
with openfile(os.path.join(self.output, file), 'w', encoding=encoding or self.getDefaultFileEncoding(file)) as f:
f.write(text)
return out
[docs] @contextlib.contextmanager
def disableLogging(self):
"""
Temporarily stops logging for the current thread.
For example::
with self.disableLogging():
self.startProcess(...)
Note that this method will do nothing if the log level if pysys is run with ``-vDEBUG`` or ``-vdisabledLogging=DEBUG``.
.. versionadded:: 1.6.0
"""
logger = logging.getLogger('pysys.disabledLogging')
saved = pysys.internal.initlogging.pysysLogHandler.getLogHandlersForCurrentThread()
if logger.isEnabledFor(logging.DEBUG):
logger.debug('Ignoring disableLogging() request as debug logging is enabled')
else:
pysys.internal.initlogging.pysysLogHandler.setLogHandlersForCurrentThread([])
try:
yield None
finally:
pysys.internal.initlogging.pysysLogHandler.setLogHandlersForCurrentThread(saved)
[docs] def copy(self, src, dest, mappers=[], encoding=None, symlinks=False, ignoreIf=None, skipMappersIf=None, overwrite=None):
"""Copy a directory or a single text or binary file, optionally tranforming the contents by filtering each line through a list of mapping functions.
If any mappers are provided, the file is copied in text mode and
each mapper is given the chance to modify or omit each line.
If no mappers are provided, the file is copied in binary mode.
For example::
self.copy('output-raw.txt', 'output-processed.txt', encoding='utf-8',
mappers=[
lambda line: None if ('Timestamp: ' in line) else line,
lambda line: line.replace('foo', 'bar'),
pysys.mappers.IncludeLinesBetween('Error message .*:', stopBefore='^$'),
pysys.mappers.RegexReplace(pysys.mappers.RegexReplace.DATETIME_REGEX, '<timestamp>'),
])
In addition to the file contents the attributes such as modification time and
executable permission will be copied where possible.
This function is useful for creating a modified version of an
output file that's more suitable for later validation steps such as
diff-ing, and also for copying required files from the input to the
output directory.
It can also be used for copying a whole directory,
similar to ``shutil.copytree`` but with the advantages of support
for long paths on Windows, better error safety, and that relative paths
are evaluated relative to the self.output directory (which is both convenient, and
safer than shutil's evaluation relative to the current working
directory).
For example::
self.copy('src.txt', 'dest.txt') # copies to outputdir/dest.txt
self.copy('src.txt', self.output) # copies to outputdir/src.txt, since self.output is an existing directory
self.copy('src.txt', 'foo/') # copies to outputdir/foo/src.txt since destination ends with a slash
self.copy('srcdirname', 'foo/') # copies to outputdir/foo/srcdirname since destination ends with a slash
Usually custom mappers are simple functions or lambdas, however for advanced use cases you can
additionally provide ``mapper.fileStarted([self,] srcPath, destPath, srcFile, destFile)`` and/or
``mapper.fileFinished(...)`` methods to allow stateful operations, or to perform extra read/write operations
before lines are read/written. For example::
class CustomLineMapper(object):
def fileStarted(self, srcPath, destPath, srcFile, destFile):
self.src = os.path.basename(srcPath)
def __call__(self, line):
return '"'+self.src+'": '+line
def fileFinished(self, srcPath, destPath, srcFile, destFile):
destFile.write('\\n' + 'footer added by CustomLineMapper')
self.copy('src.txt', 'dest.txt', mappers=[CustomLineMapper()])
.. versionchanged:: 1.6.0
Ability to copy directories was added, along with the ``overwrite=``, ``symlinks=``,
``ignoreIf=`` and ``skipMappersIf=`` arguments.
:param str src: The source filename or directory, which can be an absolute path, or
a path relative to the ``self.output`` directory.
Use ``src=self.input+'/myfile'`` if you wish to copy a file from the test
input directory.
:param str dest: The destination file or directory name, which can be an absolute path, or
a path relative to the `self.output` directory. Destination file(s) are overwritten if the
dest already exists.
The dest and src can be the same for file copies (but not directory copies).
As a convenience to avoid repeating the same text in the src and destination,
if the dest ends with a slash, or the src is a file and the dest is an existing directory,
the dest is taken as a parent directory into which the src will be copied in retaining its current name.
:param bool overwrite: If True, source files will be allowed to
overwrite destination files, if False an exception will be raised if a destination file already exists.
By default overwrite=None which means it's enabled for single file copy() but disabled for directory copies.
:param List[callable[str]->str] mappers: A list of filter functions that will be applied,
in order, to map each line from source to destination. Each function accepts a string for
the current line as input and returns either a string to write or
None if the line is to be omitted. Any ``None`` items in the mappers list will be ignored.
See `pysys.mappers` for some useful predefined mappers such as `pysys.mappers.IncludeLinesBetween`
and `pysys.mappers.RegexReplace`.
If present the ``mapper.fileStarted(...)`` and/or ``mapper.fileFinished(...)`` methods will be called on each
mapper in the list at the start and end of each file; see above for an example.
Do not share mapper instances across multiple tests or threads as this can cause race conditions.
:param str encoding: The encoding to use to open the file (only used if mappers are provided; if not, it is
opened in binary mode).
The default value is None which indicates that the decision will be delegated
to the L{getDefaultFileEncoding()} method.
:param bool symlinks: Set to True if symbolic links in the source tree should be represented as symbolic
links in the destination (rather than just being copied).
:param callable[str]->bool ignoreIf: A callable that accepts a source path and returns True if
this file/directory should be omitted from a directory copy.
For example: ``ignoreIf=lambda src: src.endswith(('.tmp', '.log')))``
:param callable[str]->bool skipMappersIf: A callable that accepts a source path and returns True if
this file should be copied in binary mode (as if no mappers had been specified).
For example: ``skipMappersIf=lambda src: not src.endswith(('.xml', '.properties')))``
:return str: the absolute path of the destination file.
"""
origdest = dest
src = toLongPathSafe(os.path.join(self.output, src))
srcIsDir = os.path.isdir(src)
dest = toLongPathSafe(os.path.join(self.output, dest)).rstrip('/\\')
if origdest.endswith((os.sep, '/', '\\')) or (not srcIsDir and os.path.isdir(dest)): dest = toLongPathSafe(dest+os.sep+os.path.basename(src))
if src == dest and not srcIsDir:
dest = src+'__pysys_copy.tmp'
renameDestAtEnd = True
else:
renameDestAtEnd = False
assert src != dest, 'Source and destination directory cannot be the same'
if overwrite is None: overwrite = not srcIsDir
if srcIsDir:
destexists = os.path.isdir(dest)
if not destexists: self.mkdir(dest)
scandir = getattr(os, 'scandir', None)
for e in (os.listdir(src) if scandir is None else scandir(src)):
if scandir is None: # for python2 support
path = os.path.join(src, e)
isdir = os.path.isdir(path)
islink = os.path.islink(path)
else:
isdir = e.is_dir()
islink = e.is_symlink()
path = e.path
if ignoreIf is not None and ignoreIf(path): continue
if islink and symlinks:
linkdest = dest+os.sep+os.path.basename(path)
os.symlink(os.readlink(path), linkdest)
shutil.copystat(path, linkdest, **({} if PY2 else {'follow_symlinks':False}))
continue
self.copy(path, dest+os.sep+os.path.basename(path),
# must pass all other arguments through
mappers=mappers, encoding=encoding, symlinks=symlinks, ignoreIf=ignoreIf, skipMappersIf=skipMappersIf, overwrite=overwrite)
return dest
if skipMappersIf is not None and skipMappersIf(src): mappers = None
if not overwrite and os.path.exists(dest):
raise Exception('copy() will not overwrite an existing file unless overwrite=True: %s'%dest)
if not mappers:
# simple binary copy
shutil.copyfile(src, dest)
else:
if None in mappers: mappers = [m for m in mappers if m]
with openfile(src, 'r', encoding=encoding or self.getDefaultFileEncoding(src)) as srcf:
with openfile(dest, 'w', encoding=encoding or self.getDefaultFileEncoding(dest)) as destf:
# give mappers a change to setup initial state and/or read/write from the source and dest files
for mapper in mappers:
fn = getattr(mapper, 'fileStarted', None)
if fn: fn(src, dest, srcf, destf)
for line in srcf:
for mapper in mappers:
line = mapper(line)
if line is None: break
if line is not None: destf.write(line)
for mapper in mappers:
fn = getattr(mapper, 'fileFinished', None)
if fn: fn(src, dest, srcf, destf)
shutil.copystat(src, dest)
if renameDestAtEnd:
os.remove(src)
os.rename(dest, src)
return dest