valitydev/salt
14
0
Fork 0
mirror of https://github.com/valitydev/salt.git synced 2024-11-08 01:18:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add transport factory unit test for send_to retry/failure

Browse Source
This commit is contained in:
Bo Maryniuk 2018-01-19 15:56:02 +01:00
parent 05a663a4c6
commit fe26b59308
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tests/unit/utils/test_ssdp.py
View File

@ -158,3 +158,24 @@ class SSDPFactoryTestCase(TestCase):
assert factory.transport.sendto.mock_calls[0][2]['addr'] == '10.10.10.10'
assert factory.log.debug.called
assert factory.log.debug.mock_calls[0][1][0] == 'Sent successfully'
@patch('salt.utils.ssdp.time.sleep', MagicMock())
def test_transport_sendto_retry(self):
'''
Test transport send_to.
:return:
'''
transport = MagicMock()
transport.sendto = MagicMock(side_effect=SSDPBaseTestCase.exception_attr_error)
log = MagicMock()
factory = ssdp.SSDPFactory()
with patch.object(factory, 'transport', transport), patch.object(factory, 'log', log):
data = {'some': 'data'}
addr = '10.10.10.10'
factory._sendto(data=data, addr=addr)
assert factory.transport.sendto.called
assert ssdp.time.sleep.called
assert ssdp.time.sleep.call_args[0][0] > 0 and ssdp.time.sleep.call_args[0][0] < 0.5
assert factory.log.debug.called
assert 'Permission error' in factory.log.debug.mock_calls[0][1][0]