7dbd17a5aa
Former-commit-id:133dc97f67[formerlya02aeb236c] [formerly9f19e3f712] [formerly133dc97f67[formerlya02aeb236c] [formerly9f19e3f712] [formerly06a8b51d6d[formerly9f19e3f712[formerly 64fa9254b946eae7e61bbc3f513b7c3696c4f54f]]]] Former-commit-id:06a8b51d6dFormer-commit-id:9bb8decbcf[formerly8e80217e59] [formerly377dcd10b9[formerly3360eb6c5f]] Former-commit-id:377dcd10b9Former-commit-id:e2ecdcfe33
79 lines
2.2 KiB
Python
Executable file
79 lines
2.2 KiB
Python
Executable file
import sys
|
|
import unittest
|
|
from optparse import OptionParser
|
|
from nose.config import Config
|
|
from nose.plugins.capture import Capture
|
|
|
|
class TestCapturePlugin(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self._stdout = sys.stdout
|
|
|
|
def tearDown(self):
|
|
sys.stdout = self._stdout
|
|
|
|
def test_enabled_by_default(self):
|
|
c = Capture()
|
|
assert c.enabled
|
|
|
|
def test_can_be_disabled(self):
|
|
c = Capture()
|
|
parser = OptionParser()
|
|
c.addOptions(parser)
|
|
options, args = parser.parse_args(['test_can_be_disabled',
|
|
'-s'])
|
|
c.configure(options, Config())
|
|
assert not c.enabled
|
|
|
|
c = Capture()
|
|
options, args = parser.parse_args(['test_can_be_disabled_long',
|
|
'--nocapture'])
|
|
c.configure(options, Config())
|
|
assert not c.enabled
|
|
|
|
env = {'NOSE_NOCAPTURE': 1}
|
|
c = Capture()
|
|
parser = OptionParser()
|
|
c.addOptions(parser, env)
|
|
options, args = parser.parse_args(['test_can_be_disabled'])
|
|
c.configure(options, Config())
|
|
assert not c.enabled
|
|
|
|
c = Capture()
|
|
parser = OptionParser()
|
|
c.addOptions(parser)
|
|
|
|
options, args = parser.parse_args(['test_can_be_disabled'])
|
|
c.configure(options, Config())
|
|
assert c.enabled
|
|
|
|
def test_captures_stdout(self):
|
|
c = Capture()
|
|
c.start()
|
|
print "Hello"
|
|
c.end()
|
|
self.assertEqual(c.buffer, "Hello\n")
|
|
|
|
def test_format_error(self):
|
|
class Dummy:
|
|
pass
|
|
d = Dummy()
|
|
c = Capture()
|
|
c.start()
|
|
try:
|
|
print "Oh my!"
|
|
raise Exception("boom")
|
|
except:
|
|
err = sys.exc_info()
|
|
formatted = c.formatError(d, err)
|
|
ec, ev, tb = err
|
|
(fec, fev, ftb) = formatted
|
|
# print fec, fev, ftb
|
|
|
|
self.assertEqual(ec, fec)
|
|
self.assertEqual(tb, ftb)
|
|
assert 'Oh my!' in fev, "Output not found in error message"
|
|
assert 'Oh my!' in d.capturedOutput, "Output not attached to test"
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|