summaryrefslogtreecommitdiffstats
path: root/testsuite/common.py
diff options
context:
space:
mode:
authorChris St. Pierre <chris.a.st.pierre@gmail.com>2014-12-09 11:10:24 -0600
committerChris St. Pierre <chris.a.st.pierre@gmail.com>2015-02-18 08:24:24 -0600
commit89e7afbf74ffbbb54dd892bf2c4245aedee2a832 (patch)
treeed4623b5c6e39d80c132e9f21f3d4804cabd93be /testsuite/common.py
parent64b458b380620f84843b1841b441745a0984946f (diff)
downloadbcfg2-89e7afbf74ffbbb54dd892bf2c4245aedee2a832.tar.gz
bcfg2-89e7afbf74ffbbb54dd892bf2c4245aedee2a832.tar.bz2
bcfg2-89e7afbf74ffbbb54dd892bf2c4245aedee2a832.zip
Remove blanket excepts from plugins and lint
This removes most blanket except: clauses from all plugins, including the base plugin libraries, and bcfg2-lint. The few that remain should all be necessary. Most of the changes were quite minor, but this did require some restructuring of the CfgPrivateKeyCreator; as a result, the tests for that module were rewritten.
Diffstat (limited to 'testsuite/common.py')
-rw-r--r--testsuite/common.py27
1 files changed, 27 insertions, 0 deletions
diff --git a/testsuite/common.py b/testsuite/common.py
index fc2397560..4c7337e0d 100644
--- a/testsuite/common.py
+++ b/testsuite/common.py
@@ -14,6 +14,7 @@ import sys
import codecs
import lxml.etree
import Bcfg2.Options
+import Bcfg2.Utils
from mock import patch, MagicMock, _patch, DEFAULT
try:
from unittest2 import skip, skipIf, skipUnless, TestCase
@@ -119,6 +120,29 @@ else:
return codecs.unicode_escape_decode(s)[0]
+class MockExecutor(object):
+ """mock object for :class:`Bcfg2.Utils.Executor` objects."""
+ def __init__(self, timeout=None):
+ self.timeout = timeout
+
+ # variables that can be set to control the result returned
+ self.stdout = ''
+ self.stderr = ''
+ self.retval = 0
+
+ # variables that record how run() was called
+ self.calls = []
+
+ def run(self, command, inputdata=None, timeout=None, **kwargs):
+ self.calls.append({"command": command,
+ "inputdata": inputdata,
+ "timeout": timeout or self.timeout,
+ "kwargs": kwargs})
+
+ return Bcfg2.Utils.ExecutorResult(self.stdout, self.stderr,
+ self.retval)
+
+
class Bcfg2TestCase(TestCase):
""" Base TestCase class that inherits from
:class:`unittest.TestCase`. This class adds
@@ -138,6 +162,9 @@ class Bcfg2TestCase(TestCase):
if cls.capture_stderr:
sys.stderr = cls._stderr
+ if hasattr(TestCase, "assertCountEqual"):
+ assertItemsEqual = assertCountEqual
+
def assertXMLEqual(self, el1, el2, msg=None):
""" Test that the two XML trees given are equal. """
if msg is None: