-
-
Notifications
You must be signed in to change notification settings - Fork 96
/
request_response_test.py
78 lines (66 loc) · 2.93 KB
/
request_response_test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
"""Unit test for KNX/IP Disconnect Request/Response."""
import asyncio
import unittest
from unittest.mock import MagicMock, patch
from xknx import XKNX
from xknx.io import UDPClient
from xknx.io.request_response import RequestResponse
from xknx.knxip import DisconnectResponse, KNXIPBody
class AsyncMock(MagicMock):
"""Async Mock."""
# pylint: disable=invalid-overridden-method
async def __call__(self, *args, **kwargs):
return super().__call__(*args, **kwargs)
class TestConnectResponse(unittest.TestCase):
"""Test class for xknx/io/Disconnect objects."""
def setUp(self):
"""Set up test class."""
self.loop = asyncio.new_event_loop()
asyncio.set_event_loop(self.loop)
def tearDown(self):
"""Tear down test class."""
self.loop.close()
def test_create_knxipframe_err(self):
"""Test if create_knxipframe of base class raises an exception."""
xknx = XKNX()
udp_client = UDPClient(xknx, ("192.168.1.1", 0), ("192.168.1.2", 1234))
request_response = RequestResponse(xknx, udp_client, DisconnectResponse)
request_response.timeout_in_seconds = 0
with self.assertRaises(NotImplementedError):
self.loop.run_until_complete(request_response.start())
@patch("logging.Logger.debug")
@patch(
"xknx.io.request_response.RequestResponse.send_request", new_callable=AsyncMock
)
def test_request_response_timeout(self, _send_request_mock, logger_debug_mock):
"""Test RequestResponse: timeout. No callback shall be left."""
xknx = XKNX()
udp_client = UDPClient(xknx, ("192.168.1.1", 0), ("192.168.1.2", 1234))
requ_resp = RequestResponse(xknx, udp_client, KNXIPBody)
requ_resp.response_received_or_timeout.wait = MagicMock(
side_effect=asyncio.TimeoutError()
)
self.loop.run_until_complete(requ_resp.start())
# Debug message was logged
logger_debug_mock.assert_called_once_with(
"Error: KNX bus did not respond in time (%s secs) to request of type '%s'",
1.0,
"RequestResponse",
)
# Callback was removed again
self.assertEqual(udp_client.callbacks, [])
@patch(
"xknx.io.request_response.RequestResponse.send_request", new_callable=AsyncMock
)
def test_request_response_cancelled(self, _send_request_mock):
"""Test RequestResponse: task cancelled. No callback shall be left."""
xknx = XKNX()
udp_client = UDPClient(xknx, ("192.168.1.1", 0), ("192.168.1.2", 1234))
requ_resp = RequestResponse(xknx, udp_client, KNXIPBody)
requ_resp.response_received_or_timeout.wait = MagicMock(
side_effect=asyncio.CancelledError()
)
with self.assertRaises(asyncio.CancelledError):
self.loop.run_until_complete(requ_resp.start())
# Callback was removed again
self.assertEqual(udp_client.callbacks, [])