hQ' ddlmZddlmZddlmZddlmZddlm Z m Z  ddl m Z ddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZdZeeGddeZee GddeZGdde eZGddeZGddeZdZ dZ!y#e$r edwxYw#e$r ddl m Z YwxYw))IHostnameResolverz=Twisted 17.1.0 or later required for autobahn.twisted.testing)Deferred) IPv4Address)HostResolution) ISSLTransportIReactorPluggableNameResolver)MemoryReactorClock)iosim)directlyProvides implementer)IWebSocketClientAgent)_TwistedWebSocketClientAgent)WebSocketServerProtocol)WebSocketServerFactory) create_pumpercreate_memory_agentMemoryReactorClockResolverceZdZddZy)_StaticTestResolverct|}|j||jtdd|dk(rdn||j y)zR Implement IHostnameResolver which always returns 127.0.0.1:31337 TCP 127.0.0.1rizN)rresolutionBeganaddressResolvedrresolutionComplete)selfreceiverhostName portNumber resolutions ]/var/www/Befach/backend/env/lib/python3.12/site-packages/autobahn/twisted/testing/__init__.pyresolveHostNamez#_StaticTestResolver.resolveHostName@sL$H-   ,  {Z1_E* U  ##%N)r)__name__ __module__ __qualname__r#r$r"rr>s &r$rc*eZdZdZdZedZdZy)_TestNameResolverz9 A test version of IReactorPluggableNameResolver NcP|jt|_|jSN) _resolverrrs r" nameResolverz_TestNameResolver.nameResolverTs >> !02DN~~r$c,|j}||_|Sr,)r-)rresolverolds r"installNameResolverz%_TestNameResolver.installNameResolverZsnn! r$)r%r&r'__doc__r-propertyr/r3r(r$r"r*r*Ls%I  r$r*ceZdZdZy)rzY Combine MemoryReactor, Clock and an IReactorPluggableNameResolver together. N)r%r&r'r4r(r$r"rr`s  r$rceZdZdZdZddZy)_TwistedWebMemoryAgenta A testing agent which will hook up an instance of `server_protocol` for every client that is created via the `open` API call. :param reactor: the reactor to use for tests (usually an instance of MemoryReactorClockResolver) :param pumper: an implementation IPumper (e.g. as returned by `create_pumper`) :param server_protocol: the server-side WebSocket protocol class to instantiate (e.g. a subclass of `WebSocketServerProtocol` c||_||_||_t|j|_t |_t|_yr,) _reactor_server_protocol_pumperr_agentset_pumpsdict_servers)rreactorpumperserver_protocols r"__init__z_TwistedWebMemoryAgent.__init__xs< / 34==A e  r$Nc|jd}jj|||}|r!jjd\}}}} } } njj d\}}}} } t dd|} t ddd} jtddt_ tjd | | }|jd}tj|d | | }|r t|tt|ttj |||}j"j%|fd }|j'||S) a} Implement IWebSocketClientAgent with in-memory transports. :param transport_config: a string starting with 'wss://' or 'ws://' :param options: a dict containing options :param protocol_class: the client protocol class to instantiate (or `None` for defaults, which is to use `WebSocketClientProtocol`) zwss://)protocol_classrrrfactoryNT)isServer hostAddress peerAddressFc&j|<|Sr,)rA)protorrDs r" add_mappingz0_TwistedWebMemoryAgent.open..add_mappings#2DMM% Lr$) startswithr=openr: sslClients tcpClientsrr;getattrrrIr FakeTransport buildProtocolr rconnectr<add addCallback)rtransport_configoptionsrG is_securereal_client_protocolhostportrIcontext_factorytimeout bindAddressserver_addressclient_addressserver_transportclientProtocolclient_transportpumprOrDs` @r"rQz_TwistedWebMemoryAgent.opensp%//9  ${{// g) 0  IMIaIabdIe FD$';8< 8P8PQS8T 5D$+$UK>$UK?//1 ?It 4 <&<&>O # .. d&ND!..t4 .. U&ND  -} = -} =}} -~?OQ   ((5##r$r,)r%r&r'r4rErQr(r$r"r8r8hs 6$r$r8c4eZdZdZdZdZdZdZdZdZ y) _Kalamazooa Feeling whimsical about class names, see https://en.wikipedia.org/wiki/Handcar This is 'an IOPump pumper', an object which causes a series of IOPumps it is monitoring to do their I/O operations periodically. This needs the 'real' reactor which trial drives, because reasons: - so @inlineCallbacks / async-def functions work (if I could explain exactly why here, I would) - we need to 'break the loop' of synchronous calls somewhere and polluting the tests themselves with that is bad - get rid of e.g. .flush() calls in tests themselves (thus 'teaching' the tests about details of I/O scheduling that they shouldn't know). cXt|_d|_g|_ddlm}||_y)NFr)rB)r>r?_pumping_waiting_for_stoptwisted.internetrB_global_reactor)rglobal_reactors r"rEz_Kalamazoo.__init__s%e  !#>-r$c:|jj|y)zp Add a new IOPump. It will be removed when both its client and server are disconnected. N)r?rX)rps r"rXz_Kalamazoo.adds r$cx|jryd|_|jjd|jy)z Begin triggering I/O in all IOPump instances we have. We will keep periodically 'pumping' our IOPumps until `.stop()` is called. Call from `setUp()` for example. NTr)rlro callLater _pump_oncer.s r"startz_Kalamazoo.starts/ ==   &&q$//:r$c|jst|jr.t}|jj |d|_|St}|j d|S)z :returns: a Deferred that fires when we have stopped pump()-ing Call from `tearDown()`, for example. FN)rllenrmrappendcallbackrds r"stopz_Kalamazoo.stopsW ==C 6 67 A  " " ) )! ,!DMH J 4r$c|jr7|j|jjd|jy|j D]}|j dg|_y)z{ flush all data from all our IOPump instances and schedule another iteration on the global reactor g?N)rl_flushrortrurmrzr{s r"ruz_Kalamazoo._pump_oncesR == KKM  * *3 @++ 4 ,%'D "r$c|j}tx}|_|D]P}|j|jjr|j jr@|j |Ry)zE Flush all data between pending client/server pairs. N)r?r>flushclientIO disconnectedserverIOrX)r old_pumps new_pumpsrrs r"rz_Kalamazoo._flushsVKK "%%' DKA GGIzz&&1::+B+B MM!  r$N) r%r&r'r4rErXrvr}rurr(r$r"rjrjs%&. ;  ( r$rjctS)z4 return a new instance implementing IPumper )rjr(r$r"rrs  <r$c,|t}t|||S)a) return a new instance implementing `IWebSocketClientAgent`. connection attempts will be satisfied by traversing the Upgrade request path starting at `resource` to find a `WebSocketResource` and then exchange data between client and server using purely in-memory buffers. )rr8)rBrCrDs r"rrs1 !'6? CCr$N)"twisted.internet.interfacesr ImportErrortwisted.internet.deferrtwisted.internet.addressrtwisted.internet._resolverrrrtwisted.internet.testingr twisted.test.proto_helpers twisted.testr zope.interfacer r autobahn.websocket.interfacesr autobahn.twisted.websocketrrr__all__objectrr*rr8rjrrr(r$r"rs<= ,05T>;8?C>=   && &  & *+,& !35F P$2P$fSSl Dq G >=>sB&B7&B47 CC