def test_a(self):
dig_mock = MagicMock(
return_value={
"pid": 3656,
"retcode": 0,
"stderr": "",
"stdout": (
"74.125.193.104\n"
"74.125.193.105\n"
"74.125.193.99\n"
"74.125.193.106\n"
"74.125.193.103\n"
"74.125.193.147"
),
}
)
with patch.dict(dig.__salt__, {"cmd.run_all": dig_mock}):
self.assertEqual(
dig.A("www.google.com"),
[
"74.125.193.104",
"74.125.193.105",
"74.125.193.99",
"74.125.193.106",
"74.125.193.103",
"74.125.193.147",
],
)
def test_a(self):
dig.__salt__ = {}
dig_mock = MagicMock(
return_value={
'pid': 3656,
'retcode': 0,
'stderr': '',
'stdout': '74.125.193.104\n'
'74.125.193.105\n'
'74.125.193.99\n'
'74.125.193.106\n'
'74.125.193.103\n'
'74.125.193.147'
}
)
with patch.dict(dig.__salt__, {'cmd.run_all': dig_mock}):
self.assertEqual(
dig.A('www.google.com'),
['74.125.193.104',
'74.125.193.105',
'74.125.193.99',
'74.125.193.106',
'74.125.193.103',
'74.125.193.147']
)
