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()
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()