sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2024-12-22 13:34:15 +03:00
Code

gp_pol: Test multiple values multi_sz roundtrip

Browse Source 
Signed-off-by: Gabriel Nagy <gabriel.nagy@canonical.com>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
Reviewed-by: David Mulder <dmulder@samba.org>

Autobuild-User(master): David Mulder <dmulder@samba.org>
Autobuild-Date(master): Fri Oct 27 14:45:28 UTC 2023 on atb-devel-224
This commit is contained in:
Gabriel Nagy 2023-10-27 11:21:50 +03:00 committed by David Mulder
parent d5d96bed02
commit 03af117693
1 changed files with 33 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
python/samba/tests/gpo.py
View File

@ -5027,6 +5027,20 @@ br"""
</PolFile>
"""
multiple_values_multi_sz_reg_pol = \
br"""
<?xml version="1.0" encoding="utf-8"?>
<PolFile num_entries="1" signature="PReg" version="1">
<Entry type="7" type_name="REG_MULTI_SZ">
<Key>KeyName</Key>
<ValueName>ValueName</ValueName>
<Value>Value1</Value>
<Value>Value2</Value>
<Value>Value3</Value>
</Entry>
</PolFile>
"""
def days2rel_nttime(val):
seconds = 60
minutes = 60
@ -8046,3 +8060,22 @@ class GPOTests(tests.TestCase):
expected_xml_data = re.sub(r"\s+", "", empty_multi_sz_reg_pol.decode(), flags=re.UNICODE)
actual_xml_data = re.sub(r"\s+", "", pol_xml_data, flags=re.UNICODE)
self.assertEqual(expected_xml_data, actual_xml_data, 'XML data mismatch')
def test_parser_roundtrip_multiple_values_multi_sz(self):
with TemporaryDirectory() as dname:
reg_pol_xml = os.path.join(dname, 'REGISTRY.POL.XML')
parser = GPPolParser()
try:
parser.load_xml(etree.fromstring(multiple_values_multi_sz_reg_pol.strip()))
except Exception as e:
self.fail(str(e))
parser.write_xml(reg_pol_xml)
with open(reg_pol_xml, 'r') as f:
pol_xml_data = f.read()
# Strip whitespace characters due to indentation differences
expected_xml_data = re.sub(r"\s+", "", multiple_values_multi_sz_reg_pol.decode(), flags=re.UNICODE)
actual_xml_data = re.sub(r"\s+", "", pol_xml_data, flags=re.UNICODE)
self.assertEqual(expected_xml_data, actual_xml_data, 'XML data mismatch')