diff --git a/tests/gclient_scm_test.py b/tests/gclient_scm_test.py index 20a4cbdb0..4f8b656fb 100755 --- a/tests/gclient_scm_test.py +++ b/tests/gclient_scm_test.py @@ -16,7 +16,17 @@ import __builtin__ from super_mox import mox, SuperMoxBaseTestBase, SuperMoxTestBase import gclient_scm -from gclient_test import BaseTestCase as GCBaseTestCase + + +class GCBaseTestCase(SuperMoxTestBase): + # Like unittest's assertRaises, but checks for Gclient.Error. + def assertRaisesError(self, msg, fn, *args, **kwargs): + try: + fn(*args, **kwargs) + except gclient_scm.gclient_utils.Error, e: + self.assertEquals(e.args[0], msg) + else: + self.fail('%s not raised' % msg) class BaseTestCase(GCBaseTestCase): diff --git a/tests/scm_unittest.py b/tests/scm_unittest.py index d4270bd1d..d2acc04b9 100755 --- a/tests/scm_unittest.py +++ b/tests/scm_unittest.py @@ -9,12 +9,22 @@ from shutil import rmtree import tempfile # Fixes include path. -from super_mox import mox, SuperMoxBaseTestBase +from super_mox import mox, SuperMoxBaseTestBase, SuperMoxTestBase -from gclient_test import BaseTestCase import scm +class BaseTestCase(SuperMoxTestBase): + # Like unittest's assertRaises, but checks for Gclient.Error. + def assertRaisesError(self, msg, fn, *args, **kwargs): + try: + fn(*args, **kwargs) + except scm.gclient_utils.Error, e: + self.assertEquals(e.args[0], msg) + else: + self.fail('%s not raised' % msg) + + class BaseSCMTestCase(BaseTestCase): def setUp(self): BaseTestCase.setUp(self)