o Óó·i5ã@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óN)ZsetResponseCode©Úselfr!©r(úeC:\Users\psant\Downloads\PAF backend\school\venv\Lib\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"r0)r+r,r-Ú__doc__r*r(r(r(r)r/2sr/ZwsprotoT)Ú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Ú transportr8r"ÚNonecCó ||_dSr1)r>)r'r>r(r(r)ÚnegotiationStartedWó zMyWSP.negotiationStartedcCsdSr1r(©r'r(r(r)ÚnegotiationFinishedZszMyWSP.negotiationFinishedÚreasonrcCr@r1)r=)r'rEr(r(r)ÚconnectionLost]rBzMyWSP.connectionLostÚdatar#cCó(|dkr |j d¡dS|j |¡dS)Nórequestó responseÿ)r>ÚsendBytesMessageÚ bDeferredÚcallback©r'rGr(r(r)ÚbytesMessageReceived`ózMyWSP.bytesMessageReceivedÚstrcCrH)Nr!Úresponse)r>ÚsendTextMessageÚdeferredrMrNr(r(r)ÚtextMessageReceivedfrPzMyWSP.textMessageReceivedÚpayloadcCs|j |¡dSr1)r<Úappend)r'rVr(r(r)Ú pongReceivedlszMyWSP.pongReceivedú Deferred[str]cCótƒ|_|j d¡|jS)zR Send a text message to the server and expect a response. r!N)r rTr>rSrCr(r(r)Ú sendRequestoó zMyWSP.sendRequestúDeferred[bytes]cCrZ)zS Send a bytes message to the server and expect a response. rIN)r rLr>rKrCr(r(r)Ú bytesRequestwr\zMyWSP.bytesRequest)r>r8r"r?©r"r?)rErr"r?)rGr#r"r?)rGrQr"r?)rVr#r"r?)r"rY)r"r])r+r,r-r2rÚlistr<Ú__annotations__r=rArDrFrOrUrXr[r^r(r(r(r)r9Ms         r9c@s eZdZUded<d dd„Zd S) Ú MyFactoryzWebSocketFixture[Any]Úfixturer!rr"r9cCstƒ}|jj |¡|Sr1)r9rcÚserversrW)r'r!Únewr(r(r)Ú buildProtocol‚szMyFactory.buildProtocolN)r!rr"r9)r+r,r-rarfr(r(r(r)rbs rbc@r) ÚMyClientFactoryÚurirQr"r9cCstƒSr1)r9)r'rhr(r(r)rfˆszMyClientFactory.buildProtocolN)rhrQr"r9)r+r,r-rfr(r(r(r)rg‡r.rgc@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éPr0Ú portNumberz list[WSP]rdr/Ú 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)rl) rbrcrmZputChildr6rr rprlZ listenTCPror)Úclsrkr'Z serverFactoryr(r(r)re•s ÿÿzWebSocketFixture.newúhttp://localhost:80/connectrhrQrcÃs"t |j|¡}| |j¡IdHSr1)r4rerlÚconnectrk)r'rhÚclientr(r(r)rt¥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. ©rvNzConnection not in progress.)rrlZ succeedOnce)r'rvZ completerZ succeededr(r(r)Úcomplete©s zWebSocketFixture.completeN)rkrjr"rq)rs)rhrQr"r)T)rvrwr"r)r+r,r-rrkrarrlrrmror`rdr/rpÚ classmethodrertryr(r(r(r)riŒs    riz'wsproto library required for websocketsc@sfeZdZddd„Zddd„Zddd„Zdd d „Zdd d „Zdd d„Zddd„Z ddd„Z ddd„Z dS)ÚWebSocketTestsr"r?cCsŽ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. érRN)rirergr Ú fromCoroutinertÚassertNoResultÚ assertEqualÚlenrlZ tcpServersZ tcpClientsryÚsuccessResultOfr[Úflush)r'rcÚ 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. rJN) rirergr r}rtryrr^r~r‚r)r'rcrƒr„r…Z 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"r?cóˆ d¡dS)NÚpaused©rWrC©Úeventsr(r)ÚpauseProducingæózEWebSocketTests.test_backpressure..TestProducer.pauseProducingcr‰)NÚresumedr‹rCrŒr(r)ÚresumeProducingérzFWebSocketTests.test_backpressure..TestProducer.resumeProducingcr‰)NÚstoppedr‹rCrŒr(r)Ú stopProducingìrzDWebSocketTests.test_backpressure..TestProducer.stopProducingNr_)r+r,r-rŽr‘r“r(rŒr(r)Ú TestProduceräsr”rrŠr)rŠrr’N)rirergr r}rtryr r rdr>ZattachProducerrZserverIOZproducerrŽr‘r“ÚassertIsZdetachProducer)r'rcr„r”Z 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) rirergr r}rtryrr>Zpingrr<r‚©r'rcrƒr„r…r(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)rirergr r}rtryrr•rdr=r>ZloseConnectionr‚Z 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) rirergrrlr!r~ryrrÚcoderr)r'rcZagentrRÚ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/emptyFrxN© rirergr r}rtryr~r‚ÚfailureResultOfr3©r'rcrƒr„r(r(r)Útest_connectionRefused0s   z%WebSocketTests.test_connectionRefusedcCrž)rŸzhttp://localhost/processingFrxNr 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/slowTrxr%shellosworlds helloworldNzshould be closed now)rirergr r}rtryr~rpr!Úwriter‚r¡r3ÚvaluerRZ deliverBodyr rrGÚfinishr•Z closedReasonÚtyper )r'rcrƒr„ZrejectedÚrespÚpÚreqr(r(r)Útest_connectionRefusedSlowLs,      z)WebSocketTests.test_connectionRefusedSlowNr_) r+r,r-r‡rˆr–r˜r™rr£r¤r¬r(r(r(r)r{¸s    "    r{N)@r2Ú __future__rÚ dataclassesrrÚtypingrrrZunittestrZzope.interfacer Ztwisted.internet.deferr Ztwisted.internet.errorr Ztwisted.internet.interfacesr Ztwisted.internet.testingr rZtwisted.python.failurerZtwisted.test.iosimrrZtwisted.trial.unittestrZtwisted.web._responsesrZtwisted.web.clientrrZtwisted.web.iwebrZtwisted.web.resourcerZtwisted.web.serverrrrZtwisted.web.staticrrZ shouldSkipr r/Ú __import__Ú ImportErrorZtwisted.web.websocketr3r4r5rr6r7r8r9rbrgrir{r(r(r(r)ÚsJ               ÿ$ 1+