tfQ' ddlmZn#e$r edwxYwddlmZddlmZddlmZddlm Z m Z  ddl m Z n#e$r ddl m Z YnwxYwddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZdZeeGddeZee GddeZGdde eZGddeZGddeZdZ dZ!dS))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_agentMemoryReactorClockResolverceZdZddZdS)_StaticTestResolverrct|}|||tdd|dkrdn||dS)zR Implement IHostnameResolver which always returns 127.0.0.1:31337 TCP 127.0.0.1rizN)rresolutionBeganaddressResolvedrresolutionComplete)selfreceiverhostName portNumber resolutions `/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/autobahn/twisted/testing/__init__.pyresolveHostNamez#_StaticTestResolver.resolveHostName@sr$H--   ,,,  {Z1__EE* U U    ##%%%%%N)r)__name__ __module__ __qualname__r#r$r"rr>s( & & & & & &r$rc2eZdZdZdZedZdZdS)_TestNameResolverz9 A test version of IReactorPluggableNameResolver NcD|jt|_|jSN) _resolverrrs r" nameResolverz_TestNameResolver.nameResolverTs > !022DN~r$c"|j}||_|Sr,)r-)rresolverolds r"installNameResolverz%_TestNameResolver.installNameResolverZsn! r$)r%r&r'__doc__r-propertyr/r3r(r$r"r*r*LsMI X r$r*ceZdZdZdS)rzY Combine MemoryReactor, Clock and an IReactorPluggableNameResolver together. N)r%r&r'r4r(r$r"rr`s Dr$rc eZdZdZdZddZdS)_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|_dSr,) _reactor_server_protocol_pumperr_agentset_pumpsdict_servers)rreactorpumperserver_protocols r"__init__z_TwistedWebMemoryAgent.__init__xsE / 34=AA ee  r$Nc|d}j|||}|rjjd\}}}} } } njjd\}}}} } t dd|} t ddd} tddt_ tj d | | }| d}tj |d | | }|r*t|tt|ttj|||}j|fd }|||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 peerAddressFcj|<|Sr,)rA)protorrDs r" add_mappingz0_TwistedWebMemoryAgent.open..add_mappings#2DM% 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.opens%//99  ${// g) 0    UIMIabdIe FD$';;8< 8PQS8T 5D$+$UK>>$UK??//11 ?It 4 4 <&<&>&>O # . d&NDDD!..t44 . U&NDDD  > -} = = = -} = = =} -~?OQQ        ((555##r$r,)r%r&r'r4rErQr(r$r"r8r8hsA  6$6$6$6$6$6$r$r8c6eZdZdZdZdZdZdZdZdZ dS) _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). cbt|_d|_g|_ddlm}||_dS)NFr)rB)r>r?_pumping_waiting_for_stoptwisted.internetrB_global_reactor)rglobal_reactors r"rEz_Kalamazoo.__init__s<ee  !#>>>>>>-r$c:|j|dS)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$cf|jrdSd|_|jd|jdS)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; =  F  &&q$/:::::r$c|jst|jr1t}|j|d|_|St}|d|S)z :returns: a Deferred that fires when we have stopped pump()-ing Call from `tearDown()`, for example. FN)rllenrmrappendcallbackrds r"stopz_Kalamazoo.stopsl = C 677  A  " ) )! , , ,!DMH JJ 4r$c|jr6||jd|jdS|jD]}|dg|_dS)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_oncesq = ( KKMMM  * *3 @ @ @ @ @+ ! ! 4    %'D " " "r$c|j}tx}|_|D]D}||jjr |jjr/||EdS)zE Flush all data between pending client/server pairs. N)r?r>flushclientIO disconnectedserverIOrX)r old_pumps new_pumpsrrs r"rz_Kalamazoo._flushsrK "%%%' DK  A GGIIIz& 1:+B  MM!       r$N) r%r&r'r4rErXrvr}rurr(r$r"rjrjsx&... ; ; ;    ( ( (     r$rjctS)z4 return a new instance implementing IPumper )rjr(r$r"rrs <<r$c6|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"rrs!1 !'6? C 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<======= +G   ,+++++000000555555TTTTTTTT>;;;;;;;>>>========>88888888??????CCCCCC>>>>>>======    & & & & && & &  &  *++,+&     !35F   P$P$P$P$P$2P$P$P$fSSSSSSSSl D D D D Ds ? A  A