def test_cast_vdata_reg_expand_sz(self):
'''
Test the cast_vdata function with REG_EXPAND_SZ
Should always return unicode
'''
vdata = salt.utils.stringutils.to_str('test data')
result = win_reg.cast_vdata(vdata=vdata, vtype='REG_EXPAND_SZ')
self.assertTrue(isinstance(result, six.text_type))
vdata = salt.utils.stringutils.to_bytes('test data')
result = win_reg.cast_vdata(vdata=vdata, vtype='REG_EXPAND_SZ')
self.assertTrue(isinstance(result, six.text_type))
def test_cast_vdata_reg_sz(self):
"""
Test the cast_vdata function with REG_SZ
Should always return unicode
"""
vdata = salt.utils.stringutils.to_str("test data")
result = win_reg.cast_vdata(vdata=vdata, vtype="REG_SZ")
self.assertTrue(isinstance(result, six.text_type))
vdata = salt.utils.stringutils.to_bytes("test data")
result = win_reg.cast_vdata(vdata=vdata, vtype="REG_SZ")
self.assertTrue(isinstance(result, six.text_type))
def test_cast_vdata_reg_dword(self):
'''
Test the cast_vdata function with REG_DWORD
Should always return integer
'''
vdata = 1
result = win_reg.cast_vdata(vdata=vdata, vtype='REG_DWORD')
self.assertTrue(isinstance(result, six.integer_types))
vdata = '1'
result = win_reg.cast_vdata(vdata=vdata, vtype='REG_DWORD')
self.assertTrue(isinstance(result, six.integer_types))
def test_cast_vdata_reg_qword(self):
"""
Test the cast_vdata function with REG_QWORD
Should always return a long integer
`int` is `long` is default on Py3
"""
vdata = 1
result = win_reg.cast_vdata(vdata=vdata, vtype="REG_QWORD")
self.assertTrue(isinstance(result, int))
vdata = "1"
result = win_reg.cast_vdata(vdata=vdata, vtype="REG_QWORD")
self.assertTrue(isinstance(result, int))
def test_cast_vdata_reg_binary(self):
"""
Test the cast_vdata function with REG_BINARY
Should always return binary data
"""
vdata = salt.utils.stringutils.to_bytes("test data")
result = win_reg.cast_vdata(vdata=vdata, vtype="REG_BINARY")
self.assertTrue(isinstance(result, six.binary_type))
vdata = salt.utils.stringutils.to_str("test data")
result = win_reg.cast_vdata(vdata=vdata, vtype="REG_BINARY")
self.assertTrue(isinstance(result, six.binary_type))
vdata = salt.utils.stringutils.to_unicode("test data")
result = win_reg.cast_vdata(vdata=vdata, vtype="REG_BINARY")
self.assertTrue(isinstance(result, six.binary_type))
def test_cast_vdata_reg_dword(self):
"""
Test the cast_vdata function with REG_DWORD
Should always return integer
"""
vdata = 1
expected = 1
result = win_reg.cast_vdata(vdata=vdata, vtype="REG_DWORD")
self.assertEqual(result, expected)
vdata = "1"
result = win_reg.cast_vdata(vdata=vdata, vtype="REG_DWORD")
self.assertEqual(result, expected)
vdata = "0000001"
result = win_reg.cast_vdata(vdata=vdata, vtype="REG_DWORD")
self.assertEqual(result, expected)
def test_cast_vdata_reg_dword(self):
'''
Test the cast_vdata function with REG_DWORD
Should always return integer
'''
vdata = 1
expected = 1
result = win_reg.cast_vdata(vdata=vdata, vtype='REG_DWORD')
self.assertEqual(result, expected)
vdata = '1'
result = win_reg.cast_vdata(vdata=vdata, vtype='REG_DWORD')
self.assertEqual(result, expected)
vdata = '0000001'
result = win_reg.cast_vdata(vdata=vdata, vtype='REG_DWORD')
self.assertEqual(result, expected)
def test_cast_vdata_reg_qword(self):
'''
Test the cast_vdata function with REG_QWORD
Should always return a long integer
`int` is `long` is default on Py3
'''
vdata = 1
result = win_reg.cast_vdata(vdata=vdata, vtype='REG_QWORD')
if six.PY2:
self.assertTrue(isinstance(result, long)) # pylint: disable=incompatible-py3-code,undefined-variable
else:
self.assertTrue(isinstance(result, int))
vdata = '1'
result = win_reg.cast_vdata(vdata=vdata, vtype='REG_QWORD')
if six.PY2:
self.assertTrue(isinstance(result, long)) # pylint: disable=incompatible-py3-code,undefined-variable
else:
self.assertTrue(isinstance(result, int))
def test_cast_vdata_reg_multi_sz(self):
"""
Test the cast_vdata function with REG_MULTI_SZ
Should always return a list of unicode strings
"""
vdata = [
salt.utils.stringutils.to_str("test string"),
salt.utils.stringutils.to_bytes("test bytes"),
]
result = win_reg.cast_vdata(vdata=vdata, vtype="REG_MULTI_SZ")
self.assertTrue(isinstance(result, list))
for item in result:
self.assertTrue(isinstance(item, six.text_type))
