def test_mx(self):
dig_mock = MagicMock(
return_value={
"pid": 27780,
"retcode": 0,
"stderr": "",
"stdout": (
"10 aspmx.l.google.com.\n"
"20 alt1.aspmx.l.google.com.\n"
"40 alt3.aspmx.l.google.com.\n"
"50 alt4.aspmx.l.google.com.\n"
"30 alt2.aspmx.l.google.com."
),
}
)
with patch.dict(dig.__salt__, {"cmd.run_all": dig_mock}):
self.assertEqual(
dig.MX("google.com"),
[
["10", "aspmx.l.google.com."],
["20", "alt1.aspmx.l.google.com."],
["40", "alt3.aspmx.l.google.com."],
["50", "alt4.aspmx.l.google.com."],
["30", "alt2.aspmx.l.google.com."],
],
)
def test_mx(self):
dig_mock = MagicMock(
return_value={
'pid': 27780,
'retcode': 0,
'stderr': '',
'stdout': '10 aspmx.l.google.com.\n'
'20 alt1.aspmx.l.google.com.\n'
'40 alt3.aspmx.l.google.com.\n'
'50 alt4.aspmx.l.google.com.\n'
'30 alt2.aspmx.l.google.com.'
}
)
with patch.dict(dig.__salt__, {'cmd.run_all': dig_mock}):
self.assertEqual(
dig.MX('google.com'),
[['10', 'aspmx.l.google.com.'],
['20', 'alt1.aspmx.l.google.com.'],
['40', 'alt3.aspmx.l.google.com.'],
['50', 'alt4.aspmx.l.google.com.'],
['30', 'alt2.aspmx.l.google.com.']]
)
