[thread-cert] refactor case 5.6.7 using pktverify (#5453)

This commit is contained in:
Jing Ma
2020-09-11 09:00:04 -07:00
committed by GitHub
parent 8b842d39a6
commit 5e3b169068
@@ -30,31 +30,123 @@
import unittest
import thread_cert
from pktverify.consts import MLE_ADVERTISEMENT, MLE_CHILD_ID_RESPONSE, MLE_DATA_REQUEST, MLE_DATA_RESPONSE, TLV_REQUEST_TLV, NETWORK_DATA_TLV, SOURCE_ADDRESS_TLV, LEADER_DATA_TLV, NWD_PREFIX_TLV, NWD_BORDER_ROUTER_TLV, NWD_6LOWPAN_ID_TLV
from pktverify.packet_verifier import PacketVerifier
LEADER = 1
ROUTER = 2
REED = 3
ROUTER1 = 2
ROUTER2 = 3
ROUTER3 = 4
ROUTER15 = 16
REED1 = 17
class Cert_5_6_7_NetworkDataRequestREED(thread_cert.TestCase):
TOPOLOGY = {
LEADER: {
'name': 'LEADER',
'mode': 'rsdn',
'panid': 0xface,
'whitelist': [ROUTER, REED]
'whitelist': [ROUTER1, ROUTER2, ROUTER3, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, ROUTER15]
},
ROUTER: {
ROUTER1: {
'name': 'ROUTER_1',
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
REED: {
ROUTER2: {
'name': 'ROUTER_2',
'mode': 'rsdn',
'panid': 0xface,
'router_upgrade_threshold': 0,
'router_selection_jitter': 1,
'whitelist': [LEADER, REED1]
},
ROUTER3: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
5: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
6: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
7: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
8: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
9: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
10: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
11: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
12: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
13: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
14: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
15: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
ROUTER15: {
'mode': 'rsdn',
'panid': 0xface,
'router_selection_jitter': 1,
'whitelist': [LEADER]
},
REED1: {
'name': 'REED',
'mode': 'rsdn',
'panid': 0xface,
'whitelist': [ROUTER2]
},
}
def test(self):
@@ -62,36 +154,59 @@ class Cert_5_6_7_NetworkDataRequestREED(thread_cert.TestCase):
self.simulator.go(4)
self.assertEqual(self.nodes[LEADER].get_state(), 'leader')
self.nodes[ROUTER].start()
for i in range(ROUTER1, ROUTER15 + 1):
self.nodes[i].start()
self.simulator.go(5)
self.assertEqual(self.nodes[i].get_state(), 'router')
self.nodes[REED1].start()
self.simulator.go(5)
self.assertEqual(self.nodes[ROUTER].get_state(), 'router')
self.assertEqual(self.nodes[REED1].get_state(), 'child')
self.nodes[REED].start()
self.simulator.go(5)
self.assertEqual(self.nodes[REED].get_state(), 'child')
self.nodes[ROUTER2].remove_whitelist(self.nodes[REED1].get_addr64())
self.nodes[REED1].remove_whitelist(self.nodes[ROUTER2].get_addr64())
self.nodes[LEADER].remove_whitelist(self.nodes[REED].get_addr64())
self.nodes[REED].remove_whitelist(self.nodes[LEADER].get_addr64())
self.nodes[ROUTER].add_prefix('2001:2:0:3::/64', 'paros')
self.nodes[ROUTER].register_netdata()
self.nodes[LEADER].add_prefix('2001:2:0:3::/64', 'paros')
self.nodes[LEADER].register_netdata()
# Set lowpan context of sniffer
self.simulator.set_lowpan_context(1, '2001:2:0:3::/64')
self.simulator.go(2)
self.nodes[LEADER].add_whitelist(self.nodes[REED].get_addr64())
self.nodes[REED].add_whitelist(self.nodes[LEADER].get_addr64())
self.nodes[ROUTER2].add_whitelist(self.nodes[REED1].get_addr64())
self.nodes[REED1].add_whitelist(self.nodes[ROUTER2].get_addr64())
self.simulator.go(35)
self.simulator.go(30)
self.simulator.go(600)
addrs = self.nodes[REED].get_addrs()
addrs = self.nodes[REED1].get_addrs()
self.assertTrue(any('2001:2:0:3' in addr[0:10] for addr in addrs))
for addr in addrs:
if addr[0:10] == '2001:2:0:3':
self.assertTrue(self.nodes[LEADER].ping(addr))
def verify(self, pv):
pkts = pv.pkts
pv.summary.show()
LEADER = pv.vars['LEADER']
REED = pv.vars['REED']
# Step 4: Leader multicasts a MLE Data Response with the new information
_pkt = pkts.filter_mle_cmd(MLE_CHILD_ID_RESPONSE).filter_wpan_dst64(REED).must_next()
pkts.filter_wpan_src64(LEADER).filter_mle_cmd(MLE_DATA_RESPONSE).must_next().must_verify(
lambda p: {NWD_PREFIX_TLV, NWD_BORDER_ROUTER_TLV, NWD_6LOWPAN_ID_TLV} <= set(p.thread_nwd.tlv.type))
# Step 8: REED1 MUST send a MLE Data Request to its parent to get the new Network Dataset.
# MLE Data Request includes a TLV Request TLV for Network Data TLV
pkts.filter_wpan_src64(REED).filter_mle_cmd(MLE_DATA_REQUEST).must_next().must_verify(
lambda p: {TLV_REQUEST_TLV, NETWORK_DATA_TLV} <= set(p.mle.tlv.type))
# Step 10: REED1 send MLE Advertisement
pkts.filter_wpan_src64(REED).filter_mle_cmd(MLE_ADVERTISEMENT).must_next().must_verify(
lambda p: p.mle.tlv.leader_data.data_version == _pkt.mle.tlv.leader_data.data_version + 1)
if __name__ == '__main__':
unittest.main()