def test(self, in_cksum):
inst = SCMPHeader()
src_ia = create_mock_full({"pack()": b"srIA"})
src_host = create_mock_full({"pack()": b"sHst"})
inst._src = create_mock_full({
"isd_as": src_ia,
"host": src_host
},
class_=SCIONAddr)
dst_ia = create_mock_full({"pack()": b"dsIA"})
dst_host = create_mock_full({"pack()": b"dHst"})
inst._dst = create_mock_full({
"isd_as": dst_ia,
"host": dst_host
},
class_=SCIONAddr)
inst.pack = create_mock()
inst.pack.return_value = b"packed with null checksum"
payload = b"payload"
expected_call = b"".join([
b"dsIA",
b"srIA",
b"dHst",
b"sHst",
b"\x00",
bytes([L4Proto.SCMP]),
b"packed with null checksum",
payload,
])
in_cksum.return_value = 0x3412
# Call
ntools.eq_(inst._calc_checksum(payload), bytes.fromhex("3412"))
# Tests
in_cksum.assert_called_once_with(expected_call)
def test(self, scapy_checksum):
inst = SCMPHeader()
inst._src = create_mock(["pack"], class_=SCIONAddr)
inst._src.pack.return_value = b"source address"
inst._dst = create_mock(["pack"], class_=SCIONAddr)
inst._dst.pack.return_value = b"destination address"
inst.pack = create_mock()
inst.pack.return_value = b"packed with null checksum"
payload = b"payload"
expected_call = b"".join([
b"source address", b"destination address", bytes([L4Proto.SCMP]),
b"packed with null checksum", payload,
])
scapy_checksum.return_value = 0x3412
# Call
ntools.eq_(inst._calc_checksum(payload), bytes.fromhex("1234"))
# Tests
scapy_checksum.assert_called_once_with(expected_call)