Helper method to show and hide test results.

1 files changed, 21 insertions, 0 deletions

diff --git a/testtools/tests/test_helpers.py b/testtools/tests/test_helpers.py
index c19fa89..7e30d9a 100644
--- a/testtools/tests/test_helpers.py
+++ b/testtools/tests/test_helpers.py
@@ -10,6 +10,7 @@ from testtools.matchers import (
     Is,
     Not,
     )
+from testtools.tests.helpers import hide_testtools_stack
 
 
 def check_error_callback(test, function, arg, expected_error_count,
@@ -154,6 +155,26 @@ class TestTryImports(TestCase):
             0, True)
 
 
+import testtools.matchers
+import testtools.runtest
+import testtools.testcase
+
+class TestStackHiding(TestCase):
+
+    def test_hidden_by_default(self):
+        self.assertEqual(True, getattr(testtools.matchers, '__unittest'))
+        self.assertEqual(True, getattr(testtools.runtest, '__unittest'))
+        self.assertEqual(True, getattr(testtools.testcase, '__unittest'))
+
+    def test_show_stack(self):
+        current_state = getattr(testtools.matchers, '__unittest')
+        hide_testtools_stack(False)
+        self.assertEqual(False, getattr(testtools.matchers, '__unittest'))
+        self.assertEqual(False, getattr(testtools.runtest, '__unittest'))
+        self.assertEqual(False, getattr(testtools.testcase, '__unittest'))
+        hide_testtools_stack(current_state)
+
+
 def test_suite():
     from unittest import TestLoader
     return TestLoader().loadTestsFromName(__name__)