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
|
import pytest
import zigpy_znp.types as t
import zigpy_znp.frames as frames
from zigpy_znp.exceptions import InvalidFrame
def test_general_frame():
data = b"\xaa\55\xccdata goes in here\x00\x00"
length = t.uint8_t(len(data)).serialize()
cmd = t.CommandHeader(0x0161)
frame = frames.GeneralFrame(cmd, data)
assert frame.serialize() == length + b"\x61\x01" + data
with pytest.raises(ValueError):
frames.GeneralFrame(0, b"\x00" * 251)
extra = b"the rest of the owl\x00\x00"
r, rest = frames.GeneralFrame.deserialize(length + b"\x23\x00" + data + extra)
assert rest == extra
def test_general_frame_wrong_data():
# wrong length
data = b"\xfb" + b"\x00" * 258
with pytest.raises(ValueError):
frames.GeneralFrame.deserialize(data)
# data too short
data = b"\x04\x00\x00"
with pytest.raises(InvalidFrame):
frames.GeneralFrame.deserialize(data)
def test_transport_frame():
sof = t.uint8_t(0xFE).serialize()
bad_sof = t.uint8_t(0xFF).serialize()
payload = b"\x02\x61\x01\x11\x00"
fcs = b"\x73"
bad_fcs = b"\x74"
extra = b"the rest of the owl\x00\xaa\x55\xff"
r, rest = frames.TransportFrame.deserialize(sof + payload + fcs + extra)
assert rest == extra
assert r.serialize() == sof + payload + fcs
# wrong SOF
with pytest.raises(InvalidFrame):
frames.TransportFrame.deserialize(bad_sof + payload + fcs + extra)
# bad FCS
with pytest.raises(InvalidFrame):
frames.TransportFrame.deserialize(sof + payload + bad_fcs + extra)
# extra
r, rest = frames.TransportFrame.deserialize(sof + payload + fcs + extra)
assert rest == extra
# constructor
assert frames.TransportFrame(r.payload) == r
|
