diff options
Diffstat (limited to 'tests/utils.py')
-rw-r--r-- | tests/utils.py | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/tests/utils.py b/tests/utils.py index e93c278..f49e0af 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -38,7 +38,24 @@ class BaseTestCase(testtools.TestCase): self.conf = conf self.addCleanup(self.conf.reset) self.useFixture(fixtures.FakeLogger('openstack.common')) - self.useFixture(fixtures.Timeout(30, True)) + + test_timeout = os.environ.get('OS_TEST_TIMEOUT', 0) + try: + test_timeout = int(test_timeout) + except ValueError: + # If timeout value is invalid do not set a timeout. + test_timeout = 0 + if test_timeout > 0: + self.useFixture(fixtures.Timeout(test_timeout, gentle=True)) + if (os.environ.get('OS_STDOUT_CAPTURE') == 'True' or + os.environ.get('OS_STDOUT_CAPTURE') == '1'): + stdout = self.useFixture(fixtures.StringStream('stdout')).stream + self.useFixture(fixtures.MonkeyPatch('sys.stdout', stdout)) + if (os.environ.get('OS_STDERR_CAPTURE') == 'True' or + os.environ.get('OS_STDERR_CAPTURE') == '1'): + stderr = self.useFixture(fixtures.StringStream('stderr')).stream + self.useFixture(fixtures.MonkeyPatch('sys.stderr', stderr)) + self.stubs.Set(exception, '_FATAL_EXCEPTION_FORMAT_ERRORS', True) self.useFixture(fixtures.NestedTempfile()) self.tempdirs = [] |