def test(self, ext_fv, pld_fp, hdr_fv):
inst = SCIONBasePacket()
inst.addrs = create_mock(["dst", "src"])
scmp_ext = create_mock(["EXT_TYPE"])
scmp_ext.EXT_TYPE = ExtHopByHopType.SCMP
inst.ext_hdrs = [scmp_ext, "ext1", "ext2", "ext3", "ext4"]
inst.set_payload = create_mock()
# Call
inst.convert_to_scmp_error("addr",
"class",
"type",
"pkt",
"arg1",
"arg2",
hopbyhop="hbh",
kwarg1="kwval1")
# Tests
ext_fv.assert_called_once_with(hopbyhop="hbh")
ntools.eq_(inst.ext_hdrs,
[ext_fv.return_value, "ext1", "ext2", "ext3"])
pld_fp.assert_called_once_with("class",
"type",
"pkt",
"arg1",
"arg2",
kwarg1="kwval1")
hdr_fv.assert_called_once_with("addr", inst.addrs.dst, "class", "type")
ntools.eq_(inst.l4_hdr, hdr_fv.return_value)
inst.set_payload.assert_called_once_with(pld_fp.return_value)
def test(self, raw, pld_raw):
inst = SCIONBasePacket()
inst._inner_parse = create_mock()
inst.set_payload = create_mock()
data = create_mock(["get"])
raw.return_value = data
# Call
inst._parse(b"data")
# Tests
raw.assert_called_once_with(b"data", inst.NAME, inst.MIN_LEN, min_=True)
inst._inner_parse.assert_called_once_with(data)
pld_raw.assert_called_once_with(data.get.return_value)
inst.set_payload.assert_called_once_with(pld_raw.return_value)