def _setup(self, hof_idx):
inst = SCIONPath()
inst.get_iof = create_mock()
inst.get_hof = create_mock()
inst._iof_idx = 0
inst._hof_idx = hof_idx
iofs = []
for _ in range(2):
iof = create_mock(["hops"])
iof.hops = 5
iofs.append(iof)
inst.get_iof.side_effect = iofs
return inst, iofs
def _setup(self, curr_label=SCIONPath.A_IOF, b_seg=False, c_seg=False):
inst = SCIONPath()
inst.set_of_idxs = create_mock()
inst._iof_idx = 0
inst._hof_idx = 1
inst._ofs = create_mock([
"__len__", "count", "get_idx_by_label", "get_label_by_idx",
"reverse_label", "reverse_up_flag", "swap"
])
inst._ofs.__len__.return_value = 10
inst._ofs.count.side_effect = \
lambda l: self._of_count(l, b_seg, c_seg)
inst._ofs.get_label_by_idx.return_value = curr_label
return inst
def _check(self, up, hof_idx, expected):
inst = SCIONPath()
inst._iof_idx = 0
inst._hof_idx = hof_idx
inst._ofs = create_mock(["get_by_idx"])
iof = create_mock(["hops", "up_flag"])
iof.hops = 5
iof.up_flag = up
# Call
ret = inst._get_hof_ver_normal(iof)
# Tests
if expected is None:
ntools.eq_(ret, None)
else:
ntools.eq_(ret, inst._ofs.get_by_idx.return_value)
inst._ofs.get_by_idx.assert_called_once_with(expected)
def _setup(self, xover=False, peer=False, shortcut=False, up_flag=True):
inst = SCIONPath()
inst._iof_idx = 0
inst._hof_idx = 0
iof = create_mock(["peer", "shortcut", "up_flag"])
iof.peer = peer
iof.shortcut = shortcut
iof.up_flag = up_flag
inst.get_iof = create_mock()
inst.get_iof.return_value = iof
hof = create_mock(["xover"])
hof.xover = xover
inst.get_hof = create_mock()
inst.get_hof.return_value = hof
inst._get_hof_ver_normal = create_mock()
inst._ofs = create_mock(["get_by_idx"])
return inst, iof, hof