def test_with_exts(self):
inst = SCIONExtPacket()
inst.ext_hdrs = [create_mock(["EXT_CLASS"]), 0]
# Call
ntools.eq_(inst._get_next_hdr(), inst.ext_hdrs[0].EXT_CLASS)
def test_no_exts(self):
inst = SCIONExtPacket()
inst.ext_hdrs = []
inst._l4_proto = 42
# Call
ntools.eq_(inst._get_next_hdr(), 42)