[gentoo-commits] proj/gentoopm:master commit in: gentoopm/tests/

["Michał Górny" <mgorny@gentoo.org>]

commit:     c331753e7d32c1329a53291d9d8c79dc14f2156a
Author:     Michał Górny <mgorny <AT> gentoo <DOT> org>
AuthorDate: Fri Jul 15 12:09:53 2011 +0000
Commit:     Michał Górny <mgorny <AT> gentoo <DOT> org>
CommitDate: Fri Jul 15 12:09:53 2011 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/gentoopm.git;a=commit;h=c331753e

Return PM name in test failure message.

---
 gentoopm/tests/__init__.py |   30 ++++++++++++++++++++++++++++++
 1 files changed, 30 insertions(+), 0 deletions(-)

diff --git a/gentoopm/tests/__init__.py b/gentoopm/tests/__init__.py
index e96860f..88ad284 100644
--- a/gentoopm/tests/__init__.py
+++ b/gentoopm/tests/__init__.py
@@ -18,6 +18,36 @@ class PMTestCase(unittest.TestCase):
 		assert(self._pm is None)
 		self._pm = val
 
+	def assertEqual(self, *args):
+		try:
+			unittest.TestCase.assertEqual(self, *args)
+		except AssertionError as e:
+			raise AssertionError('[%s] %s' % (self._pm.name, e.message))
+
+	def assertNotEqual(self, *args):
+		try:
+			unittest.TestCase.assertNotEqual(self, *args)
+		except AssertionError as e:
+			raise AssertionError('[%s] %s' % (self._pm.name, e.message))
+
+	def assertTrue(self, *args):
+		try:
+			unittest.TestCase.assertTrue(self, *args)
+		except AssertionError as e:
+			raise AssertionError('[%s] %s' % (self._pm.name, e.message))
+
+	def assertFalse(self, *args):
+		try:
+			unittest.TestCase.assertFalse(self, *args)
+		except AssertionError as e:
+			raise AssertionError('[%s] %s' % (self._pm.name, e.message))
+
+	def assertRaises(self, *args):
+		try:
+			unittest.TestCase.assertRaises(self, *args)
+		except AssertionError as e:
+			raise AssertionError('[%s] %s' % (self._pm.name, e.message))
+
 class PMTestSuiteFactory(object):
 	def __init__(self, pm):
 		self._pm = pm