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'] )