o i@sdZddlmZddlmZmZmZmZmZddl m Z ddl m Z m Z mZm Z ddlmZddlmZddlmZmZdd lmZmZmZmZmZdd lmZzdd l mZWn eycd ZYnweZdd 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 ur!d#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 methodNamepathfactoryrHrZunixPortrReC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/internet/test/test_unix.py _modeTestRs zUNIXFamilyMixin._modeTestN)__name__ __module__ __qualname____doc__rTrRrRrRrSrGMs rGcCsttdS)zI Return a new, unique abstract namespace path to be listened on. dN)rr hexdigest)caserRrRrS _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.suffixdirN)rr$)rNrrPrRrRrSserverl  zUNIXCreator.servercCs t||jS)z3 Construct a UNIX client endpoint. N)r#name)rNrZ serverAddressrRrRrSclientts zUNIXCreator.clientN)rUrVrWrXrr+r^rrrr__annotations__rdrgrRrRrRrSr]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)rNrjrkrRrRrS__init__s zSendFileDescriptor.__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) transportsendFileDescriptorrjrkwriteloseConnectionrNrRrRrSconnectionMadesz!SendFileDescriptor.connectionMadecCst||||_dSN)r0connectionLostreasonrNrurRrRrSrts  z!SendFileDescriptor.connectionLost)rUrVrWrXrurlrrrtrRrRrRrSri{s  ric@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"|jdur t|_|jSt|jS)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)rur waitingr!rqrRrRrSwaitForDescriptors  z'ReceiveFileDescriptor.waitForDescriptorcCs|j|d|_dS)z Fire the waiting Deferred, initialized by C{waitForDescriptor}, with the file descriptor just received. N)rxcallback)rN descriptorrRrRrSfileDescriptorReceivedrez,ReceiveFileDescriptor.fileDescriptorReceivedcCs4|jdur|jttd|dd|_dSdS)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.)rxerrbackr< ExceptionrNrkrRrRrS dataReceiveds  z"ReceiveFileDescriptor.dataReceivedcCs2t|||jdur|j|d|_||_dS)zj Fail the waiting Deferred, initialized by C{waitForDescriptor}, if there is one. N)r0rtrxr}rurvrRrRrSrts    z$ReceiveFileDescriptor.connectionLost) rUrVrWrXrurxryr|rrtrRrRrRrSrws   rwc@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}. cC|d|tdS)zs The UNIX socket created by L{IReactorUNIX.listenUNIX} is created with the mode specified. listenUNIXN)rTrr.rqrRrRrS test_modezUNIXTestsBuilder.test_mode8Abstract namespace UNIX sockets only supported on Linux.cC>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_listenOnLinuxAbstractNamespacecCs^dd}|tjd||}|t|dtWddS1s(wYdS)z L{IReactorUNIX.listenUNIX} raises L{CannotListenError} if the underlying port's createInternetSocket raises a socket error. cSstd)Nz FakeBasePort forced socket.error)OSErrorrqrRrRrSraiseSocketErrorsz=UNIXTestsBuilder.test_listenFailure..raiseSocketErrorZcreateInternetSocketznot-usedN)patchrZBasePortrI assertRaisesr%rr.)rNrrrRrRrStest_listenFailures  "z#UNIXTestsBuilder.test_listenFailurecCr)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@eZdZddZdS)z4UNIXTestsBuilder.test_addresses..SaveAddresscSs"t||d|_|dS)N)hostpeer)dictrgetPeer addressesrp)rNrmrRrRrSmakeConnection#s  zCUNIXTestsBuilder.test_addresses..SaveAddress.makeConnectionN)rUrVrWrrRrRrRrS SaveAddress" rrrN)r0r4 endpointsrLr)rNrrdrgrRrRrStest_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|||dSrs)r r rrL getsocknameZassertNotEqualfileno)r{Zreceived)rsrNrRrScheckDescriptor@s zAUNIXTestsBuilder.test_sendFileDescriptor..checkDescriptorz-Sending file descriptor encountered a problemc jSrsrmrpZignoredrdrRrSP z:UNIXTestsBuilder.test_sendFileDescriptor..N) r rbindrirrwry addCallback addErrbackr?addBothr4r)rNrgdrrR)rrrNrdrStest_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@r)zSUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..DoesNotReadcSs|jdSrs)rmpauseProducingrqrRrRrSrr\szbUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..DoesNotRead.connectionMadeNrUrVrWrrrRrRrRrS DoesNotRead[rrc@8eZdZdZddZddZddZdd Zd d Zd S) z`UNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptorsFcsPt_jdfdd}t|_jjj_jdt ddS)NTcs"jjjddS)Nx)rmrnr rrorRrqrRrSsenderfszUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.connectionMade..senderrzSend loop failure) r rmZregisterProducerr/taskrZclockstartrr?)rNrrRrqrSrrbs    zoUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.connectionMadecS |dSrs _disconnectrqrRrRrS stopProducingn znUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.stopProducingcSrrsrrqrRrRrSresumeProducingqrzpUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.resumeProducingcSsd|_|j|dS)NT)pausedrmZunregisterProducerrrqrRrRrSrts  zoUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.pauseProducingcSs$|j|j|jjdSrs)rstoprmZabortConnectionotherrqrRrRrSrys  zlUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors._disconnectN) rUrVrWrrrrrrrrRrRrRrSSendsManyFileDescriptors_s  rz*sendFileDescriptor producer was not pausedN)r0rr4r assertTruer)rNrrrdrgrRrRrS-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}. NcrrsrrrrRrSrrz=UNIXTestsBuilder.test_fileDescriptorOverrun..r)r rirrwryrappendr4rassertIsInstancer<trapr&ruvaluer')rNcargorgresultrrRrrStest_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| jrfd d |D}||| jdSdS)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|jfSrs)rst_devst_ino)rjfsrRrRrSdeviceInodeTuples zTUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..deviceInodeTuplecs eZdZddZfddZdS)zPUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeProtocolcSsg|_g|_dSrs)fdsdeviceInodesReceivedrqrRrRrSrl zYUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeProtocol.__init__cs(|j||j|t|dSrs)rrrr)rNrjrrRrSr|s  zgUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeProtocol.fileDescriptorReceivedN)rUrVrWrlr|rRrrRrS FakeProtocolsrc@r) zPUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceivercSs||_||_dSrs)r protocol)rNsktprotorRrRrSrlrzYUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver.__init__cSdSrsrRrrRrRrS _dataReceivedz^UNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver._dataReceivedcSrrsrRrqrRrRrSrrzXUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver.getHostcSrrsrRrqrRrRrSrrzXUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver.getPeercSrrsrR)rNorRrRrS _getLogPrefixrz^UNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver._getLogPrefixN) rUrVrWZ bufferSizerlrrrrrRrRrRrS FakeReceivers rssome data needs to be sentcsg|]}|qSrRrR.0rjrrRrS szNUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..N)r rZtwisted.internet.unixrtwisted.python.sendmsgrrr(r0rr addCleanuprrrZdoReadrLlenrZ assertFalseset intersectionr)rNancillaryPackerrrrrrZ sendSocketZ recvSocketrZreceiverZ fileOneFDZ fileOneNameZ fileTwoFDZ fileTwoNameZ dataToSend fdsToSend ancillary expectedCountZdeviceInodesSentrRrrS)_sendmsgMixinFileDescriptorReceivedDrivers6             z:UNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriverc&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$ttdg|Rfg}d}||fS)Niir rrrrrrRrSr sz[UNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgOneCMSG..ancillaryPackerNrrrrNrrRrrS1test_multiFileDescriptorReceivedPerRecvmsgOneCMSGs  zBUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgOneCMSGz=Multi control message ancillary sendmsg not supported on Mac.cr)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)NNr)ZReceivedMessage)rargskwargsrkrflagsrrRrSfakeRecvmsgUnsupportedAncillary8szkUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSG..fakeRecvmsgUnsupportedAncillaryrecvmsgz#received unsupported ancillary datac3s|] }|dvVqdS)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] \} } | | | krhqUq[d S|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@r)zRUNIXTestsBuilder.test_avoidLeakingFileDescriptors..RecordEndpointAddressescSs&|j|_|j|_t|dSrs)rmr hostAddressr peerAddressrirrrqrRrRrSrr_s  zaUNIXTestsBuilder.test_avoidLeakingFileDescriptors..RecordEndpointAddresses.connectionMadeNrrRrRrRrSRecordEndpointAddresses^rr rFrrz%(protocolName)s (on %(hostAddress)r) does not provide IFileDescriptorReceiver; closing file descriptor received (from %(peerAddress)r).r0)r r Z protocolNamerz3Expected event (%s) not found in logged events (%s)N)r rr>rrr@rirr0r4rr setblockingrLrecvrr r itemsgetr!r)rNrZ probeClientZ probeServerrr rdrgrZclsNameZ expectedEventZlogEventkvrRrRrS test_avoidLeakingFileDescriptorsJsH   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|_dSrs)r0rrrrqrRrRrSrrs  zYUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes..RecordEvents.connectionMadecs t||jt|dSrs)rrrrtype)Z innerSelfr{rqrRrSr|s zaUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes..RecordEvents.fileDescriptorReceivedcSs|j|dSrs)rextendrrRrRrSrszWUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes..RecordEvents.dataReceivedN)rUrVrWrrr|rrRrqrRrS RecordEventss  rrrN) rr(r0r rirr4rrLintrbytes)rNrrrdrgrRrqrS#test_descriptorDeliveredBeforeBytess z4UNIXTestsBuilder.test_descriptorDeliveredBeforeBytesN)rUrVrWrXr+r^r]rrrrBisLinuxrrrrrsendmsgSkipReasonrrrrrZisMacOSXrr rrrRrRrRrSrsL      "  0  c      %  Lrc@s8eZdZdZejfZddZee dddZ dS)UNIXDatagramTestsBuilderzE Builder defining tests relating to L{IReactorUNIXDatagram}. cCr)z The UNIX socket created by L{IReactorUNIXDatagram.listenUNIXDatagram} is created with the mode specified. listenUNIXDatagramN)rTrr-rqrRrRrStest_listenModerz(UNIXDatagramTestsBuilder.test_listenModercCr)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^cCs\tt}tddd}|||d|dz|||j|W| S| w)zj Get a UNIX port from a reactor, wrapping an already-initialized file descriptor. r_r`raFN) r rrrlistenrZadoptStreamPortrfamilyr)rNrrQZportSockrPrRrRrSgetListeningPorts    z SocketUNIXMixin.getListeningPortcC||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 NrrfrNraddressrQrRrRrSconnectToListenerz!SocketUNIXMixin.connectToListenerN)rUrVrWrXr+r*r^rrrrrhr%r+rRrRrRrSr!s  r!c@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`raN)rr)rNrrQrPrRrRrSr% s  z ListenUNIXMixin.getListeningPortcCr&r'r(r)rRrRrSr+r,z!ListenUNIXMixin.connectToListenerN)rUrVrWrXr%r+rRrRrRrSr-s r-c@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 Nr;rrf)rNrrQrRrRrS#getExpectedStartListeningLogMessage&sz6UNIXPortTestsMixin.getExpectedStartListeningLogMessagecCsdt|jdS)zJ Get the expected connection lost message for a UNIX port z (UNIX Port z Closed)Nr/)rNrrRrRrSgetExpectedConnectionLostLogMsg,sz2UNIXPortTestsMixin.getExpectedConnectionLostLogMsgN) rUrVrWr+r^rrrrrhr0r1rRrRrRrSr.#s  r.c@eZdZdZdS)UNIXPortTestsBuilderz. Tests for L{IReactorUNIX.listenUnix} NrUrVrWrXrRrRrRrSr33r3c@r2)UNIXFDPortTestsBuilderz3 Tests for L{IReactorUNIX.adoptStreamPort} Nr4rRrRrRrSr6>r5r6c@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@r)zXUNIXAdoptStreamConnectionTestsBuilder.test_buildProtocolReturnsNone..NoneFactorycSrrsrR)rNr*rRrRrS buildProtocol_rzfUNIXAdoptStreamConnectionTestsBuilder.test_buildProtocolReturnsNone..NoneFactory.buildProtocolN)rUrVrWr8rRrRrRrS NoneFactory^rr9FN) rIr rr.rr rrrradoptStreamConnectionZ assertIsNone) rNrrr9s1s2Zs1FDrQrrRrRrStest_buildProtocolReturnsNonePs    zCUNIXAdoptStreamConnectionTestsBuilder.test_buildProtocolReturnsNonecs<fdd}}j|ddd}|||dS)z> Helper method to test UNIX server addresses. cs|\}}}z9td|j}d|jj|ft|jd|jj|f|jj|jj d} |t W|j dS|j w)NrFz zAccumulatingProtocol,%s,%sr) r=rrfrLrmZ sessionnostrZlogstrrQZ peerAddressesrrrp) protocolsrgrdrZportPathr rqrRrS connectedqs$    zOUNIXAdoptStreamConnectionTestsBuilder.test_ServerAddressUNIX..connectedN) interface addressFamily)rIgetConnectedClientAndServerrZ runReactor)rNr@rrrRrqrStest_ServerAddressUNIXls  z.firstServerConnectedcsjr|Srs)runningr)r)rrRrSrszOUNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServer..stopcs|\}}||fdSrs)rz)r?rgrd)deferredrrRrSrszPUNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServer..startN)r8r protocolConnectionMadeZprotocolConnectionLostr7rrrr"rrrrrf) rNrrArBZ firstServerrgrPrEZ lostDeferredrZ startDeferredrrR)rGrrrdrSrCs4         zAUNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServerN) rUrVrWr)r*r+r^r=rDrCrRrRrRrSr7Is "r7c@s6eZdZdZefZdZeddZddZ ddZ dS) UnixClientTestsBuilderz7 Define tests for L{IReactorUNIX.connectUNIX}. NcCs|jdur t||_|jS)z Return a path usable by C{connectUNIX} and C{listenUNIX}. @return: A path instance, built with C{_abstractPath}. N)_pathr\rqrRrRrSrPs  zUnixClientTestsBuilder.pathcC||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. N)rrPrNrrQrRrRrSr# zUnixClientTestsBuilder.listencCrK)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. N)rrPrLrRrRrSconnectrMzUnixClientTestsBuilder.connect) rUrVrWrXr+r^rJpropertyrPr#rNrRrRrRrSrIs  rIposixz%UNIX sockets only supported on POSIX.c@sReZdZdZdZddZdeeegdfddfdd Zd d Z d d Z ddZ dS)RealSocketTestsz, Test real UNIX socket connections. Zunixpathcstjts Jt}t|j|}t|tj sJ||_ | |j t }t|j||jIdH|js9J||jj||j|jtjdS)z| A port created with L{IReactorTCP.listenUNIX} can be connected to with L{IReactorTCP.connectUNIX}. N)rr+ providedByrr6rrP isinstancerZPortrrZcleanUpr7rrGZ lastProtocolrZmade disconnectedZ lostReasonrrZConnectionDone)rNfrclientFrRrRrStest_closePortInProtocolFactorys    z/RealSocketTests.test_closePortInProtocolFactoryrzNreturnc stjts Jt}t}||_t|j|}| |j t }t}||_t |j|t ||gIdH\}}||||jsDJ|jsIJ|j|jdS)z Invoke the given callback with a client protocol and a server protocol which have been connected to each other. N)rr+rRrr8r rHrrPrrKr7rr"rmrp) rNrzZ serverFactoryZserverConnMaderrVZclientConnMadeserverProtocolclientProtocolrRrRrS_connectedClientAndServerTests&     z-RealSocketTests._connectedClientAndServerTestc0dtdtddffdd }|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. rYrZrXNcsn||fD]0}tj|jsJ|j}|d|d|d|d|dqdS)NrTF)r ITCPTransportrRrmrLZ getTcpNoDelayZ setTcpNoDelayrYrZprmrqrRrScheckA   z.RealSocketTests.test_tcpNoDelay..checkrDr[rNr`rRrqrStest_tcpNoDelay:s zRealSocketTests.test_tcpNoDelaycr\)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}. rYrZrXNcsn||fD]0}tj|jsJ|j}|d|d|d|d|dqdS)NFT)rr]rRrmrLZgetTcpKeepAliveZsetTcpKeepAliver^rqrRrSr`Wraz0RealSocketTests.test_tcpKeepAlive..checkrbrcrRrqrStest_tcpKeepAliveOs z!RealSocketTests.test_tcpKeepAlivecsFtjts Jt}tjd|dd|jIdH|jt j dS)zb Tests whether attempting to establish connection to non-existing UNIX path fails z non-existing)timeoutN) rr+rRrr7rZ failDeferredrurrZ ConnectError)rNrVrRrRrS test_Failinges  zRealSocketTests.test_Failing) rUrVrWrXrPrWrrDr[rdrerhrRrRrRrSrQs  rQ)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.unittestrErrrGr\r]rirwrrr!r-r.r3r6r7globalsupdateZmakeTestCaseClassesrIrQrRrRrRrSs               %>Y!-   ~-