# -*- coding: iso-8859-1 -*-
# vim: set ft=python ts=3 sw=3 expandtab:
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
#
# C E D A R
# S O L U T I O N S "Software done right."
# S O F T W A R E
#
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
#
# Copyright (c) 2004-2006,2008,2010,2015 Kenneth J. Pronovici.
# All rights reserved.
#
# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License,
# Version 2, as published by the Free Software Foundation.
#
# This program 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.
#
# Copies of the GNU General Public License are available from
# the Free Software Foundation website, http://www.gnu.org/.
#
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
#
# Author : Kenneth J. Pronovici <pronovic@ieee.org>
# Language : Python 3 (>= 3.4)
# Project : Cedar Backup, release 3
# Purpose : Provides unit-testing utilities.
#
# # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # # #
########################################################################
# Module documentation
########################################################################
"""
Provides unit-testing utilities.
These utilities are kept here, separate from util.py, because they provide
common functionality that I do not want exported "publicly" once Cedar Backup
is installed on a system. They are only used for unit testing, and are only
useful within the source tree.
Many of these functions are in here because they are "good enough" for unit
test work but are not robust enough to be real public functions. Others (like
:any:`removedir`) do what they are supposed to, but I don't want responsibility for
making them available to others.
:author: Kenneth J. Pronovici <pronovic@ieee.org>
"""
########################################################################
# Imported modules
########################################################################
import sys
import os
import tarfile
import time
import getpass
import random
import string # pylint: disable=W0402
import platform
import logging
from io import StringIO
from CedarBackup3.util import encodePath, executeCommand
from CedarBackup3.config import Config, OptionsConfig
from CedarBackup3.customize import customizeOverrides
from CedarBackup3.cli import setupPathResolver
########################################################################
# Public functions
########################################################################
##############################
# setupDebugLogger() function
##############################
[docs]def setupDebugLogger():
"""
Sets up a screen logger for debugging purposes.
Normally, the CLI functionality configures the logger so that
things get written to the right place. However, for debugging
it's sometimes nice to just get everything -- debug information
and output -- dumped to the screen. This function takes care
of that.
"""
logger = logging.getLogger("CedarBackup3")
logger.setLevel(logging.DEBUG) # let the logger see all messages
formatter = logging.Formatter(fmt="%(message)s")
handler = logging.StreamHandler(stream=sys.stdout)
handler.setFormatter(formatter)
handler.setLevel(logging.DEBUG)
logger.addHandler(handler)
#################
# setupOverrides
#################
[docs]def setupOverrides():
"""
Set up any platform-specific overrides that might be required.
When packages are built, this is done manually (hardcoded) in customize.py
and the overrides are set up in cli.cli(). This way, no runtime checks need
to be done. This is safe, because the package maintainer knows exactly
which platform (Debian or not) the package is being built for.
Unit tests are different, because they might be run anywhere. So, we
attempt to make a guess about plaform using platformDebian(), and use that
to set up the custom overrides so that platform-specific unit tests continue
to work.
"""
config = Config()
config.options = OptionsConfig()
if platformDebian():
customizeOverrides(config, platform="debian")
else:
customizeOverrides(config, platform="standard")
setupPathResolver(config)
###########################
# findResources() function
###########################
[docs]def findResources(resources, dataDirs):
"""
Returns a dictionary of locations for various resources.
Args:
resources: List of required resources
dataDirs: List of data directories to search within for resources
Returns:
Dictionary mapping resource name to resource path
Raises:
Exception: If some resource cannot be found
"""
mapping = { }
for resource in resources:
for resourceDir in dataDirs:
path = os.path.join(resourceDir, resource)
if os.path.exists(path):
mapping[resource] = path
break
else:
raise Exception("Unable to find resource [%s]." % resource)
return mapping
##############################
# commandAvailable() function
##############################
[docs]def commandAvailable(command):
"""
Indicates whether a command is available on $PATH somewhere.
This should work on both Windows and UNIX platforms.
Args:
command: Commang to search for
Returns:
Boolean true/false depending on whether command is available
"""
if "PATH" in os.environ:
for path in os.environ["PATH"].split(os.sep):
if os.path.exists(os.path.join(path, command)):
return True
return False
#######################
# buildPath() function
#######################
[docs]def buildPath(components):
"""
Builds a complete path from a list of components.
For instance, constructs ``"/a/b/c"`` from ``["/a", "b", "c",]``.
Args:
components: List of components
Returns:
String path constructed from components
Raises:
ValueError: If a path cannot be encoded properly
"""
path = components[0]
for component in components[1:]:
path = os.path.join(path, component)
return encodePath(path)
#######################
# removedir() function
#######################
[docs]def removedir(tree):
"""
Recursively removes an entire directory.
This is basically taken from an example on python.com.
Args:
tree: Directory tree to remove
Raises:
ValueError: If a path cannot be encoded properly
"""
tree = encodePath(tree)
for root, dirs, files in os.walk(tree, topdown=False):
for name in files:
path = os.path.join(root, name)
if os.path.islink(path):
os.remove(path)
elif os.path.isfile(path):
os.remove(path)
for name in dirs:
path = os.path.join(root, name)
if os.path.islink(path):
os.remove(path)
elif os.path.isdir(path):
os.rmdir(path)
os.rmdir(tree)
########################
# extractTar() function
########################
###########################
# changeFileAge() function
###########################
[docs]def changeFileAge(filename, subtract=None):
"""
Changes a file age using the ``os.utime`` function.
*Note:* Some platforms don't seem to be able to set an age precisely. As a
result, whereas we might have intended to set an age of 86400 seconds, we
actually get an age of 86399.375 seconds. When util.calculateFileAge()
looks at that the file, it calculates an age of 0.999992766204 days, which
then gets truncated down to zero whole days. The tests get very confused.
To work around this, I always subtract off one additional second as a fudge
factor. That way, the file age will be *at least* as old as requested
later on.
Args:
filename: File to operate on
subtract: Number of seconds to subtract from the current time
Raises:
ValueError: If a path cannot be encoded properly
"""
filename = encodePath(filename)
newTime = time.time() - 1
if subtract is not None:
newTime -= subtract
os.utime(filename, (newTime, newTime))
###########################
# getMaskAsMode() function
###########################
[docs]def getMaskAsMode():
"""
Returns the user's current umask inverted to a mode.
A mode is mostly a bitwise inversion of a mask, i.e. mask 002 is mode 775.
Returns:
Umask converted to a mode, as an integer
"""
umask = os.umask(0o777)
os.umask(umask)
return int(~umask & 0o777) # invert, then use only lower bytes
######################
# getLogin() function
######################
[docs]def getLogin():
"""
Returns the name of the currently-logged in user. This might fail under
some circumstances - but if it does, our tests would fail anyway.
"""
return getpass.getuser()
############################
# randomFilename() function
############################
[docs]def randomFilename(length, prefix=None, suffix=None):
"""
Generates a random filename with the given length.
Args:
length: Length of filename
@return Random filename
"""
characters = [None] * length
for i in range(length):
characters[i] = random.choice(string.ascii_uppercase)
if prefix is None:
prefix = ""
if suffix is None:
suffix = ""
return "%s%s%s" % (prefix, "".join(characters), suffix)
####################################
# failUnlessAssignRaises() function
####################################
# pylint: disable=W0613
[docs]def failUnlessAssignRaises(testCase, exception, obj, prop, value):
"""
Equivalent of ``failUnlessRaises``, but used for property assignments instead.
It's nice to be able to use ``failUnlessRaises`` to check that a method call
raises the exception that you expect. Unfortunately, this method can't be
used to check Python propery assignments, even though these property
assignments are actually implemented underneath as methods.
This function (which can be easily called by unit test classes) provides an
easy way to wrap the assignment checks. It's not pretty, or as intuitive as
the original check it's modeled on, but it does work.
Let's assume you make this method call::
testCase.failUnlessAssignRaises(ValueError, collectDir, "absolutePath", absolutePath)
If you do this, a test case failure will be raised unless the assignment::
collectDir.absolutePath = absolutePath
fails with a ``ValueError`` exception. The failure message differentiates
between the case where no exception was raised and the case where the wrong
exception was raised.
*Note:* Internally, the ``missed`` and ``instead`` variables are used rather
than directly calling ``testCase.fail`` upon noticing a problem because the
act of "failure" itself generates an exception that would be caught by the
general ``except`` clause.
Args:
testCase: PyUnit test case object (i.e. self)
exception: Exception that is expected to be raised
obj: Object whose property is to be assigned to
prop: Name of the property, as a string
value: Value that is to be assigned to the property
@see: ``unittest.TestCase.failUnlessRaises``
"""
missed = False
instead = None
try:
exec("obj.%s = value" % prop) # pylint: disable=W0122
missed = True
except exception: pass
except Exception as e:
instead = e
if missed:
testCase.fail("Expected assignment to raise %s, but got no exception." % (exception.__name__))
if instead is not None:
testCase.fail("Expected assignment to raise %s, but got %s instead." % (ValueError, instead.__class__.__name__))
###########################
# captureOutput() function
###########################
[docs]def captureOutput(c):
"""
Captures the output (stdout, stderr) of a function or a method.
Some of our functions don't do anything other than just print output. We
need a way to test these functions (at least nominally) but we don't want
any of the output spoiling the test suite output.
This function just creates a dummy file descriptor that can be used as a
target by the callable function, rather than ``stdout`` or ``stderr``.
*Note:* This method assumes that ``callable`` doesn't take any arguments
besides keyword argument ``fd`` to specify the file descriptor.
Args:
c: Callable function or method
Returns:
Output of function, as one big string
"""
fd = StringIO()
c(fd=fd)
result = fd.getvalue()
fd.close()
return result
#########################
# _isPlatform() function
#########################
def _isPlatform(name):
"""
Returns boolean indicating whether we're running on the indicated platform.
Args:
name: Platform name to check, currently one of "windows" or "macosx"
"""
if name == "windows":
return platform.platform(True, True).startswith("Windows")
elif name == "macosx":
return sys.platform == "darwin"
elif name == "debian":
return platform.platform(False, False).find("debian") > 0
elif name == "cygwin":
return platform.platform(True, True).startswith("CYGWIN")
else:
raise ValueError("Unknown platform [%s]." % name)
############################
# platformDebian() function
############################
############################
# platformMacOsX() function
############################
###########################
# runningAsRoot() function
###########################
[docs]def runningAsRoot():
"""
Returns boolean indicating whether the effective user id is root.
"""
return os.geteuid() == 0
##############################
# availableLocales() function
##############################
[docs]def availableLocales():
"""
Returns a list of available locales on the system
Returns:
List of string locale names
"""
locales = []
output = executeCommand(["locale"], [ "-a", ], returnOutput=True, ignoreStderr=True)[1]
for line in output:
locales.append(line.rstrip()) # pylint: disable=E1101
return locales