Vh;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*Z9y")#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 IsSequenceOfTceZdZdZeeeeedeee de ddfdZ ee e ddededdfd Zee deddfd Zd d Zy) StreamReceiverTestsz% Tests for L{StreamReceiver} streamsrandomreturnNct}|Dcgc]}|j}}|j|tt ||}|j D]\}}|D]} |j ||  |j||Dcic]}||j|} }t| tt|ycc}wcc}w)z All data passed to L{StreamReceiver.write} is returned by a call to L{StreamReceiver.finish} with a matching C{streamId}. N) r#openshuffledictzipitemswritefinishr rr ) selfr-r.receiver_ streamIds expectedDatastreamIdstringss actualDatas T/home/dcms/DCMS/lib/python3.12/site-packages/twisted/trial/_dist/test/test_stream.pytest_streamReceivedz'StreamReceiverTests.test_streamReceived's "#.56X]]_6 6 y!C 734 !-!3!3!5 , Hg ,x+ , , y!JSThh 99T TJH\$: ;<7Us CCdatar=ct}tt|jj ||t t y)z L{StreamReceiver.write} raises L{KeyError} if called with a streamId not associated with an open stream. N)r#r r r6 with_argsrKeyError)r8r=rCr9s rAtest_writeBadStreamIdz)StreamReceiverTests.test_writeBadStreamId?s1 "#GHNN+55hEvhGWXct}tt|jj |t t y)z L{StreamReceiver.finish} raises L{KeyError} if called with a streamId not associated with an open stream. N)r#r r r7rErrF)r8r=r9s rAtest_badFinishStreamIdz*StreamReceiverTests.test_badFinishStreamIdHs/ "#GHOO,66x@&BRSrHct}|j}|j|tt |jj |t ty)zI L{StreamReceiver.finish} removes the identified stream. N)r#r1r7r r rErrF)r8r9r=s rAtest_finishRemovesStreamz,StreamReceiverTests.test_finishRemovesStreamQsF"#==?!GHOO,66x@&BRSrH)r/N)__name__ __module__ __qualname____doc__rrrrrbytesrrBrrintstrrGrJrLrHrAr,r,"s 5vx !79-=4U +<=f=QU=.=. 8:tF|$YcYYY%Y 8:TsTtTTTrHr,cHeZdZdZeeeddZy) ChunkTestsz Tests for ``chunk``. r') min_value)rC chunkSizec tt||}t|tt t t |tt||ddy)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 )r8rCrXchunkss rA test_chunkzChunkTests.test_chunk`sTeD),-  4Y?@x~tBQx0   rHN)rMrNrOrPrrrr\rTrHrArVrV[s) (Q"78 9 rHrVc eZdZdZdeddfdZejdee e ffdZ e jde dedee e ffd Zy) AMPStreamReceiverz6 A simple AMP interface to L{StreamReceiver}. r-r/Nc||_yN)r-)r8r-s rA__init__zAMPStreamReceiver.__init__xs  rHc:d|jjiS)Nr=)r-r1r8s rA streamOpenzAMPStreamReceiver.streamOpen{sDLL--/00rHr=rCc>|jj||iSr`)r-r6)r8r=rCs rA streamWritezAMPStreamReceiver.streamWrites 8T* rH)rMrNrOrPr#rar" responderrrSobjectrdr$rRrQrfrTrHrAr^r^ss|41Df-11Cuc6k9JrHr^serverclient interactionr/c xddtffd }dtttfddffd }t|t |d|t |d}t j |}|j||jr"ttrjStd ) zO Let C{server} and C{client} exchange bytes while C{interaction} runs. Fr/c$Kd{S7wr`rT)rksrA to_coroutinezinteract..to_coroutines    s rNcd|y)NTrT)rofinishedresults rAcollect_resultz interact..collect_results rHT)isServerz'Interaction failed to produce a result.) r*rrrrr fromCoroutineaddBothflush isinstanceraiseException Exception) rirjrkrnrspump interactingrqrrs ` @@rAinteractr}sH!!% + ft,fu-  D ((8K'JJL fg &  ! ! # = >>rHceZdZdZdZdZy) InteractTestsz0 Tests for the test helper L{interact}. c Gddt}|j|5tttt |dddy#1swYyxYw)zg If the interaction results in a failure then L{interact} raises an exception. c eZdZy)6InteractTests.test_failure..ArbitraryExceptionN)rMrNrOrTrHrAArbitraryExceptionrs rHrN)rz assertRaisesr}rr)r8rs rA test_failurezInteractTests.test_failuresQ     1 2 I XZT2D2F-G H I I Is ,AAc|jt5tttt dddy#1swYyxYw)zl If the interaction fails to produce a result then L{interact} raises an exception. N)rrzr}rrrcs rAtest_incompletezInteractTests.test_incompletes9   y ) 9 XZXZ 8 9 9 9s &AAN)rMrNrOrPrrrTrHrArrs I9rHrcReZdZdZeeedeeddfdZ y) StreamTestsz Tests for L{stream}. r[r/Nc t}t}tt||t |t |}t |j|tt|y)z All of the chunks passed to L{stream} are sent in order over a stream using the given AMP connection. N) rr#r}r^r&iterr r7rr )r8r[senderr-r=s rA test_streamzStreamTests.test_streamsS  " g &vtF|0L  GNN8,c(62B.CDrH) rMrNrOrPrrrrrQrrTrHrArrs< 5? E$u+ E$ E ErHrN):rPr.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,rVr^r}rrrTrHrArs88   NN11.%*56KK* CL6T-6Tr $ 0$?Y? ? ! ?QR?B9'94E%ErH