mirror of
https://github.com/espressif/openthread.git
synced 2026-08-23 18:59:51 +00:00
[style] apply google python style guide (#4501)
This commit applies and enforces Google's python style for tests.
This commit is contained in:
Yakun Xu
@@ -45,9 +45,8 @@ class TestMacParser(unittest.TestCase):
|
||||
frame = mac802154.MacFrame()
|
||||
frame.parse(io.BytesIO(bytearray([0x12, 0x00, 0x12, 0x34, 0x56])))
|
||||
|
||||
self.assertEqual(
|
||||
mac802154.MacHeader.FrameType.ACK,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(mac802154.MacHeader.FrameType.ACK,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(True, frame.header.frame_pending)
|
||||
self.assertEqual(False, frame.header.ack_request)
|
||||
self.assertEqual(0, frame.header.frame_version)
|
||||
@@ -57,12 +56,15 @@ class TestMacParser(unittest.TestCase):
|
||||
|
||||
def test_should_parse_data_frame_with_short_addresses(self):
|
||||
frame = mac802154.MacFrame()
|
||||
frame.parse(io.BytesIO(bytearray([0x61, 0x88, 0x34, 0xce, 0xfa,
|
||||
0xad, 0xde, 0xef, 0xbe, 0x12, 0x34, 0xfe, 0xdc])))
|
||||
frame.parse(
|
||||
io.BytesIO(
|
||||
bytearray([
|
||||
0x61, 0x88, 0x34, 0xce, 0xfa, 0xad, 0xde, 0xef, 0xbe, 0x12,
|
||||
0x34, 0xfe, 0xdc
|
||||
])))
|
||||
|
||||
self.assertEqual(
|
||||
mac802154.MacHeader.FrameType.DATA,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(mac802154.MacHeader.FrameType.DATA,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(False, frame.header.frame_pending)
|
||||
self.assertEqual(True, frame.header.ack_request)
|
||||
self.assertEqual(0, frame.header.frame_version)
|
||||
@@ -77,17 +79,16 @@ class TestMacParser(unittest.TestCase):
|
||||
|
||||
def test_should_parse_data_frame_with_extended_addresses(self):
|
||||
frame = mac802154.MacFrame()
|
||||
frame.parse(io.BytesIO(bytearray([0x61, 0xcc,
|
||||
0x56,
|
||||
0xce, 0xfa,
|
||||
0xde, 0xad, 0xbe, 0xef, 0xde, 0xad, 0xbe, 0xef,
|
||||
0x0b, 0xad, 0xf0, 0x0d, 0xba, 0xd0, 0xd0, 0x0d,
|
||||
0x12, 0x34,
|
||||
0xfe, 0xdc])))
|
||||
frame.parse(
|
||||
io.BytesIO(
|
||||
bytearray([
|
||||
0x61, 0xcc, 0x56, 0xce, 0xfa, 0xde, 0xad, 0xbe, 0xef, 0xde,
|
||||
0xad, 0xbe, 0xef, 0x0b, 0xad, 0xf0, 0x0d, 0xba, 0xd0, 0xd0,
|
||||
0x0d, 0x12, 0x34, 0xfe, 0xdc
|
||||
])))
|
||||
|
||||
self.assertEqual(
|
||||
mac802154.MacHeader.FrameType.DATA,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(mac802154.MacHeader.FrameType.DATA,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(False, frame.header.frame_pending)
|
||||
self.assertEqual(True, frame.header.ack_request)
|
||||
self.assertEqual(0, frame.header.frame_version)
|
||||
@@ -95,36 +96,26 @@ class TestMacParser(unittest.TestCase):
|
||||
self.assertEqual(bytearray([0xfe, 0xdc]), frame.header.fcs)
|
||||
self.assertEqual(0xface, frame.header.dest_pan_id)
|
||||
self.assertEqual(
|
||||
bytearray(
|
||||
reversed(
|
||||
[
|
||||
0xde, 0xad, 0xbe, 0xef,
|
||||
0xde, 0xad, 0xbe, 0xef])),
|
||||
frame.header.dest_address.mac_address)
|
||||
bytearray(reversed([0xde, 0xad, 0xbe, 0xef, 0xde, 0xad, 0xbe,
|
||||
0xef])), frame.header.dest_address.mac_address)
|
||||
self.assertEqual(0xface, frame.header.src_pan_id)
|
||||
self.assertEqual(
|
||||
bytearray(
|
||||
reversed(
|
||||
[
|
||||
0x0b, 0xad, 0xf0, 0x0d,
|
||||
0xba, 0xd0, 0xd0, 0x0d])),
|
||||
frame.header.src_address.mac_address)
|
||||
bytearray(reversed([0x0b, 0xad, 0xf0, 0x0d, 0xba, 0xd0, 0xd0,
|
||||
0x0d])), frame.header.src_address.mac_address)
|
||||
|
||||
self.assertEqual(bytearray([0x12, 0x34]), frame.payload.data)
|
||||
|
||||
def test_should_parse_data_frame_with_short_and_extended_addresses(self):
|
||||
frame = mac802154.MacFrame()
|
||||
frame.parse(io.BytesIO(bytearray([0x61, 0xc8,
|
||||
0x56,
|
||||
0xce, 0xfa,
|
||||
0xad, 0xde,
|
||||
0x0b, 0xad, 0xf0, 0x0d, 0xba, 0xd0, 0xd0, 0x0d,
|
||||
0x12, 0x34,
|
||||
0xfe, 0xdc])))
|
||||
frame.parse(
|
||||
io.BytesIO(
|
||||
bytearray([
|
||||
0x61, 0xc8, 0x56, 0xce, 0xfa, 0xad, 0xde, 0x0b, 0xad, 0xf0,
|
||||
0x0d, 0xba, 0xd0, 0xd0, 0x0d, 0x12, 0x34, 0xfe, 0xdc
|
||||
])))
|
||||
|
||||
self.assertEqual(
|
||||
mac802154.MacHeader.FrameType.DATA,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(mac802154.MacHeader.FrameType.DATA,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(False, frame.header.frame_pending)
|
||||
self.assertEqual(True, frame.header.ack_request)
|
||||
self.assertEqual(0, frame.header.frame_version)
|
||||
@@ -134,29 +125,22 @@ class TestMacParser(unittest.TestCase):
|
||||
self.assertEqual(0xdead, frame.header.dest_address.rloc)
|
||||
self.assertEqual(0xface, frame.header.src_pan_id)
|
||||
self.assertEqual(
|
||||
bytearray(
|
||||
reversed(
|
||||
[
|
||||
0x0b, 0xad, 0xf0, 0x0d,
|
||||
0xba, 0xd0, 0xd0, 0x0d])),
|
||||
frame.header.src_address.mac_address)
|
||||
bytearray(reversed([0x0b, 0xad, 0xf0, 0x0d, 0xba, 0xd0, 0xd0,
|
||||
0x0d])), frame.header.src_address.mac_address)
|
||||
|
||||
self.assertEqual(bytearray([0x12, 0x34]), frame.payload.data)
|
||||
|
||||
def test_should_parse_data_frame_with_extended_and_short_addresses(self):
|
||||
frame = mac802154.MacFrame()
|
||||
frame.parse(io.BytesIO(bytearray([0x61, 0x8c,
|
||||
0x56,
|
||||
0xce, 0xfa,
|
||||
0xde, 0xad, 0xbe, 0xef,
|
||||
0xde, 0xad, 0xbe, 0xef,
|
||||
0x0d, 0xf0,
|
||||
0x12, 0x34,
|
||||
0xfe, 0xdc])))
|
||||
frame.parse(
|
||||
io.BytesIO(
|
||||
bytearray([
|
||||
0x61, 0x8c, 0x56, 0xce, 0xfa, 0xde, 0xad, 0xbe, 0xef, 0xde,
|
||||
0xad, 0xbe, 0xef, 0x0d, 0xf0, 0x12, 0x34, 0xfe, 0xdc
|
||||
])))
|
||||
|
||||
self.assertEqual(
|
||||
mac802154.MacHeader.FrameType.DATA,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(mac802154.MacHeader.FrameType.DATA,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(False, frame.header.frame_pending)
|
||||
self.assertEqual(True, frame.header.ack_request)
|
||||
self.assertEqual(0, frame.header.frame_version)
|
||||
@@ -164,12 +148,8 @@ class TestMacParser(unittest.TestCase):
|
||||
self.assertEqual(bytearray([0xfe, 0xdc]), frame.header.fcs)
|
||||
self.assertEqual(0xface, frame.header.dest_pan_id)
|
||||
self.assertEqual(
|
||||
bytearray(
|
||||
reversed(
|
||||
[
|
||||
0xde, 0xad, 0xbe, 0xef,
|
||||
0xde, 0xad, 0xbe, 0xef])),
|
||||
frame.header.dest_address.mac_address)
|
||||
bytearray(reversed([0xde, 0xad, 0xbe, 0xef, 0xde, 0xad, 0xbe,
|
||||
0xef])), frame.header.dest_address.mac_address)
|
||||
self.assertEqual(0xface, frame.header.src_pan_id)
|
||||
self.assertEqual(0xf00d, frame.header.src_address.rloc)
|
||||
|
||||
@@ -177,17 +157,15 @@ class TestMacParser(unittest.TestCase):
|
||||
|
||||
def test_should_parse_data_request_command(self):
|
||||
frame = mac802154.MacFrame()
|
||||
frame.parse(io.BytesIO(bytearray([0x63, 0x88,
|
||||
0x78,
|
||||
0xce, 0xfa,
|
||||
0xad, 0xde,
|
||||
0x0d, 0xf0,
|
||||
0x04,
|
||||
0xfe, 0xdc])))
|
||||
frame.parse(
|
||||
io.BytesIO(
|
||||
bytearray([
|
||||
0x63, 0x88, 0x78, 0xce, 0xfa, 0xad, 0xde, 0x0d, 0xf0, 0x04,
|
||||
0xfe, 0xdc
|
||||
])))
|
||||
|
||||
self.assertEqual(
|
||||
mac802154.MacHeader.FrameType.COMMAND,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(mac802154.MacHeader.FrameType.COMMAND,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(False, frame.header.frame_pending)
|
||||
self.assertEqual(True, frame.header.ack_request)
|
||||
self.assertEqual(0, frame.header.frame_version)
|
||||
@@ -202,26 +180,77 @@ class TestMacParser(unittest.TestCase):
|
||||
|
||||
def test_should_decrypt_data_frame(self):
|
||||
|
||||
mac802154.DeviceDescriptors.add(0x2001, MacAddress(bytearray(
|
||||
[0x16, 0x6e, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x07]), MacAddressType.LONG))
|
||||
mac802154.DeviceDescriptors.add(
|
||||
0x2001,
|
||||
MacAddress(
|
||||
bytearray([0x16, 0x6e, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x07]),
|
||||
MacAddressType.LONG))
|
||||
|
||||
frame = mac802154.MacFrame()
|
||||
frame.parse(io.BytesIO(bytearray([0x69, 0x98, 0x68, # FC, seq
|
||||
0xce, 0xfa, # Pan Id
|
||||
0x00, 0x20, # Dst addr
|
||||
0x01, 0x20, # Src addr
|
||||
0x0d, 0x00, 0x00, 0x00, 0x00, 0x01, # Aux Security Header
|
||||
0xb5, 0x5a, 0x0d, 0x8e, 0x18, 0x5c, 0xb1, 0x06, # Payload
|
||||
0xc4, 0x6f, 0x7d, 0x6b, 0xb5, 0x4a, 0x87, 0x14,
|
||||
0xae, 0xdd, 0x8e, 0xb7, 0x37, 0x62, 0x27, 0x48,
|
||||
0xc9, 0x53, 0x0c, 0x44, 0x31, 0x59, 0x8b, 0xa2,
|
||||
0x83, 0x59, 0xa1, 0x43, # MIC (valid)
|
||||
0x74, 0xe0, 0x2a, 0xf6,
|
||||
0x99, 0xfc]))) # FCS (valid)
|
||||
frame.parse(
|
||||
io.BytesIO(
|
||||
bytearray([
|
||||
0x69,
|
||||
0x98,
|
||||
0x68, # FC, seq
|
||||
0xce,
|
||||
0xfa, # Pan Id
|
||||
0x00,
|
||||
0x20, # Dst addr
|
||||
0x01,
|
||||
0x20, # Src addr
|
||||
0x0d,
|
||||
0x00,
|
||||
0x00,
|
||||
0x00,
|
||||
0x00,
|
||||
0x01, # Aux Security Header
|
||||
0xb5,
|
||||
0x5a,
|
||||
0x0d,
|
||||
0x8e,
|
||||
0x18,
|
||||
0x5c,
|
||||
0xb1,
|
||||
0x06, # Payload
|
||||
0xc4,
|
||||
0x6f,
|
||||
0x7d,
|
||||
0x6b,
|
||||
0xb5,
|
||||
0x4a,
|
||||
0x87,
|
||||
0x14,
|
||||
0xae,
|
||||
0xdd,
|
||||
0x8e,
|
||||
0xb7,
|
||||
0x37,
|
||||
0x62,
|
||||
0x27,
|
||||
0x48,
|
||||
0xc9,
|
||||
0x53,
|
||||
0x0c,
|
||||
0x44,
|
||||
0x31,
|
||||
0x59,
|
||||
0x8b,
|
||||
0xa2,
|
||||
0x83,
|
||||
0x59,
|
||||
0xa1,
|
||||
0x43, # MIC (valid)
|
||||
0x74,
|
||||
0xe0,
|
||||
0x2a,
|
||||
0xf6,
|
||||
0x99,
|
||||
0xfc
|
||||
]))) # FCS (valid)
|
||||
|
||||
self.assertEqual(
|
||||
mac802154.MacHeader.FrameType.DATA,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(mac802154.MacHeader.FrameType.DATA,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(False, frame.header.frame_pending)
|
||||
self.assertEqual(True, frame.header.ack_request)
|
||||
self.assertEqual(1, frame.header.frame_version)
|
||||
@@ -235,45 +264,41 @@ class TestMacParser(unittest.TestCase):
|
||||
self.assertEqual(0, frame.header.aux_sec_header.frame_counter)
|
||||
self.assertEqual(5, frame.header.aux_sec_header.security_level)
|
||||
|
||||
self.assertEqual(bytes(bytearray([0x7c, 0x77, 0x80, 0xf0,
|
||||
0x4d, 0x4d, 0x4d, 0x4d,
|
||||
0xe0, 0x04, 0x44, 0x02,
|
||||
0x44, 0x66, 0x13, 0x5f,
|
||||
0x22, 0x80, 0xb1, 0x61,
|
||||
0x02, 0x61, 0x73, 0x11,
|
||||
0x2a, 0xff, 0x01, 0x08,
|
||||
0x16, 0x6e, 0x0a, 0x00,
|
||||
0x00, 0x00, 0x00, 0x07])),
|
||||
frame.payload.data)
|
||||
self.assertEqual(
|
||||
bytes(
|
||||
bytearray([
|
||||
0x7c, 0x77, 0x80, 0xf0, 0x4d, 0x4d, 0x4d, 0x4d, 0xe0, 0x04,
|
||||
0x44, 0x02, 0x44, 0x66, 0x13, 0x5f, 0x22, 0x80, 0xb1, 0x61,
|
||||
0x02, 0x61, 0x73, 0x11, 0x2a, 0xff, 0x01, 0x08, 0x16, 0x6e,
|
||||
0x0a, 0x00, 0x00, 0x00, 0x00, 0x07
|
||||
])), frame.payload.data)
|
||||
|
||||
def test_should_decrypt_command_frame(self):
|
||||
frame = mac802154.MacFrame()
|
||||
frame.parse(io.BytesIO(bytearray([0x6b, 0xdc, 0xce, 0xce,
|
||||
0xfa, 0x02, 0x00, 0x00,
|
||||
0x00, 0x00, 0x0a, 0x6e,
|
||||
0x16, 0x03, 0x00, 0x00,
|
||||
0x00, 0x00, 0x0a, 0x6e,
|
||||
0x16, 0x0d, 0x00, 0x00,
|
||||
0x00, 0x00, 0x01, 0x04,
|
||||
0x2d, 0xbc, 0x12, 0xbe,
|
||||
0x0a, 0x4f])))
|
||||
frame.parse(
|
||||
io.BytesIO(
|
||||
bytearray([
|
||||
0x6b, 0xdc, 0xce, 0xce, 0xfa, 0x02, 0x00, 0x00, 0x00, 0x00,
|
||||
0x0a, 0x6e, 0x16, 0x03, 0x00, 0x00, 0x00, 0x00, 0x0a, 0x6e,
|
||||
0x16, 0x0d, 0x00, 0x00, 0x00, 0x00, 0x01, 0x04, 0x2d, 0xbc,
|
||||
0x12, 0xbe, 0x0a, 0x4f
|
||||
])))
|
||||
|
||||
self.assertEqual(
|
||||
mac802154.MacHeader.FrameType.COMMAND,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(mac802154.MacHeader.FrameType.COMMAND,
|
||||
frame.header.frame_type)
|
||||
self.assertEqual(False, frame.header.frame_pending)
|
||||
self.assertEqual(True, frame.header.ack_request)
|
||||
self.assertEqual(1, frame.header.frame_version)
|
||||
self.assertEqual(206, frame.header.seq)
|
||||
self.assertEqual(bytearray([0x0a, 0x4f]), frame.header.fcs)
|
||||
self.assertEqual(0xface, frame.header.dest_pan_id)
|
||||
self.assertEqual(bytearray([0x16, 0x6e, 0x0a, 0x00,
|
||||
0x00, 0x00, 0x00, 0x02]),
|
||||
frame.header.dest_address.mac_address)
|
||||
self.assertEqual(
|
||||
bytearray([0x16, 0x6e, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x02]),
|
||||
frame.header.dest_address.mac_address)
|
||||
self.assertEqual(0xface, frame.header.src_pan_id)
|
||||
self.assertEqual(bytearray([0x16, 0x6e, 0x0a, 0x00,
|
||||
0x00, 0x00, 0x00, 0x03]),
|
||||
frame.header.src_address.mac_address)
|
||||
self.assertEqual(
|
||||
bytearray([0x16, 0x6e, 0x0a, 0x00, 0x00, 0x00, 0x00, 0x03]),
|
||||
frame.header.src_address.mac_address)
|
||||
|
||||
self.assertEqual(0, frame.header.aux_sec_header.frame_counter)
|
||||
self.assertEqual(5, frame.header.aux_sec_header.security_level)
|
||||
|
||||
Reference in New Issue
Block a user