diff options
Diffstat (limited to 'lib/python2.7/site-packages/buildbot_slave-0.8.8-py2.7.egg/buildslave/test/util/misc.py')
-rw-r--r-- | lib/python2.7/site-packages/buildbot_slave-0.8.8-py2.7.egg/buildslave/test/util/misc.py | 127 |
1 files changed, 0 insertions, 127 deletions
diff --git a/lib/python2.7/site-packages/buildbot_slave-0.8.8-py2.7.egg/buildslave/test/util/misc.py b/lib/python2.7/site-packages/buildbot_slave-0.8.8-py2.7.egg/buildslave/test/util/misc.py deleted file mode 100644 index f1521769..00000000 --- a/lib/python2.7/site-packages/buildbot_slave-0.8.8-py2.7.egg/buildslave/test/util/misc.py +++ /dev/null @@ -1,127 +0,0 @@ -# This file is part of Buildbot. Buildbot is free software: you can -# redistribute it and/or modify it under the terms of the GNU General Public -# License as published by the Free Software Foundation, version 2. -# -# 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. See the GNU General Public License for more -# details. -# -# You should have received a copy of the GNU General Public License along with -# this program; if not, write to the Free Software Foundation, Inc., 51 -# Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. -# -# Copyright Buildbot Team Members - -import os -import sys -import mock -import shutil -import __builtin__ -import cStringIO - -def nl(s): - """Convert the given string to the native newline format, assuming it is - already in normal UNIX newline format (\n). Use this to create the - appropriate expectation in a failUnlessEqual""" - if not isinstance(s, basestring): - return s - return s.replace('\n', os.linesep) - -class BasedirMixin(object): - """Mix this in and call setUpBasedir and tearDownBasedir to set up - a clean basedir with a name given in self.basedir.""" - - def setUpBasedir(self): - self.basedir = "test-basedir" - if os.path.exists(self.basedir): - shutil.rmtree(self.basedir) - - def tearDownBasedir(self): - if os.path.exists(self.basedir): - shutil.rmtree(self.basedir) - -class PatcherMixin(object): - """ - Mix this in to get a few special-cased patching methods - """ - - def patch_os_uname(self, replacement): - # twisted's 'patch' doesn't handle the case where an attribute - # doesn't exist.. - if hasattr(os, 'uname'): - self.patch(os, 'uname', replacement) - else: - def cleanup(): - del os.uname - self.addCleanup(cleanup) - os.uname = replacement - - -class OpenFileMixin: - """ - Mixin for patching open() to simulate successful reads and I/O errors. - """ - def setUpOpen(self, file_contents): - """ - patch open() to return file object with provided contents. - - @param file_contents: contents that will be returned by file object's - read() method - """ - # create mocked file object that returns 'file_contents' on read() - self.fileobj = mock.Mock() - self.fileobj.read = mock.Mock(return_value=file_contents) - - # patch open() to return mocked object - self.open = mock.Mock(return_value=self.fileobj) - self.patch(__builtin__, "open", self.open) - - def setUpOpenError(self, errno, strerror="dummy-msg", - filename="dummy-file"): - """ - patch open() to raise IOError - - @param errno: exception's errno value - @param strerror: exception's strerror value - @param filename: exception's filename value - """ - self.open = mock.Mock(side_effect=IOError(errno, strerror, filename)) - self.patch(__builtin__, "open", self.open) - - def setUpReadError(self, errno, strerror="dummy-msg", - filename="dummy-file"): - """ - patch open() to return a file object that will raise IOError on read() - - @param errno: exception's errno value - @param strerror: exception's strerror value - @param filename: exception's filename value - - """ - self.fileobj = mock.Mock() - self.fileobj.read = mock.Mock(side_effect=IOError(errno, strerror, - filename)) - self.open = mock.Mock(return_value=self.fileobj) - self.patch(__builtin__, "open", self.open) - - -class StdoutAssertionsMixin(object): - """ - Mix this in to be able to assert on stdout during the test - """ - def setUpStdoutAssertions(self): - self.stdout = cStringIO.StringIO() - self.patch(sys, 'stdout', self.stdout) - - def assertWasQuiet(self): - self.assertEqual(self.stdout.getvalue(), '') - - def assertInStdout(self, exp): - self.assertIn(exp, self.stdout.getvalue()) - - def assertStdoutEqual(self, exp, msg=None): - self.assertEqual(exp, self.stdout.getvalue(), msg) - - def getStdout(self): - return self.stdout.getvalue().strip() |