o >h@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 methodNamepathfactoryrHrunixPortrS/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/internet/test/test_unix.py _modeTestRs zUNIXFamilyMixin._modeTestN)__name__ __module__ __qualname____doc__rUrSrSrSrTrGMs rGcCsttdS)zI Return a new, unique abstract namespace path to be listened on. d)rr hexdigest)caserSrSrT _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$)rNrrPrSrSrTserverl  zUNIXCreator.servercCs t||jS)z3 Construct a UNIX client endpoint. )r#name)rNr serverAddressrSrSrTclientts zUNIXCreator.clientN)rVrWrXrYrr+r_rrrr__annotations__rerirSrSrSrTr^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)rNrlrmrSrSrT__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) transportsendFileDescriptorrlrmwriteloseConnectionrNrSrSrTconnectionMadesz!SendFileDescriptor.connectionMadecCst||||_dSN)r0connectionLostreasonrNrwrSrSrTrvs  z!SendFileDescriptor.connectionLost)rVrWrXrYrwrnrtrvrSrSrSrTrk{s  rkc@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)rwr waitingr!rsrSrSrTwaitForDescriptors  z'ReceiveFileDescriptor.waitForDescriptorcCs|j|d|_dS)z Fire the waiting Deferred, initialized by C{waitForDescriptor}, with the file descriptor just received. N)rzcallback)rN descriptorrSrSrTfileDescriptorReceivedrfz,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.)rzerrbackr< ExceptionrNrmrSrSrT dataReceiveds  z"ReceiveFileDescriptor.dataReceivedcCs2t|||jdur|j|d|_||_dS)zj Fail the waiting Deferred, initialized by C{waitForDescriptor}, if there is one. N)r0rvrzrrwrxrSrSrTrvs    z$ReceiveFileDescriptor.connectionLost) rVrWrXrYrwrzr{r~rrvrSrSrSrTrys   ryc@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)rUrr.rsrSrSrT 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.rLgetHostrrNrPrportrSrSrT#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)OSErrorrsrSrSrTraiseSocketErrorsz=UNIXTestsBuilder.test_listenFailure..raiseSocketErrorcreateInternetSocketznot-usedN)patchrBasePortrI assertRaisesr%rr.)rNrrrSrSrTtest_listenFailures  "z#UNIXTestsBuilder.test_listenFailurecCr)zc L{IReactorUNIX.connectUNIX} also accepts a Linux abstract namespace path. rN)r]rI connectUNIXr,rLgetDestinationr)rNrPr connectorrSrSrT$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 addressesrr)rNrorSrSrTmakeConnection#s  zCUNIXTestsBuilder.test_addresses..SaveAddress.makeConnectionN)rVrWrXrrSrSrSrT SaveAddress" rrrN)r0r4 endpointsrLr)rNrrerirSrSrTtest_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|||dSru)r r rrL getsocknameassertNotEqualfileno)r}received)rsrNrSrTcheckDescriptor@s zAUNIXTestsBuilder.test_sendFileDescriptor..checkDescriptorz-Sending file descriptor encountered a problemc jSrurorrignoredrerSrTP z:UNIXTestsBuilder.test_sendFileDescriptor..N) r rbindrkrryr{ addCallback addErrbackr?addBothr4r)rNridrrS)rrrNrerTtest_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|jdSru)ropauseProducingrsrSrSrTrt\szbUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..DoesNotRead.connectionMadeNrVrWrXrtrSrSrSrT DoesNotRead[rrc@8eZdZdZddZddZddZdd Zd d Zd S) z`UNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptorsFcsPt_jdfdd}t|_jjj_jdt ddS)NTcs"jjjddS)Nx)rorpr rrqrSrsrSrTsenderfszUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.connectionMade..senderrzSend loop failure) r roregisterProducerr/taskrclockstartrr?)rNrrSrsrTrtbs    zoUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.connectionMadecS |dSru _disconnectrsrSrSrT stopProducingn znUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.stopProducingcSrrurrsrSrSrTresumeProducingqrzpUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.resumeProducingcSsd|_|j|dS)NT)pausedrounregisterProducerrrsrSrSrTrts  zoUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors.pauseProducingcSs$|j|j|jjdSru)rstoproabortConnectionotherrsrSrSrTrys  zlUNIXTestsBuilder.test_sendFileDescriptorTriggersPauseProducing..SendsManyFileDescriptors._disconnectN) rVrWrXrrtrrrrrSrSrSrTSendsManyFileDescriptors_s  rz*sendFileDescriptor producer was not pausedN)r0rr4r assertTruer)rNrrrerirSrSrT-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}. NcrrurrrrSrTrrz=UNIXTestsBuilder.test_fileDescriptorOverrun..r)r rkrryr{rappendr4rassertIsInstancer<trapr&rwvaluer')rNcargoriresultrrSrrTtest_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|jfSru)rst_devst_ino)rlfsrSrSrTdeviceInodeTuples zTUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..deviceInodeTuplecs eZdZddZfddZdS)zPUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeProtocolcSsg|_g|_dSru)fdsdeviceInodesReceivedrsrSrSrTrn zYUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeProtocol.__init__cs(|j||j|t|dSru)rrrr)rNrlrrSrTr~s  zgUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeProtocol.fileDescriptorReceivedN)rVrWrXrnr~rSrrSrT FakeProtocolsrc@r) zPUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceivercSs||_||_dSru)r protocol)rNsktprotorSrSrTrnrzYUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver.__init__cSdSrurSrrSrSrT _dataReceivedz^UNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver._dataReceivedcSrrurSrsrSrSrTrrzXUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver.getHostcSrrurSrsrSrSrTrrzXUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver.getPeercSrrurS)rNorSrSrT _getLogPrefixrz^UNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..FakeReceiver._getLogPrefixN) rVrWrX bufferSizernrrrrrSrSrSrT FakeReceivers rssome data needs to be sentcsg|]}|qSrSrS.0rlrrSrT szNUNIXTestsBuilder._sendmsgMixinFileDescriptorReceivedDriver..N)r rtwisted.internet.unixrtwisted.python.sendmsgrrr(r0rr addCleanuprrrdoReadrLlenr assertFalseset intersectionr)rNancillaryPackerrrrrr sendSocket recvSocketrreceiver fileOneFD fileOneName fileTwoFD fileTwoName dataToSend fdsToSend ancillary expectedCountdeviceInodesSentrSrrT)_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 rrrrr rSrTr sz[UNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgOneCMSG..ancillaryPackerNrr r rNrrSr rT1test_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. rr csfdd|D}d}||fS)Ncsg|] }ttd|fqS)irrr rSrTrszpUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgTwoCMSGs..ancillaryPacker..rrSrr rSrTrsz\UNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgTwoCMSGs..ancillaryPackerNrrrSr rT2test_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)NrrSrrSrSrTr3sz[UNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSG..ancillaryPackercsd}dg}d}|||S)Ns some data)NNr)ReceivedMessage)rargskwargsrmrflagsrrSrTfakeRecvmsgUnsupportedAncillary8szkUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSG..fakeRecvmsgUnsupportedAncillaryrecvmsgz#received unsupported ancillary datac3s|] }|dvVqdS)formatNrS)re)expectedMessagerSrT GszUUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSG..z+Expected message not found in logged eventsN) twisted.pythonrr>rrr@rr anyr)rNrreventsfoundrS)r rrT1test_multiFileDescriptorReceivedPerRecvmsgBadCMSG$s    zBUNIXTestsBuilder.test_multiFileDescriptorReceivedPerRecvmsgBadCMSGcCsddlm}|\}}g}t|j|t|jGdddt}||d}t}t ||||j | | d| d|dd }d } t|j|j| |d } |D]} | D] \} } | | | krhnq[d SqU|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|dSru)ror hostAddressr peerAddressrkrtrsrSrSrTrt_s  zaUNIXTestsBuilder.test_avoidLeakingFileDescriptors..RecordEndpointAddresses.connectionMadeNrrSrSrSrTRecordEndpointAddresses^rr)rFrrz%(protocolName)s (on %(hostAddress)r) does not provide IFileDescriptorReceiver; closing file descriptor received (from %(peerAddress)r).r0)r'r( protocolNamerz3Expected event (%s) not found in logged events (%s)N)r rr>rrr@rkrr0r4rr setblockingrLrecvrr(r'itemsgetr!r)rNr probeClient probeServerr$r)rerirclsName expectedEventlogEventkvrSrSrT test_avoidLeakingFileDescriptorsJsJ   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|_dSru)r0rtr$rsrSrSrTrts  zYUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes..RecordEvents.connectionMadecs t||jt|dSru)rrr$rtype) innerSelfr}rsrSrTr~s zaUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes..RecordEvents.fileDescriptorReceivedcSs|j|dSru)r$extendrrSrSrTrszWUNIXTestsBuilder.test_descriptorDeliveredBeforeBytes..RecordEvents.dataReceivedN)rVrWrXrtr~rrSrsrSrT RecordEventss  r:rrN) rr(r0r rkrr4rrLintr$bytes)rNr:rrerirSrsrT#test_descriptorDeliveredBeforeBytess z4UNIXTestsBuilder.test_descriptorDeliveredBeforeBytesN)rVrWrXrYr+r_r^rrrrBisLinuxrrrrrsendmsgSkipReasonrrrr risMacOSXrr&r6r>rSrSrSrTrsL      "  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)rUrr-rsrSrSrTtest_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]rIrCr-rLrrrrSrSrTrs 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`rarbF) r rrrlistenr+adoptStreamPortrfamilyr)rNrrQportSockrPrSrSrTgetListeningPorts    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 rrgrNraddressrQrSrSrTconnectToListenerz!SocketUNIXMixin.connectToListenerN)rVrWrXrYr+r*r_rrrrrjrLrRrSrSrSrTrFs  rFc@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`rarb)rr)rNrrQrPrSrSrTrL s  z ListenUNIXMixin.getListeningPortcCrMrNrOrPrSrSrTrRrSz!ListenUNIXMixin.connectToListenerN)rVrWrXrYrLrRrSrSrSrTrTs rTc@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;rrg)rNrrQrSrSrT#getExpectedStartListeningLogMessage&sz6UNIXPortTestsMixin.getExpectedStartListeningLogMessagecCsdt|jdS)zJ Get the expected connection lost message for a UNIX port z (UNIX Port z Closed)rV)rNrrSrSrTgetExpectedConnectionLostLogMsg,sz2UNIXPortTestsMixin.getExpectedConnectionLostLogMsgN) rVrWrXr+r_rrrrrjrWrXrSrSrSrTrU#s  rUc@eZdZdZdS)UNIXPortTestsBuilderz. Tests for L{IReactorUNIX.listenUnix} NrVrWrXrYrSrSrSrTrZ3rZc@rY)UNIXFDPortTestsBuilderz3 Tests for L{IReactorUNIX.adoptStreamPort} Nr[rSrSrSrTr]>r\r]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@r)zXUNIXAdoptStreamConnectionTestsBuilder.test_buildProtocolReturnsNone..NoneFactorycSrrurS)rNrQrSrSrT buildProtocol_rzfUNIXAdoptStreamConnectionTestsBuilder.test_buildProtocolReturnsNone..NoneFactory.buildProtocolN)rVrWrXr_rSrSrSrT NoneFactory^rr`FN) rIr rr.rr r+rrradoptStreamConnection assertIsNone) rNrrr`s1s2s1FDrQrrSrSrTtest_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=rrgrLro sessionnostrlogstrrQ peerAddressesrrrr) protocolsrirerportPathr(rsrSrT connectedqs$    zOUNIXAdoptStreamConnectionTestsBuilder.test_ServerAddressUNIX..connectedN) interface addressFamily)rIgetConnectedClientAndServerr runReactor)rNrmrrrSrsrTtest_ServerAddressUNIXls  z.firstServerConnectedcsjr|Sru)runningr)r)rrSrTrszOUNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServer..stopcs|\}}||fdSru)r|)rkrire)deferredrrSrTrszPUNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServer..start)r8r protocolConnectionMadeprotocolConnectionLostr7rrrr"rrrrrg) rNrrnro firstServerrirPru lostDeferredr startDeferredrrS)rwrrrerTrps4         zAUNIXAdoptStreamConnectionTestsBuilder.getConnectedClientAndServerN) rVrWrXr)r*r+r_rfrrrprSrSrSrTr^Is "r^c@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]rsrSrSrTrPs  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. )rrPrNrrQrSrSrTrH zUnixClientTestsBuilder.listencCr)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. )rrPrrSrSrTconnectrzUnixClientTestsBuilder.connect) rVrWrXrYr+r_r~propertyrPrHrrSrSrSrTr}s  r}posixz%UNIX sockets only supported on POSIX.c@sReZdZdZdZddZdeeegdfddfdd Zd d Z d d Z ddZ dS)RealSocketTestsz, Test real UNIX socket connections. unixpathcstjts 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 isinstancerPortrrcleanUpr7rrw lastProtocolrmade disconnected lostReasonrrConnectionDone)rNfrclientFrSrSrTtest_closePortInProtocolFactorys    z/RealSocketTests.test_closePortInProtocolFactoryr|Nreturnc 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+rrr8r rxrrPrrKr7rr"rorr) rNr| serverFactoryserverConnMaderrclientConnMadeserverProtocolclientProtocolrSrSrT_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. rrrNcsn||fD]0}tj|jsJ|j}|d|d|d|d|dqdS)NrTF)r ITCPTransportrrorL getTcpNoDelay setTcpNoDelayrrprorsrSrTcheckA   z.RealSocketTests.test_tcpNoDelay..checkrDrrNrrSrsrTtest_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}. rrrNcsn||fD]0}tj|jsJ|j}|d|d|d|d|dqdS)NFT)rrrrorLgetTcpKeepAlivesetTcpKeepAliverrsrSrTrWrz0RealSocketTests.test_tcpKeepAlive..checkrrrSrsrTtest_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+rrr7r failDeferredrwrr ConnectError)rNrrSrSrT test_Failinges  zRealSocketTests.test_Failing) rVrWrXrYrPrrrDrrrrrSrSrSrTrs  r)vrYhashlibrosrrrrrpprintrr r r r r structrtempfilerrtypingrrrrrunittestrr_AF_UNIX ImportErrorzope.interfacerrtwisted.internetrrrrrtwisted.internet.addressrtwisted.internet.deferr r!r"twisted.internet.endpointsr#r$twisted.internet.errorr%r&r'twisted.internet.interfacesr(r)r*r+twisted.internet.protocolr,r-r.twisted.internet.taskr/&twisted.internet.test.connectionmixinsr0r1r2r3r4#twisted.internet.test.reactormixinsr5twisted.internet.test.test_tcpr6r7r8r9r:twisted.python.compatr;twisted.python.failurer<twisted.python.filepathr=twisted.python.logr>r?r@twisted.python.reflectrAtwisted.python.runtimerBrCtwisted.test.test_tcprDtwisted.trial.unittestrErr@rGr]r^rkryrrBrFrTrUrZr]r^globalsupdatemakeTestCaseClassesr}rrSrSrSrTs               %>Y!-   ~-