From e06f1aa3f5db039fa8bf1f26412059fe99588a2b Mon Sep 17 00:00:00 2001 From: "W. Trevor King" Date: Mon, 18 Jan 2010 12:46:08 -0500 Subject: [PATCH] Add class name to StorageTestCase failure reporting --- libbe/storage/base.py | 17 +++++++++++++++++ 1 file changed, 17 insertions(+) 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() -- 2.26.2