def test_delete_key_recursive(self):
'''
Test the delete_key_recursive function
'''
try:
self.assertTrue(
win_reg.set_value(hive='HKLM',
key=FAKE_KEY,
vname='fake_name',
vdata='fake_value'))
expected = {
'Deleted': ['\\'.join(['HKLM', FAKE_KEY])],
'Failed': []
}
self.assertDictEqual(
win_reg.delete_key_recursive(hive='HKLM', key=FAKE_KEY),
expected)
finally:
win_reg.delete_key_recursive(hive='HKLM', key=FAKE_KEY)
def test_delete_value_unicode_key(self):
"""
Test the delete_value function on a unicode key
"""
try:
self.assertTrue(
win_reg.set_value(
hive="HKLM",
key="\\".join([FAKE_KEY, UNICODE_KEY]),
vname="fake_name",
vdata="junk data",
))
self.assertTrue(
win_reg.delete_value(
hive="HKLM",
key="\\".join([FAKE_KEY, UNICODE_KEY]),
vname="fake_name",
))
finally:
win_reg.delete_key_recursive(hive="HKLM", key=FAKE_KEY)
def test_pkg__get_reg_software_noremove_not_present():
search = "test_pkg_noremove_not_present"
key = "SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall\\{}".format(
search)
win_reg.set_value(hive="HKLM", key=key, vname="DisplayName", vdata=search)
win_reg.set_value(hive="HKLM",
key=key,
vname="DisplayVersion",
vdata="1.0.0")
try:
result = win_pkg._get_reg_software()
assert isinstance(result, dict)
found = False
for item in result:
if search in item:
found = True
assert found is False
finally:
win_reg.delete_key_recursive(hive="HKLM", key=key)
assert not win_reg.key_exists(hive="HKLM", key=key)
def test_set_value_default(self):
"""
Test the set_value function on the default value
"""
try:
self.assertTrue(
win_reg.set_value(hive="HKLM",
key=FAKE_KEY,
vdata="fake_default_data"))
expected = {
"hive": "HKLM",
"key": FAKE_KEY,
"success": True,
"vdata": "fake_default_data",
"vname": "(Default)",
"vtype": "REG_SZ",
}
self.assertEqual(win_reg.read_value(hive="HKLM", key=FAKE_KEY),
expected)
finally:
win_reg.delete_key_recursive(hive="HKLM", key=FAKE_KEY)
def test_delete_key_unicode(self):
'''
Test the delete_value function on value within a unicode key
'''
try:
self.assertTrue(
win_reg.set_value(hive='HKLM',
key='{0}\\{1}'.format(FAKE_KEY, UNICODE_KEY),
vname='fake_name',
vdata='fake_value'))
expected = {
'Deleted': ['HKLM\\{0}\\{1}\\'.format(FAKE_KEY, UNICODE_KEY)],
'Failed': []
}
self.assertEqual(
win_reg.delete_key_recursive(
hive='HKLM',
key='{0}\\{1}\\'.format(FAKE_KEY, UNICODE_KEY),
), expected)
finally:
win_reg.delete_key_recursive(hive='HKLM', key=FAKE_KEY)
def test_delete_key_recursive_failed_to_delete(self):
'''
Test the delete_key_recursive function on failure to delete a key
'''
try:
self.assertTrue(
win_reg.set_value(hive='HKLM',
key=FAKE_KEY,
vname='fake_name',
vdata='fake_value'))
expected = {
'Deleted': [],
'Failed': ['\\'.join(['HKLM', FAKE_KEY]) + ' Unknown error']
}
mock_error = MagicMock(side_effect=WindowsError('Unknown error')) # pylint: disable=undefined-variable
with patch('salt.utils.win_reg.win32api.RegDeleteKey', mock_error):
self.assertDictEqual(
win_reg.delete_key_recursive(hive='HKLM', key=FAKE_KEY),
expected)
finally:
win_reg.delete_key_recursive(hive='HKLM', key=FAKE_KEY)
def test_set_value_default(self):
'''
Test the set_value function on the default value
'''
try:
self.assertTrue(
win_reg.set_value(hive='HKLM',
key=FAKE_KEY,
vdata='fake_default_data'))
expected = {
'hive': 'HKLM',
'key': FAKE_KEY,
'success': True,
'vdata': 'fake_default_data',
'vname': '(Default)',
'vtype': 'REG_SZ'
}
self.assertEqual(win_reg.read_value(hive='HKLM', key=FAKE_KEY),
expected)
finally:
win_reg.delete_key_recursive(hive='HKLM', key=FAKE_KEY)
def test_delete_value_unicode(self):
'''
Test the delete_value function on a unicode value
'''
try:
self.assertTrue(
win_reg.set_value(
hive='HKLM',
key=FAKE_KEY,
vname='fake_unicode',
vdata=UNICODE_VALUE
)
)
self.assertTrue(
win_reg.delete_value(
hive='HKLM',
key=FAKE_KEY,
vname='fake_unicode'
)
)
finally:
win_reg.delete_key_recursive(hive='HKLM', key=FAKE_KEY)
def test_delete_value(self):
'''
Test the delete_value function
'''
try:
self.assertTrue(
win_reg.set_value(
hive='HKLM',
key=FAKE_KEY,
vname='fake_name',
vdata='fake_data'
)
)
self.assertTrue(
win_reg.delete_value(
hive='HKLM',
key=FAKE_KEY,
vname='fake_name'
)
)
finally:
win_reg.delete_key_recursive(hive='HKLM', key=FAKE_KEY)
def test_delete_key_recursive_failed_to_delete(self):
"""
Test the delete_key_recursive function on failure to delete a key
"""
try:
self.assertTrue(
win_reg.set_value(hive="HKLM",
key=FAKE_KEY,
vname="fake_name",
vdata="fake_value"))
expected = {
"Deleted": [],
"Failed": ["\\".join(["HKLM", FAKE_KEY]) + " Unknown error"],
}
# pylint: disable=undefined-variable
mock_error = MagicMock(side_effect=WindowsError("Unknown error"))
# pylint: enable=undefined-variable
with patch("salt.utils.win_reg.win32api.RegDeleteKey", mock_error):
self.assertDictEqual(
win_reg.delete_key_recursive(hive="HKLM", key=FAKE_KEY),
expected)
finally:
win_reg.delete_key_recursive(hive="HKLM", key=FAKE_KEY)
def test_delete_key_recursive_unicode(self):
"""
Test the delete_key_recursive function on value within a unicode key
"""
try:
self.assertTrue(
win_reg.set_value(
hive="HKLM",
key="\\".join([FAKE_KEY, UNICODE_KEY]),
vname="fake_name",
vdata="fake_value",
))
expected = {
"Deleted": ["\\".join(["HKLM", FAKE_KEY, UNICODE_KEY])],
"Failed": [],
}
self.assertDictEqual(
win_reg.delete_key_recursive(hive="HKLM",
key="\\".join(
[FAKE_KEY, UNICODE_KEY])),
expected,
)
finally:
win_reg.delete_key_recursive(hive="HKLM", key=FAKE_KEY)
def test_set_value_unicode_value(self):
'''
Test the set_value function on a unicode value
'''
try:
self.assertTrue(
win_reg.set_value(hive='HKLM',
key=FAKE_KEY,
vname='fake_unicode',
vdata=UNICODE_VALUE))
expected = {
'hive': 'HKLM',
'key': FAKE_KEY,
'success': True,
'vdata': UNICODE_VALUE,
'vname': 'fake_unicode',
'vtype': 'REG_SZ'
}
self.assertEqual(
win_reg.read_value(hive='HKLM',
key=FAKE_KEY,
vname='fake_unicode'), expected)
finally:
win_reg.delete_key_recursive(hive='HKLM', key=FAKE_KEY)
def test_set_value_unicode_value(self):
"""
Test the set_value function on a unicode value
"""
try:
self.assertTrue(
win_reg.set_value(
hive="HKLM", key=FAKE_KEY, vname="fake_unicode", vdata=UNICODE_VALUE
)
)
expected = {
"hive": "HKLM",
"key": FAKE_KEY,
"success": True,
"vdata": UNICODE_VALUE,
"vname": "fake_unicode",
"vtype": "REG_SZ",
}
self.assertEqual(
win_reg.read_value(hive="HKLM", key=FAKE_KEY, vname="fake_unicode"),
expected,
)
finally:
win_reg.delete_key_recursive(hive="HKLM", key=FAKE_KEY)
def tearDown(self):
win_reg.delete_key_recursive(hive="HKLM", key=FAKE_KEY)
def tearDown(self):
reg.delete_key_recursive(hive="HKLM", key=FAKE_KEY)
reg.delete_key_recursive(hive="HKLM",
key=FAKE_KEY,
use_32bit_registry=True)
