def setUp(self):
self.tree = RegistryTree()
self.hivefile = REG_EXAMPLE_FILE
self.hivefile_bad = "/does/not/exist.dat"
self.key_path = "/AppEvents/Schemes/Apps/Explorer/"
self.key_name = "Explorer"
self.key_path_bad = "/does/not/exist"
self.value_path = "/AppEvents/Schemes/Apps/Explorer/(default).RegSZ"
self.value_value = u"Windows Explorer"
self.value_bytes = "Windows Explorer\n"
self.value_path_bad = "/does/not/exist.RegSZ"
self.st_key = {}
self.st_key["st_mode"] = 0o40755 # drwxr-xr-x
self.st_key["st_ino"] = 0
self.st_key["st_dev"] = 0
self.st_key["st_nlink"] = 2
self.st_key["st_uid"] = 0
self.st_key["st_gid"] = 0
self.st_key["st_size"] = 0
self.st_key["st_atime"] = 0
self.st_key["st_mtime"] = 1305848118 # Unix epoch time
self.st_key["st_ctime"] = 0
self.st_value = {}
self.st_value["st_mode"] = 0o100644 # -rw-r--r--
self.st_value["st_ino"] = 0
self.st_value["st_dev"] = 0
self.st_value["st_nlink"] = 1
self.st_value["st_uid"] = 0
self.st_value["st_gid"] = 0
self.st_value["st_size"] = len(self.value_bytes)
self.st_value["st_atime"] = 0
self.st_value["st_mtime"] = 0
self.st_value["st_ctime"] = 0
class TestRegistryTree_Basic(unittest.TestCase):
"""Most basic RegistryTree test case."""
def setUp(self):
self.tree = RegistryTree()
self.hivefile = REG_EXAMPLE_FILE
self.hivefile_bad = "/does/not/exist.dat"
self.key_path = "/AppEvents/Schemes/Apps/Explorer/"
self.key_name = "Explorer"
self.key_path_bad = "/does/not/exist"
self.value_path = "/AppEvents/Schemes/Apps/Explorer/(default).RegSZ"
self.value_value = u"Windows Explorer"
self.value_bytes = "Windows Explorer\n"
self.value_path_bad = "/does/not/exist.RegSZ"
self.st_key = {}
self.st_key["st_mode"] = 0o40755 # drwxr-xr-x
self.st_key["st_ino"] = 0
self.st_key["st_dev"] = 0
self.st_key["st_nlink"] = 2
self.st_key["st_uid"] = 0
self.st_key["st_gid"] = 0
self.st_key["st_size"] = 0
self.st_key["st_atime"] = 0
self.st_key["st_mtime"] = 1305848118 # Unix epoch time
self.st_key["st_ctime"] = 0
self.st_value = {}
self.st_value["st_mode"] = 0o100644 # -rw-r--r--
self.st_value["st_ino"] = 0
self.st_value["st_dev"] = 0
self.st_value["st_nlink"] = 1
self.st_value["st_uid"] = 0
self.st_value["st_gid"] = 0
self.st_value["st_size"] = len(self.value_bytes)
self.st_value["st_atime"] = 0
self.st_value["st_mtime"] = 0
self.st_value["st_ctime"] = 0
def test_load(self):
with self.assertRaises(IOError):
self.tree.load(self.hivefile_bad)
self.tree.load(self.hivefile)
def test_key(self):
# Haven't called load() yet
with self.assertRaises(ValueError):
self.tree.key(self.key_path)
self.tree.load(self.hivefile)
# Path doesn't exist
with self.assertRaises(ValueError):
self.tree.key(self.key_path_bad)
# Can't get key on a value
for path in (self.value_path, self.value_path_bad):
with self.assertRaises(ValueError):
self.tree.key(path)
# Test an actual key that should work
key = self.tree.key(self.key_path)
self.assertEqual(key.name(), self.key_name)
def test_value(self):
# Haven't called load() yet
with self.assertRaises(ValueError):
self.tree.value(self.value_path)
self.tree.load(self.hivefile)
# Path doesn't exist
with self.assertRaises(ValueError):
self.tree.value(self.value_path_bad)
# Can't get value on a key
for path in (self.key_path, self.key_path_bad):
with self.assertRaises(ValueError):
self.tree.value(path)
# Test an actual value that should work
value = self.tree.value(self.value_path)
self.assertEqual(value.value(), self.value_value)
def test_items(self):
# Haven't called load() yet
with self.assertRaises(ValueError):
self.tree.items(self.key_path)
self.tree.load(self.hivefile)
# Path doesn't exist
with self.assertRaises(ValueError):
self.tree.items(self.key_path_bad)
# Can't list items under a value
for path in (self.value_path, self.value_path_bad):
with self.assertRaises(ValueError):
self.tree.items(path)
items = self.tree.items(self.key_path)
# TODO: check list of items
def test_bytestr(self):
self.tree.load(self.hivefile)
data = self.tree.bytestr(self.value_path)
self.assertEqual(data, self.value_bytes)
# TODO also try other data types
def test_stat(self):
# Haven't called load() yet
with self.assertRaises(ValueError):
self.tree.stat(self.key_path)
self.tree.load(self.hivefile)
# Path doesn't exist
with self.assertRaises(ValueError):
self.tree.stat(self.key_path_bad)
with self.assertRaises(ValueError):
self.tree.stat(self.value_path_bad)
# Test an actual key that should work
st_key = self.tree.stat(self.key_path)
self.assertEqual(st_key, self.st_key)
with self.assertRaises(KeyError):
st_key["does_not_exist"]
# Test an actual value that should work
st_value = self.tree.stat(self.value_path)
self.assertEqual(st_value, self.st_value)
with self.assertRaises(KeyError):
st_value["does_not_exist"]
