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
79
80
81
82
83
84
85
86
|
"""Unit test for KNX/IP SRP objects."""
import pytest
from xknx.exceptions import ConversionError, CouldNotParseKNXIP
from xknx.knxip import DIBServiceFamily, DIBTypeCode, SearchRequestParameterType
from xknx.knxip.srp import SRP
class TestKNXIPSRP:
"""Test class for KNX/IP SRP objects."""
def test_basic(self) -> None:
"""Test SRP with mac contains mac address."""
srp: SRP = SRP.with_mac_address(bytes([1, 2, 3, 4, 5, 6]))
assert len(srp) == SRP.SRP_HEADER_SIZE + 6
assert srp.type == SearchRequestParameterType.SELECT_BY_MAC_ADDRESS
srp: SRP = SRP.with_service(DIBServiceFamily.TUNNELING, 2)
assert len(srp) == SRP.SRP_HEADER_SIZE + 2
assert srp.type == SearchRequestParameterType.SELECT_BY_SERVICE
srp: SRP = SRP.with_programming_mode()
assert len(srp) == SRP.SRP_HEADER_SIZE
assert srp.type == SearchRequestParameterType.SELECT_BY_PROGRAMMING_MODE
dibs: list[DIBTypeCode] = [
DIBTypeCode.SUPP_SVC_FAMILIES,
DIBTypeCode.ADDITIONAL_DEVICE_INFO,
]
srp: SRP = SRP.request_device_description(dibs)
assert len(srp) == SRP.SRP_HEADER_SIZE + len(dibs)
assert srp.type == SearchRequestParameterType.REQUEST_DIBS
def test_invalid_payload_raises(self) -> None:
"""Test SRP with invalid data size raises."""
with pytest.raises(ConversionError):
SRP.with_mac_address(bytes([1, 2, 3, 4, 5, 6, 7, 8]))
with pytest.raises(ConversionError):
SRP(SearchRequestParameterType.SELECT_BY_SERVICE, True)
with pytest.raises(ConversionError):
SRP(SearchRequestParameterType.REQUEST_DIBS, True)
def test_invalid_payload_from_knx_raises(self) -> None:
"""Test from_knx with invalid data size raises."""
# size is too big
with pytest.raises(CouldNotParseKNXIP):
SRP.from_knx(bytes.fromhex("08 82"))
# too small
with pytest.raises(CouldNotParseKNXIP):
SRP.from_knx(bytes.fromhex("08"))
@pytest.mark.parametrize(
("srp", "expected_bytes"),
[
(
SRP.with_mac_address(bytes([1, 2, 3, 4, 5, 6])),
bytes.fromhex("08 82 01 02 03 04 05 06"),
),
(
SRP.with_service(DIBServiceFamily.TUNNELING, 2),
bytes.fromhex("04 83 04 02"),
),
(
SRP.with_programming_mode(),
bytes.fromhex("02 81"),
),
(
SRP.request_device_description(
[DIBTypeCode.SUPP_SVC_FAMILIES, DIBTypeCode.TUNNELING_INFO]
),
bytes.fromhex("04 04 02 07"),
),
(
SRP.request_device_description([DIBTypeCode.SUPP_SVC_FAMILIES]),
bytes.fromhex("04 04 02 00"),
),
],
)
def test_to_and_from_knx(self, srp: SRP, expected_bytes: bytes) -> None:
"""Test to and from KNX."""
assert bytes(srp) == expected_bytes
assert SRP.from_knx(expected_bytes) == srp
|
