Vh5@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 ddZ-dZ.Gdde&Z/Gdde&Z0 e1dddl2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9eGdde6Z:Gd d!e8e:Z;Gd"d#e5e:Ze e.d'Gd(d)eZ?y*#e=$rd$Z.Y4wxYw)+z% Tests for L{twisted.web.websocket}. ) annotations) dataclassfield)AnyGenericTypeVar)skipIf) implementer)Deferred)ConnectionDone) IPushProducer)AccumulatingProtocolMemoryReactorClock)Failure)ConnectionCompleterIOPump)SynchronousTestCase) BAD_REQUEST)AgentreadBody)IRequest)Resource) NOT_DONE_YETRequestSite)DataWSPWebSocketProtocol)boundFceZdZddZy) WeirdResourcec&|jdy)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)setResponseCodeselfrequests O/home/dcms/DCMS/lib/python3.12/site-packages/twisted/web/test/test_websocket.py render_GETzWeirdResource.render_GET#s $r$N)r(rreturnbytes)__name__ __module__ __qualname__r*r$r)r!r!"s r$r!ceZdZdZddZy)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. c||_tSN)r(rr&s r)r*zDelayedResponse.render_GET9s r$N)r(rr+int)r-r.r/__doc__r*r0r$r)r2r22s  r$r2wsproto)ConnectionRejectedWebSocketClientEndpointWebSocketClientFactoryrWebSocketResourceWebSocketServerFactoryWebSocketTransportc|eZdZUdZeeZded<dZded<ddZ dd Z dd Z dd Z dd Z dd ZddZddZy)MyWSPz Used to implement both client-side and server-side WebSocket application that will help with testing. default_factoryz list[bytes]pongsNzFailure | NonewasLostc||_yr4) transport)r'rEs r)negotiationStartedzMyWSP.negotiationStartedWs &DNr$cyr4r0r's r)negotiationFinishedzMyWSP.negotiationFinishedZs r$c||_yr4)rC)r'reasons r)connectionLostzMyWSP.connectionLost]s !DLr$c||dk(r|jjdy|jj|y)Nrequest response)rEsendBytesMessage bDeferredcallbackr'datas r)bytesMessageReceivedzMyWSP.bytesMessageReceived`s/z!//0GH''-r$c||dk(r|jjdy|jj|y)Nr(response)rEsendTextMessagedeferredrRrSs r)textMessageReceivedzMyWSP.textMessageReceivedfs.y ..z: &&t,r$c:|jj|yr4)rBappend)r'payloads r) pongReceivedzMyWSP.pongReceivedls JJ  g &r$cnt|_|jjd|jS)zR Send a text message to the server and expect a response. r()r rYrErXrHs r) sendRequestzMyWSP.sendRequestos),4:DM NN * *9 5== r$cnt|_|jjd|jS)zS Send a bytes message to the server and expect a response. rN)r rQrErPrHs r) bytesRequestzMyWSP.bytesRequestws)/7jDN NN + +J 7>> !r$)rEr=r+Noner+rc)rKrr+rc)rTr,r+rc)rTstrr+rc)r]r,r+rc)r+z Deferred[str])r+zDeferred[bytes])r-r.r/r6rlistrB__annotations__rCrFrIrLrUrZr^r`rbr0r$r)r?r?MsJ #48{8"&& '  " .  -  ' ! "r$r?c eZdZUded<ddZy) MyFactoryzWebSocketFixture[Any]fixturecdt}|jjj||Sr4)r?rjserversr\)r'r(news r) buildProtocolzMyFactory.buildProtocols&'C LL ' ' ,Jr$N)r(rr+r?)r-r.r/rgrnr0r$r)riris && r$riceZdZddZy)MyClientFactoryctSr4)r?)r'uris r)rnzMyClientFactory.buildProtocols 7Nr$N)rrrer+r?)r-r.r/rnr0r$r)rprps r$rpTceZdZUeZded<eeZded<eeZ ded<dZ d ed <ee Z d ed <ee Zd ed<eddZdddZdddZy)WebSocketFixtureWebSocketClientFactory[WSP] clientFactoryr@rreactorrresourcePr5 portNumberz list[WSP]rlr2 slowResourcec||}t}||_|jjdt ||jjdt dd|jjdt |jjd|j|jj|jt|j|j|S)Nsconnectsresources some-datazapplication/octet-streams processingsslow)rw) rirjrxputChildr;rr!r{rw listenTCPrzr)clsrvr' serverFactorys r)rmzWebSocketFixture.news=!! $  z+<]+KL  l,FG  }mo> w(9(9:  OOT$--F  r$cKtj|j|}|j|jd{S7wr4)r9rmrwconnectrv)r'rrclients r)rzWebSocketFixture.connects7(,,T\\3?^^D$6$67777s?AAAcft|j}|j|}|Jd|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. greetzConnection not in progress.)rrw succeedOnce)r'r completer succeededs r)completezWebSocketFixture.completes=( 5 )))6 $C&CC$r$N)rvrur+zWebSocketFixture[WSP])zhttp://localhost:80/connect)rrrer+r)T)rboolr+r)r-r.r/rrvrgrrwrrxrzrfrlr2r{ classmethodrmrrr0r$r)rtrtss16M.8"'8J"KG Kx8Hh8Jt4GY4$)/$JL/J  8 r$rtz'wsproto library required for websocketscTeZdZd dZd dZd dZd dZd dZd dZd dZ d dZ d d Z y ) WebSocketTestscHtjt}tj|j }|j ||jt|jjd|jt|jjd|j}|j|}|j}|j ||j|j|j|dy)z Connecting to a websocket server (installed with L{WebSocketResource}) from a websocket client (connected with L{WebSocketClientEndpoint}) results in a websocket connection. rWN)rtrmrpr fromCoroutinerassertNoResult assertEquallenrw tcpServers tcpClientsrsuccessResultOfr`flush)r'rj connectedpumpwsClient requesteds r)test_websocketzWebSocketTests.test_websockets #&&'89**7??+<=  I& W__778!< W__778!<!'' 2((*  I&  --i8*Er$cjtjt}tj|j }|j }|j|}|j}|j||j|j|j|dy)z Connecting to a websocket server and sending it a bytes message results in C{bytesMessageReceived} being called. rON) rtrmrpr rrrrrbrrr)r'rjrrr bRequesteds r)test_bytesMessagez WebSocketTests.test_bytesMessages #&&'89**7??+<= !'' 2**,  J'  --j9;RSr$ctjt}tj|j |j }gttGfdd}|}|jdjj||jg|jjj|jdg|jjj!|jddg|jjj#|jgd|j%|jj||jdjj'|j%|jjdy)zo Websocket transports can notify a producer of backpressure events via attachProducer. c2eZdZdfd Zdfd Zdfd Zy)6WebSocketTests.test_backpressure..TestProducerc(jdy)Npausedr\r'eventss r)pauseProducingzEWebSocketTests.test_backpressure..TestProducer.pauseProducings h'r$c(jdy)Nresumedrrs r)resumeProducingzFWebSocketTests.test_backpressure..TestProducer.resumeProducing i(r$c(jdy)Nstoppedrrs r) stopProducingzDWebSocketTests.test_backpressure..TestProducer.stopProducingrr$Nrd)r-r.r/rrr)rsr) TestProducerrs ( ) )r$rrrr)rrrN)rtrmrpr rrrr r rlrEattachProducerrserverIOproducerrrrassertIsdetachProducer)r'rjrr testProducerrs @r)test_backpressurez WebSocketTests.test_backpressuresW #&&'89w01! ] # ) ) $ )$~ $$33LA $ --/ (, ..0 (I!67 ,,. !AB dmm,,l;$$335 dmm,,d3r$ctjt}tj|j }|j }|j|}|jjd|j|jg|j|j|jdgy)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) rtrmrpr rrrrrEpingrrBrr'rjrrrs r) test_pingPongzWebSocketTests.test_pingPongs#&&'89**7??+<= !'' 2' ,  &2r$ctjt}tj|j }|j }|j|}|j|jdjd|j|jd|jj|j|jdjd|j|jd|j|j|jdjd|j|jdy)zu When the underlying TCP connection is lost, L{WebSocketProtocol.connectionLost} is invoked. rN)rtrmrpr rrrrrrlrCrEloseConnectionr assertIsNotrs r)test_serverConnectionLostz(WebSocketTests.test_serverConnectionLost s #&&'89**7??+<= !'' 2 gooa(00$7 h&&-))+ gooa(00$7 h&&-  +33T: ))40r$ctjt}t|j}|j dd}|j ||j|j|}|j|jtt|}|j|j|dy)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) rtrmrprrwr(rrrrcoderr)r'rjagentrWrbodys r)test_badzWebSocketTests.test_bads #&&'89goo&==)DE H%   * -{    &(O r$ctjt}tj|j d}|j d}|j||j|j|ty) 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/emptyFrN rtrmrpr rrrrrfailureResultOfr8r'rjrrs r)test_connectionRefusedz%WebSocketTests.test_connectionRefused0sl #&&'89**7??;S+TU e, I&  Y(:;r$ctjt}tj|j d}|j d}|j||j|j|ty)rzhttp://localhost/processingFrNrrs r)test_connectionRefusedWeirdz*WebSocketTests.test_connectionRefusedWeird=sq #&&'89** OO9 : e, I&  Y(:;r$ctjt}tj|j d}|j d}|j||jjjd|j|j|t}|jj}|j!t#x}|j%|j&d|jj}|jd|j|j%|j&d|jd|j|j%|j&d|j)|j+|j,d|j|j,Jd |j%|j,j.t0y) 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/slowTrr$shellosworlds helloworldNzshould be closed now)rtrmrpr rrrrr{r(writerrr8valuerW deliverBodyrrrTfinishr closedReasontyper )r'rjrrrejectedresppreqs r)test_connectionRefusedSlowz)WebSocketTests.test_connectionRefusedSlowLsy #&&'89**7??;R+ST d+ I&$$**3/ '' 3EF~~&& 2445 %""** (  * (  /  annd+ ~~)A+AA) ,,n=r$Nrd) r-r.r/rrrrrrrrrr0r$r)rrs2F$ T 4D3 1$ $ < <>r$rN)@r6 __future__r dataclassesrrtypingrrrunittestr zope.interfacer twisted.internet.deferr twisted.internet.errorr twisted.internet.interfacesr twisted.internet.testingrrtwisted.python.failurertwisted.test.iosimrrtwisted.trial.unittestrtwisted.web._responsesrtwisted.web.clientrrtwisted.web.iwebrtwisted.web.resourcertwisted.web.serverrrrtwisted.web.staticrr shouldSkipr!r2 __import__twisted.web.websocketr8r9r:rr;r<r=r?rirp ImportErrorrtrr0r$r)rs.#(((&+15M*:6..%)::# e./   H  h Ky/"!/"/"b*5107  (ws|( (V =>o>(o>?o>sJs"DDD