tests/utils.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

import os

import fixtures
import testtools


class TestCase(testtools.TestCase):
    def setUp(self):
        super(TestCase, self).setUp()
        if (os.environ.get("OS_STDOUT_NOCAPTURE") == "True" and
                os.environ.get("OS_STDOUT_NOCAPTURE") == "1"):
            stdout = self.useFixture(fixtures.StringStream("stdout")).stream
            self.useFixture(fixtures.MonkeyPatch("sys.stdout", stdout))
        if (os.environ.get("OS_STDERR_NOCAPTURE") == "True" and
                os.environ.get("OS_STDERR_NOCAPTURE") == "1"):
            stderr = self.useFixture(fixtures.StringStream("stderr")).stream
            self.useFixture(fixtures.MonkeyPatch("sys.stderr", stderr))

    def tearDown(self):
        super(TestCase, self).tearDown()