From c92db7b6dc31756fc461ea9f74e1cd675e3e160d Mon Sep 17 00:00:00 2001
From: Jelmer Vernooij <jelmer@samba.org>
Date: Sat, 19 Jun 2010 18:58:18 +0200
Subject: python: Use samba.tests.TestCase, make sure base class tearDown and
 setUp methods are called, fix formatting.

---
 source4/lib/registry/tests/bindings.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

(limited to 'source4/lib/registry')

diff --git a/source4/lib/registry/tests/bindings.py b/source4/lib/registry/tests/bindings.py
index b5115aefa4..d01ebff909 100644
--- a/source4/lib/registry/tests/bindings.py
+++ b/source4/lib/registry/tests/bindings.py
@@ -18,11 +18,11 @@
 #
 
 import os
-import unittest
 from samba import registry
 import samba.tests
 
-class HelperTests(unittest.TestCase):
+class HelperTests(samba.tests.TestCase):
+
     def test_predef_to_name(self):
         self.assertEquals("HKEY_LOCAL_MACHINE", 
                           registry.get_predef_name(0x80000002))
@@ -33,6 +33,7 @@ class HelperTests(unittest.TestCase):
 
 
 class HiveTests(samba.tests.TestCaseInTempDir):
+
     def setUp(self):
         super(HiveTests, self).setUp()
         self.hive_path = os.path.join(self.tempdir, "ldb_new.ldb")
@@ -41,6 +42,7 @@ class HiveTests(samba.tests.TestCaseInTempDir):
     def tearDown(self):
         del self.hive
         os.unlink(self.hive_path)
+        super(HiveTests, self).tearDown()
 
     def test_ldb_new(self):
         self.assertTrue(self.hive is not None)
@@ -52,6 +54,7 @@ class HiveTests(samba.tests.TestCaseInTempDir):
     #    self.hive.del_value("FOO")
 
 
-class RegistryTests(unittest.TestCase):
+class RegistryTests(samba.tests.TestCase):
+
     def test_new(self):
         self.registry = registry.Registry()
-- 
cgit