U ucQ'@sZzddlmZWnek r,edYnXddlmZddlmZddlmZddlm Z m Z zddl m Z Wn ek rddl m Z YnXddlmZdd lmZmZdd lmZdd lmZdd lmZdd lmZdZeeGdddeZee GdddeZGddde eZGdddeZGdddeZddZ ddZ!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_agentMemoryReactorClockResolverc@seZdZdddZdS)_StaticTestResolverrcCs<t|}|||tdd|dkr(dn||dS)zR Implement IHostnameResolver which always returns 127.0.0.1:31337 TCP 127.0.0.1rizN)rZresolutionBeganZaddressResolvedrZresolutionComplete)selfZreceiverZhostNameZ portNumber resolutionrjc:\users\psant\downloads\hotstarbackend\school\venv\Lib\site-packages\autobahn\twisted\testing\__init__.pyresolveHostName@s  z#_StaticTestResolver.resolveHostNameN)r)__name__ __module__ __qualname__rrrrrr>src@s(eZdZdZdZeddZddZdS)_TestNameResolverz9 A test version of IReactorPluggableNameResolver NcCs|jdkrt|_|jSN) _resolverrrrrr nameResolverTs z_TestNameResolver.nameResolvercCs|j}||_|Sr )r!)rZresolveroldrrrinstallNameResolverZsz%_TestNameResolver.installNameResolver)rrr__doc__r!propertyr#r%rrrrrLs  rc@seZdZdZdS)rzY Combine MemoryReactor, Clock and an IReactorPluggableNameResolver together. N)rrrr&rrrrr`src@s"eZdZdZddZdddZdS)_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` cCs2||_||_||_t|j|_t|_t|_dSr ) _reactor_server_protocol_pumperr _agentset_pumpsdict_servers)rreactorpumperserver_protocolrrr__init__xs  z_TwistedWebMemoryAgent.__init__Ncs|d}jj|||d}|r:jjd\}}}} } } njjd\}}}} } tdd|} tddd} tdddkrt _ t j d | | d }| d}t j |d | | d }|rt|tt|tt |||}j|fd d }|||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)ZisServerZ hostAddressZ peerAddressFcsj|<|Sr )r0)protorr3rr add_mappings z0_TwistedWebMemoryAgent.open..add_mapping) startswithr,openr)Z sslClientsZ tcpClientsrr*getattrrr7r Z FakeTransportZ buildProtocolr rconnectr+addZ addCallback)rZtransport_configoptionsr5Z is_secureZreal_client_protocolhostportr7Zcontext_factorytimeoutZ bindAddressZserver_addressZclient_addressZserver_transportZclientProtocolZclient_transportZpumpr:rr9rr<sP        z_TwistedWebMemoryAgent.open)N)rrrr&r4r<rrrrr(hs r(c@s@eZdZdZddZddZddZdd Zd d Zd d Z 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). cCs*t|_d|_g|_ddlm}||_dS)NFr)r1)r-r._pumping_waiting_for_stoptwisted.internetr1_global_reactor)rZglobal_reactorrrrr4s  z_Kalamazoo.__init__cCs|j|dS)zp Add a new IOPump. It will be removed when both its client and server are disconnected. N)r.r?)rprrrr?sz_Kalamazoo.addcCs$|jr dSd|_|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)rErH callLater _pump_oncer"rrrstartsz_Kalamazoo.startcCs@|jst|jr,t}|j|d|_|St}|d|S)z :returns: a Deferred that fires when we have stopped pump()-ing Call from `tearDown()`, for example. FN)rElenrFrappendcallbackrdrrrstops  z_Kalamazoo.stopcCs@|jr ||jd|jn|jD]}|dq&g|_dS)z{ flush all data from all our IOPump instances and schedule another iteration on the global reactor g?N)rE_flushrHrJrKrFrOrPrrrrKs   z_Kalamazoo._pump_oncecCsD|j}t}|_|D](}||jjr4|jjr4q||qdS)zE Flush all data between pending client/server pairs. N)r.r-flushZclientIO disconnectedZserverIOr?)rZ old_pumpsZ new_pumpsrIrrrrSs z_Kalamazoo._flushN) rrrr&r4r?rLrRrKrSrrrrrDs  rDcCstS)z4 return a new instance implementing IPumper )rDrrrrrsrcCs|dkr 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. N)rr()r1r2r3rrrrs rN)"Ztwisted.internet.interfacesr ImportErrorZtwisted.internet.deferrZtwisted.internet.addressrZtwisted.internet._resolverrrrZtwisted.internet.testingrZtwisted.test.proto_helpersZ twisted.testr Zzope.interfacer r Zautobahn.websocket.interfacesr Zautobahn.twisted.websocketr rr__all__objectrrrr(rDrrrrrrs:          SV