def _setup(self, src_type):
inst = SCIONAddrHdr()
inst.calc_lens = create_mock()
inst.update = create_mock()
data = create_mock(["get", "pop"])
data.get.side_effect = "src addr", "dst addr"
data.pop.side_effect = None, None
src = create_mock(["__len__", "host"])
src.__len__.return_value = 3
src.host = create_mock(["TYPE"])
src.host.TYPE = src_type
return inst, data, src, "dst"
def _setup(self, src_type, haddr, saddr):
inst = SCIONAddrHdr()
inst.calc_lens = create_mock()
inst.update = create_mock()
data = create_mock_full({
"pop()...": ("1-1", "2-2", "dst host", "src host"),
})
src_host = create_mock_full({"TYPE": src_type})
src = create_mock_full({"host": src_host})
haddr.side_effect = (
create_mock_full({"LEN": 12}, return_value="dst haddr"),
create_mock_full({"LEN": 34}, return_value="src haddr"),
)
saddr.side_effect = "dst", src
return inst, data, src, "dst"
def _check(self, type_lens, exp_total, exp_pad, calc_len, calc_padding):
calc_len.side_effect = type_lens
calc_padding.return_value = exp_pad
# Call
results = SCIONAddrHdr.calc_lens(1, 2)
# Tests
assert_these_calls(calc_len, [call(1), call(2)])
calc_padding.assert_called_once_with(sum(type_lens),
SCIONAddrHdr.BLK_SIZE)
ntools.eq_(results, (exp_total, exp_pad))
def test(self):
inst = SCIONAddrHdr()
inst.update = create_mock()
inst.src = "src"
inst.dst = "dst"
# Call
inst.reverse()
# Tests
ntools.eq_(inst.src, "dst")
ntools.eq_(inst.dst, "src")
inst.update.assert_called_once_with()
def test(self):
inst = SCIONAddrHdr()
inst.update = create_mock()
inst.src = create_mock(["pack"])
inst.src.pack.return_value = b"src saddr"
inst.dst = create_mock(["pack"])
inst.dst.pack.return_value = b"dst saddr"
inst._total_len = 24
inst._pad_len = 6
expected = b"src saddr" b"dst saddr"
expected += bytes(inst._pad_len)
# Call
ntools.eq_(inst.pack(), expected)
# Tests
inst.update.assert_called_once_with()
def test(self):
inst = SCIONAddrHdr()
inst.calc_lens = create_mock_full(return_value=(1, 3))
inst.src = create_mock_full({"host": create_mock(["TYPE"])})
inst.dst = create_mock_full({"host": create_mock(["TYPE"])})
# Call
inst.update()
# Tests
inst.calc_lens.assert_called_once_with(inst.dst.host.TYPE,
inst.src.host.TYPE)
ntools.eq_(inst._total_len, 1)
ntools.eq_(inst._pad_len, 3)
def test(self):
inst = SCIONAddrHdr()
inst.update = create_mock()
dst_ia = create_mock_full({"pack()": b"dsIA"})
dst_host = create_mock_full({"pack()": b"dst H"})
inst.dst = create_mock_full({"isd_as": dst_ia, "host": dst_host})
src_ia = create_mock_full({"pack()": b"srIA"})
src_host = create_mock_full({"pack()": b"src H"})
inst.src = create_mock_full({"isd_as": src_ia, "host": src_host})
inst._total_len = 24
inst._pad_len = 6
expected = b"dsIA" b"srIA" b"dst H" b"src H"
expected += bytes(inst._pad_len)
# Call
ntools.eq_(inst.pack(), expected)
# Tests
inst.update.assert_called_once_with()
def test(self):
inst = SCIONAddrHdr()
inst._total_len = 42
# Call
ntools.eq_(len(inst), 42)