From: Benjamin Peterson <benjamin@python.org>
Date: Sun, 7 Jun 2009 23:12:44 +0000 (+0000)
Subject: backport r73273
X-Git-Tag: v2.7a1~1008
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=64092a56ec2f9278cbee033fc123352cbb74aaa4;p=python

backport r73273
---

diff --git a/Lib/test/test_winreg.py b/Lib/test/test_winreg.py
index 1eaa93047e..4453afe375 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")