saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-17 10:10:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Use MagicMock

Browse source
This commit is contained in:
Gareth J. Greenaway 2019-01-24 14:51:17 -08:00
parent 172ad2e021
commit ad64efbabe
No known key found for this signature in database
GPG key ID: 10B62F8A7CAD7A41
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tests/unit/utils/test_network.py
View file

@ -10,7 +10,6 @@ from tests.support.unit import skipIf
from tests.support.unit import TestCase
from tests.support.mock import (
MagicMock,
Mock,
mock_open,
patch,
NO_MOCK,
@ -278,7 +277,7 @@ class NetworkTestCase(TestCase):
]
for host in hosts:
log.debug('=== h %s ===', host)
with patch.object(socket, 'getaddrinfo', Mock(return_value=host['mocked'])):
with patch.object(socket, 'getaddrinfo', MagicMock(return_value=host['mocked'])):
with patch('socket.socket', MockSocket):
ret = network.dns_check(host['host'], host['port'])
self.assertEqual(ret, host['ret'])