U ¹~Hj5ã@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&ƒZ0z e1dƒWne2k rRdZ.YnbXddl3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m: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@seZdZdddœdd„ZdS)Ú WeirdResourcerÚbytes©ÚrequestÚreturncCs| 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ó)ZsetResponseCode©Úselfr"©r(úC/tmp/pip-unpacked-wheel-ewod1tgb/twisted/web/test/test_websocket.pyÚ render_GET#s zWeirdResource.render_GETN)Ú__name__Ú __module__Ú __qualname__r*r(r(r(r)r"src@s eZdZdZdddœ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Úintr!cCs ||_tS©N)r"rr&r(r(r)r*9szDelayedResponse.render_GETN)r+r,r-Ú__doc__r*r(r(r(r)r.2sr.ZwsprotoT)ÚConnectionRejectedÚWebSocketClientEndpointÚWebSocketClientFactoryrÚWebSocketResourceÚWebSocketServerFactoryÚWebSocketTransportc@sªeZdZUdZeedZded<dZded<dd d œd d „Z d d œdd„Z dd dœdd„Z dd dœdd„Z dd dœdd„Z dd dœdd„Zdd œdd „Zd!d œ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ÚwasLostr7ÚNone)Ú transportr#cCs ||_dSr0)r>)r'r>r(r(r)ÚnegotiationStartedWszMyWSP.negotiationStarted©r#cCsdSr0r(©r'r(r(r)ÚnegotiationFinishedZszMyWSP.negotiationFinishedr)Úreasonr#cCs ||_dSr0)r<)r'rCr(r(r)ÚconnectionLost]szMyWSP.connectionLostr )Údatar#cCs&|dkr|j d¡n |j |¡dS)Nórequestó responseÿ)r>ÚsendBytesMessageÚ bDeferredÚcallback©r'rEr(r(r)ÚbytesMessageReceived`szMyWSP.bytesMessageReceivedÚstrcCs&|dkr|j d¡n |j |¡dS)Nr"Úresponse)r>ÚsendTextMessageÚdeferredrJrKr(r(r)ÚtextMessageReceivedfszMyWSP.textMessageReceived)Úpayloadr#cCs|j |¡dSr0)r;Úappend)r'rRr(r(r)Ú pongReceivedlszMyWSP.pongReceivedz Deferred[str]cCstƒ|_|j d¡|jS)zR Send a text message to the server and expect a response. r")r rPr>rOrAr(r(r)Ú sendRequestos zMyWSP.sendRequestzDeferred[bytes]cCstƒ|_|j d¡|jS)zS Send a bytes message to the server and expect a response. rF)r rIr>rHrAr(r(r)Ú bytesRequestws zMyWSP.bytesRequest)r+r,r-r1rÚlistr;Ú__annotations__r<r?rBrDrLrQrTrUrVr(r(r(r)r8Ms  r8c@s&eZdZUded<dddœdd„ZdS) Ú MyFactoryzWebSocketFixture[Any]Úfixturerr8r!cCstƒ}|jj |¡|Sr0)r8rZÚserversrS)r'r"Únewr(r(r)Ú buildProtocol‚szMyFactory.buildProtocolN)r+r,r-rXr]r(r(r(r)rYs rYc@seZdZdddœdd„ZdS)ÚMyClientFactoryrMr8©Úurir#cCstƒSr0)r8)r'r`r(r(r)r]ˆszMyClientFactory.buildProtocolN)r+r,r-r]r(r(r(r)r^‡sr^c@s¨eZdZUeƒZded<eedZded<eedZ ded<dZ d ed <ee dZ d ed <ee dZd ed<edddœdd„ƒZd dddœdd„Zd!dddœdd„ZdS)"ÚWebSocketFixturezWebSocketClientFactory[WSP]Ú clientFactoryr9rÚreactorrÚresourceéPr/Ú portNumberz list[WSP]r[r.Ú slowResourcezWebSocketFixture[WSP])rbr#cCsz||ƒ}tƒ}||_|j dt|ƒ¡|j dtddƒ¡|j dtƒ¡|j d|j¡|j  |j t |j|jd¡|S)Nsconnectsresources some-datazapplication/octet-streams processingsslow)rc) rYrZrdZputChildr5rrrgrcZ listenTCPrfr)Úclsrbr'Z serverFactoryr(r(r)r\•sÿÿzWebSocketFixture.newúhttp://localhost:80/connectrMrr_cÃs t |j|¡}| |j¡IdHSr0)r3r\rcÚconnectrb)r'r`Úclientr(r(r)rj¥szWebSocketFixture.connectTÚboolr)Úgreetr#cCs*t|jƒ}|j|d}|dk s&tdƒ‚|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. ©rmNzConnection not in progress.)rrcZ succeedOnceÚAssertionError)r'rmZ completerZ succeededr(r(r)Úcomplete©s  zWebSocketFixture.completeN)ri)T)r+r,r-rrbrXrrcrrdrfrWr[r.rgÚ classmethodr\rjrpr(r(r(r)raŒs  raz'wsproto library required for websocketsc@sŠeZdZddœdd„Zddœdd„Zddœdd„Zddœd d „Zddœd d „Zddœd d„Zddœdd„Z ddœdd„Z ddœdd„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. érNN)rar\r^r Ú fromCoroutinerjÚassertNoResultÚ assertEqualÚlenrcZ tcpServersZ tcpClientsrpÚsuccessResultOfrUÚflush)r'rZÚ 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. rGN) rar\r^r rtrjrprxrVruryrv)r'rZrzr{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 ¡|  ˆdddg¡| |j j|¡|j dj  ¡| |j jd¡dS)zo Websocket transports can notify a producer of backpressure events via attachProducer. csBeZdZddœ‡fdd„ Zddœ‡fdd„ Zddœ‡fdd„ Zd S) z6WebSocketTests.test_backpressure..TestProducerr=r@csˆ d¡dS)NÚpaused©rSrA©Úeventsr(r)ÚpauseProducingæszEWebSocketTests.test_backpressure..TestProducer.pauseProducingcsˆ d¡dS)NÚresumedrrAr‚r(r)ÚresumeProducingészFWebSocketTests.test_backpressure..TestProducer.resumeProducingcsˆ d¡dS)NÚstoppedrrAr‚r(r)Ú stopProducingìszDWebSocketTests.test_backpressure..TestProducer.stopProducingN)r+r,r-r„r†rˆr(r‚r(r)Ú TestProduceräsr‰rr€r…r‡N)rar\r^r rtrjrpr r r[r>ZattachProducerrvZserverIOZproducerr„r†rˆÚassertIsZdetachProducer)r'rZr{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) rar\r^r rtrjrprxr>Zpingrvr;ry©r'rZrzr{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)rar\r^r rtrjrprxrŠr[r<r>ZloseConnectionryZ 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) rar\r^rrcr"rurprxrvÚcoderr)r'rZZagentrNÚrÚbodyr(r(r)Útest_bads     ÿzWebSocketTests.test_badcCsJt 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/emptyFrnN© rar\r^r rtrjrpruryÚfailureResultOfr2©r'rZrzr{r(r(r)Útest_connectionRefused0s    z%WebSocketTests.test_connectionRefusedcCsJt tƒ¡}t | d¡¡}|jdd}| |¡| ¡|  |t ¡dS)r“zhttp://localhost/processingFrnNr”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¡|  ¡|jdk sòtd ƒ‚| |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/slowTrnr%shellosworlds helloworldNzshould be closed now)rar\r^r rtrjrprurgr"Úwriteryr•r2ÚvaluerNZ deliverBodyr rvrEÚfinishrŠZ closedReasonroÚtyper )r'rZrzr{ZrejectedÚrespÚpÚreqr(r(r)Útest_connectionRefusedSlowLs,      z)WebSocketTests.test_connectionRefusedSlowN) r+r,r-r~rr‹rrŽr’r—r˜r r(r(r(r)rr¸s" rrN)@r1Ú __future__rZ 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 shouldSkiprr.Ú __import__Ú ImportErrorZtwisted.web.websocketr2r3r4rr5r6r7r8rYr^rarrr(r(r(r)ÚsF                $ 1+