def testMultiAdd(self, winreg, build_info):
# Mock add registry keys
kpath = (r'SOFTWARE\Microsoft\Windows NT\CurrentVersion'
r'\SoftwareProtectionPlatform')
args = [
'HKLM', kpath, 'KeyManagementServiceName',
'kms-server.example.com', 'REG_SZ', False
]
skv = winreg.return_value.SetKeyValue
ra = registry.RegAdd(args, build_info)
ra.Run()
skv.assert_called_with(key_path=kpath,
key_name='KeyManagementServiceName',
key_value='kms-server.example.com',
key_type='REG_SZ',
use_64bit=False)
# Missing arguments
args = [
'HKLM', kpath, 'KeyManagementServiceName', 'kms-server.example.com'
]
ra = registry.RegAdd(args, build_info)
self.assertRaises(registry.ActionError, ra.Run)
# Multiple missing arguments
args = [[
'HKLM', kpath, 'KeyManagementServiceName',
'kms-server.example.com', 'REG_SZ'
], [
'HKLM', kpath, 'KeyManagementServiceName', 'kms-server.example.com'
]]
ra = registry.MultiRegAdd(args, build_info)
self.assertRaises(registry.ActionError, ra.Run)
def testMultiAdd(self, sv, build_info):
ra = registry.RegAdd(ARGS, build_info)
ra.Run()
sv.assert_called_with(NAME, VALUE, ROOT, PATH, TYPE, USE_64)
# Missing arguments
args = [ROOT, PATH, NAME, VALUE]
ra = registry.RegAdd(args, build_info)
self.assertRaises(registry.ActionError, ra.Run)
# Multiple missing arguments
args = [ARGS, [ROOT, PATH, NAME, VALUE]]
ra = registry.MultiRegAdd(args, build_info)
self.assertRaises(registry.ActionError, ra.Run)
def testMultiAddValidation(self):
# Valid calls
r = registry.MultiRegAdd(
[['HKLM', 'SOFTWARE/fake', 'foo', 'bazzz', 'REG_SZ'],
['HKLM', 'SOFTWARE/boo', 'fake', 100, 'REG_DWORD']], None)
r.Validate()
def testMultiAddValidation(self): # Valid calls r = registry.MultiRegAdd([ARGS, [ROOT, PATH, NAME, 100, 'REG_DWORD']], None) r.Validate()