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)