a )¶(bÿã@s¢dZddlmZddlmZddlmZddlmZddl m Z ddl m Z m Z ddlmZee ƒGd d „d ƒƒZee eƒee ƒGd d „d eƒƒZee eƒd S)z$ Utilities for Twisted.names tests. é)Ú randrange)Ú implementer)Ú verifyClass)Ú IPv4Address)Úsucceed)Ú IReactorUDPÚ IUDPTransport)ÚClockc@sJeZdZdZdd„Zdd„Zdd„Zdd d „Zd d „Zd d„Z dd„Z dS)ÚMemoryDatagramTransporta¬ This L{IUDPTransport} implementation enforces the usual connection rules and captures sent traffic in a list for later inspection. @ivar _host: The host address to which this transport is bound. @ivar _protocol: The protocol connected to this transport. @ivar _sentPackets: A C{list} of two-tuples of the datagrams passed to C{write} and the addresses to which they are destined. @ivar _connectedTo: L{None} if this transport is unconnected, otherwise an address to which all traffic is supposedly sent. @ivar _maxPacketSize: An C{int} giving the maximum length of a datagram which will be successfully handled by C{write}. cCs"||_||_g|_d|_||_dS©N)Ú_hostÚ _protocolÚ _sentPacketsÚ _connectedToÚ_maxPacketSize)ÚselfÚhostÚprotocolÚ maxPacketSize©rú;lib/python3.9/site-packages/twisted/names/test/test_util.pyÚ__init__&s z MemoryDatagramTransport.__init__cCstdg|j¢RŽS)z_ Return the address which this transport is pretending to be bound to. ZUDP)rr ©rrrrÚgetHost-szMemoryDatagramTransport.getHostcCs |jdurtdƒ‚||f|_dS)z> Connect this transport to the given address. NzAlready connected)rÚ ValueError)rrÚportrrrÚconnect4s zMemoryDatagramTransport.connectNcCsH|dur|j}|durtdƒ‚t|ƒ|jkr4tdƒ‚|j ||f¡dS)z* Send the given datagram. NzNeed an addresszPacket too big)rrÚlenrrÚappend)rZdatagramZaddrrrrÚwrite<szMemoryDatagramTransport.writecCs|j ¡tdƒS)z+ Shut down this transport. N)r Z stopProtocolrrrrrÚ stopListeningHs z%MemoryDatagramTransport.stopListeningcCsdS©zC Dummy implementation to satisfy L{IUDPTransport}. Nr)rZenabledrrrÚsetBroadcastAllowedOsz+MemoryDatagramTransport.setBroadcastAllowedcCsdSr!rrrrrÚgetBroadcastAllowedUsz+MemoryDatagramTransport.getBroadcastAllowed)N) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrrr r"r#rrrrr s r c@s"eZdZdZdd„Zd dd„ZdS) Ú MemoryReactoraO An L{IReactorTime} and L{IReactorUDP} provider. Time is controlled deterministically via the base class, L{Clock}. UDP is handled in-memory by connecting protocols to instances of L{MemoryDatagramTransport}. @ivar udpPorts: A C{dict} mapping port numbers to instances of L{MemoryDatagramTransport}. cCst |¡i|_dSr )r rÚudpPortsrrrrrls zMemoryReactor.__init__Úé cCsZ|dkr tddƒ}||jvrq q||jvr2tdƒ‚t||f||ƒ}||j|<| |¡|S)zR Pretend to bind a UDP port and connect the given protocol to it. réizAddress in use)rr)rr ZmakeConnection)rrrZ interfacerZ transportrrrÚ listenUDPps     zMemoryReactor.listenUDPN)r*r+)r$r%r&r'rr-rrrrr(_s r(N)r'ZrandomrZzope.interfacerZzope.interface.verifyrZtwisted.internet.addressrZtwisted.internet.deferrZtwisted.internet.interfacesrrZtwisted.internet.taskr r r(rrrrÚs      G !