Initial commit for the Automated Installer Testing Framework.

chrome/test/mini_installer/registry_verifier.py

+# Copyright 2013 The Chromium Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+import _winreg
+import settings
+
+
+def VerifyRegistryEntries(entries):
+  for key, entry in entries.iteritems():
+    if VerifyRegistryEntry(key, entry):
+      print "Passed"
+    else:
+      print "Failed"
+
+
+def RootKeyConstant(key):
+  if key == "HKEY_CLASSES_ROOT":
+    return _winreg.HKEY_CLASSES_ROOT
+  if key == "HKEY_CURRENT_USER":
+    return _winreg.HKEY_CURRENT_USER
+  if key == "HKEY_LOCAL_MACHINE":
+    return _winreg.HKEY_LOCAL_MACHINE
+  if key == "HKEY_USERS":
+    return _winreg.HKEY_USERS
+  raise Exception("Unknown registry key")
+
+
+def VerifyRegistryEntry(key, entry):
+  expected = entry["expected"]
+  print settings.PRINT_VERIFIER_PREFIX + key,
+  if expected:
+    print "exists...",
+  else:
+    print "doesn't exist...",
+  try:
+    root_key, sub_key = key.split("\\", 1)
+    reg_key = _winreg.OpenKey(RootKeyConstant(root_key),
+                              sub_key, 0, _winreg.KEY_READ)
+    if not expected:
+      return False
+    if "value" in entry:
+      # TODO(sukolsak): implement value
+      pass
+    return True
+  except WindowsError:
+    return not expected