From 64ea644a483e695a3d6689ba931ca557c3757625 Mon Sep 17 00:00:00 2001
From: Brian Warner <warner@lothar.com>
Date: Sun, 8 Apr 2007 13:02:13 -0700
Subject: [PATCH] test_util: add more coverage for assertutil.py

---
 src/allmydata/test/test_util.py | 64 +++++++++++++++++++++++++++++++++
 1 file changed, 64 insertions(+)

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' <type 'str'>", m)
+        m = self.should_assert(f, False, "message1", othermsg=12)
+        self.failUnlessEqual("'message1' <type 'str'>, othermsg: 12 <type 'int'>", m)
+        m = self.should_assert(f, False, othermsg="message2")
+        self.failUnlessEqual("othermsg: 'message2' <type 'str'>", 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' <type 'str'>", m)
+        m = self.should_assert(f, False, "message1", othermsg=12)
+        self.failUnlessEqual("precondition: 'message1' <type 'str'>, othermsg: 12 <type 'int'>", m)
+        m = self.should_assert(f, False, othermsg="message2")
+        self.failUnlessEqual("precondition: othermsg: 'message2' <type 'str'>", 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' <type 'str'>", m)
+        m = self.should_assert(f, False, "message1", othermsg=12)
+        self.failUnlessEqual("postcondition: 'message1' <type 'str'>, othermsg: 12 <type 'int'>", m)
+        m = self.should_assert(f, False, othermsg="message2")
+        self.failUnlessEqual("postcondition: othermsg: 'message2' <type 'str'>", m)
+
-- 
2.45.2