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)