Yf;dZddlmZddlmZmZmZmZmZddl 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+ddl0m1Z1m2Z2edZ3Gdde*Z4Gdde*Z5Gdde#Z6dededee3de3fdZ7Gdde*Z8Gd 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 IsSequenceOfTcJeZdZdZeeeeedeee de ddfdZ ee e ddededdfd Zee deddfd Zd d ZdS) StreamReceiverTestsz% Tests for L{StreamReceiver} streamsrandomreturnNc t fd|D}||tt||}|D] \}}|D]} ||!|| fd|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}. c8g|]}S)open).0_receivers V/var/www/html/env/lib/python3.11/site-packages/twisted/trial/_dist/test/test_stream.py z;StreamReceiverTests.test_streamReceived...s!666X]]__666c<i|]}||Sr2)finish)r4streamIdr6s r7 z;StreamReceiverTests.test_streamReceived..;s'TTThh 9 9TTTr9N) r#shuffledictzipitemswriter rr ) selfr-r. streamIds expectedDatar<stringss actualDatar6s @r7test_streamReceivedz'StreamReceiverTests.test_streamReceived's "##6666g666  y!!!C 73344 !-!3!3!5!5 , , Hg , ,x++++ , y!!!TTTT)TTT JH\$:$: ; ;<<<< U55vvxx ! !7799--=4U +<=f=QU===.-=. U88::ttF||$$YcYYYYY%$Y U88::TsTtTTTTTTTTTTr9r,cbeZdZdZeeeddZdS) ChunkTestsz Tests for ``chunk``. r') min_value)rJ chunkSizec tt||}t|tt t t |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 r r)r rr(r )rCrJr]chunkss r7 test_chunkzChunkTests.test_chunk`seD),,--  4Y??@@x~~tBQBx00        r9N)rSrTrUrVrrrrar2r9r7r[r[[s\ U((Q"7"7"7888  98   r9r[c eZdZdZdeddfdZejdee e ffdZ e jde dedee e ffd ZdS) AMPStreamReceiverz6 A simple AMP interface to L{StreamReceiver}. r-r/Nc||_dSN)r-)rCr-s r7__init__zAMPStreamReceiver.__init__xs  r9c8d|jiS)Nr<)r-r3rCs r7 streamOpenzAMPStreamReceiver.streamOpen{sDL--//00r9r<rJc<|j||iSre)r-rB)rCr<rJs r7 streamWritezAMPStreamReceiver.streamWrites 8T*** r9)rSrTrUrVr#rfr" responderrrYobjectrir$rXrWrkr2r9r7rcrcss41Df-1111Cuc6k9Jr9rcserverclient interactionr/c ddtffd }dtttfddffd }t|t |d|t |d}t j|}|||r+ttr Std ) zO Let C{server} and C{client} exchange bytes while C{interaction} runs. Fr/cKd{VSrer2)rpsr7 to_coroutinezinteract..to_coroutines        r9rNcd|dS)NTr2)rtfinishedresults r7collect_resultz interact..collect_resultsr9T)isServerz'Interaction failed to produce a result.) r*rrrrr fromCoroutineaddBothflush isinstanceraiseException Exception) rnrorprsrxpump interactingrvrws ` @@r7interactrsH!!!!!!!% + ft,,,fu---   D (88K'''JJLLL fg & & $  ! ! # # # = > >>r9ceZdZdZdZdZdS) InteractTestsz0 Tests for the test helper L{interact}. c Gddt}||5tttt |ddddS#1swxYwYdS)zg If the interaction results in a failure then L{interact} raises an exception. ceZdZdS)6InteractTests.test_failure..ArbitraryExceptionN)rSrTrUr2r9r7ArbitraryExceptionrs Dr9rN)r assertRaisesrrr)rCrs r7 test_failurezInteractTests.test_failures           1 2 2 I I XZZT2D2D2F2F-G-G H H H I I I I I I I I I I I I I I I I I Is?A55A9<A9c|t5tttt ddddS#1swxYwYdS)zl If the interaction fails to produce a result then L{interact} raises an exception. N)rrrrrrhs r7test_incompletezInteractTests.test_incompletes   y ) ) 9 9 XZZXZZ 8 8 8 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9 9s6AA"%A"N)rSrTrUrVrrr2r9r7rrs? I I I99999r9rcreZdZdZeeedeeddfdZ dS) StreamTestsz Tests for L{stream}. r`r/Nc *t}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#rrcr&iterr r;rr )rCr`senderr-r<s r7 test_streamzStreamTests.test_streamsz  "" g & &vtF||0L0L   GNN8,,c(62B2B.C.CDDDDDr9) rSrTrUrVrrrrrWrr2r9r7rrsq U55?? E$u+ E$ E E E E E Er9rN):rVr.rtypingrrrrrhamcrestr r r r r rrr hypothesisrhypothesis.strategiesrrrrrrtwisted.internet.deferrrtwisted.internet.interfacesrtwisted.internet.protocolrtwisted.protocols.amprtwisted.python.failurertwisted.test.iosimrrtwisted.trial.unittestr r&r"r#r$r%matchersr(r)r*r,r[rcrrrr2r9r7rs88888888888888                    NNNNNNNNNNNNNNNN11111111111111......%%%%%%******55555555666666KKKKKKKKKKKKKK******** GCLL6T6T6T6T6T-6T6T6Tr     $   0$?Y? ? ! ?QR????B99999'9994EEEEE%EEEEEr9