Exemple #1
0
 def test(self):
     inst = SCIONBasePacket()
     inst.addrs = create_mock(["update"])
     inst._update_cmn_hdr = create_mock()
     # Call
     inst.update()
     # Tests
     inst.addrs.update.assert_called_once_with()
     inst._update_cmn_hdr.assert_called_once_with()
Exemple #2
0
 def test(self, update):
     hdr_exp = b"cmn hdr" b"addrs" b"path"
     total_exp = hdr_exp + b"inner pack"
     inst = SCIONBasePacket()
     inst.update = create_mock()
     inst.cmn_hdr = create_mock(["hdr_len", "pack", "total_len"])
     inst.cmn_hdr.hdr_len = len(hdr_exp)
     inst.cmn_hdr.total_len = len(total_exp)
     inst.cmn_hdr.pack.return_value = b"cmn hdr"
     inst.addrs = create_mock(["pack"])
     inst.addrs.pack.return_value = b"addrs"
     inst.path = create_mock(["pack"])
     inst.path.pack.return_value = b"path"
     inst._inner_pack = create_mock()
     inst._inner_pack.return_value = b"inner pack"
     # Call
     ntools.eq_(inst.pack(), total_exp)
     # Tests
     inst.update.assert_called_once_with()