modules.network: update unit test

This commit is contained in:
Justin Findlay 2019-04-02 20:43:03 -07:00
parent 5b64fc054c
commit 2f49231233
No known key found for this signature in database
GPG key ID: 423725EEF5A74B6F

View file

@ -113,14 +113,15 @@ class NetworkTestCase(TestCase, LoaderModuleMockMixin):
'''
Test for Performs a traceroute to a 3rd party host
'''
with patch.object(salt.utils.path, 'which', side_effect=[False, True]):
self.assertListEqual(network.traceroute('host'), [])
with patch('salt.utils.path.which', MagicMock(return_value='traceroute')):
with patch.dict(network.__salt__, {'cmd.run': MagicMock(return_value='')}):
self.assertListEqual(network.traceroute('gentoo.org'), [])
with patch.object(salt.utils.network, 'sanitize_host',
return_value='A'):
return_value='gentoo.org'):
with patch.dict(network.__salt__, {'cmd.run':
MagicMock(return_value="")}):
self.assertListEqual(network.traceroute('host'), [])
MagicMock(return_value='')}):
self.assertListEqual(network.traceroute('gentoo.org'), [])
def test_dig(self):
'''