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 unit test for outdated timestamp at no reply

Browse Source
This commit is contained in:
Bo Maryniuk 2018-01-22 17:40:02 +01:00
parent 24ded041af
commit 6565c16be4
1 changed files with 25 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -15,6 +15,7 @@ from tests.support.mock import (
import salt.exceptions import salt.exceptions
import salt.state import salt.state
from salt.utils import ssdp from salt.utils import ssdp
import datetime
try: try:
import pytest import pytest
@ -229,3 +230,27 @@ class SSDPFactoryTestCase(TestCase):
assert 'Received invalid timestamp in package' in factory.log.debug.call_args[0][0] assert 'Received invalid timestamp in package' in factory.log.debug.call_args[0][0]
assert factory._sendto.called assert factory._sendto.called
assert '{}:E:Invalid timestamp'.format(signature) == factory._sendto.call_args[0][0] assert '{}:E:Invalid timestamp'.format(signature) == factory._sendto.call_args[0][0]
def test_datagram_signature_outdated_timestamp_quiet(self):
'''
Test if datagram processing reacts on outdated message (more than 20 seconds). Quiet mode.
:return:
'''
factory = ssdp.SSDPFactory()
signature = ssdp.SSDPBase.DEFAULTS[ssdp.SSDPBase.SIGNATURE]
data = '{}{}'.format(signature, '1516623820')
addr = '10.10.10.10', 'foo.suse.de'
ahead_dt = datetime.datetime.fromtimestamp(1516623841)
curnt_dt = datetime.datetime.fromtimestamp(1516623820)
delta = datetime.timedelta(0, 20)
with patch.object(factory, 'log', MagicMock()), patch.object(factory, '_sendto'), \
patch('salt.utils.ssdp.datetime.datetime', MagicMock()), \
patch('salt.utils.ssdp.datetime.datetime.now', MagicMock(return_value=ahead_dt)), \
patch('salt.utils.ssdp.datetime.datetime.fromtimestamp', MagicMock(return_value=curnt_dt)), \
patch('salt.utils.ssdp.datetime.timedelta', MagicMock(return_value=delta)):
factory.datagram_received(data=data, addr=addr)
assert factory.log.debug.called
assert not factory.disable_hidden
assert not factory._sendto.called
assert 'Received outdated package' in factory.log.debug.call_args[0][0]