From 64092a56ec2f9278cbee033fc123352cbb74aaa4 Mon Sep 17 00:00:00 2001 From: Benjamin Peterson Date: Sun, 7 Jun 2009 23:12:44 +0000 Subject: [PATCH] backport r73273 --- Lib/test/test_winreg.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/Lib/test/test_winreg.py b/Lib/test/test_winreg.py index 1eaa93047e7..4453afe375b 100644 --- a/Lib/test/test_winreg.py +++ b/Lib/test/test_winreg.py @@ -35,6 +35,27 @@ if test_support.have_unicode: class WinregTests(unittest.TestCase): remote_name = None + def setUp(self): + # Make sure that the test key is absent when the test + # starts. + self.delete_tree(HKEY_CURRENT_USER, test_key_name) + + def delete_tree(self, root, subkey): + try: + hkey = OpenKey(root, subkey, KEY_ALL_ACCESS) + except WindowsError: + # subkey does not exist + return + while True: + try: + subsubkey = EnumKey(hkey, 0) + except WindowsError: + # no more subkeys + break + self.delete_tree(hkey, subsubkey) + CloseKey(hkey) + DeleteKey(root, subkey) + def WriteTestData(self, root_key): # Set the default value for this key. SetValue(root_key, test_key_name, REG_SZ, "Default value")