o ¯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ú>/usr/lib/python3/dist-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. ÚUDP)rr ©rrrrÚgetHost-szMemoryDatagramTransport.getHostcCs |jdur tdƒ‚||f|_dS)z> Connect this transport to the given address. NzAlready connected)rÚ ValueError)rrÚportrrrÚconnect4s zMemoryDatagramTransport.connectNcCsH|dur|j}|durtdƒ‚t|ƒ|jkrtdƒ‚|j ||f¡dS)z* Send the given datagram. NzNeed an addresszPacket too big)rrÚlenrrÚappend)rÚdatagramÚaddrrrrÚwrite<szMemoryDatagramTransport.writecCs|j ¡tdƒS)z+ Shut down this transport. N)r Ú stopProtocolrrrrrÚ stopListeningHs z%MemoryDatagramTransport.stopListeningcCódS©zC Dummy implementation to satisfy L{IUDPTransport}. Nr)rÚenabledrrrÚsetBroadcastAllowedOóz+MemoryDatagramTransport.setBroadcastAllowedcCr%r&rrrrrÚgetBroadcastAllowedUr)z+MemoryDatagramTransport.getBroadcastAllowedr ) Ú__name__Ú __module__Ú __qualname__Ú__doc__rrrr"r$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__Úé cCs\|dkr tddƒ}||jvrnq||jvrtdƒ‚t||f||ƒ}||j|<| |¡|S)zR Pretend to bind a UDP port and connect the given protocol to it. rTéizAddress in use)rr0rr ÚmakeConnection)rrrÚ interfacerÚ transportrrrÚ listenUDPps  ý   zMemoryReactor.listenUDPN)r1r2)r+r,r-r.rr7rrrrr/_s r/N)r.ÚrandomrÚzope.interfacerÚzope.interface.verifyrÚtwisted.internet.addressrÚtwisted.internet.deferrÚtwisted.internet.interfacesrrÚtwisted.internet.taskr r r/rrrrÚs       G!