From: W. Trevor King <wking@drexel.edu>
Date: Mon, 18 Jan 2010 17:46:08 +0000 (-0500)
Subject: Add class name to StorageTestCase failure reporting
X-Git-Tag: 1.0.0~59^2~47^2~12
X-Git-Url: http://git.tremily.us/?a=commitdiff_plain;h=e06f1aa3f5db039fa8bf1f26412059fe99588a2b;p=be.git

Add class name to StorageTestCase failure reporting
---

diff --git a/libbe/storage/base.py b/libbe/storage/base.py
index 10649a8..202305b 100644
--- a/libbe/storage/base.py
+++ b/libbe/storage/base.py
@@ -533,6 +533,23 @@ if TESTING == True:
             super(StorageTestCase, self).__init__(*args, **kwargs)
             self.dirname = None
 
+        # this class will be the basis of tests for several classes,
+        # so make sure we print the name of the class we're dealing with.
+        def fail(self, msg=None):
+            """Fail immediately, with the given message."""
+            raise self.failureException, \
+                '(%s) %s' % (self.Class.__name__, msg)
+
+        def failIf(self, expr, msg=None):
+            "Fail the test if the expression is true."
+            if expr: raise self.failureException, \
+                '(%s) %s' % (self.Class.__name__, msg)
+
+        def failUnless(self, expr, msg=None):
+            """Fail the test unless the expression is true."""
+            if not expr: raise self.failureException, \
+                '(%s) %s' % (self.Class.__name__, msg)
+
         def setUp(self):
             """Set up test fixtures for Storage test case."""
             super(StorageTestCase, self).setUp()