diff options
author | Chris St. Pierre <chris.a.st.pierre@gmail.com> | 2015-02-18 08:47:30 -0600 |
---|---|---|
committer | Chris St. Pierre <chris.a.st.pierre@gmail.com> | 2015-02-18 08:47:30 -0600 |
commit | db4a3b0426f4912639c142366ff27594d54ce360 (patch) | |
tree | fe258d4b663b9607230e456cfa25cdfe29cb3a70 /testsuite/common.py | |
parent | ed9920711a6020fa01d564db34e0ee4800718cc6 (diff) | |
parent | 89e7afbf74ffbbb54dd892bf2c4245aedee2a832 (diff) | |
download | bcfg2-db4a3b0426f4912639c142366ff27594d54ce360.tar.gz bcfg2-db4a3b0426f4912639c142366ff27594d54ce360.tar.bz2 bcfg2-db4a3b0426f4912639c142366ff27594d54ce360.zip |
Merge pull request #250 from stpierre/blanket-except-plugins-lint
Remove blanket excepts from plugins and lint
Diffstat (limited to 'testsuite/common.py')
-rw-r--r-- | testsuite/common.py | 27 |
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: |