1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-22 13:34:15 +03:00

Handle skips when running on python2.6.

Change-Id: I8b0a15760a72f41800d23150232c2b0e59e32c32
Signed-off-by: Jelmer Vernooij <jelmer@samba.org>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
This commit is contained in:
Jelmer Vernooij 2015-02-04 16:40:29 +01:00 committed by Andrew Bartlett
parent 02da1b3798
commit 1988e11585

View File

@ -25,6 +25,7 @@ from samba import param
from samba.samdb import SamDB from samba.samdb import SamDB
from samba import credentials from samba import credentials
import subprocess import subprocess
import sys
import tempfile import tempfile
import unittest import unittest
@ -54,32 +55,79 @@ class TestCase(unittest.TestCase):
return cmdline_credentials return cmdline_credentials
# These functions didn't exist before Python2.7: # These functions didn't exist before Python2.7:
if not getattr(unittest.TestCase, "skipTest", None): if sys.version_info < (2, 7):
import warnings
def skipTest(self, reason): def skipTest(self, reason):
raise SkipTest(reason) raise SkipTest(reason)
if not getattr(unittest.TestCase, "assertIs", None):
def assertIs(self, a, b): def assertIs(self, a, b):
self.assertTrue(a is b) self.assertTrue(a is b)
if not getattr(unittest.TestCase, "assertIsNot", None):
def assertIsNot(self, a, b): def assertIsNot(self, a, b):
self.assertTrue(a is not b) self.assertTrue(a is not b)
if not getattr(unittest.TestCase, "assertIsInstance", None):
def assertIsInstance(self, a, b): def assertIsInstance(self, a, b):
self.assertTrue(isinstance(a, b)) self.assertTrue(isinstance(a, b))
if not getattr(unittest.TestCase, "addCleanup", None):
def addCleanup(self, fn, *args, **kwargs): def addCleanup(self, fn, *args, **kwargs):
self._cleanups = getattr(self, "_cleanups", []) + [ self._cleanups = getattr(self, "_cleanups", []) + [
(fn, args, kwargs)] (fn, args, kwargs)]
def _addSkip(self, result, reason):
addSkip = getattr(result, 'addSkip', None)
if addSkip is not None:
addSkip(self, reason)
else:
warnings.warn("TestResult has no addSkip method, skips not reported",
RuntimeWarning, 2)
result.addSuccess(self)
def run(self, result=None): def run(self, result=None):
ret = super(TestCase, self).run(result=result) if result is None: result = self.defaultTestResult()
for (fn, args, kwargs) in reversed(getattr(self, "_cleanups", [])): result.startTest(self)
fn(*args, **kwargs) testMethod = getattr(self, self._testMethodName)
return ret try:
try:
self.setUp()
except SkipTest, e:
self._addSkip(result, str(e))
return
except KeyboardInterrupt:
raise
except:
result.addError(self, self._exc_info())
return
ok = False
try:
testMethod()
ok = True
except SkipTest, e:
self._addSkip(result, str(e))
return
except self.failureException:
result.addFailure(self, self._exc_info())
except KeyboardInterrupt:
raise
except:
result.addError(self, self._exc_info())
try:
self.tearDown()
except SkipTest, e:
self._addSkip(result, str(e))
except KeyboardInterrupt:
raise
except:
result.addError(self, self._exc_info())
ok = False
for (fn, args, kwargs) in reversed(getattr(self, "_cleanups", [])):
fn(*args, **kwargs)
if ok: result.addSuccess(self)
finally:
result.stopTest(self)
class LdbTestCase(TestCase): class LdbTestCase(TestCase):