def test_data2group(self):
right = [
["10 mbox.example.com"],
[
"10 mbox1.example.com",
"20 mbox2.example.com",
"20 mbox3.example.com",
"30 mbox4.example.com",
"30 mbox5.example.com",
"30 mbox6.example.com",
],
]
rschema = OrderedDict((("prio", int), ("srvr", str),))
results = [
OrderedDict([(10, ["mbox.example.com"])]),
OrderedDict(
[
(10, ["mbox1.example.com"]),
(20, ["mbox2.example.com", "mbox3.example.com"]),
(
30,
["mbox4.example.com", "mbox5.example.com", "mbox6.example.com"],
),
]
),
]
for rdata, res in zip(right, results):
group = _data2rec_group(rschema, rdata, "prio")
self.assertEqual(group, res)
def test_data2group(self):
right = [
['10 mbox.example.com'],
[
'10 mbox1.example.com',
'20 mbox2.example.com',
'20 mbox3.example.com',
'30 mbox4.example.com',
'30 mbox5.example.com',
'30 mbox6.example.com',
],
]
rschema = OrderedDict((
('prio', int),
('srvr', str),
))
results = [
OrderedDict([(10, [{'srvr': 'mbox.example.com'}])]),
OrderedDict([
(10, [{'srvr': 'mbox1.example.com'}]),
(20, [{'srvr': 'mbox2.example.com'}, {'srvr': 'mbox3.example.com'}]),
(30, [{'srvr': 'mbox4.example.com'}, {'srvr': 'mbox5.example.com'}, {'srvr': 'mbox6.example.com'}])]
),
]
for rdata, res in zip(right, results):
group = _data2rec_group(rschema, rdata, 'prio')
self.assertEqual(group, res)
