o 9»Bi5ã@sædZddlmZddlmZmZddlmZmZm Z ddl m Z ddl m Z ddlmZddlmZdd lmZdd lmZmZdd lmZdd lmZmZdd lmZddlmZddl m!Z!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(m)Z)m*Z*ddl+m,Z,e dddZ-dZ.Gdd„de&ƒZ/Gdd„de&ƒZ0ze1dƒWn e2y§dZ.Yn1wddl3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:eGdd „d e7ƒƒZ;Gd!d"„d"e9e;ƒZe e.d'ƒGd(d)„d)eƒƒZ?d*S)+z% Tests for L{twisted.web.websocket}. é)Ú annotations)Ú dataclassÚfield)ÚAnyÚGenericÚTypeVar)ÚskipIf)Ú implementer)ÚDeferred)ÚConnectionDone)Ú IPushProducer)ÚAccumulatingProtocolÚMemoryReactorClock)ÚFailure)ÚConnectionCompleterÚIOPump)ÚSynchronousTestCase)Ú BAD_REQUEST)ÚAgentÚreadBody)ÚIRequest)ÚResource)Ú NOT_DONE_YETÚRequestÚSite)ÚDataÚWSPÚWebSocketProtocol)ÚboundFc@óeZdZddd„ZdS) Ú WeirdResourceÚrequestrÚreturnÚbytescCs| d¡dS)a¯ Per U{the wsproto documentation }: - The only scenario in which the caller receives a RejectConnection with C{has_body == False} is if the peer violates sends an informational status code (1xx) other than 101 This is a weird edge case so we provoke it. éfó)ÚsetResponseCode©Úselfr!©r)ú_/var/www/html/Trade-python/venv/lib/python3.10/site-packages/twisted/web/test/test_websocket.pyÚ render_GET#s zWeirdResource.render_GETN)r!rr"r#)Ú__name__Ú __module__Ú __qualname__r+r)r)r)r*r "ór c@seZdZdZd dd„ZdS) ÚDelayedResponsez¾ A resource that will not respond to a C{GET} request right away. You will need to use the C{request.write(b'RESPONSE BODY')} and C{request.finish()} to trigger a response. r!rr"ÚintcCs ||_tS©N)r!rr'r)r)r*r+9szDelayedResponse.render_GETN)r!rr"r1)r,r-r.Ú__doc__r+r)r)r)r*r02sr0ÚwsprotoT)ÚConnectionRejectedÚWebSocketClientEndpointÚWebSocketClientFactoryrÚWebSocketResourceÚWebSocketServerFactoryÚWebSocketTransportc@s€eZdZUdZeedZded<dZded<d$d d „Z d%dd„Z d&dd„Z d'dd„Z d(dd„Z d)dd„Zd*dd „Zd+d"d#„ZdS),ÚMyWSPz Used to implement both client-side and server-side WebSocket application that will help with testing. ©Údefault_factoryz list[bytes]ÚpongsNzFailure | NoneÚwasLostÚ transportr:r"ÚNonecCó ||_dSr2)r@)r(r@r)r)r*ÚnegotiationStartedWó zMyWSP.negotiationStartedcCsdSr2r)©r(r)r)r*ÚnegotiationFinishedZszMyWSP.negotiationFinishedÚreasonrcCrBr2)r?)r(rGr)r)r*ÚconnectionLost]rDzMyWSP.connectionLostÚdatar#cCó(|dkr |j d¡dS|j |¡dS)Nórequestó responseÿ)r@ÚsendBytesMessageÚ bDeferredÚcallback©r(rIr)r)r*ÚbytesMessageReceived`ózMyWSP.bytesMessageReceivedÚstrcCrJ)Nr!Úresponse)r@ÚsendTextMessageÚdeferredrOrPr)r)r*ÚtextMessageReceivedfrRzMyWSP.textMessageReceivedÚpayloadcCs|j |¡dSr2)r>Úappend)r(rXr)r)r*Ú pongReceivedlszMyWSP.pongReceivedú Deferred[str]cCótƒ|_|j d¡|jS)zR Send a text message to the server and expect a response. r!)r rVr@rUrEr)r)r*Ú sendRequestoó zMyWSP.sendRequestúDeferred[bytes]cCr\)zS Send a bytes message to the server and expect a response. rK)r rNr@rMrEr)r)r*Ú bytesRequestwr^zMyWSP.bytesRequest)r@r:r"rA©r"rA)rGrr"rA)rIr#r"rA)rIrSr"rA)rXr#r"rA)r"r[)r"r_)r,r-r.r3rÚlistr>Ú__annotations__r?rCrFrHrQrWrZr]r`r)r)r)r*r;Ms         r;c@s eZdZUded<d dd„Zd S) Ú MyFactoryzWebSocketFixture[Any]Úfixturer!rr"r;cCstƒ}|jj |¡|Sr2)r;reÚserversrY)r(r!Únewr)r)r*Ú buildProtocol‚szMyFactory.buildProtocolN)r!rr"r;)r,r-r.rcrhr)r)r)r*rds rdc@r) ÚMyClientFactoryÚurirSr"r;cCstƒSr2)r;)r(rjr)r)r*rhˆszMyClientFactory.buildProtocolN)rjrSr"r;)r,r-r.rhr)r)r)r*ri‡r/ric@s–eZdZUeƒZded<eedZded<eedZ ded<dZ d ed <ee dZ d ed <ee dZd ed<ed dd„ƒZd!d"dd„Zd#d$dd„ZdS)%ÚWebSocketFixtureúWebSocketClientFactory[WSP]Ú clientFactoryr<rÚreactorrÚresourceéPr1Ú portNumberz list[WSP]rfr0Ú slowResourcer"úWebSocketFixture[WSP]cCsz||ƒ}tƒ}||_|j dt|ƒ¡|j dtddƒ¡|j dtƒ¡|j d|j¡|j  |j t |j|jd¡|S)Nsconnectsresources some-datazapplication/octet-streams processingsslow)rn) rdreroÚputChildr8rr rrrnÚ listenTCPrqr)Úclsrmr(Ú serverFactoryr)r)r*rg•s ÿÿzWebSocketFixture.newúhttp://localhost:80/connectrjrSrcÃs"t |j|¡}| |j¡IdHSr2)r6rgrnÚconnectrm)r(rjÚclientr)r)r*ry¥s€zWebSocketFixture.connectTÚgreetÚboolrcCs*t|jƒ}|j|d}|dusJdƒ‚|S)aP There should be a single websocket connection in progress; complete it. @param greet: Should we immediately issue a greeting, i.e. deliver any pending buffered TCP data, upon connection? If C{False}, the caller must call C{flush} on the result to deliver any pending data. ©r{NzConnection not in progress.)rrnÚ succeedOnce)r(r{Ú completerÚ succeededr)r)r*Úcomplete©s zWebSocketFixture.completeN)rmrlr"rs)rx)rjrSr"r)T)r{r|r"r)r,r-r.rrmrcrrnrrorqrbrfr0rrÚ classmethodrgryrr)r)r)r*rkŒs    rkz'wsproto library required for websocketsc@sfeZdZddd„Zddd„Zddd„Zdd d „Zdd d „Zdd d„Zddd„Z ddd„Z ddd„Z dS)ÚWebSocketTestsr"rAcCsŽt tƒ¡}t | ¡¡}| |¡| t|j j ƒd¡| t|j j ƒd¡|  ¡}|  |¡}| ¡}| |¡| ¡| |  |¡d¡dS)zÏ Connecting to a websocket server (installed with L{WebSocketResource}) from a websocket client (connected with L{WebSocketClientEndpoint}) results in a websocket connection. érTN)rkrgrir Ú fromCoroutineryÚassertNoResultÚ assertEqualÚlenrnÚ tcpServersÚ tcpClientsrÚsuccessResultOfr]Úflush)r(reÚ connectedÚpumpÚwsClientÚ requestedr)r)r*Útest_websocketºs    zWebSocketTests.test_websocketcCs\t tƒ¡}t | ¡¡}| ¡}| |¡}| ¡}|  |¡|  ¡|  | |¡d¡dS)zŠ Connecting to a websocket server and sending it a bytes message results in C{bytesMessageReceived} being called. rLN) rkrgrir r…ryrr‹r`r†rŒr‡)r(rerrŽrÚ bRequestedr)r)r*Útest_bytesMessageÌs   z WebSocketTests.test_bytesMessagecsêt tƒ¡}t | ¡¡| ¡}g‰ttƒG‡fdd„dƒƒ}|ƒ}|j dj   |¡|  ˆg¡|j j ¡|  ˆdg¡|j j ¡|  ˆddg¡|j j ¡|  ˆgd¢¡| |j j|¡|j dj  ¡| |j jd¡dS)zo Websocket transports can notify a producer of backpressure events via attachProducer. cs6eZdZd ‡fdd„ Zd ‡fdd„ Zd ‡fdd„ Zd S) z6WebSocketTests.test_backpressure..TestProducerr"rAcóˆ d¡dS)NÚpaused©rYrE©Úeventsr)r*ÚpauseProducingæózEWebSocketTests.test_backpressure..TestProducer.pauseProducingcr”)NÚresumedr–rEr—r)r*ÚresumeProducingéršzFWebSocketTests.test_backpressure..TestProducer.resumeProducingcr”)NÚstoppedr–rEr—r)r*Ú stopProducingìršzDWebSocketTests.test_backpressure..TestProducer.stopProducingNra)r,r-r.r™rœržr)r—r)r*Ú TestProduceräsrŸrr•r›)r•r›rN)rkrgrir r…ryrr r rfr@ÚattachProducerr‡ÚserverIOÚproducerr™rœržÚassertIsÚdetachProducer)r(rerŽrŸÚ testProducerr)r—r*Útest_backpressureÚs$      z WebSocketTests.test_backpressurecCsbt tƒ¡}t | ¡¡}| ¡}| |¡}|j  d¡|  |j g¡|  ¡|  |j dg¡dS)a6 Internally, Twisted's websocket resource responds to all ping requests with a pong as required by the spec, so peers can issue a C{ping} and receive a C{pong} message with the same payload; our websocket transport also implements a C{ping} method to send that message. s123N) rkrgrir r…ryrr‹r@Úpingr‡r>rŒ©r(rerrŽrr)r)r*Ú test_pingPongüs   zWebSocketTests.test_pingPongcCs¨t tƒ¡}t | ¡¡}| ¡}| |¡}| |j dj d¡| |j d¡|j   ¡| |j dj d¡| |j d¡|  ¡| |j dj d¡| |j d¡dS)zu When the underlying TCP connection is lost, L{WebSocketProtocol.connectionLost} is invoked. rN)rkrgrir r…ryrr‹r£rfr?r@ÚloseConnectionrŒÚ assertIsNotr¨r)r)r*Útest_serverConnectionLost s   z(WebSocketTests.test_serverConnectionLostcCsjt tƒ¡}t|jƒ}| dd¡}| |¡| ¡| |¡}|  |j t ¡t |ƒ}|  | |¡d¡dS)z« Attempting to issue an C{HTTP GET} against a websocket server (installed with L{WebSocketResource}) results in a C{BAD_REQUEST} response. sGETshttp://localhost/connects$websocket protocol negotiation errorN) rkrgrirrnr!r†rr‹r‡Úcoderr)r(reÚagentrTÚrÚbodyr)r)r*Útest_bads      ÿzWebSocketTests.test_badcCóJt tƒ¡}t | d¡¡}|jdd}| |¡| ¡|  |t ¡dS)úÁ Attempting to connect to a regular HTTP resource that does not support websockets will result in the Deferred returned from L{WebSocketClientEndpoint} failing. zhttp://localhost/emptyFr}N© rkrgrir r…ryrr†rŒÚfailureResultOfr5©r(rerrŽr)r)r*Útest_connectionRefused0s   z%WebSocketTests.test_connectionRefusedcCr²)r³zhttp://localhost/processingFr}Nr´r¶r)r)r*Útest_connectionRefusedWeird=s ÿ  z*WebSocketTests.test_connectionRefusedWeirdcCst tƒ¡}t | d¡¡}|jdd}| |¡|jj   d¡|  ¡|  |t ¡}|jj}| tƒ}¡| |jd¡|jj }|  d¡|  ¡| |jd¡|  d¡|  ¡| |jd¡| ¡| |jd¡|  ¡|jdusyJd ƒ‚| |jjt¡dS) z§ When the client connection is refused by an asynchronous HTTP response, the websocket client will not be notified until the response arrives. zhttp://localhost/slowTr}r%shellosworlds helloworldNzshould be closed now)rkrgrir r…ryrr†rrr!ÚwriterŒrµr5ÚvaluerTÚ deliverBodyr r‡rIÚfinishr£Ú closedReasonÚtyper )r(rerrŽÚrejectedÚrespÚpÚreqr)r)r*Útest_connectionRefusedSlowLs,      z)WebSocketTests.test_connectionRefusedSlowNra) r,r-r.r‘r“r¦r©r¬r±r·r¸rÃr)r)r)r*rƒ¸s    "    rƒN)@r3Ú __future__rÚ dataclassesrrÚtypingrrrÚunittestrÚzope.interfacer Útwisted.internet.deferr Útwisted.internet.errorr Útwisted.internet.interfacesr Útwisted.internet.testingr rÚtwisted.python.failurerÚtwisted.test.iosimrrÚtwisted.trial.unittestrÚtwisted.web._responsesrÚtwisted.web.clientrrÚtwisted.web.iwebrÚtwisted.web.resourcerÚtwisted.web.serverrrrÚtwisted.web.staticrrÚ shouldSkipr r0Ú __import__Ú ImportErrorÚtwisted.web.websocketr5r6r7rr8r9r:r;rdrirkrƒr)r)r)r*ÚsJ               ÿ$ 1+