U c6{i@sdZddlmZddlmZmZmZmZmZddl m Z ddl m Z m Z mZm Z ddlmZddlmZddlmZmZdd lmZmZmZmZmZdd lmZzdd l mZWnek rd ZYnXeZdd l m!Z!m"Z"ddl#m$Z$m%Z%m&Z&m'Z'm(Z(ddl)m*Z*ddl+m,Z,m-Z-m.Z.ddl/m0Z0m1Z1ddl2m3Z3m4Z4m5Z5ddl6m7Z7m8Z8m9Z9m:Z:ddl;mZ>ddl?m@Z@ddlAmBZBmCZCmDZDmEZEmFZFddlGmHZHddlImJZJmKZKmLZLmMZMmNZNddlOmPZPddlQmRZRddlSmTZTddlUmVZVmWZWmXZXddlYmZZZddl[m\Z\m]Z]ddl^m_Z_dd l`maZaeZd!Zbd"ZceZd!d k rDd#ZcGd$d%d%Zdd&d'ZeGd(d)d)eDZfGd*d+d+eBZge"e7Gd,d-d-eBZhGd.d/d/edeHeCZiGd0d1d1edeHZjGd2d3d3ZkGd4d5d5ZlGd6d7d7ZmGd8d9d9elemeHeMZnGd:d;d;ekemeHeMZoGdd?d?eHeEZteqretsee]d@kdAGdBdCdCeaZud S)Dz/ Tests for implementations of L{IReactorUNIX}. )md5)closefstatstatunlinkurandom)pformat)AF_INET SOCK_STREAM SOL_SOCKETsocket)S_IMODE)pack)mkstempmktemp)AnyCallableOptionalSequenceType)skipIf)AF_UNIXN) Interface implementer)baseerror interfacesreactortcp) UNIXAddress)Deferredfail gatherResults)UNIXClientEndpointUNIXServerEndpoint)CannotListenErrorConnectionClosedFileDescriptorOverrun)IFileDescriptorReceiver IReactorFDSetIReactorSocket IReactorUNIX) ClientFactoryDatagramProtocol ServerFactory) LoopingCall)ConnectableProtocolConnectionTestsMixinEndpointCreatorStreamClientTestsMixinrunProtocolsWithReactor)ReactorBuilder)ClosingFactoryMyClientFactoryMyServerFactoryStreamTransportTestsMixinWriteSequenceTestsMixin) nativeString)Failure)_coerceToFilesystemEncoding) addObservererrremoveObserver) requireModule)platform platformType)AccumulatingProtocol)TestCaseztwisted.python.sendmsgz>sendmsg extension unavailable, extended UNIX features disabledc@seZdZdZddZdS)UNIXFamilyMixinzK Test-helper defining mixin for things related to AF_UNIX sockets. cCsBd}|}t|||||d}||tt|j|dS)z} Assert that the mode of the created unix socket is set to the mode specified to the reactor method. i)modeN) buildReactorgetattr stopListening assertEqualr rst_mode)self methodNamepathfactoryrHrZunixPortrRC/tmp/pip-unpacked-wheel-hfbvu0bz/twisted/internet/test/test_unix.py _modeTestRs zUNIXFamilyMixin._modeTestN)__name__ __module__ __qualname____doc__rTrRrRrRrSrGMsrGcCsttdS)zI Return a new, unique abstract namespace path to be listened on. d)rr hexdigest)ZcaserRrRrS _abstractPath^sr[c@s>eZdZUdZejfZeee e e d<ddZ ddZ dS) UNIXCreatorz( Create UNIX socket end points. requiredInterfacescCstddd}t||S)z3 Construct a UNIX server endpoint. .sock.suffixdir)rr$)rNrrPrRrRrSserverls zUNIXCreator.servercCs t||jS)z3 Construct a UNIX client endpoint. )r#name)rNrZ serverAddressrRrRrSclienttszUNIXCreator.clientN)rUrVrWrXrr+r]rrrr__annotations__rcrerRrRrRrSr\es r\c@s,eZdZdZdZddZddZddZdS) SendFileDescriptorz L{SendFileDescriptorAndBytes} sends a file descriptor and optionally some normal bytes and then closes its connection. @ivar reason: The reason the connection was lost, after C{connectionLost} is called. NcCs||_||_dS)z @param fd: A C{int} giving a file descriptor to send over the connection. @param data: A C{str} giving data to send over the connection, or L{None} if no data is to be sent. N)fddata)rNrhrirRrRrS__init__szSendFileDescriptor.__init__cCs0|j|j|jr"|j|j|jdS)zn Send C{self.fd} and, if it is not L{None}, C{self.data}. Then close the connection. N) transportsendFileDescriptorrhriwriteloseConnectionrNrRrRrSconnectionMadesz!SendFileDescriptor.connectionMadecCst||||_dSN)r0connectionLostreasonrNrsrRrRrSrrs z!SendFileDescriptor.connectionLost)rUrVrWrXrsrjrprrrRrRrRrSrg{s   rgc@s8eZdZdZdZdZddZddZddZd d Z dS) ReceiveFileDescriptora} L{ReceiveFileDescriptor} provides an API for waiting for file descriptors to be received. @ivar reason: The reason the connection was lost, after C{connectionLost} is called. @ivar waiting: A L{Deferred} which fires with a file descriptor once one is received, or with a failure if the connection is lost with no descriptor arriving. NcCs&|jdkrt|_|jSt|jSdS)z Return a L{Deferred} which will fire with the next file descriptor received, or with a failure if the connection is or has already been lost. N)rsr waitingr!rorRrRrSwaitForDescriptors z'ReceiveFileDescriptor.waitForDescriptorcCs|j|d|_dS)z Fire the waiting Deferred, initialized by C{waitForDescriptor}, with the file descriptor just received. N)rvcallback)rN descriptorrRrRrSfileDescriptorReceiveds z,ReceiveFileDescriptor.fileDescriptorReceivedcCs0|jdk r,|jttd|dd|_dS)a_ Fail the waiting Deferred, if it has not already been fired by C{fileDescriptorReceived}. The bytes sent along with a file descriptor are guaranteed to be delivered to the protocol's C{dataReceived} method only after the file descriptor has been delivered to the protocol's C{fileDescriptorReceived}. NzReceived bytes (z) before descriptor.)rverrbackr< ExceptionrNrirRrRrS dataReceiveds  z"ReceiveFileDescriptor.dataReceivedcCs2t|||jdk r(|j|d|_||_dS)zj Fail the waiting Deferred, initialized by C{waitForDescriptor}, if there is one. N)r0rrrvr{rsrtrRrRrSrrs    z$ReceiveFileDescriptor.connectionLost) rUrVrWrXrsrvrwrzr~rrrRrRrRrSrus  ruc@seZdZdZefZeZddZe e dddZ ddZ e e dd d Zd d Ze e ed dZe e eddZe e eddZddZe e eddZe e de e eddZe e eddZe e eddZe e eddZd S)!UNIXTestsBuilderz= Builder defining tests relating to L{IReactorUNIX}. cCs|d|tdS)zs The UNIX socket created by L{IReactorUNIX.listenUNIX} is created with the mode specified. listenUNIXN)rTrr.rorRrRrS test_modeszUNIXTestsBuilder.test_mode8Abstract namespace UNIX sockets only supported on Linux.cCs>t|}|}|d|t}||td|dS)z On Linux, a UNIX socket path may begin with C{''} to indicate a socket in the abstract namespace. L{IReactorUNIX.listenUNIX} accepts such a path. N)r[rIrr.rLgetHostrrNrPrportrRrRrS#test_listenOnLinuxAbstractNamespaces z4UNIXTestsBuilder.test_listenOnLinuxAbstractNamespacec CsHdd}|tjd||}|t|dtW5QRXdS)z L{IReactorUNIX.listenUNIX} raises L{CannotListenError} if the underlying port's createInternetSocket raises a socket error. cSs tddS)Nz FakeBasePort forced socket.error)OSErrorrorRrRrSraiseSocketErrorsz=UNIXTestsBuilder.test_listenFailure..raiseSocketErrorZcreateInternetSocketznot-usedN)patchrZBasePortrI assertRaisesr%rr.)rNrrrRrRrStest_listenFailures  z#UNIXTestsBuilder.test_listenFailurecCs>t|}|}|d|t}||td|dS)zc L{IReactorUNIX.connectUNIX} also accepts a Linux abstract namespace path. rN)r[rI connectUNIXr,rLZgetDestinationr)rNrPrZ connectorrRrRrS$test_connectToLinuxAbstractNamespace s z5UNIXTestsBuilder.test_connectToLinuxAbstractNamespacecCs`Gdddt}|}|}t||||j||jd|jd||jd|jddS)z A client's transport's C{getHost} and C{getPeer} return L{UNIXAddress} instances which have the filesystem path of the host and peer ends of the connection. c@seZdZddZdS)z4UNIXTestsBuilder.test_addresses..SaveAddresscSs"t||d|_|dS)N)hostpeer)dictrgetPeer addressesrn)rNrkrRrRrSmakeConnection#s zCUNIXTestsBuilder.test_addresses..SaveAddress.makeConnectionN)rUrVrWrrRrRrRrS SaveAddress"srrrN)r0r4 endpointsrLr)rNrrcrerRrRrStest_addressess zUNIXTestsBuilder.test_addressescsddlmtdtdt}|}fdd}|||t d| fdd t |j d S) z L{IUNIXTransport.sendFileDescriptor} accepts an integer file descriptor and sends a copy of it to the process reading from the connection. r)fromfd)rFrjunkcs@|tt}t|||dSrq)r r rrL getsocknameZassertNotEqualfileno)ryZreceived)rsrNrRrScheckDescriptor@s zAUNIXTestsBuilder.test_sendFileDescriptor..checkDescriptorz-Sending file descriptor encountered a problemcs jSrqrkrnZignoredrcrRrSPz:UNIXTestsBuilder.test_sendFileDescriptor..N) r rbindrgrrurw addCallback addErrbackr?addBothr4r)rNredrrR)rrrNrcrStest_sendFileDescriptor1s    z(UNIXTestsBuilder.test_sendFileDescriptorcCsTGdddt}Gdddt}|}|}||_t||||j||jddS)z If a L{IUNIXTransport.sendFileDescriptor} call fills up the send buffer, any registered producer is paused. c@seZdZddZdS)zSUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..DoesNotReadcSs|jdSrq)rkpauseProducingrorRrRrSrp\szbUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..DoesNotRead.connectionMadeNrUrVrWrprRrRrRrS DoesNotRead[src@s8eZdZdZddZddZddZdd Zd d Zd S) z`UNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptorsFcsPt_jdfdd}t|_jjj_jdt ddS)NTcs"jjjddS)Nx)rkrlr rrmrRrorRrSsenderfszUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.connectionMade..senderrzSend loop failure) r rkZregisterProducerr/taskrZclockstartrr?)rNrrRrorSrpbs    zoUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.connectionMadecSs |dSrq _disconnectrorRrRrS stopProducingnsznUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.stopProducingcSs |dSrqrrorRrRrSresumeProducingqszpUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.resumeProducingcSsd|_|j|dS)NT)pausedrkZunregisterProducerrrorRrRrSrts zoUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.pauseProducingcSs$|j|j|jjdSrq)rstoprkZabortConnectionotherrorRrRrSrys  zlUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors._disconnectN) rUrVrWrrprrrrrRrRrRrSSendsManyFileDescriptors_s  rz*sendFileDescriptor producer was not pausedN)r0rr4r assertTruer)rNrrrcrerRrRrS-test_sendFileDescriptorTriggersPauseProducingTsz>UNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducingcst}t|dt}g}|}||j|fddt|||j| |dt |d t | j jtdS)ag If L{IUNIXTransport.sendFileDescriptor} is used to queue a greater number of file descriptors than the number of bytes sent using L{ITransport.write}, the connection is closed and the protocol connected to the transport has its C{connectionLost} method called with a failure wrapping L{FileDescriptorOverrun}. Ncs jSrqrrrrRrSrrz=UNIXTestsBuilder.test_fileDescriptorOverrun..r)r rgrrurwrappendr4rassertIsInstancer<trapr&rsvaluer')rNcargoreresultrrRrrStest_fileDescriptorOverruns  z+UNIXTestsBuilder.test_fileDescriptorOverruncs>ddlm}ddlm}ddlm}ddttGfdddt}Gd d d |}|t t \}}| |j | |j |} ||| } t \} } t \} }| t| | t|d }| | g}||\}}||||| |t| j||t|t| j| jr:fd d |D}||| jdS)a Drive _SendmsgMixin via sendmsg socket calls to check that L{IFileDescriptorReceiver.fileDescriptorReceived} is called once for each file descriptor received in the ancillary messages. @param ancillaryPacker: A callable that will be given a list of two file descriptors and should return a two-tuple where: The first item is an iterable of zero or more (cmsg_level, cmsg_type, cmsg_data) tuples in the same order as the given list for actual sending via sendmsg; the second item is an integer indicating the expected number of FDs to be received. r socketpair) _SendmsgMixinsendmsgcSst|}|j|jfSrq)rst_devst_ino)rhfsrRrRrSdeviceInodeTupleszTUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..deviceInodeTuplecs eZdZddZfddZdS)zPUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeProtocolcSsg|_g|_dSrq)fdsdeviceInodesReceivedrorRrRrSrjszYUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeProtocol.__init__cs(|j||j|t|dSrq)rrrr)rNrhrrRrSrzs zgUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeProtocol.fileDescriptorReceivedN)rUrVrWrjrzrRrrRrS FakeProtocolsrc@s8eZdZdZddZddZddZdd Zd d Zd S) zPUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceivercSs||_||_dSrq)r protocol)rNsktprotorRrRrSrjszYUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver.__init__cSsdSrqrRr}rRrRrS _dataReceivedsz^UNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver._dataReceivedcSsdSrqrRrorRrRrSrszXUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver.getHostcSsdSrqrRrorRrRrSrszXUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver.getPeercSsdSrqrR)rNorRrRrS _getLogPrefixsz^UNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver._getLogPrefixN) rUrVrWZ bufferSizerjrrrrrRrRrRrS FakeReceivers rssome data needs to be sentcsg|] }|qSrRrR.0rhrrRrS szNUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..N)r rZtwisted.internet.unixrtwisted.python.sendmsgrrr(r0rr addCleanuprrrZdoReadrLlenrZ assertFalseset intersectionr)rNancillaryPackerrrrrrZ sendSocketZ recvSocketrZreceiverZ fileOneFDZ fileOneNameZ fileTwoFDZ fileTwoNameZ dataToSend fdsToSend ancillary expectedCountZdeviceInodesSentrRrrS)_sendmsgMixinFileDescriptorReceivedDrivers4             z:UNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDrivercs&ddlmfdd}||dS)z _SendmsgMixin handles multiple file descriptors per recvmsg, calling L{IFileDescriptorReceiver.fileDescriptorReceived} once per received file descriptor. Scenario: single CMSG with two FDs. r SCM_RIGHTScs ttd|fg}d}||fS)Nii)rr rrrrrrRrSr sz[UNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgOneCMSG..ancillaryPackerNrrrrNrrRrrS1test_multiFileDescriptorReceivedPerRecvmsgOneCMSGs  zBUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgOneCMSGz=Multi control message ancillary sendmsg not supported on Mac.cs&ddlmfdd}||dS)z _SendmsgMixin handles multiple file descriptors per recvmsg, calling L{IFileDescriptorReceiver.fileDescriptorReceived} once per received file descriptor. Scenario: two CMSGs with one FD each. rrcsfdd|D}d}||fS)Ncsg|]}ttd|fqS)irrrrRrSrszpUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgTwoCMSGs..ancillaryPacker..rrRrrrRrSrsz\UNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgTwoCMSGs..ancillaryPackerNrrrRrrS2test_multiFileDescriptorReceivedPerRecvmsgTwoCMSGss  zCUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgTwoCMSGscs~ddlmdd}fdd}g}t|j|t|j|d|||dtfd d |D}| |d d S) z _SendmsgMixin handles multiple file descriptors per recvmsg, calling L{IFileDescriptorReceiver.fileDescriptorReceived} once per received file descriptor. Scenario: unsupported CMSGs. rrcSsg}d}||fS)NrrRrrRrRrSr3sz[UNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSG..ancillaryPackercsd}dg}d}|||S)Ns some data)NNrr)ZReceivedMessage)rargskwargsrirflagsrrRrSfakeRecvmsgUnsupportedAncillary8szkUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSG..fakeRecvmsgUnsupportedAncillaryrecvmsgz#received unsupported ancillary datac3s|]}|dkVqdS)formatNrR)re)expectedMessagerRrS GszUUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSG..z+Expected message not found in logged eventsN) Ztwisted.pythonrr>rrr@rranyr)rNrreventsfoundrR)rrrS1test_multiFileDescriptorReceivedPerRecvmsgBadCMSG$s    zBUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSGcCsddlm}|\}}g}t|j|t|jGdddt}||d}t}t ||||j | | d| d|dd }d } t|j|j| |d } |D],} | D]\} } | | | krqqqq|d | t|fd S)z If associated with a protocol which does not provide L{IFileDescriptorReceiver}, file descriptors received by the L{IUNIXTransport} implementation are closed and a warning is emitted. rrc@seZdZddZdS)zRUNIXTestsBuilder.test_avoidLeakingFileDescriptors..RecordEndpointAddressescSs&|j|_|j|_t|dSrq)rkr hostAddressr peerAddressrgrprorRrRrSrp_s  zaUNIXTestsBuilder.test_avoidLeakingFileDescriptors..RecordEndpointAddresses.connectionMadeNrrRrRrRrSRecordEndpointAddresses^srrFrrz%(protocolName)s (on %(hostAddress)r) does not provide IFileDescriptorReceiver; closing file descriptor received (from %(peerAddress)r).r0)rrZ protocolNamerz3Expected event (%s) not found in logged events (%s)N)r rr>rrr@rgrr0r4rr setblockingrLrecvrrritemsgetr!r)rNrZ probeClientZ probeServerrrrcrerZclsNameZ expectedEventZlogEventkvrRrRrS test_avoidLeakingFileDescriptorsJsF    z1UNIXTestsBuilder.test_avoidLeakingFileDescriptorscsvttGfdddt}t}t|d}|}t||jt |j ddt |j dddS)z L{IUNIXTransport.sendFileDescriptor} sends file descriptors before L{ITransport.write} sends normal bytes. cs(eZdZddZfddZddZdS)zJUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes..RecordEventscSst|g|_dSrq)r0rprrorRrRrSrps zYUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes..RecordEvents.connectionMadecs t||jt|dSrq)rrrrtype)Z innerSelfryrorRrSrzs zaUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes..RecordEvents.fileDescriptorReceivedcSs|j|dSrq)rextendr}rRrRrSr~szWUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes..RecordEvents.dataReceivedN)rUrVrWrprzr~rRrorRrS RecordEventss rrrN) rr(r0r rgrr4rrLintrbytes)rNrrrcrerRrorS#test_descriptorDeliveredBeforeBytess z4UNIXTestsBuilder.test_descriptorDeliveredBeforeBytesN)rUrVrWrXr+r]r\rrrrBisLinuxrrrrrsendmsgSkipReasonrrrrrZisMacOSXrrrr rRrRrRrSrsJ    "  0  c      %  L rc@s8eZdZdZejfZddZee dddZ dS)UNIXDatagramTestsBuilderzE Builder defining tests relating to L{IReactorUNIXDatagram}. cCs|d|tdS)z The UNIX socket created by L{IReactorUNIXDatagram.listenUNIXDatagram} is created with the mode specified. listenUNIXDatagramN)rTrr-rorRrRrStest_listenModesz(UNIXDatagramTestsBuilder.test_listenModercCs>t|}|}|d|t}||td|dS)z On Linux, a UNIX socket path may begin with C{''} to indicate a socket in the abstract namespace. L{IReactorUNIX.listenUNIXDatagram} accepts such a path. rN)r[rIrr-rLrrrrRrRrSrs zeZdZUdZeefZeee e e d<ddZ ddZ dS)SocketUNIXMixinzb Mixin which uses L{IReactorSocket.adoptStreamPort} to hand out listening UNIX ports. r]cCsZtt}tddd}|||d|dz|||j |WS|XdS)zj Get a UNIX port from a reactor, wrapping an already-initialized file descriptor. r^r_r`FN) r rrrlistenrrZadoptStreamPortrfamily)rNrrQZportSockrPrRrRrSgetListeningPorts    z SocketUNIXMixin.getListeningPortcCs||j|SaZ Connect to a listening UNIX socket. @param reactor: The reactor under test. @type reactor: L{IReactorUNIX} @param address: The listening's address. @type address: L{UNIXAddress} @param factory: The client factory. @type factory: L{ClientFactory} @return: The connector rrdrNraddressrQrRrRrSconnectToListenersz!SocketUNIXMixin.connectToListenerN)rUrVrWrXr+r*r]rrrrrfrrrRrRrRrSrs rc@s eZdZdZddZddZdS)ListenUNIXMixinzZ Mixin which uses L{IReactorTCP.listenUNIX} to hand out listening UNIX ports. cCstddd}|||S)z0 Get a UNIX port from a reactor r^r_r`)rr)rNrrQrPrRrRrSr s z ListenUNIXMixin.getListeningPortcCs||j|SrrrrRrRrSrsz!ListenUNIXMixin.connectToListenerN)rUrVrWrXrrrRrRrRrSrsrc@s8eZdZUefZeeeee d<ddZ ddZ dS)UNIXPortTestsMixinr]cCs|dt|jS)zZ Get the message expected to be logged when a UNIX port starts listening. z starting on r;rrd)rNrrQrRrRrS#getExpectedStartListeningLogMessage&sz6UNIXPortTestsMixin.getExpectedStartListeningLogMessagecCsdt|jdS)zJ Get the expected connection lost message for a UNIX port z (UNIX Port z Closed)r)rNrrRrRrSgetExpectedConnectionLostLogMsg,sz2UNIXPortTestsMixin.getExpectedConnectionLostLogMsgN) rUrVrWr+r]rrrrrfrrrRrRrRrSr#s rc@seZdZdZdS)UNIXPortTestsBuilderz. Tests for L{IReactorUNIX.listenUnix} NrUrVrWrXrRrRrRrSr 3sr c@seZdZdZdS)UNIXFDPortTestsBuilderz3 Tests for L{IReactorUNIX.adoptStreamPort} Nr!rRrRrRrSr">sr"c@s.eZdZeeefZddZddZddZ dS)%UNIXAdoptStreamConnectionTestsBuilderc Cs~|}ddlm}Gdddt}|tt\}}|d||j||j| }|}| |t|}| |dS)z {IReactorSocket.adoptStreamConnection} returns None if the given factory's buildProtocol returns None. rrc@seZdZddZdS)zXUNIXAdoptStreamConnectionTestsBuilder.test_buildProtocolReturnsNone..NoneFactorycSsdSrqrR)rNrrRrRrS buildProtocol_szfUNIXAdoptStreamConnectionTestsBuilder.test_buildProtocolReturnsNone..NoneFactory.buildProtocolN)rUrVrWr$rRrRrRrS NoneFactory^sr%FN) rIr rr.rr rrrradoptStreamConnectionZ assertIsNone) rNrrr%s1s2Zs1FDrQrrRrRrStest_buildProtocolReturnsNonePs     zCUNIXAdoptStreamConnectionTestsBuilder.test_buildProtocolReturnsNonecs<fdd}}j|ddd}|||dS)z> Helper method to test UNIX server addresses. cs|\}}}zftd|j}d|jj|ft|jd|jj|f|jj|j j d} |t W5|jXdS)NrFz zAccumulatingProtocol,%s,%sr) rkrnr=rrdrLZ sessionnostrZlogstrrQZ peerAddressesrr) protocolsrercrZportPathrrorRrS connectedqs$    zOUNIXAdoptStreamConnectionTestsBuilder.test_ServerAddressUNIX..connectedN) interface addressFamily)rIgetConnectedClientAndServerrZ runReactor)rNr,rrrRrorStest_ServerAddressUNIXls  z.firstServerConnectedcsjr|Srq)Zrunningr)r)rrRrSrszOUNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServer..stopcs|\}}||fdSrq)rx)r+rerc)deferredrrRrSrszPUNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServer..start)r8r protocolConnectionMadeZprotocolConnectionLostr7rrrr"rrrrrd) rNrr-r.Z firstServerrerPr1Z lostDeferredrZ startDeferredrrR)r2rrrcrSr/s4         zAUNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServerN) rUrVrWr)r*r+r]r)r0r/rRrRrRrSr#Is"r#c@s6eZdZdZefZdZeddZddZ ddZ dS) UnixClientTestsBuilderz7 Define tests for L{IReactorUNIX.connectUNIX}. NcCs|jdkrt||_|jS)z Return a path usable by C{connectUNIX} and C{listenUNIX}. @return: A path instance, built with C{_abstractPath}. N)_pathr[rorRrRrSrPs  zUnixClientTestsBuilder.pathcCs||j|S)z Start an UNIX server with the given C{factory}. @param reactor: The reactor to create the UNIX port in. @param factory: The server factory. @return: A UNIX port instance. )rrPrNrrQrRrRrSrs zUnixClientTestsBuilder.listencCs||j|S)z Start an UNIX client with the given C{factory}. @param reactor: The reactor to create the connection in. @param factory: The client factory. @return: A UNIX connector instance. )rrPr6rRrRrSconnects zUnixClientTestsBuilder.connect) rUrVrWrXr+r]r5propertyrPrr7rRrRrRrSr4s  r4posixz%UNIX sockets only supported on POSIX.c@sPeZdZdZdZddZeeegdfddddZd d Z d d Z d dZ dS)RealSocketTestsz, Test real UNIX socket connections. Zunixpathcstjtstt}t|j|}t|t j s4t||_ | |j t}t|j||jIdH|jspt||jj||j|jtjdS)z| A port created with L{IReactorTCP.listenUNIX} can be connected to with L{IReactorTCP.connectUNIX}. N)rr+ providedByrAssertionErrorr6rrP isinstancerZPortrrZcleanUpr7rr2Z lastProtocolrZmadeZ disconnectedZ lostReasonrrZConnectionDone)rNfrclientFrRrRrStest_closePortInProtocolFactorys    z/RealSocketTests.test_closePortInProtocolFactoryN)rxreturnc stjtstt}t}||_t|j |}| |j t }t}||_t |j |t||gIdH\}}||||jst|jst|j|jdS)z Invoke the given callback with a client protocol and a server protocol which have been connected to each other. N)rr+r;rr<r8r r3rrPrrKr7rr"rkrn) rNrxZ serverFactoryZserverConnMaderr?ZclientConnMadeserverProtocolclientProtocolrRrRrS_connectedClientAndServerTests$     z-RealSocketTests._connectedClientAndServerTestcs*ttddfdd }|IdHdS)z The transport of a protocol connected with L{IReactorTCP.connectUNIX} or L{IReactor.TCP.listenUNIX} can have L{ITCPTransport.getTcpNoDelay} or L{ITCPTransport.setTcpNoDelay} called without error. NrBrCrAcsn||fD]`}tj|jst|j}|d|d|d|d|dqdS)NrTF)r ITCPTransportr;rkr<rLZ getTcpNoDelayZ setTcpNoDelayrBrCprkrorRrScheckAs   z.RealSocketTests.test_tcpNoDelay..checkrDrDrNrIrRrorStest_tcpNoDelay:s  zRealSocketTests.test_tcpNoDelaycs*ttddfdd }|IdHdS)a The transport of a protocol connected with L{IReactorTCP.connectUNIX} or L{IReactor.TCP.listenUNIX} can have its I{SO_KEEPALIVE} state inspected and manipulated with L{ITCPTransport.getTcpKeepAlive} and L{ITCPTransport.setTcpKeepAlive}. NrEcsn||fD]`}tj|jst|j}|d|d|d|d|dqdS)NFT)rrFr;rkr<rLZgetTcpKeepAliveZsetTcpKeepAliverGrorRrSrIWs   z0RealSocketTests.test_tcpKeepAlive..checkrJrKrRrorStest_tcpKeepAliveOs  z!RealSocketTests.test_tcpKeepAlivecsDtjtstt}tjd|dd|jIdH|j t j dS)zb Tests whether attempting to establish connection to non-existing UNIX path fails z non-existing)timeoutN) rr+r;rr<r7rZ failDeferredrsrrZ ConnectError)rNr?rRrRrS test_Failinges  zRealSocketTests.test_Failing) rUrVrWrXrPr@rrDrDrLrMrPrRrRrRrSr:s r:)vrXhashlibrosrrrrrpprintrr r r r r structrtempfilerrtypingrrrrrZunittestrrZ_AF_UNIX ImportErrorZzope.interfacerrZtwisted.internetrrrrrZtwisted.internet.addressrZtwisted.internet.deferr r!r"Ztwisted.internet.endpointsr#r$Ztwisted.internet.errorr%r&r'Ztwisted.internet.interfacesr(r)r*r+Ztwisted.internet.protocolr,r-r.Ztwisted.internet.taskr/Z&twisted.internet.test.connectionmixinsr0r1r2r3r4Z#twisted.internet.test.reactormixinsr5Ztwisted.internet.test.test_tcpr6r7r8r9r:Ztwisted.python.compatr;Ztwisted.python.failurer<Ztwisted.python.filepathr=Ztwisted.python.logr>r?r@Ztwisted.python.reflectrAZtwisted.python.runtimerBrCZtwisted.test.test_tcprDZtwisted.trial.unittestrErr rGr[r\rgrurrrrrr r"r#globalsupdateZmakeTestCaseClassesr4r:rRrRrRrSs               %>Y!-    ~-