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