From: Brian Warner Date: Sun, 8 Apr 2007 20:02:13 +0000 (-0700) Subject: test_util: add more coverage for assertutil.py X-Git-Url: https://git.rkrishnan.org/%5B/(%5B%5E?a=commitdiff_plain;h=64ea644a483e695a3d6689ba931ca557c3757625;p=tahoe-lafs%2Ftahoe-lafs.git test_util: add more coverage for assertutil.py --- diff --git a/src/allmydata/test/test_util.py b/src/allmydata/test/test_util.py index 94536e96..1c14457c 100644 --- a/src/allmydata/test/test_util.py +++ b/src/allmydata/test/test_util.py @@ -4,6 +4,7 @@ def foo(): pass # keep the line number constant from twisted.trial import unittest from allmydata.util import bencode, idlib, humanreadable, mathutil +from allmydata.util import assertutil class IDLib(unittest.TestCase): @@ -206,3 +207,66 @@ class Math(unittest.TestCase): self.failUnlessEqual(f([0,0,0,4]), 1) self.failUnlessAlmostEqual(f([0.0, 1.0, 1.0]), .666666666666) + +class Asserts(unittest.TestCase): + def should_assert(self, func, *args, **kwargs): + try: + func(*args, **kwargs) + except AssertionError, e: + return str(e) + except Exception, e: + self.fail("assert failed with non-AssertionError: %s" % e) + self.fail("assert was not caught") + + def should_not_assert(self, func, *args, **kwargs): + if "re" in kwargs: + regexp = kwargs["re"] + del kwargs["re"] + try: + func(*args, **kwargs) + except AssertionError, e: + self.fail("assertion fired when it should not have: %s" % e) + except Exception, e: + self.fail("assertion (which shouldn't have failed) failed with non-AssertionError: %s" % e) + return # we're happy + + + def test_assert(self): + f = assertutil._assert + self.should_assert(f) + self.should_assert(f, False) + self.should_not_assert(f, True) + + m = self.should_assert(f, False, "message") + self.failUnlessEqual(m, "'message' ", m) + m = self.should_assert(f, False, "message1", othermsg=12) + self.failUnlessEqual("'message1' , othermsg: 12 ", m) + m = self.should_assert(f, False, othermsg="message2") + self.failUnlessEqual("othermsg: 'message2' ", m) + + def test_precondition(self): + f = assertutil.precondition + self.should_assert(f) + self.should_assert(f, False) + self.should_not_assert(f, True) + + m = self.should_assert(f, False, "message") + self.failUnlessEqual("precondition: 'message' ", m) + m = self.should_assert(f, False, "message1", othermsg=12) + self.failUnlessEqual("precondition: 'message1' , othermsg: 12 ", m) + m = self.should_assert(f, False, othermsg="message2") + self.failUnlessEqual("precondition: othermsg: 'message2' ", m) + + def test_postcondition(self): + f = assertutil.postcondition + self.should_assert(f) + self.should_assert(f, False) + self.should_not_assert(f, True) + + m = self.should_assert(f, False, "message") + self.failUnlessEqual("postcondition: 'message' ", m) + m = self.should_assert(f, False, "message1", othermsg=12) + self.failUnlessEqual("postcondition: 'message1' , othermsg: 12 ", m) + m = self.should_assert(f, False, othermsg="message2") + self.failUnlessEqual("postcondition: othermsg: 'message2' ", m) +