AppPkg/Applications/Python: Add Python 2.7.2 sources since the release of Python 2.7.3 made them unavailable from the python.org web site.
These files are a subset of the python-2.7.2.tgz distribution from python.org. Changed files from PyMod-2.7.2 have been copied into the corresponding directories of this tree, replacing the original files in the distribution. Signed-off-by: daryl.mcdaniel@intel.com git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@13197 6f19259b-4bc3-4df7-8a09-765794883524
This commit is contained in:
darylm503
416
AppPkg/Applications/Python/Python-2.7.2/Lib/test/test_winreg.py
Normal file
416
AppPkg/Applications/Python/Python-2.7.2/Lib/test/test_winreg.py
Normal file
@ -0,0 +1,416 @@
|
||||
# Test the windows specific win32reg module.
|
||||
# Only win32reg functions not hit here: FlushKey, LoadKey and SaveKey
|
||||
|
||||
import os, sys
|
||||
import unittest
|
||||
from test import test_support
|
||||
threading = test_support.import_module("threading")
|
||||
from platform import machine
|
||||
|
||||
# Do this first so test will be skipped if module doesn't exist
|
||||
test_support.import_module('_winreg')
|
||||
# Now import everything
|
||||
from _winreg import *
|
||||
|
||||
try:
|
||||
REMOTE_NAME = sys.argv[sys.argv.index("--remote")+1]
|
||||
except (IndexError, ValueError):
|
||||
REMOTE_NAME = None
|
||||
|
||||
# tuple of (major, minor)
|
||||
WIN_VER = sys.getwindowsversion()[:2]
|
||||
# Some tests should only run on 64-bit architectures where WOW64 will be.
|
||||
WIN64_MACHINE = True if machine() == "AMD64" else False
|
||||
|
||||
# Starting with Windows 7 and Windows Server 2008 R2, WOW64 no longer uses
|
||||
# registry reflection and formerly reflected keys are shared instead.
|
||||
# Windows 7 and Windows Server 2008 R2 are version 6.1. Due to this, some
|
||||
# tests are only valid up until 6.1
|
||||
HAS_REFLECTION = True if WIN_VER < (6, 1) else False
|
||||
|
||||
test_key_name = "SOFTWARE\\Python Registry Test Key - Delete Me"
|
||||
# On OS'es that support reflection we should test with a reflected key
|
||||
test_reflect_key_name = "SOFTWARE\\Classes\\Python Test Key - Delete Me"
|
||||
|
||||
test_data = [
|
||||
("Int Value", 45, REG_DWORD),
|
||||
("String Val", "A string value", REG_SZ),
|
||||
("StringExpand", "The path is %path%", REG_EXPAND_SZ),
|
||||
("Multi-string", ["Lots", "of", "string", "values"], REG_MULTI_SZ),
|
||||
("Raw Data", ("binary"+chr(0)+"data"), REG_BINARY),
|
||||
("Big String", "x"*(2**14-1), REG_SZ),
|
||||
("Big Binary", "x"*(2**14), REG_BINARY),
|
||||
]
|
||||
|
||||
if test_support.have_unicode:
|
||||
test_data += [
|
||||
(unicode("Unicode Val"), unicode("A Unicode value"), REG_SZ,),
|
||||
("UnicodeExpand", unicode("The path is %path%"), REG_EXPAND_SZ),
|
||||
("Multi-unicode", [unicode("Lots"), unicode("of"), unicode("unicode"),
|
||||
unicode("values")], REG_MULTI_SZ),
|
||||
("Multi-mixed", [unicode("Unicode"), unicode("and"), "string",
|
||||
"values"], REG_MULTI_SZ),
|
||||
]
|
||||
|
||||
class BaseWinregTests(unittest.TestCase):
|
||||
|
||||
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 _write_test_data(self, root_key, CreateKey=CreateKey):
|
||||
# Set the default value for this key.
|
||||
SetValue(root_key, test_key_name, REG_SZ, "Default value")
|
||||
key = CreateKey(root_key, test_key_name)
|
||||
# Create a sub-key
|
||||
sub_key = CreateKey(key, "sub_key")
|
||||
# Give the sub-key some named values
|
||||
|
||||
for value_name, value_data, value_type in test_data:
|
||||
SetValueEx(sub_key, value_name, 0, value_type, value_data)
|
||||
|
||||
# Check we wrote as many items as we thought.
|
||||
nkeys, nvalues, since_mod = QueryInfoKey(key)
|
||||
self.assertEqual(nkeys, 1, "Not the correct number of sub keys")
|
||||
self.assertEqual(nvalues, 1, "Not the correct number of values")
|
||||
nkeys, nvalues, since_mod = QueryInfoKey(sub_key)
|
||||
self.assertEqual(nkeys, 0, "Not the correct number of sub keys")
|
||||
self.assertEqual(nvalues, len(test_data),
|
||||
"Not the correct number of values")
|
||||
# Close this key this way...
|
||||
# (but before we do, copy the key as an integer - this allows
|
||||
# us to test that the key really gets closed).
|
||||
int_sub_key = int(sub_key)
|
||||
CloseKey(sub_key)
|
||||
try:
|
||||
QueryInfoKey(int_sub_key)
|
||||
self.fail("It appears the CloseKey() function does "
|
||||
"not close the actual key!")
|
||||
except EnvironmentError:
|
||||
pass
|
||||
# ... and close that key that way :-)
|
||||
int_key = int(key)
|
||||
key.Close()
|
||||
try:
|
||||
QueryInfoKey(int_key)
|
||||
self.fail("It appears the key.Close() function "
|
||||
"does not close the actual key!")
|
||||
except EnvironmentError:
|
||||
pass
|
||||
|
||||
def _read_test_data(self, root_key, OpenKey=OpenKey):
|
||||
# Check we can get default value for this key.
|
||||
val = QueryValue(root_key, test_key_name)
|
||||
self.assertEqual(val, "Default value",
|
||||
"Registry didn't give back the correct value")
|
||||
|
||||
key = OpenKey(root_key, test_key_name)
|
||||
# Read the sub-keys
|
||||
with OpenKey(key, "sub_key") as sub_key:
|
||||
# Check I can enumerate over the values.
|
||||
index = 0
|
||||
while 1:
|
||||
try:
|
||||
data = EnumValue(sub_key, index)
|
||||
except EnvironmentError:
|
||||
break
|
||||
self.assertIn(data, test_data,
|
||||
"Didn't read back the correct test data")
|
||||
index = index + 1
|
||||
self.assertEqual(index, len(test_data),
|
||||
"Didn't read the correct number of items")
|
||||
# Check I can directly access each item
|
||||
for value_name, value_data, value_type in test_data:
|
||||
read_val, read_typ = QueryValueEx(sub_key, value_name)
|
||||
self.assertEqual(read_val, value_data,
|
||||
"Could not directly read the value")
|
||||
self.assertEqual(read_typ, value_type,
|
||||
"Could not directly read the value")
|
||||
sub_key.Close()
|
||||
# Enumerate our main key.
|
||||
read_val = EnumKey(key, 0)
|
||||
self.assertEqual(read_val, "sub_key", "Read subkey value wrong")
|
||||
try:
|
||||
EnumKey(key, 1)
|
||||
self.fail("Was able to get a second key when I only have one!")
|
||||
except EnvironmentError:
|
||||
pass
|
||||
|
||||
key.Close()
|
||||
|
||||
def _delete_test_data(self, root_key):
|
||||
key = OpenKey(root_key, test_key_name, 0, KEY_ALL_ACCESS)
|
||||
sub_key = OpenKey(key, "sub_key", 0, KEY_ALL_ACCESS)
|
||||
# It is not necessary to delete the values before deleting
|
||||
# the key (although subkeys must not exist). We delete them
|
||||
# manually just to prove we can :-)
|
||||
for value_name, value_data, value_type in test_data:
|
||||
DeleteValue(sub_key, value_name)
|
||||
|
||||
nkeys, nvalues, since_mod = QueryInfoKey(sub_key)
|
||||
self.assertEqual(nkeys, 0, "subkey not empty before delete")
|
||||
self.assertEqual(nvalues, 0, "subkey not empty before delete")
|
||||
sub_key.Close()
|
||||
DeleteKey(key, "sub_key")
|
||||
|
||||
try:
|
||||
# Shouldnt be able to delete it twice!
|
||||
DeleteKey(key, "sub_key")
|
||||
self.fail("Deleting the key twice succeeded")
|
||||
except EnvironmentError:
|
||||
pass
|
||||
key.Close()
|
||||
DeleteKey(root_key, test_key_name)
|
||||
# Opening should now fail!
|
||||
try:
|
||||
key = OpenKey(root_key, test_key_name)
|
||||
self.fail("Could open the non-existent key")
|
||||
except WindowsError: # Use this error name this time
|
||||
pass
|
||||
|
||||
def _test_all(self, root_key):
|
||||
self._write_test_data(root_key)
|
||||
self._read_test_data(root_key)
|
||||
self._delete_test_data(root_key)
|
||||
|
||||
class LocalWinregTests(BaseWinregTests):
|
||||
|
||||
def test_registry_works(self):
|
||||
self._test_all(HKEY_CURRENT_USER)
|
||||
|
||||
def test_registry_works_extended_functions(self):
|
||||
# Substitute the regular CreateKey and OpenKey calls with their
|
||||
# extended counterparts.
|
||||
# Note: DeleteKeyEx is not used here because it is platform dependent
|
||||
cke = lambda key, sub_key: CreateKeyEx(key, sub_key, 0, KEY_ALL_ACCESS)
|
||||
self._write_test_data(HKEY_CURRENT_USER, cke)
|
||||
|
||||
oke = lambda key, sub_key: OpenKeyEx(key, sub_key, 0, KEY_READ)
|
||||
self._read_test_data(HKEY_CURRENT_USER, oke)
|
||||
|
||||
self._delete_test_data(HKEY_CURRENT_USER)
|
||||
|
||||
def test_connect_registry_to_local_machine_works(self):
|
||||
# perform minimal ConnectRegistry test which just invokes it
|
||||
h = ConnectRegistry(None, HKEY_LOCAL_MACHINE)
|
||||
self.assertNotEqual(h.handle, 0)
|
||||
h.Close()
|
||||
self.assertEqual(h.handle, 0)
|
||||
|
||||
def test_inexistant_remote_registry(self):
|
||||
connect = lambda: ConnectRegistry("abcdefghijkl", HKEY_CURRENT_USER)
|
||||
self.assertRaises(WindowsError, connect)
|
||||
|
||||
def test_expand_environment_strings(self):
|
||||
r = ExpandEnvironmentStrings(u"%windir%\\test")
|
||||
self.assertEqual(type(r), unicode)
|
||||
self.assertEqual(r, os.environ["windir"] + "\\test")
|
||||
|
||||
def test_context_manager(self):
|
||||
# ensure that the handle is closed if an exception occurs
|
||||
try:
|
||||
with ConnectRegistry(None, HKEY_LOCAL_MACHINE) as h:
|
||||
self.assertNotEqual(h.handle, 0)
|
||||
raise WindowsError
|
||||
except WindowsError:
|
||||
self.assertEqual(h.handle, 0)
|
||||
|
||||
def test_changing_value(self):
|
||||
# Issue2810: A race condition in 2.6 and 3.1 may cause
|
||||
# EnumValue or QueryValue to throw "WindowsError: More data is
|
||||
# available"
|
||||
done = False
|
||||
|
||||
class VeryActiveThread(threading.Thread):
|
||||
def run(self):
|
||||
with CreateKey(HKEY_CURRENT_USER, test_key_name) as key:
|
||||
use_short = True
|
||||
long_string = 'x'*2000
|
||||
while not done:
|
||||
s = 'x' if use_short else long_string
|
||||
use_short = not use_short
|
||||
SetValue(key, 'changing_value', REG_SZ, s)
|
||||
|
||||
thread = VeryActiveThread()
|
||||
thread.start()
|
||||
try:
|
||||
with CreateKey(HKEY_CURRENT_USER,
|
||||
test_key_name+'\\changing_value') as key:
|
||||
for _ in range(1000):
|
||||
num_subkeys, num_values, t = QueryInfoKey(key)
|
||||
for i in range(num_values):
|
||||
name = EnumValue(key, i)
|
||||
QueryValue(key, name[0])
|
||||
finally:
|
||||
done = True
|
||||
thread.join()
|
||||
with OpenKey(HKEY_CURRENT_USER, test_key_name, 0, KEY_ALL_ACCESS) as key:
|
||||
DeleteKey(key, 'changing_value')
|
||||
DeleteKey(HKEY_CURRENT_USER, test_key_name)
|
||||
|
||||
def test_long_key(self):
|
||||
# Issue2810, in 2.6 and 3.1 when the key name was exactly 256
|
||||
# characters, EnumKey threw "WindowsError: More data is
|
||||
# available"
|
||||
name = 'x'*256
|
||||
try:
|
||||
with CreateKey(HKEY_CURRENT_USER, test_key_name) as key:
|
||||
SetValue(key, name, REG_SZ, 'x')
|
||||
num_subkeys, num_values, t = QueryInfoKey(key)
|
||||
EnumKey(key, 0)
|
||||
finally:
|
||||
with OpenKey(HKEY_CURRENT_USER, test_key_name, 0, KEY_ALL_ACCESS) as key:
|
||||
DeleteKey(key, name)
|
||||
DeleteKey(HKEY_CURRENT_USER, test_key_name)
|
||||
|
||||
def test_dynamic_key(self):
|
||||
# Issue2810, when the value is dynamically generated, these
|
||||
# throw "WindowsError: More data is available" in 2.6 and 3.1
|
||||
EnumValue(HKEY_PERFORMANCE_DATA, 0)
|
||||
QueryValueEx(HKEY_PERFORMANCE_DATA, None)
|
||||
|
||||
# Reflection requires XP x64/Vista at a minimum. XP doesn't have this stuff
|
||||
# or DeleteKeyEx so make sure their use raises NotImplementedError
|
||||
@unittest.skipUnless(WIN_VER < (5, 2), "Requires Windows XP")
|
||||
def test_reflection_unsupported(self):
|
||||
try:
|
||||
with CreateKey(HKEY_CURRENT_USER, test_key_name) as ck:
|
||||
self.assertNotEqual(ck.handle, 0)
|
||||
|
||||
key = OpenKey(HKEY_CURRENT_USER, test_key_name)
|
||||
self.assertNotEqual(key.handle, 0)
|
||||
|
||||
with self.assertRaises(NotImplementedError):
|
||||
DisableReflectionKey(key)
|
||||
with self.assertRaises(NotImplementedError):
|
||||
EnableReflectionKey(key)
|
||||
with self.assertRaises(NotImplementedError):
|
||||
QueryReflectionKey(key)
|
||||
with self.assertRaises(NotImplementedError):
|
||||
DeleteKeyEx(HKEY_CURRENT_USER, test_key_name)
|
||||
finally:
|
||||
DeleteKey(HKEY_CURRENT_USER, test_key_name)
|
||||
|
||||
|
||||
@unittest.skipUnless(REMOTE_NAME, "Skipping remote registry tests")
|
||||
class RemoteWinregTests(BaseWinregTests):
|
||||
|
||||
def test_remote_registry_works(self):
|
||||
remote_key = ConnectRegistry(REMOTE_NAME, HKEY_CURRENT_USER)
|
||||
self._test_all(remote_key)
|
||||
|
||||
|
||||
@unittest.skipUnless(WIN64_MACHINE, "x64 specific registry tests")
|
||||
class Win64WinregTests(BaseWinregTests):
|
||||
|
||||
def test_reflection_functions(self):
|
||||
# Test that we can call the query, enable, and disable functions
|
||||
# on a key which isn't on the reflection list with no consequences.
|
||||
with OpenKey(HKEY_LOCAL_MACHINE, "Software") as key:
|
||||
# HKLM\Software is redirected but not reflected in all OSes
|
||||
self.assertTrue(QueryReflectionKey(key))
|
||||
self.assertEqual(None, EnableReflectionKey(key))
|
||||
self.assertEqual(None, DisableReflectionKey(key))
|
||||
self.assertTrue(QueryReflectionKey(key))
|
||||
|
||||
@unittest.skipUnless(HAS_REFLECTION, "OS doesn't support reflection")
|
||||
def test_reflection(self):
|
||||
# Test that we can create, open, and delete keys in the 32-bit
|
||||
# area. Because we are doing this in a key which gets reflected,
|
||||
# test the differences of 32 and 64-bit keys before and after the
|
||||
# reflection occurs (ie. when the created key is closed).
|
||||
try:
|
||||
with CreateKeyEx(HKEY_CURRENT_USER, test_reflect_key_name, 0,
|
||||
KEY_ALL_ACCESS | KEY_WOW64_32KEY) as created_key:
|
||||
self.assertNotEqual(created_key.handle, 0)
|
||||
|
||||
# The key should now be available in the 32-bit area
|
||||
with OpenKey(HKEY_CURRENT_USER, test_reflect_key_name, 0,
|
||||
KEY_ALL_ACCESS | KEY_WOW64_32KEY) as key:
|
||||
self.assertNotEqual(key.handle, 0)
|
||||
|
||||
# Write a value to what currently is only in the 32-bit area
|
||||
SetValueEx(created_key, "", 0, REG_SZ, "32KEY")
|
||||
|
||||
# The key is not reflected until created_key is closed.
|
||||
# The 64-bit version of the key should not be available yet.
|
||||
open_fail = lambda: OpenKey(HKEY_CURRENT_USER,
|
||||
test_reflect_key_name, 0,
|
||||
KEY_READ | KEY_WOW64_64KEY)
|
||||
self.assertRaises(WindowsError, open_fail)
|
||||
|
||||
# Now explicitly open the 64-bit version of the key
|
||||
with OpenKey(HKEY_CURRENT_USER, test_reflect_key_name, 0,
|
||||
KEY_ALL_ACCESS | KEY_WOW64_64KEY) as key:
|
||||
self.assertNotEqual(key.handle, 0)
|
||||
# Make sure the original value we set is there
|
||||
self.assertEqual("32KEY", QueryValue(key, ""))
|
||||
# Set a new value, which will get reflected to 32-bit
|
||||
SetValueEx(key, "", 0, REG_SZ, "64KEY")
|
||||
|
||||
# Reflection uses a "last-writer wins policy, so the value we set
|
||||
# on the 64-bit key should be the same on 32-bit
|
||||
with OpenKey(HKEY_CURRENT_USER, test_reflect_key_name, 0,
|
||||
KEY_READ | KEY_WOW64_32KEY) as key:
|
||||
self.assertEqual("64KEY", QueryValue(key, ""))
|
||||
finally:
|
||||
DeleteKeyEx(HKEY_CURRENT_USER, test_reflect_key_name,
|
||||
KEY_WOW64_32KEY, 0)
|
||||
|
||||
@unittest.skipUnless(HAS_REFLECTION, "OS doesn't support reflection")
|
||||
def test_disable_reflection(self):
|
||||
# Make use of a key which gets redirected and reflected
|
||||
try:
|
||||
with CreateKeyEx(HKEY_CURRENT_USER, test_reflect_key_name, 0,
|
||||
KEY_ALL_ACCESS | KEY_WOW64_32KEY) as created_key:
|
||||
# QueryReflectionKey returns whether or not the key is disabled
|
||||
disabled = QueryReflectionKey(created_key)
|
||||
self.assertEqual(type(disabled), bool)
|
||||
# HKCU\Software\Classes is reflected by default
|
||||
self.assertFalse(disabled)
|
||||
|
||||
DisableReflectionKey(created_key)
|
||||
self.assertTrue(QueryReflectionKey(created_key))
|
||||
|
||||
# The key is now closed and would normally be reflected to the
|
||||
# 64-bit area, but let's make sure that didn't happen.
|
||||
open_fail = lambda: OpenKeyEx(HKEY_CURRENT_USER,
|
||||
test_reflect_key_name, 0,
|
||||
KEY_READ | KEY_WOW64_64KEY)
|
||||
self.assertRaises(WindowsError, open_fail)
|
||||
|
||||
# Make sure the 32-bit key is actually there
|
||||
with OpenKeyEx(HKEY_CURRENT_USER, test_reflect_key_name, 0,
|
||||
KEY_READ | KEY_WOW64_32KEY) as key:
|
||||
self.assertNotEqual(key.handle, 0)
|
||||
finally:
|
||||
DeleteKeyEx(HKEY_CURRENT_USER, test_reflect_key_name,
|
||||
KEY_WOW64_32KEY, 0)
|
||||
|
||||
|
||||
def test_main():
|
||||
test_support.run_unittest(LocalWinregTests, RemoteWinregTests,
|
||||
Win64WinregTests)
|
||||
|
||||
if __name__ == "__main__":
|
||||
if not REMOTE_NAME:
|
||||
print "Remote registry calls can be tested using",
|
||||
print "'test_winreg.py --remote \\\\machine_name'"
|
||||
test_main()
|
||||
Reference in New Issue
Block a user