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)