U 2ܟj^<@spddlmZddlmZddlmZmZddlmZddl m Z e ej GdddZ Gdd d e jZd S) ) annotations) interface)ipraw) components)unittestc@sXeZdZdddddZdddddddddddddddd d d Zd d dd ddZdS) MyProtocolz(list[tuple[bytes, dict[str, str | int]]]None) expectingreturncCst||_dSN)listr )selfr r=/tmp/pip-unpacked-wheel-4b529toj/twisted/pair/test/test_ip.py__init__szMyProtocol.__init__bytesintstr)datapartialsourcedestprotocolversionihltostot_len fragment_idfragment_offset dont_fragmentmore_fragmentsttlr c Cs|jstd|jd\}}tt|}t}|D]6}||||ks8td|d||d||q8||kstd|d|dS)Nz(Got a packet when not expecting anymore.rz Expected =z, got )r AssertionErrorpopr sortedkeyslocals)rrrrrrrrrrrrr r!r"Z expectDataZexpectKwZ expectKwKeysZlocalVariableskrrrdatagramReceivedszMyProtocol.datagramReceivedobject)numprotor cCsdSr r)rr,r-rrraddProto.szMyProtocol.addProtoN)__name__ __module__ __qualname__rr*r.rrrrr s*rc@seZdZddddZddddZddddZddd d Zddd d Zddd dZddddZ ddddZ ddddZ dS)IPTestsr )r cCsjt}tddddddddd d d dd d d fg}|d||jdddddd|jrftd|jdSNfoobarr1.2.3.45.6.7.8 rrrrrrrrrrr r!r"TޭFEfoobardummyrrrr5Should not expect any more packets, but still want %rr IPProtocolrr.r*r r$rr-p1rrrtestPacketParsing4s>  zIPTests.testPacketParsingcCst}tddddddddd d d dd d d fdd ddddddd d d dd d d fg}|d||jdddddd|jdd dddd|jrtd|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|jrtd|j|jrtd|jdSr3rErr-rHp2rrrtestMultipleSameProtossl   zIPTests.testMultipleSameProtoscCs4t}tg}|d||jdddddddS)Nr>rArrBrC)rrFrr.r*rGrrrtestWrongProtoNotSeens  zIPTests.testWrongProtoNotSeencCs8t}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|jrtd|j|jr4td|jdS)Nr4rr5r6r7r8r9r:r;r<r=r>r?r@rJrKrL sTޭ FE quuxrBrCrArMsTޭ FEfoobarrDrErOrrr testDemuxings( (       zIPTests.testDemuxingcCs<t}z|ddWntjk r.Yn XtddS)z2Adding a wrong level protocol raises an exception.*Z silliness2addProto must raise an exception for bad protocolsN)rrFr.rZ CannotAdaptr$rerrrtestAddingBadProtos_WrongLevels z&IPTests.testAddingBadProtos_WrongLevelc CsZt}z|dtgWn0tk rL}z|jdkr:nW5d}~XYn XtddS)z=Adding a protocol with a negative number raises an exception.)z'Added protocol must be positive or zeroNrVrrFr.r TypeErrorargsr$rWrrrtestAddingBadProtos_TooSmalls z$IPTests.testAddingBadProtos_TooSmallc CsZt}z|dtgWn0tk rL}z|jdkr:nW5d}~XYn XtddS)=2**32 raises an exception.lz"Added protocol must fit in 32 bitsNrVr[rWrrrtestAddingBadProtos_TooBigs z"IPTests.testAddingBadProtos_TooBigc CsZt}z|dtgWn0tk rL}z|jdkr:nW5d}~XYn XtddS)r_lr`NrVr[rWrrrtestAddingBadProtos_TooBig2s z#IPTests.testAddingBadProtos_TooBig2N) r/r0r1rIrNrQrRrTrYr^rarbrrrrr23s0RL#   r2N) __future__rZzoperZ twisted.pairrrZtwisted.pythonrZ twisted.trialrZ implementerZIRawDatagramProtocolrZTestCaser2rrrrs     &