o >h;@sdZddlmZddlmZmZmZmZmZddl m Z m Z m Z m Z mZmZmZmZddlmZddlmZmZmZmZmZmZddlmZmZddlmZdd l m!Z!dd l"m#Z#dd l$m%Z%dd l&m'Z'm(Z(dd l)m*Z*ddl+m,Z,m-Z-m.Z.m/Z/m+Z+ddl0m1Z1m2Z2edZ3Gddde*Z4Gddde*Z5Gddde#Z6dededee3de3fddZ7Gdd d e*Z8Gd!d"d"e*Z9d#S)$z* Tests for L{twisted.trial._dist.stream}. )Random) AwaitableDictListTypeVarUnion)all_of assert_thatcallingequal_to has_lengthis_less_than_or_equal_toraises)given)binaryintegersjustlistsrandomstext)Deferredfail) IProtocol)Protocol)AMP)Failure) FakeTransportconnect)SynchronousTestCase) StreamOpenStreamReceiver StreamWritechunkstream)HasSum IsSequenceOfTc@seZdZdZeeeeedeee de ddfddZ ee e dd ededdfd d Zee d eddfd d ZdddZdS)StreamReceiverTestsz% Tests for L{StreamReceiver} streamsrandomreturnNc stfdd|D}||tt||}|D]\}}|D]}||q"q||fdd|D}t|tt|dS)z All data passed to L{StreamReceiver.write} is returned by a call to L{StreamReceiver.finish} with a matching C{streamId}. csg|]}qS)open).0_receiverr./var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/trial/_dist/test/test_stream.py .sz;StreamReceiverTests.test_streamReceived..csi|]}||qSr.)finish)r0streamIdr2r.r4 ;sz;StreamReceiverTests.test_streamReceived..N) r"shuffledictzipitemswriter r r ) selfr+r, streamIds expectedDatar7stringss actualDatar.r2r4test_streamReceived's  z'StreamReceiverTests.test_streamReceiveddatar7cCs&t}tt|j||ttdS)z L{StreamReceiver.write} raises L{KeyError} if called with a streamId not associated with an open stream. N)r"r r r= with_argsrKeyError)r>r7rEr3r.r.r4test_writeBadStreamId?s z)StreamReceiverTests.test_writeBadStreamIdcCs$t}tt|j|ttdS)z L{StreamReceiver.finish} raises L{KeyError} if called with a streamId not associated with an open stream. N)r"r r r6rFrrG)r>r7r3r.r.r4test_badFinishStreamIdHsz*StreamReceiverTests.test_badFinishStreamIdcCs6t}|}||tt|j|ttdS)zI L{StreamReceiver.finish} removes the identified stream. N)r"r/r6r r rFrrG)r>r3r7r.r.r4test_finishRemovesStreamQs z,StreamReceiverTests.test_finishRemovesStream)r-N)__name__ __module__ __qualname____doc__rrrrrbytesrrDrrintstrrHrIrJr.r.r.r4r*"s r*c@s,eZdZdZeeedddddZdS) ChunkTestsz Tests for ``chunk``. r&) min_value)rE chunkSizec Cs@tt||}t|tttt|tt||dddS)z L{chunk} returns an iterable of L{str} where each element is no longer than the given limit. The concatenation of the strings is also equal to the original input string. Nr) listr$r rr(r rr'r )r>rErTchunksr.r.r4 test_chunk`s zChunkTests.test_chunkN)rKrLrMrNrrrrWr.r.r.r4rR[srRc @sbeZdZdZdeddfddZejdee e ffddZ e jd e d edee e ffd d ZdS) AMPStreamReceiverz6 A simple AMP interface to L{StreamReceiver}. r+r-NcCs ||_dSN)r+)r>r+r.r.r4__init__xs zAMPStreamReceiver.__init__cCsd|jiS)Nr7)r+r/r>r.r.r4 streamOpen{szAMPStreamReceiver.streamOpenr7rEcCs|j||iSrY)r+r=)r>r7rEr.r.r4 streamWriteszAMPStreamReceiver.streamWrite)rKrLrMrNr"rZr! responderrrQobjectr\r#rPrOr]r.r.r.r4rXss$rXserverclient interactionr-csddtffdd }dtttfddffdd }t|t|d d |t|dd }t|}|||rFt trD St d ) zO Let C{server} and C{client} exchange bytes while C{interaction} runs. Fr-cs IdHSrYr.r.)rbr.r4 to_coroutines zinteract..to_coroutinerNcs d|dS)NTr.)rd)finishedresultr.r4collect_resultsz interact..collect_resultT)isServerz'Interaction failed to produce a result.) r)rrrrr fromCoroutineaddBothflush isinstanceraiseException Exception)r`rarbrcrgpump interactingr.)rerbrfr4interacts"      rqc@s eZdZdZddZddZdS) InteractTestsz0 Tests for the test helper L{interact}. cCsVGdddt}||tttt|WddS1s$wYdS)zg If the interaction results in a failure then L{interact} raises an exception. c@s eZdZdS)z6InteractTests.test_failure..ArbitraryExceptionN)rKrLrMr.r.r.r4ArbitraryExceptionsrsN)rn assertRaisesrqrr)r>rsr.r.r4 test_failures "zInteractTests.test_failurecCsB|tttttWddS1swYdS)zl If the interaction fails to produce a result then L{interact} raises an exception. N)rtrnrqrrr[r.r.r4test_incompletes "zInteractTests.test_incompleteN)rKrLrMrNrurvr.r.r.r4rrs rrc@s4eZdZdZeeedeeddfddZ dS) StreamTestsz Tests for L{stream}. rVr-NcCsBt}t}tt||t|t|}t||tt |dS)z All of the chunks passed to L{stream} are sent in order over a stream using the given AMP connection. N) rr"rqrXr%iterr r6r r )r>rVsenderr+r7r.r.r4 test_streams zStreamTests.test_stream) rKrLrMrNrrrrrOrzr.r.r.r4rws rwN):rNr,rtypingrrrrrhamcrestrr r r r r rr hypothesisrhypothesis.strategiesrrrrrrtwisted.internet.deferrrtwisted.internet.interfacesrtwisted.internet.protocolrtwisted.protocols.amprtwisted.python.failurertwisted.test.iosimrrtwisted.trial.unittestrr%r!r"r#r$matchersr'r(r)r*rRrXrqrrrwr.r.r.r4s, (      9!