Esempio n. 1
0
 def test_full(self, raw):
     inst = SCIONPath()
     inst._parse_iof = create_mock()
     inst._parse_hofs = create_mock()
     inst._init_of_idxs = create_mock()
     iof_list = []
     for i in 2, 4, 6:
         iof = create_mock(["hops", "shortcut"])
         iof.hops = i
         iof.shortcut = False
         iof_list.append(iof)
     inst._parse_iof.side_effect = iof_list
     data = create_mock()
     data.side_effect = ("A IOF", "A HOFS", "B IOF", "B HOFS", "C IOF",
                         "C HOFS")
     raw.return_value = data
     # Call
     inst._parse("data")
     # Tests
     assert_these_calls(inst._parse_iof, [
         call(data, inst.A_IOF),
         call(data, inst.B_IOF),
         call(data, inst.C_IOF)
     ])
     assert_these_calls(inst._parse_hofs, [
         call(data, inst.A_HOFS, 2),
         call(data, inst.B_HOFS, 4),
         call(data, inst.C_HOFS, 6)
     ])
     inst._init_of_idxs.assert_called_once_with()
Esempio n. 2
0
 def test(self, iof):
     inst = SCIONPath()
     data = create_mock(["pop"])
     inst._ofs = create_mock(["set"])
     # Call
     ntools.eq_(inst._parse_iof(data, "label"), iof.return_value)
     # Tests
     data.pop.assert_called_once_with(iof.LEN)
     iof.assert_called_once_with(data.pop.return_value)
     inst._ofs.set.assert_called_once_with("label", [iof.return_value])