U  W[Kã@s´dZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZeeƒGd d „d eƒƒZe eeƒeeƒGd d „d eƒƒZe eeƒdS)z$ Utilities for Twisted.names tests. é)ÚdivisionÚabsolute_import)Ú randrange)Ú implementer)Ú verifyClass)Ú IPv4Address)Úsucceed)ÚClock)Ú IReactorUDPÚ IUDPTransportc@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__cCstd|jžŽS)z_ Return the address which this transport is pretending to be bound to. ÚUDP)r)rr©rrrrÚgetHost/szMemoryDatagramTransport.getHostcCs |jdk rtdƒ‚||f|_dS)z> Connect this transport to the given address. NzAlready connected)rÚ ValueError)rrÚportrrrÚconnect7s zMemoryDatagramTransport.connectNcCsH|dkr|j}|dkrtdƒ‚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)rZ stopProtocolrrrrrÚ stopListeningMs z%MemoryDatagramTransport.stopListeningcCsdS©zC Dummy implementation to satisfy L{IUDPTransport}. Nr)rÚenabledrrrÚsetBroadcastAllowedUsz+MemoryDatagramTransport.setBroadcastAllowedcCsdSr$rrrrrÚgetBroadcastAllowed\sz+MemoryDatagramTransport.getBroadcastAllowed)N) Ú__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ÚudpPortsrrrrrss zMemoryReactor.__init__Úé cCsZ|dkr tddƒ}||jkrq q||jkr2tdƒ‚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Ú listenUDPxs   ÿ  zMemoryReactor.listenUDPN)r.r/)r(r)r*r+rr1rrrrr,gs r,N)r+Z __future__rrZrandomrZzope.interfacerZzope.interface.verifyrZtwisted.internet.addressrZtwisted.internet.deferrZtwisted.internet.taskr Ztwisted.internet.interfacesr r Úobjectr r,rrrrÚs      L !