From 89e7afbf74ffbbb54dd892bf2c4245aedee2a832 Mon Sep 17 00:00:00 2001 From: "Chris St. Pierre" Date: Tue, 9 Dec 2014 11:10:24 -0600 Subject: 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. --- testsuite/common.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) (limited to 'testsuite/common.py') 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: -- cgit v1.2.3-1-g7c22