Use samba TestCase so we get all compatibility functions on Python < 2.7.

Change-Id: Iba87e3c8fa9331c4d5438ab60a8385379da634d7
Signed-Off-By: Jelmer Vernooij <jelmer@samba.org>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>

diff --git a/python/samba/tests/__init__.py b/python/samba/tests/__init__.py
index 480b2c813524b96e257d113185c54fdc687693e4..b73308ecd49629ea963d634c0007d8616e4e3cde 100644 (file)
--- a/python/samba/tests/__init__.py
+++ b/python/samba/tests/__init__.py
@@ -81,7 +81,7 @@ class TestCase(unittest.TestCase):
                 fn(*args, **kwargs)
 
 
-class LdbTestCase(unittest.TestCase):
+class LdbTestCase(TestCase):
     """Trivial test case for running tests against a LDB."""
 
     def setUp(self):

diff --git a/selftest/tests/test_samba.py b/selftest/tests/test_samba.py
index 4528213c4e5b4d8605855b45c70f5cff04e2bccc..23de0d34c736024e9ed30cad41fedbdf669fc5b1 100644 (file)
--- a/selftest/tests/test_samba.py
+++ b/selftest/tests/test_samba.py
@@ -21,7 +21,7 @@
 
 from cStringIO import StringIO
 
-from unittest import TestCase
+from samba.tests import TestCase
 
 from selftest.target.samba import (
     bindir_path,

diff --git a/selftest/tests/test_socket_wrapper.py b/selftest/tests/test_socket_wrapper.py
index 5862f1edf8a910fb58623c48805084446d166007..e278fdddfa6fabeb5abf7b7828ee4a2d1d717089 100644 (file)
--- a/selftest/tests/test_socket_wrapper.py
+++ b/selftest/tests/test_socket_wrapper.py
@@ -19,7 +19,7 @@
 
 """Tests for selftest/socket_wrapper."""
 
-from unittest import TestCase
+from samba.tests import TestCase
 
 from selftest import socket_wrapper
 

diff --git a/selftest/tests/test_target.py b/selftest/tests/test_target.py
index 0d081ceadb053493f066221238c17bf9a7b481ac..ed9b064961efb14d0a7e74b748668e769392a6b0 100644 (file)
--- a/selftest/tests/test_target.py
+++ b/selftest/tests/test_target.py
@@ -29,7 +29,7 @@ from selftest.target import (
     UnsupportedEnvironment,
     )
 
-from unittest import TestCase
+from samba.tests import TestCase
 
 
 class DummyEnvironment(Environment):