Patch #1006: port test_winreg to unittest.
This commit is contained in:
Georg Brandl
parent
94bda3a586
commit
3376a9a711
|
|
@ -1,3 +0,0 @@
|
|||
test_winreg
|
||||
Local registry tests worked
|
||||
Remote registry calls can be tested using 'test_winreg.py --remote \\machine_name'
|
||||
|
|
@ -3,8 +3,9 @@
|
|||
|
||||
from _winreg import *
|
||||
import os, sys
|
||||
import unittest
|
||||
|
||||
from test.test_support import verify, have_unicode
|
||||
from test import test_support
|
||||
|
||||
test_key_name = "SOFTWARE\\Python Registry Test Key - Delete Me"
|
||||
|
||||
|
|
@ -17,140 +18,158 @@ test_data = [
|
|||
("Big String", "x"*(2**14-1), REG_SZ),
|
||||
("Big Binary", "x"*(2**14), REG_BINARY),
|
||||
]
|
||||
if 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),
|
||||
|
||||
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),
|
||||
]
|
||||
|
||||
def WriteTestData(root_key):
|
||||
# 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)
|
||||
verify(nkeys==1, "Not the correct number of sub keys")
|
||||
verify(nvalues==1, "Not the correct number of values")
|
||||
nkeys, nvalues, since_mod = QueryInfoKey(sub_key)
|
||||
verify(nkeys==0, "Not the correct number of sub keys")
|
||||
verify(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)
|
||||
raise RuntimeError, "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)
|
||||
raise RuntimeError, "It appears the key.Close() function does not close the actual key!"
|
||||
except EnvironmentError:
|
||||
pass
|
||||
|
||||
def ReadTestData(root_key):
|
||||
# Check we can get default value for this key.
|
||||
val = QueryValue(root_key, test_key_name)
|
||||
verify(val=="Default value", "Registry didn't give back the correct value")
|
||||
|
||||
key = OpenKey(root_key, test_key_name)
|
||||
# Read the sub-keys
|
||||
sub_key = OpenKey(key, "sub_key")
|
||||
# Check I can enumerate over the values.
|
||||
index = 0
|
||||
while 1:
|
||||
try:
|
||||
data = EnumValue(sub_key, index)
|
||||
except EnvironmentError:
|
||||
break
|
||||
verify(data in test_data, "Didn't read back the correct test data")
|
||||
index = index + 1
|
||||
verify(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)
|
||||
verify(read_val==value_data and read_typ == value_type, \
|
||||
"Could not directly read the value" )
|
||||
sub_key.Close()
|
||||
# Enumerate our main key.
|
||||
read_val = EnumKey(key, 0)
|
||||
verify(read_val == "sub_key", "Read subkey value wrong")
|
||||
try:
|
||||
EnumKey(key, 1)
|
||||
verify(0, "Was able to get a second key when I only have one!")
|
||||
except EnvironmentError:
|
||||
pass
|
||||
|
||||
key.Close()
|
||||
|
||||
def DeleteTestData(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)
|
||||
verify(nkeys==0 and 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")
|
||||
verify(0, "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)
|
||||
verify(0, "Could open the non-existent key")
|
||||
except WindowsError: # Use this error name this time
|
||||
pass
|
||||
|
||||
def TestAll(root_key):
|
||||
WriteTestData(root_key)
|
||||
ReadTestData(root_key)
|
||||
DeleteTestData(root_key)
|
||||
|
||||
# Test on my local machine.
|
||||
TestAll(HKEY_CURRENT_USER)
|
||||
print "Local registry tests worked"
|
||||
try:
|
||||
remote_name = sys.argv[sys.argv.index("--remote")+1]
|
||||
except (IndexError, ValueError):
|
||||
class WinregTests(unittest.TestCase):
|
||||
remote_name = None
|
||||
|
||||
if remote_name is not None:
|
||||
def WriteTestData(self, root_key):
|
||||
# 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.assertEquals(nkeys, 1, "Not the correct number of sub keys")
|
||||
self.assertEquals(nvalues, 1, "Not the correct number of values")
|
||||
nkeys, nvalues, since_mod = QueryInfoKey(sub_key)
|
||||
self.assertEquals(nkeys, 0, "Not the correct number of sub keys")
|
||||
self.assertEquals(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 ReadTestData(self, root_key):
|
||||
# Check we can get default value for this key.
|
||||
val = QueryValue(root_key, test_key_name)
|
||||
self.assertEquals(val, "Default value",
|
||||
"Registry didn't give back the correct value")
|
||||
|
||||
key = OpenKey(root_key, test_key_name)
|
||||
# Read the sub-keys
|
||||
sub_key = OpenKey(key, "sub_key")
|
||||
# Check I can enumerate over the values.
|
||||
index = 0
|
||||
while 1:
|
||||
try:
|
||||
data = EnumValue(sub_key, index)
|
||||
except EnvironmentError:
|
||||
break
|
||||
self.assertEquals(data in test_data, True,
|
||||
"Didn't read back the correct test data")
|
||||
index = index + 1
|
||||
self.assertEquals(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.assertEquals(read_val, value_data,
|
||||
"Could not directly read the value")
|
||||
self.assertEquals(read_typ, value_type,
|
||||
"Could not directly read the value")
|
||||
sub_key.Close()
|
||||
# Enumerate our main key.
|
||||
read_val = EnumKey(key, 0)
|
||||
self.assertEquals(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 DeleteTestData(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.assertEquals(nkeys, 0, "subkey not empty before delete")
|
||||
self.assertEquals(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 TestAll(self, root_key):
|
||||
self.WriteTestData(root_key)
|
||||
self.ReadTestData(root_key)
|
||||
self.DeleteTestData(root_key)
|
||||
|
||||
def testLocalMachineRegistryWorks(self):
|
||||
self.TestAll(HKEY_CURRENT_USER)
|
||||
|
||||
def testConnectRegistryToLocalMachineWorks(self):
|
||||
# perform minimal ConnectRegistry test which just invokes it
|
||||
h = ConnectRegistry(None, HKEY_LOCAL_MACHINE)
|
||||
h.Close()
|
||||
|
||||
def testRemoteMachineRegistryWorks(self):
|
||||
if not self.remote_name:
|
||||
raise test_support.TestSkipped("Remote machine name "
|
||||
"not specified.")
|
||||
remote_key = ConnectRegistry(self.remote_name, HKEY_CURRENT_USER)
|
||||
self.TestAll(remote_key)
|
||||
|
||||
def test_main():
|
||||
test_support.run_unittest(WinregTests)
|
||||
|
||||
if __name__ == "__main__":
|
||||
try:
|
||||
remote_key = ConnectRegistry(remote_name, HKEY_CURRENT_USER)
|
||||
except EnvironmentError, exc:
|
||||
print "Could not connect to the remote machine -", exc.strerror
|
||||
remote_key = None
|
||||
if remote_key is not None:
|
||||
TestAll(remote_key)
|
||||
print "Remote registry tests worked"
|
||||
else:
|
||||
print "Remote registry calls can be tested using",
|
||||
print "'test_winreg.py --remote \\\\machine_name'"
|
||||
# perform minimal ConnectRegistry test which just invokes it
|
||||
h = ConnectRegistry(None, HKEY_LOCAL_MACHINE)
|
||||
h.Close()
|
||||
WinregTests.remote_name = sys.argv[sys.argv.index("--remote")+1]
|
||||
except (IndexError, ValueError):
|
||||
print "Remote registry calls can be tested using",
|
||||
print "'test_winreg.py --remote \\\\machine_name'"
|
||||
WinregTests.remote_name = None
|
||||
test_main()
|
||||
|
|
|
|||
Loading…
Reference in New Issue