def test(self):
inst = SCIONBasePacket()
cmn_hdr = create_mock([
"__len__",
"src_addr_type",
"dst_addr_type",
"addrs_len",
"hdr_len",
"total_len",
"set_of_idxs",
"next_hdr",
"update",
])
cmn_hdr.__len__.return_value = 2
addrs = create_mock(["__len__", "src_type", "dst_type"])
addrs.__len__.return_value = 4
path = create_mock(["__len__", "get_of_idxs"])
path.__len__.return_value = 8
path.get_of_idxs.return_value = 3, 7
inst.cmn_hdr = cmn_hdr
inst.addrs = addrs
inst.path = path
inst._get_offset_len = create_mock()
inst._get_offset_len.return_value = 42
inst._get_next_hdr = create_mock()
# Call
inst._update_cmn_hdr()
# Tests
ntools.eq_(cmn_hdr.src_addr_type, addrs.src_type.return_value)
ntools.eq_(cmn_hdr.dst_addr_type, addrs.dst_type.return_value)
ntools.eq_(cmn_hdr.addrs_len, 4)
ntools.eq_(cmn_hdr.hdr_len, 2 + 4 + 8)
ntools.eq_(cmn_hdr.total_len, 14 + 42)
cmn_hdr.set_of_idxs.assert_called_once_with(3, 7)
ntools.eq_(cmn_hdr.next_hdr, inst._get_next_hdr.return_value)
def test(self):
inst = SCIONBasePacket()
cmn_hdr = create_mock([
"__len__", "src_addr_type", "dst_addr_type", "addrs_len", "hdr_len",
"hdr_len_bytes", "bytes_to_hdr_len", "total_len", "set_of_idxs", "next_hdr", "update",
])
cmn_hdr.__len__.return_value = SCIONCommonHdr.LEN
cmn_hdr.bytes_to_hdr_len = lambda x: x//8
cmn_hdr.hdr_len_bytes.return_value = 32
addrs = create_mock(["__len__", "src_type", "dst_type"])
addrs.__len__.return_value = 16
path = create_mock(["__len__", "get_of_idxs"])
path.__len__.return_value = 8
path.get_of_idxs.return_value = 3, 7
inst.cmn_hdr = cmn_hdr
inst.addrs = addrs
inst.path = path
inst._get_offset_len = create_mock()
inst._get_offset_len.return_value = 42
inst._get_next_hdr = create_mock()
# Call
inst._update_cmn_hdr()
# Tests
ntools.eq_(cmn_hdr.src_addr_type, addrs.src_type.return_value)
ntools.eq_(cmn_hdr.dst_addr_type, addrs.dst_type.return_value)
ntools.eq_(cmn_hdr.addrs_len, 16)
ntools.eq_(cmn_hdr.hdr_len, (8 + 16 + 8)//8)
ntools.eq_(cmn_hdr.total_len, 32 + 42)
cmn_hdr.set_of_idxs.assert_called_once_with(3, 7)
ntools.eq_(cmn_hdr.next_hdr, inst._get_next_hdr.return_value)
def test(self):
inst = SCIONBasePacket()
inst.addrs = create_mock(["update"])
inst._update_cmn_hdr = create_mock()
# Call
inst.update()
# Tests
inst.addrs.update.assert_called_once_with()
inst._update_cmn_hdr.assert_called_once_with()