def test_full(self, p_cls): msg = p_cls.new_message.return_value pcbms = [] for i in range(3): pcbms.append(create_mock_full({"p": "pcbm %d" % i})) exts = [] exts.append( create_mock_full({ "EXT_TYPE": ASMExtType.ROUTING_POLICY, "p": { "polType": RoutingPolType.ALLOW_AS, "itf": 0, "isdases": [_ISD_AS1] } })) # Call ASMarking.from_values(_ISD_AS1, 2, 3, pcbms, "root", "mtu", exts, ifid_size=14) # Tests p_cls.new_message.assert_called_once_with(isdas=_ISD_AS1, trcVer=2, certVer=3, ifIDSize=14, hashTreeRoot="root", mtu="mtu") msg.init.assert_called_once_with("hops", 3) for i, pcbm in enumerate(msg.pcbms): ntools.eq_("pcbm %d" % i, pcbm)
def test_full(self, p_cls): msg = p_cls.new_message.return_value pcbms = [] for i in range(3): pcbms.append(create_mock_full({"p": "pcbm %d" % i})) cchain = create_mock_full({"pack()": "cchain"}) # Call ASMarking.from_values(_ISD_AS1, 2, 3, pcbms, "root", "mtu", cchain, ifid_size=14) # Tests p_cls.new_message.assert_called_once_with(isdas=_ISD_AS1, trcVer=2, certVer=3, ifIDSize=14, hashTreeRoot="root", mtu="mtu", chain="cchain") msg.init.assert_called_once_with("pcbms", 3) for i, pcbm in enumerate(msg.pcbms): ntools.eq_("pcbm %d" % i, pcbm)
def test_full(self, p_cls): msg = p_cls.new_message.return_value pcbms = [] for i in range(3): pcbms.append(create_mock_full({"p": "pcbm %d" % i})) cchain = create_mock_full({"pack()": "cchain"}) revs = [] for i in range(2): revs.append(create_mock_full({"pack()": "rev %d" % i})) # Call ASMarking.from_values("isdas", 2, 3, pcbms, "eg rev token", "mtu", cchain, ifid_size=14, rev_infos=revs) # Tests p_cls.new_message.assert_called_once_with(isdas="isdas", trcVer=2, certVer=3, ifIDSize=14, egRevToken="eg rev token", mtu="mtu", chain="cchain") msg.init.assert_called_once_with("pcbms", 3) msg.exts.init.assert_called_once_with("revInfos", 2) for i, pcbm in enumerate(msg.pcbms): ntools.eq_("pcbm %d" % i, pcbm) for i, rev in enumerate(msg.exts.revInfos): ntools.eq_("rev %d" % i, rev)
def _create_asm(self, in_if, out_if, ts, prev_hof): pcbms = list(self._create_pcbms(in_if, out_if, ts, prev_hof)) if not pcbms: return None chain = self._get_my_cert() _, cert_ver = chain.get_leaf_isd_as_ver() return ASMarking.from_values(self.addr.isd_as, self._get_my_trc().version, cert_ver, pcbms, self.topology.mtu)
def _create_asm(self, in_if, out_if, ts, prev_hof): pcbms = list(self._create_pcbms(in_if, out_if, ts, prev_hof)) exts = self._create_asm_exts() chain = self._get_my_cert() _, cert_ver = chain.get_leaf_isd_as_ver() return ASMarking.from_values(self.addr.isd_as, self._get_my_trc().version, cert_ver, pcbms, self._get_if_rev_token(out_if), self.topology.mtu, chain, **exts)