o i^<@spddlmZddlmZddlmZmZddlmZddl m Z e ej GdddZ Gdd d e jZd S) ) annotations) interface)ipraw) components)unittestc@s*eZdZd ddZd!ddZd"ddZdS)# MyProtocol expecting(list[tuple[bytes, dict[str, str | int]]]returnNonecCst||_dSN)listr )selfr r_C:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/pair/test/test_ip.py__init__szMyProtocol.__init__databytespartialintsourcestrdestprotocolversionihltostot_len fragment_idfragment_offset dont_fragmentmore_fragmentsttlc Cs|jsJd|jd\}}tt|}t}|D]}||||ks7Jd|d||d||q||ksFJd|d|dS)Nz(Got a packet when not expecting anymore.rz Expected =z, got )r poprsortedkeyslocals)rrrrrrrrrrrr r!r"r#Z expectDataZexpectKwZ expectKwKeysZlocalVariableskrrrdatagramReceiveds zMyProtocol.datagramReceivednumobjectprotocCsdSr r)rr+r-rrraddProto.szMyProtocol.addProtoN)r r r r )rrrrrrrrrrrrrrrrrrrrr rr!rr"rr#rr r )r+r,r-r,r r )__name__ __module__ __qualname__rr*r.rrrrr s  rc@sfeZdZdddZdddZdddZdd d Zdd d Zdd dZdddZ dddZ dddZ dS)IPTestsr r cCsjt}tddddddddd d d dd d d fg}|d||jdddddd|jr3Jd|jdSNfoobarr1.2.3.45.6.7.8 rrrrrrrrrr r!r"r#TޭFEfoobardummyrrrr5Should not expect any more packets, but still want %rr IPProtocolrr.r*r rr-p1rrrtestPacketParsing4s>  zIPTests.testPacketParsingcCst}tddddddddd d d dd d d fdd ddddddd d d dd d d fg}|d||jdddddd|jdd dddd|jrNJd|jdS)Nr4rr5r6r7r8r9r:r;r<r=r>r?r@quux5.4.3.26.7.8.9rArBrCTޭFE quuxrDrErGrrrtestMultiplePacketsdsl (  zIPTests.testMultiplePacketscCst}tddddddddd d d dd d d fg}tddddddddd d d dd d d fg}|d||d||jdddddd|jrNJd|j|jrXJd|jdSr3rErr-rHp2rrrtestMultipleSameProtossl   zIPTests.testMultipleSameProtoscCs4t}tg}|d||jdddddddS)Nr>rArrBrC)rrFrr.r*rGrrrtestWrongProtoNotSeens   zIPTests.testWrongProtoNotSeencCs4t}tddddddddd d d dd d d fdd ddddddd d d dd d d fg}|d|tdd ddddddd d d dd d d fddddddddd d d dd d d fg}|d||jdd dddd|jdddddd|jdd dddd|jdddddd|jrJd|j|jrJd|jdS)Nr4rr5r6r7r8r9r:r;r<r=r>r?r@rJrKrL sTޭ FE quuxrBrCrArMsTޭ FEfoobarrDrErOrrr testDemuxings ( (    zIPTests.testDemuxingcCs6t}z |ddWtdtjyYdSw)z2Adding a wrong level protocol raises an exception.*Z silliness2addProto must raise an exception for bad protocolsN)rrFr.rZ CannotAdaptAssertionErrorrerrrtestAddingBadProtos_WrongLevelsz&IPTests.testAddingBadProtos_WrongLevelc CXt}z |dtgWtdty+}z|jdkrnWYd}~dSd}~ww)z=Adding a protocol with a negative number raises an exception.)z'Added protocol must be positive or zeroNrVrrFr.r TypeErrorargsrWrXrrrtestAddingBadProtos_TooSmall z$IPTests.testAddingBadProtos_TooSmallc Cr[)=2**32 raises an exception.lz"Added protocol must fit in 32 bitsNrVr]rXrrrtestAddingBadProtos_TooBigraz"IPTests.testAddingBadProtos_TooBigc Cr[)rblrcNrVr]rXrrrtestAddingBadProtos_TooBig2raz#IPTests.testAddingBadProtos_TooBig2N)r r ) r/r0r1rIrNrQrRrTrZr`rdrerrrrr23s  0 R L  #  r2N) __future__rZzoperZ twisted.pairrrZtwisted.pythonrZ twisted.trialrZ implementerZIRawDatagramProtocolrZTestCaser2rrrrs     &