3 files changed, 11 insertions, 8 deletions
diff --git a/source4/auth/credentials/tests/bindings.py b/source4/auth/credentials/tests/bindings.py
index 30120b3a60..4cb059ddf0 100644
--- a/source4/auth/credentials/tests/bindings.py
+++ b/source4/auth/credentials/tests/bindings.py
@@ -23,11 +23,13 @@ Note that this just tests the bindings work. It does not intend to test
 the functionality, that's already done in other tests.
 """
 
-import unittest
 from samba import credentials
+import samba.tests
+
+class CredentialsTests(samba.tests.TestCase):
 
-class CredentialsTests(unittest.TestCase):
     def setUp(self):
+        super(CredentialsTests, self).setUp()
         self.creds = credentials.Credentials()
 
     def test_set_username(self):
diff --git a/source4/auth/gensec/tests/bindings.py b/source4/auth/gensec/tests/bindings.py
index b9d80e4f9e..cffcc27aec 100644
--- a/source4/auth/gensec/tests/bindings.py
+++ b/source4/auth/gensec/tests/bindings.py
@@ -23,16 +23,16 @@ Note that this just tests the bindings work. It does not intend to test
 the functionality, that's already done in other tests.
 """
 
-import unittest
 from samba import gensec
-from samba.tests import env_loadparm
+import samba.tests
 
-class CredentialsTests(unittest.TestCase):
+class CredentialsTests(samba.tests.TestCase):
 
     def setUp(self):
+        super(CredentialsTests, self).setUp()
         settings = {}
         settings["target_hostname"] = "localhost"
-        settings["lp_ctx"] = env_loadparm()
+        settings["lp_ctx"] = samba.tests.env_loadparm()
         self.gensec = gensec.Security.start_client(settings)
 
     def test_info(self):
diff --git a/source4/auth/tests/bindings.py b/source4/auth/tests/bindings.py
index 15368a4bf7..58b66d3c98 100644
--- a/source4/auth/tests/bindings.py
+++ b/source4/auth/tests/bindings.py
@@ -23,10 +23,11 @@ Note that this just tests the bindings work. It does not intend to test
 the functionality, that's already done in other tests.
 """
 
-import unittest
 from samba import auth
+import samba.tests
+
+class AuthTests(samba.tests.TestCase):
 
-class AuthTests(unittest.TestCase):
     def test_system_session(self):
         auth.system_session()