Your IP : 216.73.216.85
��Yf� � + @ s d d l Z d d l m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l
Z
d d l Z d d l Z d d l
Z
d d l Z d d l Z d d l Z d d l m Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z y d d l Z Wn e k
rFd Z Yn Xy d d l Z Wn e k
rqd Z Yn Xe j Z d j d � Z y d d l Z d d l! Z! Wn e k
r�d Z d Z! Yn Xy d d l" Z" Wn e k
r�d Z" Yn Xd d � Z# d d
� Z$ e# � Z% e$ � Z& e j d � j' Z( Gd d
� d
e j) � Z* Gd d � d e j) � Z+ Gd d � d e j) � Z, Gd d � d e j) � Z- Gd d � d e j) � Z. Gd d � d � Z/ Gd d � d e* e/ � Z0 Gd d � d e+ e/ � Z1 Gd d � d e- e/ � Z2 Gd d � d e. e/ � Z3 Gd d! � d! e0 � Z4 Gd" d# � d# e j) e/ � Z5 Gd$ d% � d% e j) � Z6 Gd& d' � d' e6 � Z7 Gd( d) � d) e, e6 e/ � Z8 Gd* d+ � d+ e7 e8 � Z9 Gd, d- � d- e6 � Z: Gd. d/ � d/ e: � Z; Gd0 d1 � d1 e6 � Z< Gd2 d3 � d3 e< � Z= Gd4 d5 � d5 e< � Z> Gd6 d7 � d7 e< � Z? Gd8 d9 � d9 e< � Z@ Gd: d; � d; e@ � ZA d< d= � ZB d> d? � ZC d@ dA � ZD GdB dC � dC e j) � ZE e jF e% dD � GdE dF � dF e j) � � ZG e jF e% dD � e jF e dG � GdH dI � dI e2 � � � ZH e jF e& dJ � GdK dL � dL e j) � � ZI e jF e& dJ � e jF e dG � GdM dN � dN e3 � � � ZJ e jF e dG � GdO dP � dP e4 � � ZK e jF e dG � GdQ dR � dR e1 � � ZL GdS dT � dT e, � ZM GdU dV � dV eM � ZN GdW dX � dX eM � ZO GdY dZ � dZ eM � ZP Gd[ d\ � d\ eM � ZQ Gd] d^ � d^ eM � ZR Gd_ d` � d` eM � ZS Gda db � db eS � ZT Gdc dd � dd eT � ZU Gde df � df eT � ZV Gdg dh � dh eM � ZW Gdi dj � dj eW � ZX Gdk dl � dl eW � ZY Gdm dn � dn eN eW � ZZ Gdo dp � dp e j) � Z[ Gdq dr � dr eS � Z\ Gds dt � dt eS � Z] Gdu dv � dv eO eQ e8 e> � Z^ eC e j dw � e jF e dG � Gdx dy � dy eV e^ � � � Z_ eC e j dz � e jF e dG � Gd{ d| � d| eY e^ � � � Z` eC e j d} � e jF e dG � Gd~ d � d eZ e^ � � � Za Gd� d� � d� eO eQ e8 eA � Zb eC e j dw � e jF e jc d� � eD d� d� � e jF e dG � Gd� d� � d� eV eb � � � � � Zd eC e j dz � e jF e jc d� � eD d� d� � e jF e dG � Gd� d� � d� eY eb � � � � � Ze eC e j d} � e jF e jc d� � eD d� d� � e jF e dG � Gd� d� � d� eZ eb � � � � � Zf eC e j dz � e jF e jc d� � eC e d� � eD d� d� � e jF e dG � Gd� d� � d� e] eb � � � � � � Zg eC e j d} � e jF e jc d� � eC e d� � eD d� d� � e jF e dG � Gd� d� � d� eN e] eb � � � � � � Zh Gd� d� � d� eR e9 e= � Zi eC e j dw � e jF e dG � Gd� d� � d� eU ei � � � Zj eC e j dz � e jF e dG � Gd� d� � d� eY eX ei � � � Zk eC e j d} � e jF e dG � Gd� d� � d� eZ eX ei � � � Zl Gd� d� � d� eP eR e9 e? � Zm eC e j dw � eD d� d� d� � e jF e dG � Gd� d� � d� eU em � � � � Zn eC e j dz � eD d� d� d� � e jF e dG � Gd� d� � d� eY eX em � � � � Zo eC e j d} � eD d� d� d� � e jF e dG � Gd� d� � d� eZ eX em � � � � Zp Gd� d� � d� eR e9 e; � Zq eC e j dw � eC e d� � e jF e dG � Gd� d� � d� eU eq � � � � Zr eC e j dz � eC e d� � e jF e dG � Gd� d� � d� eY eX eq � � � � Zs eC e j d} � eC e d� � e jF e dG � Gd� d� � d� eZ eX eq � � � � Zt eC e j dw dz � eC e d� d� d� � e jF e dG � Gd� d� � d� e\ eq � � � � Zu eC e j dw d} � eC e d� d� d� � e jF e dG � Gd� d� � d� eN e\ eq � � � � Zv Gd� d� � d� e j) � Zw eC e d� � e jF ex e d� � pex e d� � d� � Gd� d� � d� ew e> � � � Zy eC e d� � e jF ex e d� � plex e d� � d� � e jF e dG � Gd� d� � d� ew e, e7 e= � � � � Zz e jF e dG � Gd� d� � d� e0 � � Z{ e jF e dG � Gd� d� � d� e5 � � Z| e jF e dG � Gd� d� � d� e0 � � Z} e jF e dG � Gd� d� � d� e4 � � Z~ Gd� d� � d� e~ � Z Gd� d� � d� e~ � Z� Gd� d� � d� e~ � Z� Gd� d� � d� e~ � Z� Gd� d� � d� e~ � Z� Gd� d� � d� e~ � Z� Gd� d� � d� e� � Z� Gd� d� � d� e� eK � Z� Gd� d� � d� e j) � Z� e jF e dG � Gd� d� � d� e* e/ � � Z� e jF e dG � Gd� d� � d� e* e/ � � Z� Gd� d� � d� e* � Z� Gd� d� � d� e+ � Z� Gd� d� � d� e j) � Z� e jF e j d� k d� � Gd� d� � d� e j) � � Z� e jF ex e d� � d� � Gd� d� � d� e j) � � Z� e jF e dG � Gd� d� � d� e4 � � Z� d� Z� d� Z� d� Z� d� d� � Z� e jF e� � d� � Gd� d� � d� e j) � � Z� e jF e� � d� � Gd� d� � d� e j) e/ � � Z� e jF e dG � Gd� d� � d� e0 � � Z� Gd� d� � d� e j) � Z� e jF ex e d� � d� � Gd� d� � d� e j) � � Z� e jF e
j� d� k d� � e jF e d� � Gd� d� � d� e* � � � Z� e jF e dG � Gd d� de0 � � Z� e jF e dG � e jF ex e
d� d� Gdd� de� � � � Z� dd� Z� e� dk r�e� � d S( � N)�support)�proxyFu Michael Gilfix was hereሴ
zutf-8c C sO y" t j t j t j t j � } Wn t t f k
r@ d SYn X| j � d S)z5Check whether CAN sockets are supported on this host.FT)�socket�PF_CAN�SOCK_RAW�CAN_RAW�AttributeError�OSError�close)�s� r �5/opt/alt/python35/lib64/python3.5/test/test_socket.py�_have_socket_can1 s "
r c C sL y t j t j t j d � } Wn t t f k
r= d SYn X| j � d S)z5Check whether RDS sockets are supported on this host.r FT)r �PF_RDS�SOCK_SEQPACKETr r r
)r r r r
�_have_socket_rds; s
r �ic @ s( e Z d Z d d � Z d d � Z d S)�
SocketTCPTestc C sA t j t j t j � | _ t j | j � | _ | j j � d S)N)r �AF_INET� SOCK_STREAM�servr � bind_port�port�listen)�selfr r r
�setUpN s zSocketTCPTest.setUpc C s | j j � d | _ d S)N)r r
)r r r r
�tearDownS s
zSocketTCPTest.tearDownN)�__name__�
__module__� __qualname__r r r r r r
r L s r c @ s( e Z d Z d d � Z d d � Z d S)�
SocketUDPTestc C s4 t j t j t j � | _ t j | j � | _ d S)N)r r �
SOCK_DGRAMr r r r )r r r r
r Y s zSocketUDPTest.setUpc C s | j j � d | _ d S)N)r r
)r r r r
r ] s
zSocketUDPTest.tearDownN)r r r r r r r r r
r W s r c sR e Z d Z d Z e rN � f d d � Z � f d d � Z � f d d � Z � S)�ThreadSafeCleanupTestCasez�Subclass of unittest.TestCase with thread-safe cleanup methods.
This subclass protects the addCleanup() and doCleanups() methods
with a recursive lock.
c s& t � j | | � t j � | _ d S)N)�super�__init__� threading�RLock�
_cleanup_lock)r �args�kwargs)� __class__r r
r$ i s z"ThreadSafeCleanupTestCase.__init__c
s( | j � t � j | | � SWd QRXd S)N)r' r# �
addCleanup)r r( r) )r* r r
r+ m s
z$ThreadSafeCleanupTestCase.addCleanupc
s( | j � t � j | | � SWd QRXd S)N)r' r# �
doCleanups)r r( r) )r* r r
r, q s
z$ThreadSafeCleanupTestCase.doCleanups)r r r �__doc__r% r$ r+ r, r r )r* r
r" a s
r" c @ sd e Z d Z d Z d Z d Z d Z e j e � Z d Z
e
d e j e
� d 7Z
d d � Z d
S) �
SocketCANTestz�To be able to run this test, a `vcan0` CAN interface can be created with
the following commands:
# modprobe vcan
# ip link add dev vcan0 type vcan
# ifconfig vcan0 up
Zvcan0� z=IB3x8sz@3I4l2I�x� c C s{ t j t j t j t j � | _ | j | j j � y | j j | j f � Wn& t k
rv | j
d | j � Yn Xd S)Nz%network interface `%s` does not exist) r r r r r r+ r
�bind� interfacer �skipTest)r r r r
r � s !
zSocketCANTest.setUpN) r r r r- r3 �bufsize�
can_frame_fmt�structZcalcsizeZcan_frame_size�bcm_cmd_msg_fmtr r r r r
r. u s
r. c @ s( e Z d Z d Z d Z d d � Z d S)�
SocketRDSTestz\To be able to run this test, the `rds` kernel module must be loaded:
# modprobe rds
i c C sp t j t j t j d � | _ | j | j j � y t j | j � | _ Wn t k
rk | j
d � Yn Xd S)Nr zunable to bind RDS socket) r r r r r+ r
r r r r r4 )r r r r
r � s
zSocketRDSTest.setUpN)r r r r- r5 r r r r r
r9 � s r9 c @ sj e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d S)�ThreadableTesta� Threadable Test class
The ThreadableTest class makes it easy to create a threaded
client/server pair from an existing unit test. To create a
new threaded class from an existing unit test, use multiple
inheritance:
class NewClass (OldClass, ThreadableTest):
pass
This class defines two new fixture functions with obvious
purposes for overriding:
clientSetUp ()
clientTearDown ()
Any new test functions within the class must then define
tests in pairs, where the test name is preceded with a
'_' to indicate the client portion of the test. Ex:
def testFoo(self):
# Server portion
def _testFoo(self):
# Client portion
Any exceptions raised by the clients during their tests
are caught and transferred to the main thread to alert
the testing framework.
Note, the server setup function cannot call any blocking
functions that rely on the client thread during setup,
unless serverExplicitReady() is called just before
the blocking call (such as in setting up a client/server
connection and performing the accept() in setUp().
c C s4 | j | _ | j | _ | j | _ | j | _ d S)N)r �_ThreadableTest__setUpr �_ThreadableTest__tearDown�_setUp� _tearDown)r r r r
r$ � s zThreadableTest.__init__c C s | j j � d S)z�This method allows the server to explicitly indicate that
it wants the client thread to proceed. This is useful if the
server is about to execute a blocking routine that is
dependent upon the client thread during its setup routine.N)� server_ready�set)r r r r
�serverExplicitReady� s z"ThreadableTest.serverExplicitReadyc C s� t j � | _ t j � | _ t j � | _ t j d � | _ d | _ | j � } | j d � } | | d d � } t
| d | � } t j | j
| f � | _ z) y | j � Wn d | _ � Yn XWd | j j � X| j j � d S)N� F�.�_T)r% �Eventr? � client_ready�done�queue�Queue�server_crashed�id�rfind�getattr�thread�start_new_thread� clientRunZ
client_threadr; r@ �wait)r Z
methodnamer Z test_methodr r r
r= � s$ zThreadableTest._setUpc C s? | j � | j j � | j j � r; | j j � } | � d S)N)r< rG rQ rH Zqsize�get)r �excr r r
r> s
zThreadableTest._tearDownc C s� | j j � | j � | j j � | j r; | j � d St | d � sV t d � � zG y | � Wn5 t k
r� } z | j
j | � WYd d } ~ Xn XWd | j � Xd S)N�__call__z%test_func must be a callable function) r? rQ � clientSetUprF r@ rJ �clientTearDown�hasattr� TypeError�
BaseExceptionrH �put)r Z test_func�er r r
rP s
'zThreadableTest.clientRunc C s t d � � d S)Nz clientSetUp must be implemented.)�NotImplementedError)r r r r
rU s zThreadableTest.clientSetUpc C s | j j � t j � d S)N)rG r@ rN �exit)r r r r
rV s
z ThreadableTest.clientTearDownN) r r r r- r$ rA r= r> rP rU rV r r r r
r: � s $ r: c @ s7 e Z d Z d d d � Z d d � Z d d � Z d S) �ThreadedTCPSocketTest�runTestc C s$ t j | d | �t j | � d S)N�
methodName)r r$ r: )r r` r r r
r$ % s z ThreadedTCPSocketTest.__init__c C s t j t j t j � | _ d S)N)r r r �cli)r r r r
rU ) s z!ThreadedTCPSocketTest.clientSetUpc C s' | j j � d | _ t j | � d S)N)ra r
r: rV )r r r r
rV , s
z$ThreadedTCPSocketTest.clientTearDownN)r r r r$ rU rV r r r r
r^ # s r^ c @ s7 e Z d Z d d d � Z d d � Z d d � Z d S) �ThreadedUDPSocketTestr_ c C s$ t j | d | �t j | � d S)Nr` )r r$ r: )r r` r r r
r$ 3 s z ThreadedUDPSocketTest.__init__c C s t j t j t j � | _ d S)N)r r r! ra )r r r r
rU 7 s z!ThreadedUDPSocketTest.clientSetUpc C s' | j j � d | _ t j | � d S)N)ra r
r: rV )r r r r
rV : s
z$ThreadedUDPSocketTest.clientTearDownN)r r r r$ rU rV r r r r
rb 1 s rb c @ s7 e Z d Z d d d � Z d d � Z d d � Z d S) �ThreadedCANSocketTestr_ c C s$ t j | d | �t j | � d S)Nr` )r. r$ r: )r r` r r r
r$ A s z ThreadedCANSocketTest.__init__c C sT t j t j t j t j � | _ y | j j | j f � Wn t k
rO Yn Xd S)N)r r r r ra r2 r3 r )r r r r
rU E s
!
z!ThreadedCANSocketTest.clientSetUpc C s' | j j � d | _ t j | � d S)N)ra r
r: rV )r r r r
rV N s
z$ThreadedCANSocketTest.clientTearDownN)r r r r$ rU rV r r r r
rc ? s rc c @ s7 e Z d Z d d d � Z d d � Z d d � Z d S) �ThreadedRDSSocketTestr_ c C s$ t j | d | �t j | � d S)Nr` )r9 r$ r: )r r` r r r
r$ U s z ThreadedRDSSocketTest.__init__c C sc t j t j t j d � | _ y, | j j t d f � | j j � | _ Wn t k
r^ Yn Xd S)Nr ) r r r ra r2 �HOST� getsockname�cli_addrr )r r r r
rU Y s
z!ThreadedRDSSocketTest.clientSetUpc C s' | j j � d | _ t j | � d S)N)ra r
r: rV )r r r r
rV d s
z$ThreadedRDSSocketTest.clientTearDownN)r r r r$ rU rV r r r r
rd S s rd c @ sU e Z d Z d Z d d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�SocketConnectedTestz�Socket tests for client-server connection.
self.cli_conn is a client socket connected to the server. The
setUp() method guarantees that it is connected to the server.
r_ c C s t j | d | �d S)Nr` )r^ r$ )r r` r r r
r$ p s z SocketConnectedTest.__init__c C s9 t j | � | j � | j j � \ } } | | _ d S)N)r^ r rA r �accept�cli_conn)r �conn�addrr r r
r s s
zSocketConnectedTest.setUpc C s' | j j � d | _ t j | � d S)N)rj r
r^ r )r r r r
r { s
z SocketConnectedTest.tearDownc C s6 t j | � | j j t | j f � | j | _ d S)N)r^ rU ra �connectre r � serv_conn)r r r r
rU � s
z SocketConnectedTest.clientSetUpc C s' | j j � d | _ t j | � d S)N)rn r
r^ rV )r r r r
rV � s
z"SocketConnectedTest.clientTearDownN) r r r r- r$ r r rU rV r r r r
rh i s rh c @ sO e Z d Z d d d � Z d d � Z d d � Z d d � Z d
d � Z d S)
�SocketPairTestr_ c C s' t j j | d | �t j | � d S)Nr` )�unittest�TestCaser$ r: )r r` r r r
r$ � s zSocketPairTest.__init__c C s t j � \ | _ | _ d S)N)r �
socketpairr ra )r r r r
r � s zSocketPairTest.setUpc C s | j j � d | _ d S)N)r r
)r r r r
r � s
zSocketPairTest.tearDownc C s d S)Nr )r r r r
rU � s zSocketPairTest.clientSetUpc C s' | j j � d | _ t j | � d S)N)ra r
r: rV )r r r r
rV � s
z SocketPairTest.clientTearDownN)r r r r$ r r rU rV r r r r
ro � s
ro c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S) �SocketTestBasez�A base class for socket tests.
Subclasses must provide methods newSocket() to return a new socket
and bindSock(sock) to bind it to an unused address.
Creates a socket self.serv and sets self.serv_addr to its address.
c C s | j � | _ | j � d S)N)� newSocketr �
bindServer)r r r r
r � s zSocketTestBase.setUpc C s& | j | j � | j j � | _ d S)z9Bind server socket and set self.serv_addr to its address.N)�bindSockr rf � serv_addr)r r r r
ru � s zSocketTestBase.bindServerc C s | j j � d | _ d S)N)r r
)r r r r
r � s
zSocketTestBase.tearDownN)r r r r- r ru r r r r r
rs � s rs c s( e Z d Z d Z � f d d � Z � S)�SocketListeningTestMixinz%Mixin to listen on the server socket.c s t � j � | j j � d S)N)r# r r r )r )r* r r
r � s
z SocketListeningTestMixin.setUp)r r r r- r r r )r* r
rx � s rx c sX e Z d Z d Z � f d d � Z d d � Z d d � Z d d � Z d
d � Z � S) �ThreadedSocketTestMixinz�Mixin to add client socket and allow client/server tests.
Client socket is self.cli and its address is self.cli_addr. See
ThreadableTest for usage information.
c s$ t � j | | � t j | � d S)N)r# r$ r: )r r( r) )r* r r
r$ � s z ThreadedSocketTestMixin.__init__c C s | j � | _ | j � d S)N)�newClientSocketra �
bindClient)r r r r
rU � s z#ThreadedSocketTestMixin.clientSetUpc C s
| j � S)z&Return a new socket for use as client.)rt )r r r r
rz � s z'ThreadedSocketTestMixin.newClientSocketc C s& | j | j � | j j � | _ d S)z8Bind client socket and set self.cli_addr to its address.N)rv ra rf rg )r r r r
r{ � s z"ThreadedSocketTestMixin.bindClientc C s' | j j � d | _ t j | � d S)N)ra r
r: rV )r r r r
rV � s
z&ThreadedSocketTestMixin.clientTearDown) r r r r- r$ rU rz r{ rV r r )r* r
ry � s ry c s^ e Z d Z d Z � f d d � Z � f d d � Z � f d d � Z � f d d � Z � S)
�ConnectedStreamTestMixinz�Mixin to allow client/server stream tests with connected client.
Server's socket representing connection to client is self.cli_conn
and client's connection to server is self.serv_conn. (Based on
SocketConnectedTest.)
c s9 t � j � | j � | j j � \ } } | | _ d S)N)r# r rA r ri rj )r rk rl )r* r r
r � s
z ConnectedStreamTestMixin.setUpc s' | j j � d | _ t � j � d S)N)rj r
r# r )r )r* r r
r � s
z!ConnectedStreamTestMixin.tearDownc s0 t � j � | j j | j � | j | _ d S)N)r# rU ra rm rw rn )r )r* r r
rU � s
z$ConnectedStreamTestMixin.clientSetUpc s' | j j � d | _ t � j � d S)N)rn r
r# rV )r )r* r r
rV � s
z'ConnectedStreamTestMixin.clientTearDown)r r r r- r r rU rV r r )r* r
r| � s
r| c s4 e Z d Z d Z � f d d � Z d d � Z � S)�UnixSocketTestBasez(Base class for Unix-domain socket tests.c s6 t j � | _ | j t j | j � t � j � d S)N)�tempfileZmkdtemp�dir_pathr+ �os�rmdirr# r )r )r* r r
r
s zUnixSocketTestBase.setUpc C s9 t j d | j � } | j | � | j t j | � d S)N�dir)r~ Zmktempr r2 r+ r �unlink)r �sock�pathr r r
rv s
zUnixSocketTestBase.bindSock)r r r r- r rv r r )r* r
r} s r} c @ s" e Z d Z d Z d d � Z d S)�UnixStreamBasez-Base class for Unix-domain SOCK_STREAM tests.c C s t j t j t j � S)N)r �AF_UNIXr )r r r r
rt s zUnixStreamBase.newSocketN)r r r r- rt r r r r
r� s r� c s: e Z d Z d Z e Z � f d d � Z d d � Z � S)� InetTestBasez!Base class for IPv4 socket tests.c s! t � j � | j d | _ d S)NrB )r# r rw r )r )r* r r
r # s
zInetTestBase.setUpc C s t j | d | j �d S)N�host)r r r� )r r� r r r
rv ' s zInetTestBase.bindSock)r r r r- re r� r rv r r )r* r
r� s r� c @ s" e Z d Z d Z d d � Z d S)� TCPTestBasez#Base class for TCP-over-IPv4 tests.c C s t j t j t j � S)N)r r r )r r r r
rt - s zTCPTestBase.newSocketN)r r r r- rt r r r r
r� * s r� c @ s" e Z d Z d Z d d � Z d S)� UDPTestBasez#Base class for UDP-over-IPv4 tests.c C s t j t j t j � S)N)r r r! )r r r r
rt 3 s zUDPTestBase.newSocketN)r r r r- rt r r r r
r� 0 s r� c @ s" e Z d Z d Z d d � Z d S)�SCTPStreamBasez;Base class for SCTP tests in one-to-one (SOCK_STREAM) mode.c C s t j t j t j t j � S)N)r r r � IPPROTO_SCTP)r r r r
rt 9 s zSCTPStreamBase.newSocketN)r r r r- rt r r r r
r� 6 s r� c @ s e Z d Z d Z e j Z d S)�
Inet6TestBasez!Base class for IPv6 socket tests.N)r r r r- r �HOSTv6r� r r r r
r� > s r� c @ s" e Z d Z d Z d d � Z d S)� UDP6TestBasez#Base class for UDP-over-IPv6 tests.c C s t j t j t j � S)N)r �AF_INET6r! )r r r r
rt F s zUDP6TestBase.newSocketN)r r r r- rt r r r r
r� C s r� c s; d d � � � � f d d � } d d � } | r7 | S| S)a Skip decorated test if condition is true, add client_skip decorator.
If the decorated object is not a class, sets its attribute
"client_skip" to a decorator which will return an empty function
if the test is to be skipped, or the original function if it is
not. This can be used to avoid running the client part of a
skipped test when using ThreadableTest.
c _ s d S)Nr )r( r) r r r
� client_passU s z%skipWithClientIf.<locals>.client_passc s= t j � � | � } t | t � s9 � f d d � | _ | S)Nc s � S)Nr )�f)r� r r
�<lambda>Z s z3skipWithClientIf.<locals>.skipdec.<locals>.<lambda>)rp �skip�
isinstance�type� client_skip)�objZretval)r� �reasonr r
�skipdecW s z!skipWithClientIf.<locals>.skipdecc S s1 t | t � p t | d � s- d d � | _ | S)Nr� c S s | S)Nr )r� r r r
r� ^ s z5skipWithClientIf.<locals>.noskipdec.<locals>.<lambda>)r� r� rW r� )r� r r r
� noskipdec\ s z#skipWithClientIf.<locals>.noskipdecr )Z conditionr� r� r� r )r� r� r
�skipWithClientIfL s r� c s@ � f d d � | D� } t | d d j d d � | D� � � S)z�Skip decorated test if obj is missing any of the given attributes.
Sets client_skip attribute as skipWithClientIf() does.
c s% g | ] } t � | � s | � q Sr )rW )�.0�name)r� r r
�
<listcomp>h s z requireAttrs.<locals>.<listcomp>z don't have z, c s s | ] } | Vq d S)Nr )r� r� r r r
� <genexpr>j s z requireAttrs.<locals>.<genexpr>)r� �join)r� Z
attributes�missingr )r� r
� requireAttrsc s r� c G s� d } d d � | D� } | rB d d j d d � | D� � } nd d d � | D� } y t j | � } Wn1 t k
r� } z t | � } WYd d } ~ Xn X| j � t | d k d j d j d
d � | D� � | � � S) a Skip decorated test if a socket cannot be created with given arguments.
When an argument is given as a string, will use the value of that
attribute of the socket module, or skip the test if it doesn't
exist. Sets client_skip attribute as skipWithClientIf() does.
Nc S s5 g | ]+ } t | t � r t t | � r | � q Sr )r� �strrW r )r� r� r r r
r� u s z!requireSocket.<locals>.<listcomp>z don't have z, c s s | ] } | Vq d S)Nr )r� r� r r r
r� x s z requireSocket.<locals>.<genexpr>c S s4 g | ]* } t | t � r* t t | � n | � q Sr )r� r� rM r )r� r� r r r
r� z s z can't create socket({0}): {1}c s s | ] } t | � Vq d S)N)r� )r� �or r r
r� � s )r� r r r� r
r� �format)r( �errr� Zcallargsr r[ r r r
�
requireSocketm s #
r� c @ s e Z d Z d d � Z d d � Z e j e d k d � d d � � Z d d
� Z d d � Z
d
d � Z d d � Z d d � Z
d d � Z e j e e d � d � e j e e d � d � d d � � � Z e j e e d � d � d d � � Z e j e e d � d � d d � � Z e j e e d! � d" � d# d$ � � Z d% d& � Z d' d( � Z e j d) d* � � Z d+ d, � Z d- d. � Z e j e e d/ � d0 � d1 d2 � � Z e j e e d3 � d4 � d5 d6 � � Z e j e e d3 � d4 � d7 d8 � � Z e j e e d9 � d: � d; d< � � Z e j e e d9 � d: � d= d> � � Z! d? d@ � Z" dA dB � Z# dC dD � Z$ dE dF � Z% dG dH � Z& dI dJ � Z' e j e( j) dK k dL � dM dN � � Z* dO dP � Z+ dQ dR � Z, e j e j- dS � dT � dU dV � � Z. dW dX � Z/ dY dZ � Z0 d[ d\ � Z1 d] d^ � Z2 d_ d` � Z3 da db � Z4 dc dd � Z5 de df � Z6 dg dh � Z7 di dj � Z8 e j dk dl � � Z9 e j e j: dm � dn do � � Z; dp dq � Z< e j= e( j) dK k dr � ds dt � � Z> e j e e( du � dv � dw dx � � Z? d S)y�GeneralModuleTestsc C sR d d l } | j t j | j k � t j � } | j | t j � | j � d S)Nr )�_socket�
assertTruer Z
SocketType�assertIsInstancer
)r r� r r r r
� test_SocketType_is_socketobject� s
z2GeneralModuleTests.test_SocketType_is_socketobjectc C s! t j t j t j � } | �� | j d | j � t | � � | j d t j t | � � | j d t j t | � � | j d t | � � | j d t | � � | j d
� | j d t | � � | j t | j � � t | � � Wd QRX| j d t | � � | j d t | � � d S) Nzfd=%iz family=%sztype=%szproto=0Zraddr� 127.0.0.1r Zladdrz[closed])r� r )
r r r �assertIn�fileno�repr� assertNotInr2 r� rf )r r r r r
� test_repr� s
)z GeneralModuleTests.test_reprNzneed _socket modulec C s� t j t j t j � } z? d | j � | j | j | j f } | j t | � | � Wd | j
� Xd | j | j | j f } | j t | � | � d S)Nz4<socket object, fd=%s, family=%s, type=%s, proto=%s>z4<socket object, fd=-1, family=%s, type=%s, proto=%s>) r� r r r r� �familyr� �proto� assertEqualr� r
)r r Zexpectedr r r
�test_csocket_repr� s " z$GeneralModuleTests.test_csocket_reprc C s� t j t j t j � } t | � } | j | j � | j � � | j � d } y | j � Wn t k
rr Yn X| j d � d S)NzSocket proxy still exists) r r r r r� r� r
�ReferenceError�fail)r r �pr r r
� test_weakref� s
z GeneralModuleTests.test_weakrefc C s� d } | j t d | d �� t � Wd QRX| j t d | d �� t j � Wd QRX| j t d | d �� t j � Wd QRXd S)Nz$Error raising socket exception (%s).�msgr z
socket.herrorzsocket.gaierror)� assertRaisesr r �herror�gaierror)r r� r r r
�testSocketError� s
z"GeneralModuleTests.testSocketErrorc K C s� t j t j t j � } | j | j � | j d � | j � } | j t � � } | j d | � Wd QRX| j
t | j � d � | j t � � } | j d | � Wd QRX| j
t | j � d � | j t � � } | j d d � Wd QRX| j
d t | j � � | j t � � } | j d d | � Wd QRX| j
t | j � d � | j t � � } | j d d | � Wd QRX| j
t | j � d � | j t � � } | j d d d � Wd QRX| j
d t | j � � | j t � � } | j d d | � Wd QRX| j
d
t | j � � | j t � � } | j d d d � Wd QRX| j
d
t | j � � | j t � � } | j d � Wd QRX| j
d t | j � � | j t � � } | j d d | d � Wd QRX| j
d
t | j � � d S)N� r u ☠z*a bytes-like object is required, not 'str'y @z.a bytes-like object is required, not 'complex's fooz not NoneType�barzan integer is requiredz (1 given)� z (4 given))r� r )r r r! r+ r
r2 rf r� rX �sendtor� r� � exceptionr� )r r Zsockname�cmr r r
�testSendtoErrors� sL
z#GeneralModuleTests.testSendtoErrorsc C s< t j t j t j t j t j t j t j t j d S)N) r r r r! r �SOCK_RDMr �
SOL_SOCKET� SO_REUSEADDR)r r r r
�testCrucialConstants� s z'GeneralModuleTests.testCrucialConstantsc C s� t j � } y t j | � } Wn t k
r@ | j d � Yn X| j | j d � d k d � y t j | � \ } } } Wn t k
r� | j d � Yn X| | g | } t j | � } | | k r� | j d | t
| � f � d S)Nzname lookup failurerC r zError resolving host to ip.z=Error testing host resolution mechanisms. (fqdn: %s, all: %s)) r � gethostname�
gethostbynamer r4 r� �find�
gethostbyaddrZgetfqdnr� r� )r ZhostnameZipZhname�aliasesZipaddrsZall_host_namesZfqhnr r r
�testHostnameRes� s
z"GeneralModuleTests.testHostnameResc C s� xI d d d d d g D]2 } | j t t j | � | j t t j | � q Wx3 t j d d g D] } | j t j | � | � q_ Wx1 t j g D]# } | j | t j | � d � q� Wd S) Nz0.1.1.~1z1+.1.1.1z::1qz::1::2z1:1:1:1:1:1:1:1:1z10.0.0.1z255.255.255.255� ) r� r r r� r� r re r� r� )r rl r� r r r
�test_host_resolution s
z'GeneralModuleTests.test_host_resolution� sethostnamez test needs socket.sethostname()r� z test needs socket.gethostname()c C s� t j � } y t j d � WnJ t k
ri } z* | j t j k rT | j d � n � WYd d } ~ Xn Xz= | j t j � d � t j d � | j t j � d � Wd t j | � Xd S)N�newztest should be run as roots barr� )r r� r� r �errnoZEPERMr4 r� )r Zoldhnr[ r r r
�test_sethostname s
z#GeneralModuleTests.test_sethostname� if_nameindexz$socket.if_nameindex() not available.c C s� t j � } x� | D]� \ } } | j | t � | j | t � | j | d � t j | � } | j | t � | j | | � t j | � } | j | t � | j | | � q Wd S)Nr ) r r� r� �intr� �
assertGreater�if_nametoindexr� �if_indextoname)r Z
interfaces�indexr� Z_index�_namer r r
�testInterfaceNameIndex3 s z)GeneralModuleTests.testInterfaceNameIndexc C s\ | j t t j d � | j t t j d � | j t t j d � | j t t j d � d S)Nr Z _DEADBEEF)r� r r r� r� rX )r r r r
� testInvalidInterfaceNameIndexC s z0GeneralModuleTests.testInvalidInterfaceNameIndex� getrefcountz test needs sys.getrefcount()c C s^ y# t j t � } t j t d � Wn4 t k
rY t j t � | k rU | j d � Yn Xd S)Nr z$socket.getnameinfo loses a reference)�sysr� r r � getnameinforX r� )r Zorigr r r
�testRefCountGetNameInfoM s
z*GeneralModuleTests.testRefCountGetNameInfoc C s- y t j d d � Wn t k
r( Yn Xd S)Nr0 r )r0 r r r )r r� r )r r r r
�testInterpreterCrashY s
z'GeneralModuleTests.testInterpreterCrashc C s� t j d t j d t j d t j d i } x� | j � D]� \ } } d | >d } x6 d D]. } | j | | @| | | | @� � | @� qX W| | � } | j | | @| � | j t | d � q7 Wd S)
N� � rB r � r� �gE#�2Tv�" � ��)r rB r� r� r� r� r� l ) r �htonl�ntohl�htons�ntohs�itemsr� r� �
OverflowError)r Zsizes�func�size�maskr Zswappedr r r
�testNtoHa s
, zGeneralModuleTests.testNtoHc C s� d d d d d d g } d d d d d d g } xB | D]: } t j | � t j | � t j | � t j | � q7 Wxf | D]^ } | j t t j | � | j t t j | � | j t t j | � | j t t j | � q| Wd S)
NrB r� � ���������r� r� r� )r r� r� r� r� r� r� )r Z good_valuesZ
bad_values�kr r r
�testNtoHErrorso s
z!GeneralModuleTests.testNtoHErrorsc C sZ | j } t j j d � s* t j d k r3 d } n d } x@ | D]2 } y t j | d
� } PWq@ t k
rq Yq@ Xq@ Wt � t j | � } | | | � y t j | d � } Wn t k
r� d } Yn X| | | � | t j | � | � | t j | d
� | � | d k r*| t j | d � | � | j t t j d � | j t t j d
� d S)N�freebsd�netbsd� gnukfreebsd�linux�darwin�daytime�qotd�domain�echoZtcpZudprB i )r� r� r )r r )r r r )r r r r� )
r� r� �platform�
startswithr Z
getservbynamer Z
getservbyportr� r� )r �eqZservicesZservicer Zport2Zudpportr r r
�
testGetServBy~ s2
z GeneralModuleTests.testGetServByc C s | j t j � d � t j � } | j | j � d � | j � t j d � | j t j � d � t j � } | j | j � d � | j � t j d � | j t j � d � t j � } | j | j � d � | j � | j t t j d � | j t t j d � d S)N�
rB Zspamr� ) r� r �getdefaulttimeout�
gettimeoutr
�setdefaulttimeoutr� �
ValueErrorrX )r r r r r
�testDefaultTimeout� s
z%GeneralModuleTests.testDefaultTimeout� inet_atonz test needs socket.inet_aton()c C s6 | j d t j d � � | j d t j d � � d S)Ns r� z0.0.0.0s �z255.255.255.255s s ����)r� r r )r r r r
� testIPv4_inet_aton_fourbytes� s z/GeneralModuleTests.testIPv4_inet_aton_fourbytes� inet_ptonz test needs socket.inet_pton()c s� d d l m } m � m � � � f d d � } � f d d � } � j d | d � � � j d | d � � � j d
| d � � � j d | d
� � � j d | d � � | | d � | | d � | | d � | | d � | | d � � j d | d � � � j d | d � � � j d
| d � � � j d | d � � | | d � | | d � | | d � | | d � | | d � d S)Nr )r r r c s
� � | � S)Nr )�a)r r r r
r� � s z5GeneralModuleTests.testIPv4toString.<locals>.<lambda>c s � j t t f | | � S)N)r� r r )r� r )r r r
r� � s s z0.0.0.0s � � z 255.0.255.0s ����z170.170.170.170s z1.2.3.4s ����z255.255.255.255z0.0.0.z 300.0.0.0za.0.0.0z 1.2.3.4.5z::1)r r r r r� )r r� �g�
assertInvalidr )r r r r
�testIPv4toString� s,
z#GeneralModuleTests.testIPv4toStringc sC y3 d d l m � m � m } | s2 � j d � Wn t k
rT � j d � Yn Xt j d k r� y � � d � WnA t k
r� } z! | j d k r� � j d � WYd d } ~ Xn X� � f d d
� � � � f d d
� } � j
d2 � d � � � j
d3 � d � � � j
d5 � d � � � j
d � d � � � j
d8 � d � � � j
d: � d � � | d � | d � | d � | d � | d � | d � | d! � | d" � | d# � � j
d< � d% � � � j
d? � d) � � � j
d* � d+ � � | d, � | d- � | d. � | d/ � | d0 � | d1 � d S)@Nr )r r� �has_ipv6zIPv6 not availablez+could not import needed symbols from socket�win32z::i&' zIPv6 might not be supportedc s
� � | � S)Nr )r )r� r r r
r� � s z5GeneralModuleTests.testIPv6toString.<locals>.<lambda>c s � j t t f � | � S)N)r� r r )r )r� r r r
r� � s s r� z0::0s � z1::s E�v� V�� �$��z$45ef:76cb:1a:56ef:afeb:bac:1924:aeaes �B
�r� s ' T zad42:abc::127:0:254:2s
� z12:a::z0x20::z:::z::0::z1::abc::z 1::abc::defz 1:2:3:4:5:6:z 1:2:3:4:5:6z1:2:3:4:5:6:7:8:z1:2:3:4:5:6:7:8:0s �*@z::254.42.23.64s Br1 s ���*@z42::a29b:254.42.23.64s B�� �����*@z"42:a8b9:0:2:ffff:a29b:254.42.23.64z255.254.253.252z 1::260.2.3.0z 1::0.be.e.0z1:2:3:4:5:6:7:1.2.3.4z ::1.2.3.4:0z0.100.200.0:3:4:5:6:7:8s s s s s s �B
� s �B
� ' T s s
s s �*@s s
B s B ���*@) r r r� r r4 � ImportErrorr� r r �winerrorr� )r r r[ r r )r� r� r r r
�testIPv6toString� s^
z#GeneralModuleTests.testIPv6toString� inet_ntopz test needs socket.inet_ntop()c sg d d l m } m � m � � � f d d � } � f d d � } � j d | d � � � j d | d � � � j d
| d � � � j d | d
� � | | d � | | d � | | d � � j d | t d � � � � j d | d � � � j d | d � � � j d
| d � � | | d � | | d � | | d � � j d | t d � � � d S)Nr )� inet_ntoar r c s
� � | � S)Nr )r )r r r r
r� * s z5GeneralModuleTests.testStringToIPv4.<locals>.<lambda>c s � j t t f | | � S)N)r� r r )r� r )r r r
r� + s z1.0.1.0s z
170.85.170.85s �U�Uz255.255.255.255s ����z1.2.3.4s s r� � r� s s s s s s )r r r r r� � bytearray)r r� r r r )r r r r
�testStringToIPv4& s$
z#GeneralModuleTests.testStringToIPv4c sc y3 d d l m � m � m } | s2 � j d � Wn t k
rT � j d � Yn Xt j d k r� y � � d � WnA t k
r� } z! | j d k r� � j d � WYd d } ~ Xn X� � f d
d � � � � f d d � } � j
d
� d � � � j
d � d � � � j
d � d � � � j
d � t d � � � | d � | d � | d � d S) Nr )r r� r zIPv6 not availablez+could not import needed symbols from socketr s r� i&' zIPv6 might not be supportedc s
� � | � S)Nr )r )r� r r r
r� Q s z5GeneralModuleTests.testStringToIPv6.<locals>.<lambda>c s � j t t f � | � S)N)r� r r )r )r� r r r
r� R s z::z::1� s z!aef:b01:506:1001:ffff:9997:55:170s
� ���� Ups � r� s s s s s s s s s ) r r r� r r4 r r� r r r r� r" )r r r[ r r )r� r� r r r
�testStringToIPv6@ s.
z#GeneralModuleTests.testStringToIPv6c C s� t j � } t j t j t j � } | j | j � | j d | f � | j � } y t j t j
� � } Wn t k
r� | j d � Yn X| j
| d d | f d | d � | j | d | � d S)Nz0.0.0.0zname lookup failurer z
%s invalidrB )r �find_unused_portr r r r+ r
r2 rf r� r� r r4 r� r� )r r r� r� Z
my_ip_addrr r r
� testSockNamed s
%z GeneralModuleTests.testSockNamec C sZ t j t j t j � } | j | j � | j t j t j � } | j | d k d � d S)Nr zinitial mode is reuse) r r r r+ r
�
getsockoptr� r� � assertFalse)r r� �reuser r r
�testGetSockOptv s z!GeneralModuleTests.testGetSockOptc C ss t j t j t j � } | j | j � | j t j t j d � | j t j t j � } | j | d k d � d S)NrB r zfailed to set reuse mode)
r r r r+ r
�
setsockoptr� r� r) r* )r r� r+ r r r
�testSetSockOpt~ s
z!GeneralModuleTests.testSetSockOptc C sI t j t j t j � } | j d � | j � | j t | j d � d S)NrB s spam)r r r �
settimeoutr
r� r �send)r r� r r r
�testSendAfterClose� s
z%GeneralModuleTests.testSendAfterClosec C s� t j t j t j � } | j | j t j � t t d � rf | j | j t j t j Bt j f � n | j | j t j � | j | j d � | j
� d S)N� SOCK_CLOEXECr ) r r r r� r� rW r� r� r2 r� r
)r r� r r r
�testNewAttributes� s
z$GeneralModuleTests.testNewAttributesc C s� t j � } | j | j � t j � } | d } | d } | j t | j t | f � | j t | j t | f � x� t j
� D]s } t j � } y | j t | f � WnF t k
r� } z& | j t j
k s� | d k r� � WYd d } ~ Xq� XPq� Wd S)Ni r! )r r+ r
r r' r� r� r2 re � itertools�countr r� Z
EADDRINUSE)r r� r Zbig_portZneg_portr r[ r r r
�test_getsockaddrarg� s
z&GeneralModuleTests.test_getsockaddrarg�ntzWindows specificc C s� | j t t j d � � | j t t d � � | j t t d � � | j t t d � � | j t t d � � t j � } | j | j � | j t | j d d � | j t j d � d S)
N�ioctlZ
SIO_RCVALLZ RCVALL_ONZ
RCVALL_OFF�SIO_KEEPALIVE_VALSrB �d r� )rB r: r: ) r� rW r r+ r
r� r r8 r9 )r r r r r
�test_sock_ioctl� s z"GeneralModuleTests.test_sock_ioctlc
C sz y t j d d � WnJ t j k
r` } z' | j t j k rK | j d � � WYd d } ~ Xn Xx0 t j t d � D] } | j t | � d � qt Wt j d d � t j d d � t j d d � t j
r� t j d d � t j t d � t j t d � t j t d � t j t d t j t j � } xo | D]g \ } } } } } | j | t j � | j t
| � d � | j | t j � | j t
| � d � q2Wt j t d d
t j � } x0 | D]( \ } } } } } | j | t j � q�Wt j t d d
d
t j � t j t d d
d
d
t j � t j d d
t j t j d
t j � t j t d � } t j d t d d � } | j | | � t j t d t j � } t j t d d
t j �} | j | | � t j t d d
t j � } t j t d d t j �} | j | | � t j t d d
d
t j � } t j t d d t j �} | j | | � t j t d d
d
d
t j � } t j t d d t j �} | j | | � t j d d
t j t j d
t j � } t j d d d d
d
t j d t j d d
d t j � } | j | | � | j t t j d d � t t d � rvy# t j d d d
d
d
t j � Wn t j k
ruYn Xd S)NZ localhost�P zbuggy libc versionr! z 127.0.0.1z::1ZhttpzAddressFamily.AF_INETzSocketKind.SOCK_STREAMr r� r r� r� r� �flagsu ��AI_NUMERICSERV)r � getaddrinfor� r� Z EAI_SERVICEr4 re r� �lenr � IPV6_ENABLEDr r r� ZSOL_TCPZ
AI_PASSIVE� AF_UNSPECr� �UnicodeEncodeErrorrW r> )
r r� �infoZinfosr� r� rD Zsocktyper �br r r
�testGetaddrinfo� sr
! z"GeneralModuleTests.testGetaddrinfoc C s | j t t j d d � d S)N�mail.python.orgr )rG r )r� r r r� )r r r r
�test_getnameinfo s z#GeneralModuleTests.test_getnameinfoZnetworkznetwork is not enabledc C sd t j d � � t j d � Wd QRXd } t j | � t j | � t j | d t j t j � d S)Nz
python.orgu! испытание.pythontest.netr )r Ztransient_internetr r� Zgethostbyname_exr? rB r )r r r r r
� test_idna
s
z GeneralModuleTests.test_idnac sI t t d � s t t d � r- � j d � � f d d � } � f d d � } t j � \ } } t j t j | � } z� | r� | j d � � j t � �& t j d � | j
d
t j � Wd QRX| r t j t j | � t j d � � j t j
| j
d
t j � Wd t j d � t j t j | � | j � | j � Xd S) N�alarmrr z9signal.alarm and socket.socketpair required for this testc s � j t t j d � d S)Nr )r� r �math�acosh)r( )r r r
�
ok_handler! s z@GeneralModuleTests.check_sendall_interrupted.<locals>.ok_handlerc s" � j t t j d � d d d S)Nr rB )r� r rK rL )r( )r r r
�raising_handler# s zEGeneralModuleTests.check_sendall_interrupted.<locals>.raising_handlerg �?rB � xr )rW �signalr r4 rr �SIGALRMr/ r� �ZeroDivisionErrorrJ �sendallr �
SOCK_MAX_SIZE�timeoutr
)r Z with_timeoutrM rN �cr � old_alarmr )r r
�check_sendall_interrupted s*
z,GeneralModuleTests.check_sendall_interruptedc C s | j d � d S)NF)rX )r r r r
�test_sendall_interrupted: s z+GeneralModuleTests.test_sendall_interruptedc C s | j d � d S)NT)rX )r r r r
�%test_sendall_interrupted_with_timeout= s z8GeneralModuleTests.test_sendall_interrupted_with_timeoutc C s� t j t j t j � } t | � } | j t � � } d } t j � Wd QRX| j | t | j
j d � � t j t j t j � } | j d � } t | � } d } t j � | j t � � d } t j � Wd QRXd S)Nr �rb)
r r r r� Z assertWarns�ResourceWarningr Z
gc_collectr� r� Zwarningr( �makefile)r r� �rr� r� r r r
�test_dealloc_warn@ s
z$GeneralModuleTests.test_dealloc_warnc
C sU t j t j t j � �6 } | j d � } | j � | j t | � d � Wd QRXd S)Nr[ z <_io.BufferedReader name=-1>)r r r r] r
r� r� )r r� �fpr r r
�test_name_closed_socketioQ s
z,GeneralModuleTests.test_name_closed_socketioc C s� t j � �� } | j d d d �} | j | j � � | j | j � � | j | j � � | j � | j t | j � | j t | j � | j t | j � Wd QRXd S)Nr[ � bufferingr )
r r] r� �readabler* �writable�seekabler
r� r )r r� r` r r r
� test_unusable_closed_socketioW s
z0GeneralModuleTests.test_unusable_closed_socketioc C sq xj d D]b } | j d | � �G t j � �3 } | j | � � } | j | j | � Wd QRXWd QRXWd QRXq Wd S)Nr^ r[ �rw�w�wb�mode)r^ r[ rg rh ri )�subTestr r] r� rj )r rj r� r` r r r
�test_makefile_modeb s
z%GeneralModuleTests.test_makefile_modec C sl xe d D]] } | j d | � �B t j � �. } | j t d � � | j | � Wd QRXWd QRXWd QRXq Wd S)N�rtr0 �+r rj z invalid mode)rm r0 rn r )rk r �assertRaisesRegexr r] )r rj r� r r r
�test_makefile_invalid_modei s
z-GeneralModuleTests.test_makefile_invalid_modec
C s� t j � } | �< x4 t t j d � D] } | j t t j | | � q' WWd QRXx} t t j d � D]h } t j t j t j | � � } | j | t j � t j t j t j
| � � } | j | t j
� qe Wd S)NrB ) r �range�pickleZHIGHEST_PROTOCOLr� rX �dumps�loadsr r� r )r r� Zprotocolr� r� r r r
� test_picklep s $ z GeneralModuleTests.test_picklec C s� xP d D]H } t j t j t j � �' } | j t d f � | j | � Wd QRXq Wt j t j t j � �$ } | j t d f � | j � Wd QRXd S)Nr rB r� )r r� )r r r r2 re r )r Zbacklog�srvr r r
�test_listen_backlog{ s
z&GeneralModuleTests.test_listen_backlogc C sb d d l } t j t j t j � } | j t d f � | j t | j | j d � | j
� d S)Nr rB ) � _testcapir r r r2 re r� r� r �INT_MAXr
)r rx rv r r r
� test_listen_backlog_overflow� s
z/GeneralModuleTests.test_listen_backlog_overflowz IPv6 required for this test.c
C sm | j t t j t j d d f d � t j t j t j � �) } | j t | j t j d d f � Wd QRXd S)Nr l �� r i����) r� r� r r� r r� r� r r2 )r r r r r
�
test_flowinfo� s z GeneralModuleTests.test_flowinfoc
C sX t j t j t j � �9 } | j t | j � d � | j t | j � d � Wd QRXd S)NzAddressFamily.AF_INETzSocketKind.SOCK_STREAM)r r r r� r� r� r� )r r r r r
�test_str_for_enums� s z%GeneralModuleTests.test_str_for_enumszWill not work on Windowsc
C sw t j � \ } } | j t j | � t j d d d d d | � �- } | j | j d � | j | j d � Wd QRXd S)Nr� i�� r� i4 r� ) r~ �mkstempr+ r� r� r r� r� r� )r �fdr� r r r r
� test_uknown_socket_family_repr� s
!z1GeneralModuleTests.test_uknown_socket_family_repr�sendfileztest needs os.sendfile()c C s� Gd d � d � } t j � �� } t j t j t j � } t j | � | j t j � � | j | | � � Wd QRX| j t � � | j | d � � Wd QRX| j t
� � | j | d � � Wd QRXWd QRXd S)Nc @ s( e Z d Z d d � Z d d � Z d S)z<GeneralModuleTests.test__sendfile_use_sendfile.<locals>.Filec S s
| | _ d S)N)r~ )r r~ r r r
r$ � s zEGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.__init__c S s | j S)N)r~ )r r r r
r� � s zCGeneralModuleTests.test__sendfile_use_sendfile.<locals>.File.filenoN)r r r r$ r� r r r r
�File� s r� r� i� lC ) r r� �open�curdir�O_RDONLYr
r� �_GiveupOnSendfile�_sendfile_use_sendfiler� rX )r r� r� r~ r r r
�test__sendfile_use_sendfile� s
z.GeneralModuleTests.test__sendfile_use_sendfile)@r r r r� r� rp �
skipUnlessr� r� r� r� r� r� r� r� rW r r� r� r� r� r� r� r� r � cpython_onlyr� r
r r r r r# r& r( r, r. r1 r3 r6 r� r� r; rF rH Zis_resource_enabledrI rX rY rZ r_ ra rf rl rp ru rw rz rA r{ r| �skipIfr r� r r r r
r� � st $
, * ' =# ' K
! 'r� z!SocketCan required for this test.c @ s� e Z d Z d d � Z e j e e d � d � d d � � Z d d � Z e j e e d � d � d d
� � Z
d d � Z d
d � Z e j e e d � d � d d � � Z
e j e e d � d � d d � � Z d S)� BasicCANTestc C s t j t j t j d S)N)r �AF_CANr r )r r r r
r� � s z!BasicCANTest.testCrucialConstants�CAN_BCMz&socket.CAN_BCM required for this test.c C s_ t j t j t j t j t j t j t j t j t j t j
t j t j t j
d S)N)r r� ZCAN_BCM_TX_SETUPZCAN_BCM_TX_DELETEZCAN_BCM_TX_READ�CAN_BCM_TX_SENDZCAN_BCM_RX_SETUPZCAN_BCM_RX_DELETEZCAN_BCM_RX_READZCAN_BCM_TX_STATUSZCAN_BCM_TX_EXPIREDZCAN_BCM_RX_STATUSZCAN_BCM_RX_TIMEOUTZCAN_BCM_RX_CHANGED)r r r r
�testBCMConstants� s z BasicCANTest.testBCMConstantsc C s, t j t j t j t j � � } Wd QRXd S)N)r r r r )r r r r r
�testCreateSocket� s !z BasicCANTest.testCreateSocketc C s, t j t j t j t j � � } Wd QRXd S)N)r r r! r� )r r r r r
�testCreateBCMSocket� s !z BasicCANTest.testCreateBCMSocketc C s9 t j t j t j t j � � } | j d � Wd QRXd S)Nr� )r� )r r r r r2 )r r r r r
� testBindAny� s !zBasicCANTest.testBindAnyc
C sL t j t j t j t j � �' } | j t d | j d d f � Wd QRXd S)Nzinterface name too longr0 i )r r r r ro r r2 )r r r r r
�testTooLongInterfaceName� s ! z%BasicCANTest.testTooLongInterfaceName�CAN_RAW_LOOPBACKz/socket.CAN_RAW_LOOPBACK required for this test.c
C sx t j t j t j t j � �S } xI d D]A } | j t j t j | � | j | | j t j t j � � q( WWd QRXd S)Nr rB )r rB ) r r r r r- � SOL_CAN_RAWr� r� r) )r r Zloopbackr r r
� testLoopback� s !
zBasicCANTest.testLoopback�CAN_RAW_FILTERz-socket.CAN_RAW_FILTER required for this test.c
C s� d \ } } t j d | | � } t j t j t j t j � �d } | j t j t j | � | j | | j
t j t j d � � | j t j t j t | � � Wd QRXd S)N� � z=IIr1 )r� r� ) r7 �packr r r r r- r� r� r� r) r" )r �can_idZcan_maskZ
can_filterr r r r
�
testFilter� s ! zBasicCANTest.testFilterN)r r r r� rp r� rW r r� r� r� r� r� r� r� r r r r
r� � s r� z!Threading required for this test.c @ s� e Z d Z d d d � Z e d d � � Z e d d � � Z d d � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z e
j e e d � d � d d � � Z e
j e e d � d � d d � � Z d S)�CANTestr_ c C s t j | d | �d S)Nr` )rc r$ )r r` r r r
r$ s zCANTest.__init__c C s7 t | � } | j d d � } t j | j | | | � S)zBuild a CAN frame.r1 s )r@ �ljustr7 r� r6 )�clsr� �data�can_dlcr r r
�build_can_frame s zCANTest.build_can_framec C s5 t j | j | � \ } } } | | | d | � f S)zDissect a CAN frame.N)r7 Zunpackr6 )r� �framer� r� r� r r r
�dissect_can_frame s zCANTest.dissect_can_framec C s` | j j | j � \ } } | j | j | � | j | d | j � | j | d t j � d S)Nr rB )r �recvfromr5 r� �cfr3 r r� )r r� rl r r r
�
testSendFrame s zCANTest.testSendFramec C s, | j d d � | _ | j j | j � d S)Nr s )r� r� ra r0 )r r r r
�_testSendFrame s zCANTest._testSendFramec C s2 | j j | j � \ } } | j | j | � d S)N)r r� r5 r� r� )r r� rl r r r
�testSendMaxFrame" s zCANTest.testSendMaxFramec C s, | j d d � | _ | j j | j � d S)Nr s r1 s )r� r� ra r0 )r r r r
�_testSendMaxFrame& s zCANTest._testSendMaxFramec C s` | j j | j � \ } } | j | j | � | j j | j � \ } } | j | j | � d S)N)r r� r5 r� �cf1�cf2)r r� rl r r r
�testSendMultiFrames* s zCANTest.testSendMultiFramesc C sT | j d d � | _ | j j | j � | j d d � | _ | j j | j � d S)N� s D3"� s �"3)r� r� ra r0 r� )r r r r
�_testSendMultiFrames1 s z CANTest._testSendMultiFramesr� z&socket.CAN_BCM required for this test.c C sp | j j | j � \ } } | j | j | � | j | � \ } } } | j | j | � | j | j | � d S)N)ra r� r5 r� r� r� r� r� )r r� rl r� r� r� r r r
�_testBCM8 s
zCANTest._testBCMc C s t j t j t j t j � } | j | j � | j | j f � d | _ t d d d g � | _
| j | j | j
� | _ t j
} d } d } d } } } } d } d }
t j | j | | | | | | | | |
�
} | | j } | j | � }
| j |
t | � � d S)Ni# � � �� r i" rB )r r r! r� r+ r
rm r3 r� �bytesr� r� r� r� r7 r� r8 r0 r� r@ )r ZbcmZopcoder= r5 Z
ival1_secondsZ
ival1_usecZ
ival2_secondsZ
ival2_usecZ
bcm_can_idZnframes�headerZheader_plus_frameZ
bytes_sentr r r
�testBCMA s2
zCANTest.testBCMN)r r r r$ � classmethodr� r� r� r� r� r� r� r� rp r� rW r r� r� r r r r
r� s r� z#RDS sockets required for this test.c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� BasicRDSTestc C s t j t j d S)N)r ZAF_RDSr )r r r r
r� d s z!BasicRDSTest.testCrucialConstantsc C s) t j t j t j d � � } Wd QRXd S)Nr )r r r )r r r r r
r� h s z BasicRDSTest.testCreateSocketc C sa d } t j t j t j d � �9 } | j t j t j | � | j t j t j | � Wd QRXd S)Ni @ r )r r r r- r� � SO_RCVBUFZ SO_SNDBUF)r r5 r r r r
�testSocketBufferSizel s z!BasicRDSTest.testSocketBufferSizeN)r r r r� r� r� r r r r
r� a s r� c s� e Z d Z d d d � Z � f d d � Z d d � Z d d � Z d
d � Z d d
� Z e e
j
d � d d � � Z e e
j
d � d d � � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z � S) �RDSTestr_ c C s t j | d | �d S)Nr` )rd r$ )r r` r r r
r$ w s zRDSTest.__init__c s t � j � t j � | _ d S)N)r# r r% rE �evt)r )r* r r
r z s
z
RDSTest.setUpc C sE | j j | j � \ } } | j | j | � | j | j | � d S)N)r r� r5 r� r� rg )r r� rl r r r
�testSendAndRecv~ s zRDSTest.testSendAndRecvc C s/ d | _ | j j | j d t | j f � d S)Ns spamr )r� ra r� re r )r r r r
�_testSendAndRecv� s zRDSTest._testSendAndRecvc C sf | j j | j t j � \ } } | j | j | � | j j | j � \ } } | j | j | � d S)N)r r� r5 r �MSG_PEEKr� r� )r r� rl r r r
�testPeek� s !zRDSTest.testPeekc C s/ d | _ | j j | j d t | j f � d S)Ns spamr )r� ra r� re r )r r r r
� _testPeek� s zRDSTest._testPeek�recvmsgc C s8 | j j | j � \ } } } } | j | j | � d S)N)r r� r5 r� r� )r r� �ancdataZ msg_flagsrl r r r
�testSendAndRecvMsg� s !zRDSTest.testSendAndRecvMsg�sendmsgc C s9 d d | _ | j j | j g f d t | j f � d S)Ns hello r r )r� ra r� re r )r r r r
�_testSendAndRecvMsg� s
zRDSTest._testSendAndRecvMsgc C s` | j j | j � \ } } | j | j | � | j j | j � \ } } | j | j | � d S)N)r r� r5 r� �data1�data2)r r� rl r r r
�testSendAndRecvMulti� s z RDSTest.testSendAndRecvMultic C sZ d | _ | j j | j d t | j f � d | _ | j j | j d t | j f � d S)Ns baconr s egg)r� ra r� re r r� )r r r r
�_testSendAndRecvMulti� s " z RDSTest._testSendAndRecvMultic C sl t j | j g g g d � \ } } } | j | j | � | j j | j � \ } } | j | j | � d S)Ng @)�selectr r� r� r5 r� r� )r r^ rh r0 r� rl r r r
�
testSelect� s 'zRDSTest.testSelectc C s/ d | _ | j j | j d t | j f � d S)Ns selectr )r� ra r� re r )r r r r
� _testSelect� s zRDSTest._testSelectc C s | j j � d S)N)r� rQ )r r r r
�testCongestion� s zRDSTest.testCongestionc C s� d | _ | j j d � y | j j t j t j d � Wn t k
rM Yn X| j t � �E } z- x& | j j | j d t
| j f � qf WWd | j j
� XWd QRX| j | j j t j � t j | j g g g d � \ } } } | j | j | � d S)Ns fillFi @ r g @)r� ra � setblockingr- r r� r� r r� r� re r r� r@ r� r� r� ZENOBUFSr� r r� )r r� r^ rh r0 r r r
�_testCongestion� s
*'zRDSTest._testCongestion)r r r r$ r r� r� r� r� r� r r� r� r� r� r� r� r� r� r r )r* r
r� s s r� c @ s e Z d Z d d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d! � Z d" d# � Z e j e � Z e j d$ d% � � Z d& d' � Z d( d) � Z d* S)+� BasicTCPTestr_ c C s t j | d | �d S)Nr` )rh r$ )r r` r r r
r$ � s zBasicTCPTest.__init__c C s& | j j d � } | j | t � d S)Ni )rj �recvr� �MSG)r r� r r r
�testRecv� s zBasicTCPTest.testRecvc C s | j j t � d S)N)rn r0 r� )r r r r
� _testRecv� s zBasicTCPTest._testRecvc C sL | j j t t � d � } | j j d � } | | } | j | t � d S)Nr� i )rj r� r@ r� r� )r �seg1�seg2r� r r r
�testOverFlowRecv� s
z BasicTCPTest.testOverFlowRecvc C s | j j t � d S)N)rn r0 r� )r r r r
�_testOverFlowRecv� s z BasicTCPTest._testOverFlowRecvc C s, | j j d � \ } } | j | t � d S)Ni )rj r� r� r� )r r� rl r r r
� testRecvFrom� s zBasicTCPTest.testRecvFromc C s | j j t � d S)N)rn r0 r� )r r r r
�
_testRecvFrom� s zBasicTCPTest._testRecvFromc C sX | j j t t � d � \ } } | j j d � \ } } | | } | j | t � d S)Nr� i )rj r� r@ r� r� )r r� rl r� r� r r r
�testOverFlowRecvFrom� s "
z!BasicTCPTest.testOverFlowRecvFromc C s | j j t � d S)N)rn r0 r� )r r r r
�_testOverFlowRecvFrom� s z"BasicTCPTest._testOverFlowRecvFromc C sH d } x' | j j d � } | s" P| | 7} q W| j | d d � d S)N� i � fi )rj r� r� )r r� �readr r r
� testSendAll� s zBasicTCPTest.testSendAllc C s d d } | j j | � d S)Nr� i )rn rS )r Z big_chunkr r r
� _testSendAll s
zBasicTCPTest._testSendAllc C sp | j j � } t j | t j t j � } | j | j � | j | t j � | j d � } | j
| t � d S)Ni ) rj r� r Zfromfdr r r+ r
r� r� r� r� )r r~ r� r� r r r
�
testFromFd s zBasicTCPTest.testFromFdc C s | j j t � d S)N)rn r0 r� )r r r r
� _testFromFd s zBasicTCPTest._testFromFdc C sB | j j � } | j | j � | j d � } | j | t � d S)Ni )rj �dupr+ r
r� r� r� )r r� r� r r r
�testDup s zBasicTCPTest.testDupc C s | j j t � d S)N)rn r0 r� )r r r r
�_testDup s zBasicTCPTest._testDupc C s3 | j j d � } | j | t � | j j � d S)Ni )rj r� r� r� rG rQ )r r� r r r
� testShutdown s zBasicTCPTest.testShutdownc C s$ | j j t � | j j d � d S)Nr� )rn r0 r� �shutdown)r r r r
�
_testShutdown& s zBasicTCPTest._testShutdownc C st d d l } | j j t � | j t | j j | j d � | j t | j j d | j d � | j j d � d S)Nr rB r� ) rx rn r0 r� r� r� r� ry �UINT_MAX)r rx r r r
�_testShutdown_overflow, s z#BasicTCPTest._testShutdown_overflowc C s� | j j � } | j j � } | j | | � | j | j j � | j t | j j d � | j j � t
j
t
j t
j d | �} | j
| j � | j d � } | j | t � d S)Ni r� )rj r� �detachr� r� �_closedr� r r� r
r r r r+ r� )r r� r� r� r� r r r
�
testDetach7 s
zBasicTCPTest.testDetachc C s | j j t � d S)N)rn r0 r� )r r r r
� _testDetachG s zBasicTCPTest._testDetachN)r r r r$ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r� ZtestShutdown_overflowr� r� r� r r r r
r� � s*
r� c @ sg e Z d Z d d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d S)� BasicUDPTestr_ c C s t j | d | �d S)Nr` )rb r$ )r r` r r r
r$ M s zBasicUDPTest.__init__c C s, | j j t t � � } | j | t � d S)N)r r� r@ r� r� )r r� r r r
�testSendtoAndRecvP s z BasicUDPTest.testSendtoAndRecvc C s# | j j t d t | j f � d S)Nr )ra r� r� re r )r r r r
�_testSendtoAndRecvU s z BasicUDPTest._testSendtoAndRecvc C s2 | j j t t � � \ } } | j | t � d S)N)r r� r@ r� r� )r r� rl r r r
r� X s zBasicUDPTest.testRecvFromc C s# | j j t d t | j f � d S)Nr )ra r� r� re r )r r r r
r� ] s zBasicUDPTest._testRecvFromc C s | j t | j j d � d S)NrB r� )r� r r r� )r r r r
�testRecvFromNegative` s z!BasicUDPTest.testRecvFromNegativec C s# | j j t d t | j f � d S)Nr )ra r� r� re r )r r r r
�_testRecvFromNegatived s z"BasicUDPTest._testRecvFromNegativeN)
r r r r$ r� r� r� r� r� r� r r r r
r� J s r� c s� e Z d Z d Z � f d d � Z d d � Z f Z d d � Z d d � Z d
d � Z d d
� Z
d Z x$ d D] Z e e
e e d � OZ qs Wd Z d Z d Z d d d d d d � Z � S)�SendrecvmsgBaseg @c s t j � | _ t � j � d S)N)r% rE �
misc_eventr# r )r )r* r r
r � s zSendrecvmsgBase.setUpc C s | j j | � S)N)�cli_sockr0 )r r� r r r
� sendToServer� s z SendrecvmsgBase.sendToServerc G s' | j j | | j t | � d � � S)N)r� r� �sendmsg_to_server_defaultsr@ )r r( r r r
�sendmsgToServer� s z SendrecvmsgBase.sendmsgToServerc G s# | j | | � } | j | � | S)N)r� �registerRecvmsgResult)r r� r5 r( �resultr r r
� doRecvmsg� s
zSendrecvmsgBase.doRecvmsgc C s d S)Nr )r r� r r r
r� � s z%SendrecvmsgBase.registerRecvmsgResultc C s | j | | � d S)N)r� )r �addr1�addr2r r r
�checkRecvmsgAddress� s z#SendrecvmsgBase.checkRecvmsgAddressr �
MSG_CTRUNC�MSG_OOBNc
C s� | j } | j } | r5 | | j O} | | j O} n& | d k r[ | | j O} | | j O} | | M} | | M} | | O} | | O} | | @| @} | r� t d j | � � � | | B| @} | j | | @| | @� d S)Nz6contradictory set, unset requirements for flags {0:#x})�msg_flags_common_set�msg_flags_common_unset�msg_flags_eor_indicator�msg_flags_non_eor_indicator� Exceptionr� r� )
r r= �eor�checkset�
checkunset�ignoreZ
defaultsetZ defaultunsetZinbothr� r r r
�
checkFlags� s$
zSendrecvmsgBase.checkFlags)r r )r r r � fail_timeoutr r� r� r� r� r� r r r� rM r r r r r r r )r* r
r� � s
r� c @ s e Z d Z d d � Z d S)�RecvmsgIntoMixinc G s t | � } | j | g | � } | j | � | j | d d � | j | d | � t | d | d � � f | d d � S)Nr rB )r" � recvmsg_intor� �assertGreaterEqual�assertLessEqualr� )r r� r5 r( �bufr� r r r
r� � s
zRecvmsgIntoMixin.doRecvmsgN)r r r r� r r r r
r � s r c s( e Z d Z e � f d d � � Z � S)�SendrecvmsgDgramFlagsBasec s t � j t j BS)N)r# r r � MSG_TRUNC)r )r* r r
r s z5SendrecvmsgDgramFlagsBase.msg_flags_non_eor_indicator)r r r �propertyr r r )r* r
r s r c s( e Z d Z e � f d d � � Z � S)�SendrecvmsgSCTPFlagsBasec s t � j t j BS)N)r# r r ZMSG_EOR)r )r* r r
r s z0SendrecvmsgSCTPFlagsBase.msg_flags_eor_indicator)r r r r r r r )r* r
r
s r c @ sR e Z d Z e d d � � Z e d d � � Z e d d � � Z d d � Z d S)
� SendrecvmsgConnectionlessBasec C s | j S)N)r )r r r r
� serv_sock s z'SendrecvmsgConnectionlessBase.serv_sockc C s | j S)N)ra )r r r r
r� s z&SendrecvmsgConnectionlessBase.cli_sockc C s g g d | j f S)Nr )rw )r r r r
r� " s z8SendrecvmsgConnectionlessBase.sendmsg_to_server_defaultsc C s | j j | | j � S)N)r� r� rw )r r� r r r
r� & s z*SendrecvmsgConnectionlessBase.sendToServerN)r r r r r r� r� r� r r r r
r s r c @ s@ e Z d Z e d d � � Z e d d � � Z d d � Z d S)�SendrecvmsgConnectedBasec C s | j S)N)rj )r r r r
r 0 s z"SendrecvmsgConnectedBase.serv_sockc C s | j S)N)rn )r r r r
r� 4 s z!SendrecvmsgConnectedBase.cli_sockc C s d S)Nr )r r r r r r
r 8 s z,SendrecvmsgConnectedBase.checkRecvmsgAddressN)r r r r r r� r r r r r
r * s r c s" e Z d Z � f d d � Z � S)� SendrecvmsgServerTimeoutBasec s$ t � j � | j j | j � d S)N)r# r r r/ r )r )r* r r
r A s
z"SendrecvmsgServerTimeoutBase.setUp)r r r r r r )r* r
r > s r c @ s e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z e e d � d d � � Z e j d d! � � Z d" d# � Z d$ d% � Z d& d' � Z d( d) � Z d* S)+� SendmsgTestsc C s& | j | j j t t � � t � d S)N)r� r r� r@ r� )r r r r
� testSendmsgJ s zSendmsgTests.testSendmsgc C s& | j | j t g � t t � � d S)N)r� r� r� r@ )r r r r
� _testSendmsgN s zSendmsgTests._testSendmsgc C s& | j | j j t t � � t � d S)N)r� r r� r@ r� )r r r r
�testSendmsgDataGeneratorQ s z%SendmsgTests.testSendmsgDataGeneratorc C s3 | j | j d d � t g D� � t t � � d S)Nc s s | ] } | Vq d S)Nr )r� r� r r r
r� V s z9SendmsgTests._testSendmsgDataGenerator.<locals>.<genexpr>)r� r� r� r@ )r r r r
�_testSendmsgDataGeneratorU s "z&SendmsgTests._testSendmsgDataGeneratorc C s& | j | j j t t � � t � d S)N)r� r r� r@ r� )r r r r
� testSendmsgAncillaryGeneratorY s z*SendmsgTests.testSendmsgAncillaryGeneratorc C s6 | j | j t g d d � g D� � t t � � d S)Nc s s | ] } | Vq d S)Nr )r� r� r r r
r� ^ s z>SendmsgTests._testSendmsgAncillaryGenerator.<locals>.<genexpr>)r� r� r� r@ )r r r r
� _testSendmsgAncillaryGenerator] s %z+SendmsgTests._testSendmsgAncillaryGeneratorc C s& | j | j j t t � � t � d S)N)r� r r� r@ r� )r r r r
�testSendmsgArraya s z SendmsgTests.testSendmsgArrayc C s2 | j | j t j d t � g � t t � � d S)N�B)r� r� �arrayr� r@ )r r r r
�_testSendmsgArraye s !z SendmsgTests._testSendmsgArrayc C s& | j | j j t t � � t � d S)N)r� r r� r@ r� )r r r r
�testSendmsgGatheri s z SendmsgTests.testSendmsgGatherc C s= | j | j t d d � t d d � g � t t � � d S)Nr� )r� r� r� r@ )r r r r
�_testSendmsgGatherm s z SendmsgTests._testSendmsgGatherc C s | j | j j d � d � d S)Ni� s done)r� r r� )r r r r
�testSendmsgBadArgsp s z SendmsgTests.testSendmsgBadArgsc C s� | j t | j j � | j t | j d � | j t | j t � � | j t | j t � g � | j t | j t t � g � | j t | j t g t � � | j t | j t g g t � � | j t | j t g g d t � � | j d � d S)Ns not in an iterabler s done)r� rX r� r� r� �objectr� r� )r r r r
�_testSendmsgBadArgst s
z SendmsgTests._testSendmsgBadArgsc C s | j | j j d � d � d S)Ni� s done)r� r r� )r r r r
�testSendmsgBadCmsg� s z SendmsgTests.testSendmsgBadCmsgc C s� | j t | j t g t � g � | j t | j t g t � d d f g � | j t | j t g d t � d f g � | j t | j t g d d t � f g � | j t | j t g d g � | j t | j t g d g � | j d � d S)Nr � data�* s done)r r )r r r. r/ )r� rX r� r� r+ r� )r r r r
�_testSendmsgBadCmsg� s z SendmsgTests._testSendmsgBadCmsg�
CMSG_SPACEc C s | j | j j d � d � d S)Ni� s done)r� r r� )r r r r
�testSendmsgBadMultiCmsg� s z$SendmsgTests.testSendmsgBadMultiCmsgc C s[ | j t | j t g d d d g � | j t | j t g d t � g � | j d � d S)Nr r� s done)r r r� )r� rX r� r� r+ r� )r r r r
�_testSendmsgBadMultiCmsg� s
z%SendmsgTests._testSendmsgBadMultiCmsgc C s | j | j j d � d � d S)Ni� s done)r� r r� )r r r r
�testSendmsgExcessCmsgReject� s z(SendmsgTests.testSendmsgExcessCmsgRejectc
C se t t d � sT | j t � � } | j t g d d g � Wd QRX| j | j j � | j d � d S)Nr1 r r� s done)r r r� )r r r� )
rW r r� r r� r� � assertIsNoner� r� r� )r r� r r r
� _testSendmsgExcessCmsgReject� s
z)SendmsgTests._testSendmsgExcessCmsgRejectc C s d S)Nr )r r r r
�testSendmsgAfterClose� s z"SendmsgTests.testSendmsgAfterClosec C s* | j j � | j t | j t g � d S)N)r� r
r� r r� r� )r r r r
�_testSendmsgAfterClose� s
z#SendmsgTests._testSendmsgAfterCloseN)r r r r r r r! r" r# r$ r' r( r) r* r, r- r0 r� r r2 r� r3 r4 r6 r7 r8 r r r r
r F s( r c @ sy e Z d Z d d � Z d d � Z d d � Z d d � Z e e j d k d
� d d � � Z
e
j d
d � � Z d S)�SendmsgStreamTestsc C s& | j | j j t t � � t � d S)N)r� r r� r@ r� )r r r r
�testSendmsgExplicitNoneAddr� s z.SendmsgStreamTests.testSendmsgExplicitNoneAddrc C s/ | j | j t g g d d � t t � � d S)Nr )r� r� r� r@ )r r r r
� _testSendmsgExplicitNoneAddr� s z/SendmsgStreamTests._testSendmsgExplicitNoneAddrc C sC | j | j j d � d d � | j | j j d | j � � d S)Ni � arU )r� r r� r� r� rQ r )r r r r
�testSendmsgTimeout� s z%SendmsgStreamTests.testSendmsgTimeoutc C s^ zI | j j d � | j t j � � x | j d d g � q) WWd QRXWd | j j � Xd S)Ng� ��Q��?r< i )r� r/ r� r rU r� r� r@ )r r r r
�_testSendmsgTimeout� s #z&SendmsgStreamTests._testSendmsgTimeoutr z<MSG_DONTWAIT not known to work on this platform when sendingc C sC | j | j j d � d d � | j | j j d | j � � d S)Ni r< rU )r� r r� r� r� rQ r )r r r r
�testSendmsgDontWait� s z&SendmsgStreamTests.testSendmsgDontWaitc C sx zc | j t � �+ } x! | j d d g g t j � q WWd QRX| j | j j t j t j f � Wd | j
j � Xd S)Nr< i ) r� r r� r Z MSG_DONTWAITr� r� r� ZEAGAINZ EWOULDBLOCKr� r@ )r r� r r r
�_testSendmsgDontWait� s (z'SendmsgStreamTests._testSendmsgDontWaitN> r )
r r r r: r; r= r> r� r� r r? r� r@ r r r r
r9 � s r9 c @ s( e Z d Z d d � Z d d � Z d S)�SendmsgConnectionlessTestsc C s d S)Nr )r r r r
�testSendmsgNoDestAddr� s z0SendmsgConnectionlessTests.testSendmsgNoDestAddrc C sE | j t | j j t g � | j t | j j t g g d d � d S)Nr )r� r r� r� r� )r r r r
�_testSendmsgNoDestAddr� s
z1SendmsgConnectionlessTests._testSendmsgNoDestAddrN)r r r rB rC r r r r
rA � s rA c @ sQ e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z e j
d
� d d � � Z e j
d
� d d � � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d! � Z e e d" � d# d$ � � Z e j d% d& � � Z e e j d' � d( d) � � Z e j d* d+ � � Z d, S)-�RecvmsgGenericTestsc C sq | j | j t t � � \ } } } } | j | t � | j | | j � | j | g � | j | d d �d S)Nr
T)r� r r@ r� r� r rg r )r r� r� r= rl r r r
� testRecvmsg s
'z RecvmsgGenericTests.testRecvmsgc C s | j t � d S)N)r� r� )r r r r
� _testRecvmsg s z RecvmsgGenericTests._testRecvmsgc C sw | j | j t t � d d � \ } } } } | j | t � | j | | j � | j | g � | j | d d �d S)Nr r
T)r� r r@ r� r� r rg r )r r� r� r= rl r r r
�testRecvmsgExplicitDefaults s !z/RecvmsgGenericTests.testRecvmsgExplicitDefaultsc C s | j t � d S)N)r� r� )r r r r
� _testRecvmsgExplicitDefaults s z0RecvmsgGenericTests._testRecvmsgExplicitDefaultsc C su | j | j t t � d � \ } } } } | j | t � | j | | j � | j | g � | j | d d �d S)Nr/ r
T)r� r r@ r� r� r rg r )r r� r� r= rl r r r
�testRecvmsgShorter s z&RecvmsgGenericTests.testRecvmsgShorterc C s | j t � d S)N)r� r� )r r r r
�_testRecvmsgShorter$ s z'RecvmsgGenericTests._testRecvmsgShorterr1 c C s | j | j t t � d � \ } } } } | j | t d d � � | j | | j � | j | g � | j | d d �d S)Nr� r
Fr� )r� r r@ r� r� r rg r )r r� r� r= rl r r r
�testRecvmsgTrunc) s z$RecvmsgGenericTests.testRecvmsgTruncc C s | j t � d S)N)r� r� )r r r r
�_testRecvmsgTrunc3 s z%RecvmsgGenericTests._testRecvmsgTruncc C st | j | j t t � d � \ } } } } | j | t � | j | | j � | j | g � | j | d d �d S)NrB r
T)r� r r@ r� r� r rg r )r r� r� r= rl r r r
� testRecvmsgShortAncillaryBuf7 s z0RecvmsgGenericTests.testRecvmsgShortAncillaryBufc C s | j t � d S)N)r� r� )r r r r
� _testRecvmsgShortAncillaryBuf@ s z1RecvmsgGenericTests._testRecvmsgShortAncillaryBufc C st | j | j t t � d � \ } } } } | j | t � | j | | j � | j | g � | j | d d �d S)Ni ( r
T)r� r r@ r� r� r rg r )r r� r� r= rl r r r
�testRecvmsgLongAncillaryBufC s z/RecvmsgGenericTests.testRecvmsgLongAncillaryBufc C s | j t � d S)N)r� r� )r r r r
� _testRecvmsgLongAncillaryBufL s z0RecvmsgGenericTests._testRecvmsgLongAncillaryBufc C s- | j j � | j t | j | j d � d S)Ni )r r
r� r r� )r r r r
�testRecvmsgAfterCloseO s
z)RecvmsgGenericTests.testRecvmsgAfterClosec C s d S)Nr )r r r r
�_testRecvmsgAfterCloseT s z*RecvmsgGenericTests._testRecvmsgAfterClosec C sN z9 | j j d � | j t j | j | j t t � � Wd | j j � Xd S)Ng� ��Q��?)
r r/ r� r rU r� r@ r� r� r@ )r r r r
�testRecvmsgTimeoutW s
z&RecvmsgGenericTests.testRecvmsgTimeoutc C s# | j | j j d | j � � d S)NrU )r� r� rQ r )r r r r
�_testRecvmsgTimeout` s z'RecvmsgGenericTests._testRecvmsgTimeoutr� c C s} | j | j t t � d d t j � \ } } } } | j | t d d � � | j | | j � | j | g � | j | d d d t
t d d � �| j | j t t � d t j � \ } } } } | j | t � | j | | j � | j | g � | j | d d �| j | j t t � � \ } } } } | j | t � | j | | j � | j | g � | j | d d �d S) Nr� r r
Fr
r Tr� ) r� r r@ r� r r� r� r rg r rM )r r� r� r= rl r r r
�testRecvmsgPeekc s( 'z#RecvmsgGenericTests.testRecvmsgPeekc C s | j t � d S)N)r� r� )r r r r
�_testRecvmsgPeek� s z$RecvmsgGenericTests._testRecvmsgPeekr� c C s� | j j | j � | j | j t t � � \ } } } } | j | t � | j | | j � | j | g � | j | d d �d S)Nr
T)
r r/ r r� r@ r� r� r rg r )r r� r� r= rl r r r
�testRecvmsgFromSendmsg� s 'z*RecvmsgGenericTests.testRecvmsgFromSendmsgc C s= | j | j t d d � t d d � g � t t � � d S)Nr� )r� r� r� r@ )r r r r
�_testRecvmsgFromSendmsg� s z+RecvmsgGenericTests._testRecvmsgFromSendmsgN) r r r rE rF rG rH rI rJ r Zrequires_freebsd_versionrK rL rM rN rO rP rQ rR rS rT r� r rU r� rV rW rX r r r r
rD s(
# rD c @ sO e Z d Z e j d � d d � � Z d d � Z d d � Z d d � Z d
S) �RecvmsgGenericStreamTestsz)fails intermittently when run within Kojic C sk | j | j d � \ } } } } | j | d � | j | | j � | j | g � | j | d d �d S)Ni r� r
)r� r r� r rg r )r r� r� r= rl r r r
�testRecvmsgEOF� s
!z(RecvmsgGenericStreamTests.testRecvmsgEOFc C s | j j � d S)N)r� r
)r r r r
�_testRecvmsgEOF� s z)RecvmsgGenericStreamTests._testRecvmsgEOFc C s� | j | j t t � d � \ } } } } | j | | j � | j | g � | j | d d �| j | j d � \ } } } } | j | | j � | j | g � | j | d d �| | } | j | t � d S)Nr� r
Fi T)r� r r@ r� r rg r� r )r r� r� r= rl r� r� r r r
�testRecvmsgOverflow� s !
z-RecvmsgGenericStreamTests.testRecvmsgOverflowc C s | j t � d S)N)r� r� )r r r r
�_testRecvmsgOverflow� s z.RecvmsgGenericStreamTests._testRecvmsgOverflowN) r r r rp Z_skipInRpmBuildrZ r[ r\ r] r r r r
rY � s rY c @ s( e Z d Z d d � Z d d � Z d S)� RecvmsgTestsc C sh | j t | j j � | j t | j j d d d � | j t | j j t t � d d � | j t | j j t d � g d d � | j t | j j t � d d � | j t | j j t t � t � d � | j t | j j t t � d t � � | j j t t � d d � \ } } } } | j | t � | j
| | j � | j | g � | j | d d �d S)NrB r r r
Tr� r� )
r� rX r r� r r@ r� r" r+ r� r rg r )r r� r� r= rl r r r
�testRecvmsgBadArgs� s$
*z RecvmsgTests.testRecvmsgBadArgsc C s | j t � d S)N)r� r� )r r r r
�_testRecvmsgBadArgs� s z RecvmsgTests._testRecvmsgBadArgsN)r r r r_ r` r r r r
r^ � s r^ c @ sp e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d S)�RecvmsgIntoTestsc C s� t t t � � } | j t | j j � | j t | j j t t � d d � | j t | j j | d d � | j t | j j t � g d d � | j t | j j d g d d � | j t | j j | t � g d d � | j t | j j | g d d � | j t | j j | g t � d � | j t | j j | g d t � � | j j | g d d � \ } } } } | j | t t � � | j | t t � � | j
| | j � | j | g � | j | d d �d S)Nr s I'm not writablerB r
Tr� )
r" r@ r� r� rX r r r+ r r� r rg r )r r �nbytesr� r= rl r r r
�testRecvmsgIntoBadArgs� s0
'z'RecvmsgIntoTests.testRecvmsgIntoBadArgsc C s | j t � d S)N)r� r� )r r r r
�_testRecvmsgIntoBadArgs� s z(RecvmsgIntoTests._testRecvmsgIntoBadArgsc C s� t t t � � } | j j d d � | g D� � \ } } } } | j | t t � � | j | t t � � | j | | j � | j | g � | j | d d �d S)Nc s s | ] } | Vq d S)Nr )r� r� r r r
r�
s z<RecvmsgIntoTests.testRecvmsgIntoGenerator.<locals>.<genexpr>r
T) r" r@ r� r r r� r rg r )r r rb r� r= rl r r r
�testRecvmsgIntoGenerator� s %z)RecvmsgIntoTests.testRecvmsgIntoGeneratorc C s | j t � d S)N)r� r� )r r r r
�_testRecvmsgIntoGenerator
s z*RecvmsgIntoTests._testRecvmsgIntoGeneratorc C s� t j d d g t t � � } | j j | g � \ } } } } | j | t t � � | j | j � t � | j | | j � | j | g � | j | d d �d S)Nr% r r
T)
r& r@ r� r r r� �tobytesr rg r )r r rb r� r= rl r r r
�testRecvmsgIntoArray
s !z%RecvmsgIntoTests.testRecvmsgIntoArrayc C s | j t � d S)N)r� r� )r r r r
�_testRecvmsgIntoArray
s z&RecvmsgIntoTests._testRecvmsgIntoArrayc C s� t d � } t d � } t d � } | j j | t | � d d � | g � \ } } } } | j | t d � � | j | t d � � | j | t d � � | j | t d � � | j | | j � | j | g � | j | d
d �d S) Ns ----s
0123456789s --------------r� � s Mary had a little lambs Marys
01 had a 9s little lamb---r
T) r" r r �
memoryviewr� r@ r rg r )r Zb1Zb2Zb3rb r� r= rl r r r
�testRecvmsgIntoScatter
s .z'RecvmsgIntoTests.testRecvmsgIntoScatterc C s | j d � d S)Ns Mary had a little lamb)r� )r r r r
�_testRecvmsgIntoScatter'
s z(RecvmsgIntoTests._testRecvmsgIntoScatterN) r r r rc rd re rf rh ri rl rm r r r r
ra � s
ra c @ s� e Z d Z y d d l Z Wn e k
r6 d Z Yn Xe d e j � Z e e d � d d � � Z
e e d � d d � � Z d S)
�CmsgMacroTestsr Ni����CMSG_LENc C s | j t j d � d } t t d � � t t | d | � � } | j t j d � t j d � j d � xM | D]E } t j | � } | j | t j d � | � | j | | j � qs W| j
t t j d � | j
t t j | � | j
t t j t j
� d S)Nr rB i r r� r� )�socklen_t_limitr ro �listrq r� r& �itemsizer� r r� r� r� �maxsize)r �toobig�values�n�retr r r
� testCMSG_LEN8
s ))
zCmsgMacroTests.testCMSG_LENr1 c C s: | j t j d � d } t t d � � t t | d | � � } t j d � } | j | t j d � j d � x| | D]t } t j | � } | j | | � | j | t j | � � | j | | t j d � � | j
| | j � | } qy W| j t t j d � | j t t j | � | j t t j t
j � d S)NrB i r r r� r� )rp r r1 rq rq r� r& rr r ro r r� r� r� rs )r rt ru Zlastrv rw r r r
�testCMSG_SPACEM
s )
z CmsgMacroTests.testCMSG_SPACE) r r r rx r rp �minry r� r rx ry r r r r
rn +
s
rn c @ s� e Z d Z de Z d d � Z d d � Z d d � Z d d � Z d
d � Z d d
d d � Z d d � Z
d d � Z d d � Z d d � Z
e e d � d d � � Z e j d d � � Z d d � Z d d � Z e j e j d! k d" � e j e j j d# � d$ � e e d � d% d& � � � � Z e j e j e j d! k d" � e j e j j d# � d$ � d' d( � � � � Z e j e j d! k d" � e j e j j d# � d$ � e e d � d) d* � � � � Z e j e j e j d! k d" � e j e j j d# � d$ � d+ d, � � � � Z d- d. � Z e j e j d! k d/ � d0 d1 � � Z d2 d3 � Z d4 d5 � Z! d6 d7 � Z" e e d � d8 d9 � � Z# e# j d: d; � � Z$ d
d< d= � Z% d> d? � Z& d@ dA � Z' dB dC � Z( dD dE � Z) dF dG � Z* dH dI � Z+ dJ dK � Z, dL dM � Z- dN dO � Z. dP dQ � Z/ d
dR dS � Z0 dT dU � Z1 dV dW � Z2 dX dY � Z3 dZ d[ � Z4 d\ d] � Z5 d^ d_ � Z6 d` da � Z7 db dc � Z8 dd S)f�
SCMRightsTestiUU c C s� g } xu t | � D]g } t j � \ } } | j t j | � | j t j | � t j | t | � j � � | j
| � q W| S)N) rq r~ r} r+ r� r� r
�writer� �encode�append)r rv �fdsr r~ r� r r r
�newFDsn
s zSCMRightsTest.newFDsc C s_ xX t | � D]J \ } } t j | d t j � | j t j | d � t | � j � � q
Wd S)Nr i )� enumerater� �lseek�SEEK_SETr� r� r� r} )r r rv r~ r r r
�checkFDsz
s zSCMRightsTest.checkFDsc C s | j | j | � d S)N)r+ �closeRecvmsgFDs)r r� r r r
r� �
s z#SCMRightsTest.registerRecvmsgResultc C s� x� | d D]� \ } } } | t j k r | t j k r t j d � } | j | d t | � t | � | j � � x | D] } t j | � q| Wq Wd S)NrB r ) r r� �
SCM_RIGHTSr& � frombytesr@ rr r� r
)r Zrecvmsg_result�
cmsg_level� cmsg_type� cmsg_datar r~ r r r
r� �
s "
z SCMRightsTest.closeRecvmsgFDsc C sP | j | j t g t j t j t j d | j | � � f g � t t � � d S)Nr ) r� r� r� r r� r� r& r� r@ )r rv r r r
�createAndSendFDs�
s !z SCMRightsTest.createAndSendFDsrB r c C s@ | \ } } } } | j | t � | j | | j � | j | d d d t j d | �| j | t � | j t
| � | � t j d � } x� | D]� }
| j |
t � |
\ } } }
| j | t j
� | j | t j � | j |
t � | j t
|
� t d � | j |
� q� W| j t
| � | � | j | � d S)Nr
Tr r
r r )r� r� r rg r r r r� rq r r@ r& �tupler� r� r� �
SIZEOF_INTr� r� )r Znumfdsr� �maxcmsgs� ignoreflagsr� r� r= rl r �itemr� r� r� r r r
�checkRecvmsgFDs�
s$
z SCMRightsTest.checkRecvmsgFDsc C s, | j d | j | j t t � d � � d S)NrB i ( )r� r� r r@ r� )r r r r
�testFDPassSimple�
s z SCMRightsTest.testFDPassSimplec C sV | j | j t g t j t j t j d | j d � � j � f g � t t � � d S)Nr rB )
r� r� r� r r� r� r& r� rg r@ )r r r r
�_testFDPassSimple�
s 'z SCMRightsTest._testFDPassSimplec C s, | j d | j | j t t � d � � d S)Nr� i ( )r� r� r r@ r� )r r r r
�testMultipleFDPass�
s z SCMRightsTest.testMultipleFDPassc C s | j d � d S)Nr� )r� )r r r r
�_testMultipleFDPass�
s z!SCMRightsTest._testMultipleFDPassr1 c C s9 | j d | j | j t t � t j d t � � � d S)Nr� )r� r� r r@ r� r r1 r� )r r r r
�testFDPassCMSG_SPACE�
s z"SCMRightsTest.testFDPassCMSG_SPACEc C s | j d � d S)Nr� )r� )r r r r
�_testFDPassCMSG_SPACE�
s z#SCMRightsTest._testFDPassCMSG_SPACEc C sB | j d | j | j t t � t j d t � � d t j �d S)NrB r� r� ) r� r� r r@ r� r ro r� r )r r r r
�testFDPassCMSG_LEN�
s z SCMRightsTest.testFDPassCMSG_LENc C s | j d � d S)NrB )r� )r r r r
�_testFDPassCMSG_LEN�
s z!SCMRightsTest._testFDPassCMSG_LENr zskipping, see issue #12958�aixzskipping, see issue #22397c C s2 | j d | j | j t t � d � d d �d S)Nr� i ( r� )r� r� r r@ r� )r r r r
�testFDPassSeparate�
s z SCMRightsTest.testFDPassSeparatec C s� | j d � \ } } | j | j t g t j t j t j d | g � f t j t j t j d | g � f g � t t � � d S)Nr� r ) r� r� r� r� r r� r� r& r@ )r �fd0�fd1r r r
�_testFDPassSeparate�
s z!SCMRightsTest._testFDPassSeparatec C sQ | j d | j | j t t � t j t � t j t � � d d d t j �d S)Nr� r� r� )
r� r� r r@ r� r r1 r� ro r )r r r r
�testFDPassSeparateMinSpace�
s
z(SCMRightsTest.testFDPassSeparateMinSpacec C s� | j d � \ } } | j | j t g t j t j t j d | g � f t j t j t j d | g � f g � t t � � d S)Nr� r ) r� r� r� r� r r� r� r& r@ )r r� r� r r r
�_testFDPassSeparateMinSpace s z)SCMRightsTest._testFDPassSeparateMinSpacec C s� y | j | g | � } WnJ t k
re } z* | j | j t � | j | g � } WYd d } ~ Xn X| j | t | � � d S)N)r� r r� r� r� r� r@ )r r� r� rb r[ r r r
�sendAncillaryIfPossible s %z%SCMRightsTest.sendAncillaryIfPossiblezskipping, see issue #24725c C s5 | j d | j | j t t � d � d t j �d S)Nr i ( r� )r� r� r r@ r� r r )r r r r
�testFDPassEmpty s z SCMRightsTest.testFDPassEmptyc C s& | j t t j t j d f g � d S)Nr� )r� r� r r� r� )r r r r
�_testFDPassEmpty( s z SCMRightsTest._testFDPassEmptyc C s� | j | j t t � d � \ } } } } | j | t � | j | | j � | j | d d d t j �| j
t | � d � xS | D]K \ } } } | j | t j � | j | t j � | j
t | � t � q� Wd S)Ni ( r
Tr
rB )r� r r@ r� r� r rg r r r r r� r� �
assertLessr� )r r� r� r= rl r� r� r� r r r
�testFDPassPartialInt- s z"SCMRightsTest.testFDPassPartialIntc C sH | j t t j t j t j d | j g � j � d d � f g � d S)Nr rB r� )r� r� r r� r� r& �badfdrg )r r r r
�_testFDPassPartialInt: s
z#SCMRightsTest._testFDPassPartialIntc C s# | j | j t t � d � \ } } } } | j | t � | j | | j � | j | d d d t j �| j
t | � d � t j d � } xk | D]c \ } } } | j | t j � | j | t j
� | j | d t | � t | � | j � � q� W| j
t | � d � | j | � d S)Ni ( r
Tr
r� r )r� r r@ r� r� r rg r r r r r& r� r� r� rr r� ) r r� r� r= rl r r� r� r� r r r
� testFDPassPartialIntInMiddleA s &z*SCMRightsTest.testFDPassPartialIntInMiddlec C s� | j d � \ } } | j t t j t j t j d | | j g � j � d d � f t j t j t j d | g � f g � d S)Nr� r rB r� ) r� r� r� r r� r� r& r� rg )r r� r� r r r
� _testFDPassPartialIntInMiddleT s +z+SCMRightsTest._testFDPassPartialIntInMiddlec C sk | \ } } } } | j | t � | j | | j � | j | g � | j | d d d t j d | �d S)Nr
Tr r
)r� r� r rg r r r )r r� r� r� r� r= rl r r r
�checkTruncatedHeader` s z"SCMRightsTest.checkTruncatedHeaderc C s/ | j | j | j t t � � d t j �d S)Nr� )r� r� r r@ r� r r )r r r r
�testCmsgTruncNoBufSizej s !z$SCMRightsTest.testCmsgTruncNoBufSizec C s | j d � d S)NrB )r� )r r r r
�_testCmsgTruncNoBufSizes s z%SCMRightsTest._testCmsgTruncNoBufSizec C s2 | j | j | j t t � d � d t j �d S)Nr r� )r� r� r r@ r� r r )r r r r
�testCmsgTrunc0v s $z SCMRightsTest.testCmsgTrunc0c C s | j d � d S)NrB )r� )r r r r
�_testCmsgTrunc0{ s z SCMRightsTest._testCmsgTrunc0c C s) | j | j | j t t � d � � d S)NrB )r� r� r r@ r� )r r r r
�testCmsgTrunc1� s z SCMRightsTest.testCmsgTrunc1c C s | j d � d S)NrB )r� )r r r r
�_testCmsgTrunc1� s z SCMRightsTest._testCmsgTrunc1c C s- | j | j | j t t � t d � � d S)Nr� )r� r� r r@ r� r� )r r r r
�testCmsgTrunc2Int� s z SCMRightsTest.testCmsgTrunc2Intc C s | j d � d S)NrB )r� )r r r r
�_testCmsgTrunc2Int� s z SCMRightsTest._testCmsgTrunc2Intc C s6 | j | j | j t t � t j d � d � � d S)Nr rB )r� r� r r@ r� r ro )r r r r
�testCmsgTruncLen0Minus1� s z%SCMRightsTest.testCmsgTruncLen0Minus1c C s | j d � d S)NrB )r� )r r r r
�_testCmsgTruncLen0Minus1� s z&SCMRightsTest._testCmsgTruncLen0Minus1c C sN | j | j t t � | � \ } } } } | j | t � | j | | j � | j | d d d t j �| d k r� | g k r� d S| j t | � d � | d \ } } }
| j | t j
� | j | t j � | j t |
� | � | j
t |
� | � t j d � } | j |
d t |
� t |
� | j � � | j | � d S)Nr
Tr r rB r )r� r r@ r� r� r rg r r r r� r� r r r& r� rr r� ) r �ancbuf�maxdataZmindatar� r� r= rl r� r� r� r r r r
�checkTruncatedArray� s" "z!SCMRightsTest.checkTruncatedArrayc C s# | j d t j d � d d � d S)Nr� r r� )r� r ro )r r r r
�testCmsgTruncLen0� s z SCMRightsTest.testCmsgTruncLen0c C s | j d � d S)NrB )r� )r r r r
�_testCmsgTruncLen0� s z SCMRightsTest._testCmsgTruncLen0c C s' | j d t j d � d d d � d S)Nr� r rB r� )r� r ro )r r r r
�testCmsgTruncLen0Plus1� s z$SCMRightsTest.testCmsgTruncLen0Plus1c C s | j d � d S)Nr� )r� )r r r r
�_testCmsgTruncLen0Plus1� s z%SCMRightsTest._testCmsgTruncLen0Plus1c C s# | j d t j t � d t � d S)Nr� r� )r� r ro r� )r r r r
�testCmsgTruncLen1� s z SCMRightsTest.testCmsgTruncLen1c C s | j d � d S)Nr� )r� )r r r r
�_testCmsgTruncLen1� s z SCMRightsTest._testCmsgTruncLen1c C s3 | j d t j d t � d d d t d � d S)Nr� r� rB r� )r� r ro r� )r r r r
�testCmsgTruncLen2Minus1� s z%SCMRightsTest.testCmsgTruncLen2Minus1c C s | j d � d S)Nr� )r� )r r r r
�_testCmsgTruncLen2Minus1� s z&SCMRightsTest._testCmsgTruncLen2Minus1Ni����)9r r r r� r� r� r� r� r� r� r� r� r� r� r� r r� r� r� r� r� rp r� r� r r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r
r{ f
sl
!0 !
0 '
r{ c @ s� e Z d Z d Z d^ Z d d � Z d d d d � Z e e d d
� d d � � Z e j
d
d � � Z e e d d d
� d d � � Z e j
d d � � Z
e e j d � e e d d
� d d � � � Z e j
d d � � Z d d d d � Z e e d d
d d � d d � � Z e j
d d � � Z e e d d d
d d � d! d" � � Z e j
d# d$ � � Z e e j d � e e d d d
d d � d% d& � � � Z e j
d' d( � � Z e e j d � e e d d d
d d � d) d* � � � Z e j
d+ d, � � Z d d- d. � Z e e d d
� d/ d0 � � Z e j
d1 d2 � � Z e e d d
� d3 d4 � � Z e j
d5 d6 � � Z e e d d
� d7 d8 � � Z e j
d9 d: � � Z e e d d
� d; d< � � Z e j
d= d> � � Z! e e d d
� d? d@ � � Z" e" j
dA dB � � Z# e e d d
� dC dD � � Z$ e$ j
dE dF � � Z% d dG dH � Z& e e d d d
d d � dI dJ � � Z' e' j
dK dL � � Z( e e d d d
d d � dM dN � � Z) e) j
dO dP � � Z* e e d d d
d d � dQ dR � � Z+ e+ j
dS dT � � Z, e e d d d
d d � dU dV � � Z- e- j
dW dX � � Z. e e d d d
d d � dY dZ � � Z/ e/ j
d[ d\ � � Z0 d] S)_�RFC3542AncillaryTestr� rB c C sJ i } x= | D]5 \ } } } | j | | f | � | | | | f <q
W| S)N)r� )r r� �dr� r� r� r r r
�ancillaryMapping� s
z%RFC3542AncillaryTest.ancillaryMappingr� r c C si | j j t j t j d � | j j � | j | j t t � | � \ } } } } | j
| t � | j | | j � | j
| d d d t j d | �| j
t | � d � | j | d t � | d \ } } }
| j
| t j � | j
| t j � | j |
t � | j
t |
� t � t j d � } | j |
� | j | d d � | j | d | � d S)NrB r
Tr r
r r )r r- r � IPPROTO_IPV6�IPV6_RECVHOPLIMITr� r@ r� r@ r� r� r rg r r r� r� �
IPV6_HOPLIMITr� r� r& r� r r ) r �
ancbufsize�maxhopr� r� r� r= rl r� r� r� r r r r
�
checkHopLimit� s(
z"RFC3542AncillaryTest.checkHopLimitr� r� c C s | j d d � d S)Nr� i ( )r� )r r r r
�testRecvHopLimit s z%RFC3542AncillaryTest.testRecvHopLimitc C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
�_testRecvHopLimit
s z&RFC3542AncillaryTest._testRecvHopLimitr1 c C s | j d t j t � � d S)Nr� )r� r r1 r� )r r r r
�testRecvHopLimitCMSG_SPACE s z/RFC3542AncillaryTest.testRecvHopLimitCMSG_SPACEc C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
�_testRecvHopLimitCMSG_SPACE s z0RFC3542AncillaryTest._testRecvHopLimitCMSG_SPACEr� c C s | j d d d | j � d S)Nr� i ( r� )r� � hop_limit)r r r r
�testSetHopLimit! s z$RFC3542AncillaryTest.testSetHopLimitc C sl | j | j j d | j � � | j | j t g t j t j t
j
d | j g � f g � t t � � d S)NrU r )
r� r� rQ r r� r� r� r r� r� r& r� r@ )r r r r
�_testSetHopLimit( s z%RFC3542AncillaryTest._testSetHopLimitc C s� | j j t j t j d � | j j t j t j d � | j j � | j | j t t
� | � \ } } } } | j | t
� | j | | j
� | j | d d d t j d | �| j t | � d � | j | � } | t j t j f } | j t | � t � t j d � }
|
j | � | j |
d d � | j |
d d � | t j t j f } | j t | � t � t j d � }
|
j | � | j |
d d � | j |
d | � d S)
NrB r
Tr r
r� r r r� )r r- r r� r� �IPV6_RECVTCLASSr� r@ r� r@ r� r� r rg r r r� � IPV6_TCLASSr� r& r� r r r� ) r r� r� r� r� r� r= rl ZancmapZtcdatar Zhldatar r r
� checkTrafficClassAndHopLimit1 s2
z1RFC3542AncillaryTest.checkTrafficClassAndHopLimitr� r� c C s | j d d � d S)Nr� i ( )r� )r r r r
� testRecvTrafficClassAndHopLimitU s z4RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitc C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
� _testRecvTrafficClassAndHopLimit[ s z5RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitc C s! | j d t j t � d � d S)Nr� r� )r� r r1 r� )r r r r
�)testRecvTrafficClassAndHopLimitCMSG_SPACE` s z>RFC3542AncillaryTest.testRecvTrafficClassAndHopLimitCMSG_SPACEc C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
�*_testRecvTrafficClassAndHopLimitCMSG_SPACEh s z?RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitCMSG_SPACEc C s | j d d d | j � d S)Nr� i ( r� )r� r� )r r r r
� testSetTrafficClassAndHopLimitm s z3RFC3542AncillaryTest.testSetTrafficClassAndHopLimitc C s� | j | j j d | j � � | j | j t g t j t j t
j
d | j g � f t j t j t
j
d | j
g � f g � t t � � d S)NrU r )r� r� rQ r r� r� r� r r� r� r& �
traffic_classr� r� r@ )r r r r
� _testSetTrafficClassAndHopLimitv s z4RFC3542AncillaryTest._testSetTrafficClassAndHopLimitc C s | j d d d | j � d S)Nr� i ( r� )r� r� )r r r r
�testOddCmsgSize� s z$RFC3542AncillaryTest.testOddCmsgSizec C s< | j | j j d | j � � yk | j t g t j t j t j d | j
g � j � d f t j t j t j d | j
g � f g � } Wn� t k
r7} z� | j | j t � | j t g t j t j t j d | j
g � f t j t j t j d | j
g � f g � } | j | t t � � WYd d } ~ Xn Xd S)NrU r s )r� r� rQ r r� r� r r� r� r& r� rg r� r� r r� r� r� r� r@ )r rb r[ r r r
�_testOddCmsgSize� s" " % !z%RFC3542AncillaryTest._testOddCmsgSizec C s� | j j t j t j d � | j j � | d k r; f n | f } | j | j t t � | � \ } } } } | j
| t � | j | | j � | j
| g � | j
| d d d t j d | �d S)NrB r
Tr r
)r r- r r� r� r� r@ r� r@ r� r� r rg r r )r r� r� r( r� r� r= rl r r r
� checkHopLimitTruncatedHeader� s
z1RFC3542AncillaryTest.checkHopLimitTruncatedHeaderc C s | j d d d t j � d S)Nr� r� )r� r r )r r r r
r� � s z+RFC3542AncillaryTest.testCmsgTruncNoBufSizec C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
r� � s z,RFC3542AncillaryTest._testCmsgTruncNoBufSizec C s | j d d d t j � d S)Nr� r r� )r� r r )r r r r
�testSingleCmsgTrunc0� s z)RFC3542AncillaryTest.testSingleCmsgTrunc0c C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
�_testSingleCmsgTrunc0� s z*RFC3542AncillaryTest._testSingleCmsgTrunc0c C s | j d d � d S)Nr� rB )r� )r r r r
�testSingleCmsgTrunc1� s z)RFC3542AncillaryTest.testSingleCmsgTrunc1c C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
�_testSingleCmsgTrunc1� s z*RFC3542AncillaryTest._testSingleCmsgTrunc1c C s | j d d t � d S)Nr� r� )r� r� )r r r r
�testSingleCmsgTrunc2Int� s z,RFC3542AncillaryTest.testSingleCmsgTrunc2Intc C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
�_testSingleCmsgTrunc2Int� s z-RFC3542AncillaryTest._testSingleCmsgTrunc2Intc C s! | j d t j d � d � d S)Nr� r rB )r� r ro )r r r r
� testSingleCmsgTruncLen0Minus1� s z2RFC3542AncillaryTest.testSingleCmsgTruncLen0Minus1c C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
� _testSingleCmsgTruncLen0Minus1� s z3RFC3542AncillaryTest._testSingleCmsgTruncLen0Minus1c C s | j j t j t j d � | j j � | j | j t t � t j
t � d � \ } } } } | j | t � | j
| | j � | j | d d d t j �| j t | � d � | r
| d \ } } } | j | t j � | j | t j � | j t | � t � d S)NrB r
Tr r )r r- r r� r� r� r@ r� r@ r� ro r� r� r rg r r r r� r� )r r� r� r= rl r� r� r� r r r
�testSingleCmsgTruncInData� s
1 z.RFC3542AncillaryTest.testSingleCmsgTruncInDatac C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
�_testSingleCmsgTruncInData
s z/RFC3542AncillaryTest._testSingleCmsgTruncInDatac C sj | j j t j t j d � | j j t j t j d � | j j � | j | j t t
� | � \ } } } } | j | t
� | j | | j
� | j | d d d t j d | �| j t | � d � | d \ } } } | j | t j � | j | t j t j h � | j t | � t � t j d � }
|
j | � | j |
d d � | j |
d d � d S) NrB r
Tr r
r r r� )r r- r r� r� r� r� r@ r� r@ r� r� r rg r r r� r� r� r� r& r� r r ) r r� r� r� r� r= rl r� r� r� r r r r
�checkTruncatedSecondHeader
s(
z/RFC3542AncillaryTest.checkTruncatedSecondHeaderc C s# | j t j t � d t j �d S)Nr� )r� r r1 r� r )r r r r
�testSecondCmsgTrunc0*
s z)RFC3542AncillaryTest.testSecondCmsgTrunc0c C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
�_testSecondCmsgTrunc00
s z*RFC3542AncillaryTest._testSecondCmsgTrunc0c C s | j t j t � d � d S)NrB )r� r r1 r� )r r r r
�testSecondCmsgTrunc15
s z)RFC3542AncillaryTest.testSecondCmsgTrunc1c C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
�_testSecondCmsgTrunc1:
s z*RFC3542AncillaryTest._testSecondCmsgTrunc1c C s" | j t j t � d t � d S)Nr� )r� r r1 r� )r r r r
�testSecondCmsgTrunc2Int?
s z,RFC3542AncillaryTest.testSecondCmsgTrunc2Intc C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
�_testSecondCmsgTrunc2IntE
s z-RFC3542AncillaryTest._testSecondCmsgTrunc2Intc C s+ | j t j t � t j d � d � d S)Nr rB )r� r r1 r� ro )r r r r
� testSecondCmsgTruncLen0Minus1J
s z2RFC3542AncillaryTest.testSecondCmsgTruncLen0Minus1c C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
� _testSecondCmsgTruncLen0Minus1P
s z3RFC3542AncillaryTest._testSecondCmsgTruncLen0Minus1c
C s� | j j t j t j d � | j j t j t j d � | j j � | j | j t t
� t j t � t j
t � d � \ } } } } | j | t
� | j | | j � | j | d d d t j �t j t j h } | j d � \ } } } | j | t j � | j | � | j t | � t � t j d � } | j | � | j | d d � | j | d d � | r�| j d � \ } } } | j | t j � | j | � | j t | � t � | j | g � d S)NrB r
Tr r r r� ) r r- r r� r� r� r� r@ r� r@ r� r1 r� ro r� r rg r r r� r� �pop�remover& r� r r r� )
r r� r� r= rl Z
cmsg_typesr� r� r� r r r r
�testSecomdCmsgTruncInDataU
s4
/
z.RFC3542AncillaryTest.testSecomdCmsgTruncInDatac C s0 | j | j j d | j � � | j t � d S)NrU )r� r� rQ r r� r� )r r r r
�_testSecomdCmsgTruncInDataz
s z/RFC3542AncillaryTest._testSecomdCmsgTruncInDataNr� )1r r r r� r� r� r� r� r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r
r� � sp
!
! # $r� c @ s e Z d Z d S)�SendrecvmsgUDPTestBaseN)r r r r r r r
r� �
s r� r� c @ s e Z d Z d S)�SendmsgUDPTestN)r r r r r r r
r� �
s r� r� c @ s e Z d Z d S)�RecvmsgUDPTestN)r r r r r r r
r� �
s r� r c @ s e Z d Z d S)�RecvmsgIntoUDPTestN)r r r r r r r
r� �
s r� c @ s e Z d Z d d � Z d S)�SendrecvmsgUDP6TestBasec C s( | j | d d � | d d � � d S)NrB r� r� )r� )r r r r r r
r �
s z+SendrecvmsgUDP6TestBase.checkRecvmsgAddressN)r r r r r r r r
r� �
s r� z IPv6 required for this test.r� r! c @ s e Z d Z d S)�SendmsgUDP6TestN)r r r r r r r
r� �
s r� c @ s e Z d Z d S)�RecvmsgUDP6TestN)r r r r r r r
r� �
s r� c @ s e Z d Z d S)�RecvmsgIntoUDP6TestN)r r r r r r r
r� �
s r� r� c @ s e Z d Z d S)� RecvmsgRFC3542AncillaryUDP6TestN)r r r r r r r
r� �
s r� c @ s e Z d Z d S)�#RecvmsgIntoRFC3542AncillaryUDP6TestN)r r r r r r r
r� �
s r� c @ s e Z d Z d S)�SendrecvmsgTCPTestBaseN)r r r r r r r
r� �
s r� c @ s e Z d Z d S)�SendmsgTCPTestN)r r r r r r r
r� �
s r� c @ s e Z d Z d S)�RecvmsgTCPTestN)r r r r r r r
r �
s r c @ s e Z d Z d S)�RecvmsgIntoTCPTestN)r r r r r r r
r �
s r c @ s e Z d Z d S)� SendrecvmsgSCTPStreamTestBaseN)r r r r r r r
r �
s r r r r� c @ s e Z d Z d S)�SendmsgSCTPStreamTestN)r r r r r r r
r �
s r c s" e Z d Z � f d d � Z � S)�RecvmsgSCTPStreamTestc se y t t | � j � WnG t k
r` } z' | j t j k rA � | j d � WYd d } ~ Xn Xd S)Nz4sporadic ENOTCONN (kernel issue?) - see issue #13876)r# r rZ r r� �ENOTCONNr4 )r r[ )r* r r
rZ �
s z$RecvmsgSCTPStreamTest.testRecvmsgEOF)r r r rZ r r )r* r
r �
s r c s" e Z d Z � f d d � Z � S)�RecvmsgIntoSCTPStreamTestc se y t t | � j � WnG t k
r` } z' | j t j k rA � | j d � WYd d } ~ Xn Xd S)Nz4sporadic ENOTCONN (kernel issue?) - see issue #13876)r# r rZ r r� r r4 )r r[ )r* r r
rZ �
s z(RecvmsgIntoSCTPStreamTest.testRecvmsgEOF)r r r rZ r r )r* r
r �
s r c @ s e Z d Z d S)� SendrecvmsgUnixStreamTestBaseN)r r r r r r r
r s r r� c @ s e Z d Z d S)�SendmsgUnixStreamTestN)r r r r r r r
r s r c @ s e Z d Z d S)�RecvmsgUnixStreamTestN)r r r r r r r
r s r c @ s e Z d Z d S)�RecvmsgIntoUnixStreamTestN)r r r r r r r
r
s r
r� r� c @ s e Z d Z d S)�RecvmsgSCMRightsStreamTestN)r r r r r r r
r s r c @ s e Z d Z d S)� RecvmsgIntoSCMRightsStreamTestN)r r r r r r r
r % s r c s^ e Z d Z � f d d � Z d Z e e d � rH d Z d d � Z n d Z d d � Z � S)
�InterruptedTimeoutBasec sX t � j � t j t j d d � � } | j t j t j | � | j | j d � d S)Nc S s d d S)NrB r r )Zsignumr� r r r
r� : s z.InterruptedTimeoutBase.setUp.<locals>.<lambda>r )r# r rP rQ r+ �setAlarm)r Zorig_alrm_handler)r* r r
r 7 s
z InterruptedTimeoutBase.setUpg @� setitimerg�������?c C s t j t j | � d S)N)rP r � ITIMER_REAL)r �secondsr r r
r G s z InterruptedTimeoutBase.setAlarmr� c C s t j | � d S)N)rP rJ )r r r r r
r M s ) r r r r rU rW rP �
alarm_timer r r )r* r
r
2 s r
� siginterruptrJ r z+Don't have signal.alarm or signal.setitimerc s� e Z d Z � f d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z e e
j
d
� d d � � Z e e
j
d � d d � � Z � S)�InterruptedRecvTimeoutTestc s$ t � j � | j j | j � d S)N)r# r r r/ rU )r )r* r r
r Z s
z InterruptedRecvTimeoutTest.setUpc
O s: | j | j � | j t � � } | | | � Wd QRXd S)N)r r r� rR )r r� r( r) r� r r r
�checkInterruptedRecv^ s z/InterruptedRecvTimeoutTest.checkInterruptedRecvc C s | j | j j d � d S)Ni )r r r� )r r r r
�testInterruptedRecvTimeoute s z5InterruptedRecvTimeoutTest.testInterruptedRecvTimeoutc C s | j | j j t d � � d S)Ni )r r � recv_intor" )r r r r
� testInterruptedRecvIntoTimeouth s z9InterruptedRecvTimeoutTest.testInterruptedRecvIntoTimeoutc C s | j | j j d � d S)Ni )r r r� )r r r r
� testInterruptedRecvfromTimeoutk s z9InterruptedRecvTimeoutTest.testInterruptedRecvfromTimeoutc C s | j | j j t d � � d S)Ni )r r �
recvfrom_intor" )r r r r
�"testInterruptedRecvfromIntoTimeoutn s z=InterruptedRecvTimeoutTest.testInterruptedRecvfromIntoTimeoutr� c C s | j | j j d � d S)Ni )r r r� )r r r r
� testInterruptedRecvmsgTimeoutq s z8InterruptedRecvTimeoutTest.testInterruptedRecvmsgTimeoutr c C s# | j | j j t d � g � d S)Ni )r r r r" )r r r r
�!testInterruptedRecvmsgIntoTimeoutu s z<InterruptedRecvTimeoutTest.testInterruptedRecvmsgIntoTimeout)
r r r r r r r r r r� r r r r r )r* r
r S s r c s� e Z d Z � f d d � Z d d � Z d d � Z e j d d � d d
� � Z e j d d � d d � � Z e j d d � e
e j d
� d d � � � Z � S)�InterruptedSendTimeoutTestc s� t � j � | j � | _ | j | j j � t j d | j � } | j � | j
j � \ | _ } | j | j j � | j
� | j j | j � d S)N�target)r# r rt rn r+ r
r% ZThread� doConnect�startr ri rj r� r/ rU )r Z
cli_threadrl )r* r r
r � s
z InterruptedSendTimeoutTest.setUpc C s | j j | j � d S)N)rn rm rw )r r r r
r � s z$InterruptedSendTimeoutTest.doConnectc
O sA | j t � �+ } x! | j | j � | | | � q WWd QRXd S)N)r� rR r r )r r� r( r) r� r r r
�checkInterruptedSend� s z/InterruptedSendTimeoutTest.checkInterruptedSendr r� c C s | j | j j d d � d S)Nr< i )r" rn r0 )r r r r
�testInterruptedSendTimeout� s z5InterruptedSendTimeoutTest.testInterruptedSendTimeoutc C s$ | j | j j d d | j � d S)Nr< i )r" rn r� rw )r r r r
� testInterruptedSendtoTimeout� s z7InterruptedSendTimeoutTest.testInterruptedSendtoTimeoutr� c C s! | j | j j d d g � d S)Nr< i )r" rn r� )r r r r
� testInterruptedSendmsgTimeout� s z8InterruptedSendTimeoutTest.testInterruptedSendmsgTimeout)
r r r r r r" r Zrequires_mac_verr# r$ r� r r% r r )r* r
r | s
r c @ s( e Z d Z d d � Z d d � Z d S)�
TCPCloserTestc C s� | j j � \ } } | j � | j } t j | g g g d � \ } } } | j | | g � | j | j d � d � | j � | j � d S)Ng �?rB r� )r ri r
ra r� r� r� )r rk rl Zsdr� r| r� r r r
� testClose� s
$
zTCPCloserTest.testClosec C s* | j j t | j f � t j d � d S)Ng �?)ra rm re r �time�sleep)r r r r
�
_testClose� s zTCPCloserTest._testCloseN)r r r r' r* r r r r
r& � s
r& c @ ss e Z d Z d d d � Z d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z d d � Z
d S)�BasicSocketPairTestr_ c C s t j | d | �d S)Nr` )ro r$ )r r` r r r
r$ � s z BasicSocketPairTest.__init__c C s~ | j | t j � t t d � r; | j | j t j � n | j | j t j � | j | j t j � | j | j d � d S)Nr� r )
r� r rW r� r� r� r r� r r� )r r� r r r
�_check_defaults� s z#BasicSocketPairTest._check_defaultsc C s | j | j � d S)N)r, ra )r r r r
�
_testDefaults� s z!BasicSocketPairTest._testDefaultsc C s | j | j � d S)N)r, r )r r r r
� testDefaults� s z BasicSocketPairTest.testDefaultsc C s& | j j d � } | j | t � d S)Ni )r r� r� r� )r r� r r r
r� � s z BasicSocketPairTest.testRecvc C s | j j t � d S)N)ra r0 r� )r r r r
r� � s z BasicSocketPairTest._testRecvc C s | j j t � d S)N)r r0 r� )r r r r
�testSend� s z BasicSocketPairTest.testSendc C s& | j j d � } | j | t � d S)Ni )ra r� r� r� )r r� r r r
� _testSend� s z BasicSocketPairTest._testSendN) r r r r$ r, r- r. r� r� r/ r0 r r r r
r+ � s r+ c @ s e Z d Z d d d � Z d d � Z d d � Z e j d d � � Z e j e � Z e
j e e
d
� d � e j d d
d � d d � � � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# S)$�NonBlockingTCPTestsr_ c C s t j | d | �d S)Nr` )r^ r$ )r r` r r r
r$ � s z NonBlockingTCPTests.__init__c C s� | j j d � | j | j j � � | j j d � | j | j j � d � t j � } y | j j � Wn t k
r� Yn Xt j � } | j | | d k d � d S)NTFg g �?z Error setting non-blocking mode.) r r� r5 r
r� r( ri r r� )r r! �endr r r
�testSetBlocking� s
z#NonBlockingTCPTests.testSetBlockingc C s d S)Nr )r r r r
�_testSetBlocking� s z$NonBlockingTCPTests._testSetBlockingc C s� d d l } | j | j k r+ | j d � | j j d � | j | j j � d � | j j | j d � | j | j j � � d S)Nr zneeds UINT_MAX < ULONG_MAXFg rB ) rx r� Z ULONG_MAXr4 r r� r� r
r5 )r rx r r r
�testSetBlocking_overflow s
z,NonBlockingTCPTests.testSetBlocking_overflow�
SOCK_NONBLOCKz test needs socket.SOCK_NONBLOCKr� � � c C s� | j j � t j t j t j t j B� | _ t j | j � | _ | j j � t
j
� } y | j j � Wn t k
r� Yn Xt
j
� } | j
| | d k d � d S)Ng �?z&Error creating with non-blocking mode.)r r
r r r r6 r r r r r( ri r r� )r r! r2 r r r
�testInitNonBlocking s
z'NonBlockingTCPTests.testInitNonBlockingc C s d S)Nr )r r r r
�_testInitNonBlocking" s z(NonBlockingTCPTests._testInitNonBlockingc C s` | j j d � z. | j j � \ } } | j t t � � } Wd | j � | j j d � Xd S)Nr )r r/ ri r� r@ r� r
)r rk rl �messager r r
�testInheritFlags% s
z$NonBlockingTCPTests.testInheritFlagsc C sG t j d � | j j t | j f � t j d � | j j t � d S)Ng�������?g �?)r( r) ra rm re r r0 r� )r r r r
�_testInheritFlags0 s
z%NonBlockingTCPTests._testInheritFlagsc C s� | j j d � y | j j � \ } } Wn t k
r= Yn X| j d � t j | j g g g � \ } } } | j | k r� | j j � \ } } | j | j � � | j � n
| j d � d S)Nr z'Error trying to do non-blocking accept.z'Error trying to do accept after select.) r r� ri r r� r� r5 r
r
)r rk rl r� r| r� r r r
�
testAccept6 s
$
z NonBlockingTCPTests.testAcceptc C s* t j d � | j j t | j f � d S)Ng�������?)r( r) ra rm re r )r r r r
� _testAcceptG s
z NonBlockingTCPTests._testAcceptc C s# | j j � \ } } | j � d S)N)r ri r
)r rk rl r r r
� testConnectK s z NonBlockingTCPTests.testConnectc C s- | j j d � | j j t | j f � d S)Nr )ra r/ rm re r )r r r r
� _testConnectP s z NonBlockingTCPTests._testConnectc C s� | j j � \ } } | j d � y | j t t � � } Wn t k
rO Yn X| j d � t j | g g g � \ } } } | | k r� | j t t � � } | j � | j
| t � n
| j d � d S)Nr z%Error trying to do non-blocking recv.z0Error during select call to non-blocking socket.) r ri r� r� r@ r� r r� r� r
r� )r rk rl r� r� r| r� r r r
r� T s
!
z NonBlockingTCPTests.testRecvc C s: | j j t | j f � t j d � | j j t � d S)Ng�������?)ra rm re r r( r) r0 r� )r r r r
r� f s
z NonBlockingTCPTests._testRecvN)r r r r$ r3 r4 r r� r5 Z_testSetBlocking_overflowrp r� rW r �requires_linux_versionr9 r: r< r= r> r? r@ rA r� r� r r r r
r1 � s" $ r1 c @ su e Z d Z d Z d; Z d Z d Z d Z d Z e Z
d Z e Z d d d
� Z
d d � Z d
d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z! d3 d4 � Z" d5 d6 � Z# d7 d8 � Z$ d9 d: � Z% d S)<�FileObjectClassTestCaseao Unit tests for the object returned by socket.makefile()
self.read_file is the io object returned by makefile() on
the client connection. You can read from this file to
get output from the server.
self.write_file is the io object returned by makefile() on the
server connection. You can write to this file to send output
to the client.
rB zutf-8�strictNr[ ri r_ c C s t j | d | �d S)Nr` )rh r$ )r r` r r r
r$ � s z FileObjectClassTestCase.__init__c C s{ d d � t d � D� \ | _ | _ | _ | _ t j | � | j j | j | j
d | j d | j d | j
�| _ d S)Nc S s g | ] } t j � � q Sr )r% rE )r� r r r r
r� � s z1FileObjectClassTestCase.setUp.<locals>.<listcomp>r� �encoding�errors�newline)rq �evt1�evt2�
serv_finished� cli_finishedrh r rj r] � read_moder5 rE rF rG � read_file)r r r r
r � s 1
z FileObjectClassTestCase.setUpc C sG | j j � | j j � | j | j j � d | _ t j | � d S)N)rJ r@ rM r
r� �closedrh r )r r r r
r � s
z FileObjectClassTestCase.tearDownc C sJ t j | � | j j | j | j d | j d | j d | j �| _ d S)NrE rF rG )
rh rU rn r] �
write_moder5 rE rF rG �
write_file)r r r r
rU � s
z#FileObjectClassTestCase.clientSetUpc C sG | j j � | j j � | j | j j � d | _ t j | � d S)N)rK r@ rP r
r� rN rh rV )r r r r
rV � s
z&FileObjectClassTestCase.clientTearDownc C s� | j j d � | j j d � | j t j | j j d � | j t � � } | j j d � Wd QRX| j d t | j
� � d S)NrB r� z!cannot read from timed out object) rj r/ rM r� r� r rU r r� r� r� )r Zctxr r r
�testReadAfterTimeout� s z,FileObjectClassTestCase.testReadAfterTimeoutc C s; | j j | j d d � � | j j � | j j � d S)Nr r� )rP r| � write_msg�flushrJ rQ )r r r r
�_testReadAfterTimeout� s
z-FileObjectClassTestCase._testReadAfterTimeoutc C sR | j j t | j � d � } | j j d � } | | } | j | | j � d S)Nr� )rM r� r@ �read_msgr� )r � first_segZ
second_segr� r r r
�
testSmallRead� s
z%FileObjectClassTestCase.testSmallReadc C s$ | j j | j � | j j � d S)N)rP r| rR rS )r r r r
�_testSmallRead� s z&FileObjectClassTestCase._testSmallReadc C s& | j j � } | j | | j � d S)N)rM r� r� rU )r r� r r r
� testFullRead� s z$FileObjectClassTestCase.testFullReadc C s$ | j j | j � | j j � d S)N)rP r| rR r
)r r r r
�
_testFullRead� s z%FileObjectClassTestCase._testFullReadc C sS t | j � � } x' | j j d � } | s. P| | 7} q W| j | | j � d S)NrB )r� rU rM r� r� )r r �charr r r
�testUnbufferedRead� s z*FileObjectClassTestCase.testUnbufferedReadc C s$ | j j | j � | j j � d S)N)rP r| rR rS )r r r r
�_testUnbufferedRead� s z+FileObjectClassTestCase._testUnbufferedReadc C s& | j j � } | j | | j � d S)N)rM �readliner� rU )r �liner r r
� testReadline� s z$FileObjectClassTestCase.testReadlinec C s$ | j j | j � | j j � d S)N)rP r| rR rS )r r r r
�
_testReadline� s z%FileObjectClassTestCase._testReadlinec C s3 | j j � | j j � } | j | | j � d S)N)rj r
rM r� r� rU )r r� r r r
�testCloseAfterMakefile� s
z.FileObjectClassTestCase.testCloseAfterMakefilec C s$ | j j | j � | j j � d S)N)rP r| rR rS )r r r r
�_testCloseAfterMakefile� s z/FileObjectClassTestCase._testCloseAfterMakefilec C sZ | j j � | j j t t � � } t | j t � rC | j � } | j
| | j � d S)N) rM r
rj r� r@ r� r� rU r� �decoder� )r r� r r r
� testMakefileAfterMakefileClose� s
z6FileObjectClassTestCase.testMakefileAfterMakefileClosec C s$ | j j | j � | j j � d S)N)rP r| rR rS )r r r r
� _testMakefileAfterMakefileClose� s z7FileObjectClassTestCase._testMakefileAfterMakefileClosec C s | j | j j � d S)N)r� rM rN )r r r r
�testClosedAttr� s z&FileObjectClassTestCase.testClosedAttrc C s | j | j j � d S)N)r� rP rN )r r r r
�_testClosedAttr� s z'FileObjectClassTestCase._testClosedAttrc C s< | j | j j | j � | j | j j | j j � � d S)N)r� rM rj rL r� rj r� )r r r r
�testAttributes� s z&FileObjectClassTestCase.testAttributesc C s< | j | j j | j � | j | j j | j j � � d S)N)r� rP rj rO r� rn r� )r r r r
�_testAttributes s z'FileObjectClassTestCase._testAttributesc C sJ | j j � | j t | j j � | j j � | j t | j j � d S)N)rM r
r� r r� rj r rf )r r r r
�
testRealClose s
z%FileObjectClassTestCase.testRealClosec C s d S)Nr )r r r r
�_testRealClose s z&FileObjectClassTestCase._testRealCloser� )&r r r r- r5 rE rF rG rL r� rU rO rR r$ r r rU rV rQ rT rW rX rY rZ r\ r] r` ra rb rc re rf rg rh ri rj rk rl r r r r
rC k sD
rC c @ s� e Z d Z d Z d Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z
d d � Z d d � Z d d � Z
d d � Z d S)�!UnbufferedFileObjectClassTestCasea� Repeat the tests from FileObjectClassTestCase with bufsize==0.
In this case (and in this case only), it should be possible to
create a file object, read a line from it, create another file
object, read another line from it, without loss of data in the
first file object's buffer. Note that http.client relies on this
when reading multiple requests from the same socket.r c C sh | j j � } | j | d | j � | j j d d � | _ | j j � } | j | d | j � d S)Ns A. r[ r s B. )rM r^ r� rR rj r] )r r_ r r r
�testUnbufferedReadline s
z8UnbufferedFileObjectClassTestCase.testUnbufferedReadlinec C s? | j j d | j � | j j d | j � | j j � d S)Ns A. s B. )rP r| rR rS )r r r r
�_testUnbufferedReadline# s z9UnbufferedFileObjectClassTestCase._testUnbufferedReadlinec C s\ | j j � | j j d � } | j | | j � | j j � | j t | j j d � d S)Ni )rj r
r� r� rU rM r� r )r r� r r r
�testMakefileClose( s
z3UnbufferedFileObjectClassTestCase.testMakefileClosec C s$ | j j | j � | j j � d S)N)rP r| rR rS )r r r r
�_testMakefileClose1 s z4UnbufferedFileObjectClassTestCase._testMakefileClosec C sI t j | j � } | j j � t j | j � } | j | d | � d S)NrB )r� r� rj rM r
r� )r Zrefcount_beforeZrefcount_afterr r r
� testMakefileCloseSocketDestroy5 s
z@UnbufferedFileObjectClassTestCase.testMakefileCloseSocketDestroyc C s d S)Nr )r r r r
� _testMakefileCloseSocketDestroy; s zAUnbufferedFileObjectClassTestCase._testMakefileCloseSocketDestroyc C sf | j j d � | j | j j t d � � d � | j | j j t | j � d � d � | j j
� | j j d � | j j t | j � d � } | d k r� t
j d � | j j t | j � d � } t d � } | j j | � } | j | d � | | d | � } | j | | j � | j | j j t d � � d � | j | j j d � d � d S)NFr r� g �?g �?r� rB )rj r� r� rM �readintor" r� r@ rU rH r@ rI rQ r( r) )r rV r rv r� r r r
�testSmallReadNonBlockingB s ")
"z:UnbufferedFileObjectClassTestCase.testSmallReadNonBlockingc C sQ | j j d � | j j | j � | j j � | j j � | j j d � d S)Ng �?g @) rH rQ rP r| rR rS rI r@ rJ )r r r r
�_testSmallReadNonBlockingU s
z;UnbufferedFileObjectClassTestCase._testSmallReadNonBlockingc C s- | j j d � | j r) | j | j � d S)Ng @)rK rQ � serv_skippedr4 )r r r r
�testWriteNonBlocking^ s z6UnbufferedFileObjectClassTestCase.testWriteNonBlockingc C s� d | _ | j j d � d t j } d } | j j | � } | j | d � xL t | � D]5 } | j j | � } | d k r� P| j | d � q[ Wd | _ d S)NFrO r r z$failed to saturate the socket buffer) rw rn r� r rT rP r| r� rq )r ZBIGZLIMITrv r r r r
�_testWriteNonBlockinge s
z7UnbufferedFileObjectClassTestCase._testWriteNonBlockingN)r r r r- r5 rn ro rp rq rr rs ru rv rx ry r r r r
rm s rm c @ s e Z d Z d Z d S)�#LineBufferedFileObjectClassTestCaserB N)r r r r5 r r r r
rz { s rz c @ s e Z d Z d Z d S)�$SmallBufferedFileObjectClassTestCaser� N)r r r r5 r r r r
r{ � s r{ c @ s= e Z d Z d Z d Z e j d � Z d Z e Z d Z
d S)�"UnicodeReadFileObjectClassTestCasez=Tests for socket.makefile() in text mode (rather than binary)r^ zutf-8ri r� N) r r r r- rL r� rd rU rO rR rG r r r r
r| � s r| c @ s= e Z d Z d Z d Z e Z d Z e j d � Z d Z
d S)�#UnicodeWriteFileObjectClassTestCasez=Tests for socket.makefile() in text mode (rather than binary)r[ rh zutf-8r� N) r r r r- rL r� rU rO rd rR rG r r r r
r} � s r} c @ sF e Z d Z d Z d Z e j d � Z d Z e j d � Z d Z
d S)�'UnicodeReadWriteFileObjectClassTestCasez=Tests for socket.makefile() in text mode (rather than binary)r^ zutf-8rh r� N) r r r r- rL r� rd rU rO rR rG r r r r
r~ � s r~ c @ s" e Z d Z d Z d d � Z d S)�NetworkConnectionTestzProve network connection.c C s+ t j t | j f � | _ | j | _ d S)N)r �create_connectionre r ra rn )r r r r
rU � s z!NetworkConnectionTest.clientSetUpN)r r r r- rU r r r r
r � s r c @ s e Z d Z d Z d S)�
BasicTCPTest2zLTests that NetworkConnection does not break existing TCP functionality.
N)r r r r- r r r r
r� � s r� c @ sb e Z d Z Gd d � d e j � Z e j d d � � Z d d � Z d d � Z d d
� Z
d S) �NetworkConnectionNoServerc @ s e Z d Z d d � Z d S)z$NetworkConnectionNoServer.MockSocketc G s t j d � � d S)Nz timed out)r rU )r r( r r r
rm � s z,NetworkConnectionNoServer.MockSocket.connectN)r r r rm r r r r
�
MockSocket� s r� c c s/ t j } | j t _ z d VWd | t _ Xd S)z*Return a socket which times out on connectN)r r� )r Z
old_socketr r r
�mocked_socket_module� s
z.NetworkConnectionNoServer.mocked_socket_modulec
C s} t j � } t j t j t j � } | j | j � | j t � � } | j t
| f � Wd QRX| j | j j
t
j � d S)N)r r' r r r r+ r
r� r rm re r� r� r� � ECONNREFUSED)r r ra r� r r r
� test_connect� s z&NetworkConnectionNoServer.test_connectc
C s} t j � } | j t � � } t j t | f � Wd QRXt j g } t t d � rc | j
t j � | j | j
j | � d S)N� ENETUNREACH)r r' r� r r r� re r� r� rW r~ r� r� r� )r r r� Zexpected_errnosr r r
�test_create_connection� s z0NetworkConnectionNoServer.test_create_connectionc C sE | j � �2 | j t j � � t j t d f � Wd QRXWd QRXd S)Ni� )r� r� r rU r� re )r r r r
� test_create_connection_timeout� s
z8NetworkConnectionNoServer.test_create_connection_timeoutN) r r r r r� �
contextlib�contextmanagerr� r� r� r� r r r r
r� � s
r� c @ s� e Z d Z d d d � Z d d � Z d d � Z d d � Z e Z d
d � Z e Z d d
� Z
e Z d d � Z e Z
d d � Z e Z d d � Z e Z d d � Z d S)� NetworkConnectionAttributesTestr_ c C s$ t j | d | �t j | � d S)Nr` )r r$ r: )r r` r r r
r$ � s z(NetworkConnectionAttributesTest.__init__c C s t j � | _ d S)N)r r' � source_port)r r r r
rU � s z+NetworkConnectionAttributesTest.clientSetUpc C s' | j j � d | _ t j | � d S)N)ra r
r: rV )r r r r
rV � s
z.NetworkConnectionAttributesTest.clientTearDownc C s# | j j � \ } } | j � d S)N)r ri r
)r rk rl r r r
� _justAccept� s z+NetworkConnectionAttributesTest._justAcceptc C sN t j t | j f d d �| _ | j | j j � | j | j j d � d S)NrU � r� ) r r� re r ra r+ r
r� r� )r r r r
� _testFamily� s !z+NetworkConnectionAttributesTest._testFamilyc C sg t j t | j f d d d d | j f �| _ | j | j j � | j | j j � d | j � d S)NrU r� Zsource_addressr� rB )
r r� re r r� ra r+ r
r� rf )r r r r
�_testSourceAddress s z2NetworkConnectionAttributesTest._testSourceAddressc C s� | j t j � d k � t j d � z2 t j t | j f � | _ | j | j j � Wd t j d � X| j
| j j � d � d S)Nr/ ) r� r r r r� re r ra r+ r
r� r
)r r r r
�_testTimeoutDefault s
z3NetworkConnectionAttributesTest._testTimeoutDefaultc C s� | j t j � d k � t j d � z8 t j t | j f d d �| _ | j | j j � Wd t j d � X| j
| j j � d � d S)Nr� rU ) r� r r r r� re r ra r+ r
r� r
)r r r r
�_testTimeoutNone s
!z0NetworkConnectionAttributesTest._testTimeoutNonec C s> t j t | j f d d �| _ | j | j j � d � d S)NrU r� )r r� re r ra r� r
)r r r r
�_testTimeoutValueNamed& s !z6NetworkConnectionAttributesTest._testTimeoutValueNamedc C sN t j t | j f d � | _ | j | j j � | j | j j � d � d S)Nr� ) r r� re r ra r+ r
r� r
)r r r r
�_testTimeoutValueNonamed+ s z8NetworkConnectionAttributesTest._testTimeoutValueNonamedN)r r r r$ rU rV r� Z
testFamilyr� ZtestSourceAddressr� ZtestTimeoutDefaultr� ZtestTimeoutNoner� ZtestTimeoutValueNamedr� ZtestTimeoutValueNonamedr� r r r r
r� � s r� c @ sa e Z d Z d d d � Z d d � Z d d � Z d d � Z e Z d
d � Z d d
� Z d S)� NetworkConnectionBehaviourTestr_ c C s$ t j | d | �t j | � d S)Nr` )r r$ r: )r r` r r r
r$ 3 s z'NetworkConnectionBehaviourTest.__init__c C s d S)Nr )r r r r
rU 7 s z*NetworkConnectionBehaviourTest.clientSetUpc C s' | j j � d | _ t j | � d S)N)ra r
r: rV )r r r r
rV : s
z-NetworkConnectionBehaviourTest.clientTearDownc C sC | j j � \ } } | j | j � t j d � | j d � d S)Nr� s done!)r ri r+ r
r( r) r0 )r rk rl r r r
�testInsideTimeout? s
z0NetworkConnectionBehaviourTest.testInsideTimeoutc C sB t j t | j f � | _ } | j d � } | j | d � d S)Nr! s done!)r r� re r ra r� r� )r r� r� r r r
�_testInsideTimeoutF s z1NetworkConnectionBehaviourTest._testInsideTimeoutc sH t j t | j f d d �| _ � | j t j � f d d � � d S)NrU rB c s
� j d � S)Nr! )r� r )r� r r
r� M s zDNetworkConnectionBehaviourTest._testOutsideTimeout.<locals>.<lambda>)r r� re r ra r� rU )r r )r� r
�_testOutsideTimeoutK s %z2NetworkConnectionBehaviourTest._testOutsideTimeoutN)
r r r r$ rU rV r� ZtestOutsideTimeoutr� r� r r r r
r� 0 s r� c @ sO e Z d Z d d � Z d d � Z e j e e d � d � d d � � Z d S)
�TCPTimeoutTestc s, � f d d � } � j t j | d � d S)Nc s! � j j d � � j j � d S)Ng �?)r r/ ri )r( r) )r r r
�
raise_timeoutS s z4TCPTimeoutTest.testTCPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (TCP))r� r rU )r r� r )r r
�testTCPTimeoutR s z TCPTimeoutTest.testTCPTimeoutc C s� d } y# | j j d � | j j � } WnM t j k
rM | j d � Yn, t k
rd d } Yn | j d � Yn X| s� | j d � d S)NFg z%caught timeout instead of error (TCP)Tz!caught unexpected exception (TCP)z3accept() returned success when we did not expect it)r r/ ri r rU r� r )r �ok�foor r r
�testTimeoutZeroY s
z TCPTimeoutTest.testTimeoutZerorJ ztest needs signal.alarm()c s? | j j d � Gd d � d t � � � f d d � } t j t j | � } z� y� t j d � z� y | j j � } Wnh t j k
r� | j d � YnT � k
r� YnC | j d t
j � d d � t j
� f � Yn X| j d � Wd t j d
� XWn � k
r"| j d � Yn XWd t j t j | � Xd S) Ng @c @ s e Z d Z d S)z4TCPTimeoutTest.testInterruptedTimeout.<locals>.AlarmN)r r r r r r r
�Alarmn s r� c s
� � d S)Nr )rP r� )r� r r
�
alarm_handlerp s z<TCPTimeoutTest.testInterruptedTimeout.<locals>.alarm_handlerr� z caught timeout instead of Alarmz3caught other exception instead of Alarm: %s(%s):
%sznothing caughtr zgot Alarm in wrong place)r r/ r rP rQ rJ ri r rU r� r� �exc_info� traceback�
format_exc)r r� rW r� r )r� r
�testInterruptedTimeoutg s(
*
z%TCPTimeoutTest.testInterruptedTimeoutN)
r r r r� r� rp r� rW rP r� r r r r
r� P s r� c @ s( e Z d Z d d � Z d d � Z d S)�UDPTimeoutTestc s, � f d d � } � j t j | d � d S)Nc s$ � j j d � � j j d � d S)Ng �?i )r r/ r� )r( r) )r r r
r� � s z4UDPTimeoutTest.testUDPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (UDP))r� r rU )r r� r )r r
�testUDPTimeout� s z UDPTimeoutTest.testUDPTimeoutc C s� d } y& | j j d � | j j d � } WnM t j k
rP | j d � Yn, t k
rg d } Yn | j d � Yn X| s� | j d � d S)NFg i z%caught timeout instead of error (UDP)Tz!caught unexpected exception (UDP)z1recv() returned success when we did not expect it)r r/ r� r rU r� r )r r� r� r r r
r� � s
z UDPTimeoutTest.testTimeoutZeroN)r r r r� r� r r r r
r� � s r� c @ s e Z d Z d d � Z d S)�TestExceptionsc C se | j t t t � � | j t t j t � � | j t t j t � � | j t t j t � � d S)N)r� �
issubclassr r r r� r� rU )r r r r
�testExceptionTree� s z TestExceptions.testExceptionTreeN)r r r r� r r r r
r� � s r� r zLinux specific testc @ sR e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
d � Z d S)
�TestLinuxAbstractNamespace�l c C s� d } t j t j t j � �� } | j | � | j � t j t j t j � �` } | j | j � � | j � d �3 } | j | j � | � | j | j � | � Wd QRXWd QRXWd QRXd S)Ns python-test-hello �r )
r r� r r2 r rm rf ri r� Z getpeername)r �address�s1�s2Zs3r r r
�testLinuxAbstractNamespace� s
z5TestLinuxAbstractNamespace.testLinuxAbstractNamespacec
C s^ d d | j d } t j t j t j � �* } | j | � | j | j � | � Wd QRXd S)Ns � hrB )�
UNIX_PATH_MAXr r� r r2 r� rf )r r� r r r r
� testMaxName� s
z&TestLinuxAbstractNamespace.testMaxNamec C sM d d | j } t j t j t j � � } | j t | j | � Wd QRXd S)N� �h)r� r r� r r� r r2 )r r� r r r r
�testNameOverflow� s z+TestLinuxAbstractNamespace.testNameOverflowc
C sQ t j t j t j � } z' | j d � | j | j � d � Wd | j � Xd S)Nz
python test s
python test )r r� r r2 r� rf r
)r r r r r
� testStrName� s
z&TestLinuxAbstractNamespace.testStrNamec
C sO t j t j t j � �0 } | j t d � � | j | j � d � Wd QRXd S)Ns
python test )r r� r r2 r" r� rf )r r r r r
�testBytearrayName� s z,TestLinuxAbstractNamespace.testBytearrayNameN) r r r r� r� r� r� r� r� r r r r
r� � s r� ztest needs socket.AF_UNIXc @ s| e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d S)�TestUnixDomainc C s t j t j t j � | _ d S)N)r r� r r� )r r r r
r � s zTestUnixDomain.setUpc C s | j j � d S)N)r� r
)r r r r
r � s zTestUnixDomain.tearDownc C sI y t j | � SWn1 t k
rD | j d j | t j � � � Yn Xd S)NzBPathname {0!a} cannot be represented in file system encoding {1!r})r� �fsencoderC r4 r� r� �getfilesystemencoding)r r� r r r
�encoded� s
zTestUnixDomain.encodedc C sk y | j | � WnS t k
rf } z3 t | � d k rQ | j d j | � � n � WYd d } ~ Xn Xd S)NzAF_UNIX path too longz6Pathname {0!a} is too long to serve as an AF_UNIX path)r2 r r� r4 r� )r r� r� r[ r r r
r2 � s
zTestUnixDomain.bindc C s | j | j j � d � d S)Nr� )Nr� )r� r� rf )r r r r
� testUnbound� s zTestUnixDomain.testUnboundc C sX t j j t j � } | j | j | � | j t j | � | j | j j
� | � d S)N) r� r� �abspathr �TESTFNr2 r� r+ r� r� rf )r r� r r r
� testStrAddr� s zTestUnixDomain.testStrAddrc C sa t j j t j � } | j | j | j | � � | j t j | � | j
| j j � | � d S)N) r� r� r� r r� r2 r� r� r+ r� r� rf )r r� r r r
�
testBytesAddr� s z TestUnixDomain.testBytesAddrc C ss t j j t j � } | j | � } | j | j | j d d � � | j t j
| � | j | j j � | � d S)N�ascii�surrogateescape)
r� r� r� r ZTESTFN_UNICODEr� r2 r� rd r+ r� r� rf )r r� rE r r r
�testSurrogateescapeBind s
z&TestUnixDomain.testSurrogateescapeBindc C st t j d k r | j d � t j j t j � } | j | j | � | j t j | � | j
| j j � | � d S)Nz!No unencodable filename available) r ZTESTFN_UNENCODABLEr4 r� r� r� r2 r� r+ r� r� rf )r r� r r r
�testUnencodableAddr
s
z"TestUnixDomain.testUnencodableAddrN) r r r r r r� r2 r� r� r� r� r� r r r r
r� � s r� c @ s� e Z d Z d Z d d d � Z d d � Z d d � Z d d
� Z e Z d d � Z e Z
d
d � Z d d � Z d d � Z
e Z d d � Z e Z d d � Z d d � Z d d � Z e Z d S) � BufferIOTestzF
Test the buffer versions of socket.recv() and socket.send().
r_ c C s t j | d | �d S)Nr` )rh r$ )r r` r r r
r$ s zBufferIOTest.__init__c C s} t j d d g t t � � } | j j | � } | j | t t � � | j � } | d t t � � } | j | t � d S)Nr% r )r& r@ r� rj r r� rg )r r rb r� r r r
�testRecvIntoArray s z BufferIOTest.testRecvIntoArrayc C s t t � } | j j | � d S)N)r� r� rn r0 )r r r r r
�_testRecvIntoArray' s z BufferIOTest._testRecvIntoArrayc C s^ t d � } | j j | � } | j | t t � � | d t t � � } | j | t � d S)Ni )r" rj r r� r@ r� )r r rb r� r r r
�testRecvIntoBytearray+ s
z"BufferIOTest.testRecvIntoBytearrayc C sd t d � } | j j t | � � } | j | t t � � | d t t � � } | j | t � d S)Ni )r" rj r rk r� r@ r� )r r rb r� r r r
�testRecvIntoMemoryview4 s
z#BufferIOTest.testRecvIntoMemoryviewc C s� t j d d g t t � � } | j j | � \ } } | j | t t � � | j � } | d t t � � } | j | t � d S)Nr% r )r& r@ r� rj r r� rg )r r rb rl r� r r r
�testRecvFromIntoArray= s z"BufferIOTest.testRecvFromIntoArrayc C s t t � } | j j | � d S)N)r� r� rn r0 )r r r r r
�_testRecvFromIntoArrayE s z#BufferIOTest._testRecvFromIntoArrayc C sd t d � } | j j | � \ } } | j | t t � � | d t t � � } | j | t � d S)Ni )r" rj r r� r@ r� )r r rb rl r� r r r
�testRecvFromIntoBytearrayI s
z&BufferIOTest.testRecvFromIntoBytearrayc C sj t d � } | j j t | � � \ } } | j | t t � � | d t t � � } | j | t � d S)Ni )r" rj r rk r� r@ r� )r r rb rl r� r r r
�testRecvFromIntoMemoryviewR s
z'BufferIOTest.testRecvFromIntoMemoryviewc C s, t d � } | j t | j j | d � d S)Nr1 i )r" r� r rj r )r r r r r
�testRecvFromIntoSmallBuffer[ s z(BufferIOTest.testRecvFromIntoSmallBufferc C s | j j t � d S)N)rn r0 r� )r r r r
� _testRecvFromIntoSmallBuffer` s z)BufferIOTest._testRecvFromIntoSmallBufferc C s0 t � } | j j | � | j j | d � d S)Nr )r" rj r )r r r r r
�testRecvFromIntoEmptyBufferc s z(BufferIOTest.testRecvFromIntoEmptyBufferN)r r r r- r$ r� r� r� Z_testRecvIntoBytearrayr� Z_testRecvIntoMemoryviewr� r� r� Z_testRecvFromIntoBytearrayr� Z_testRecvFromIntoMemoryviewr� r� r� Z _testRecvFromIntoEmptyBufferr r r r
r� s$ r� i� �� �� c C s{ t t d � s d Sy t d � } Wn t t t f k
rD d SYn X| �) x! | D] } | j d � rS d SqS WWd QRXd S)z�Check if the TIPC module is loaded
The TIPC module is not loaded automatically on Ubuntu and probably
other Linux distros.
�AF_TIPCFz
/proc/modulesztipc TN)rW r r� �FileNotFoundError�IsADirectoryError�PermissionErrorr )r� r_ r r r
�isTipcAvailableo s
r� z6TIPC module is not loaded, please 'sudo modprobe tipc'c @ s e Z d Z d d � Z d S)�TIPCTestc C s t j t j t j � } t j t j t j � } | j | j � | j | j � | j t j t j d � t j t t
t f } | j | � t j
t t
t t t
d � d f } | j t | � | j d � \ } } | j | j � | � | j | t � d S)NrB r� r i )r r� r� r+ r
r- r� r� �TIPC_ADDR_NAMESEQ�
TIPC_STYPE�
TIPC_LOWER�
TIPC_UPPERr2 �TIPC_ADDR_NAMEr� r� r� r� r� rf )r rv ra �srvaddrZsendaddrr� Zrecvaddrr r r
�testRDM� s
zTIPCTest.testRDMN)r r r r� r r r r
r� � s r� c @ sO e Z d Z d d d � Z d d � Z d d � Z d d � Z d
d � Z d S)
�TIPCThreadableTestr_ c C s' t j j | d | �t j | � d S)Nr` )rp rq r$ r: )r r` r r r
r$ � s zTIPCThreadableTest.__init__c C s� t j t j t j � | _ | j | j j � | j j t j t j d � t j t
t t f } | j j
| � | j j � | j � | j j � \ | _ | _ | j | j j � d S)NrB )r r� r rv r+ r
r- r� r� r� r� r� r� r2 r rA ri rk �connaddr)r r� r r r
r � s
zTIPCThreadableTest.setUpc C s� t j d � t j t j t j � | _ | j | j j � t j t t
t t t
d � d f } | j j
| � | j j � | _ d S)Ng�������?r� r )r( r) r r� r ra r+ r
r� r� r� r� r� rm rf �cliaddr)r rl r r r
rU � s
z TIPCThreadableTest.clientSetUpc C s< | j j d � } | j | t � | j | j | j � d S)Ni )rk r� r� r� r� r� )r r� r r r
�
testStream� s z TIPCThreadableTest.testStreamc C s! | j j t � | j j � d S)N)ra r0 r� r
)r r r r
� _testStream� s z TIPCThreadableTest._testStreamN)r r r r$ r rU r� r� r r r r
r� � s
r� c @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �ContextManagersTestc C s� t j � � } | j | j � Wd QRX| j | j � t j � � } | j � Wd QRX| j | j � t j � � } | j t | j d � Wd QRX| j | j � d S)Ns foo)r r* r� r� r
r� r rS )r r� r r r
�_testSocketClass� s z$ContextManagersTest._testSocketClassc C sE | j j � \ } } | j | j � | j d � } | j | � d S)Ni )r ri r+ r
r� rS )r rk rl r� r r r
�testCreateConnectionBase� s z,ContextManagersTest.testCreateConnectionBasec
C sr | j j � } t j | � �= } | j | j � | j d � | j | j d � d � Wd QRX| j | j � d S)Ns fooi )
r rf r r� r* r� rS r� r� r� )r r� r� r r r
�_testCreateConnectionBase� s
z-ContextManagersTest._testCreateConnectionBasec C sE | j j � \ } } | j | j � | j d � } | j | � d S)Ni )r ri r+ r
r� rS )r rk rl r� r r r
�testCreateConnectionClose� s z-ContextManagersTest.testCreateConnectionClosec C s\ | j j � } t j | � � } | j � Wd QRX| j | j � | j t | j d � d S)Ns foo)
r rf r r� r
r� r� r� r rS )r r� r� r r r
�_testCreateConnectionClose� s
z.ContextManagersTest._testCreateConnectionCloseN)r r r r� r� r� r� r� r r r r
r� � s
r� c @ s� e Z d Z e j e e d � d � e j d d d � d d � � � Z d d � Z
d
d � Z d d
� Z e j
e d k d � d d � � Z e j
e d k d � d d � � Z e j e e d � d � d d � � Z d S)�InheritanceTestr2 zSOCK_CLOEXEC not definedr� r7 r8 c
C sW t j t j t j t j B� �1 } | j | j t j @� | j | j � � Wd QRXd S)N)r r r r2 r� r� r* �get_inheritable)r r r r r
�test_SOCK_CLOEXEC� s z!InheritanceTest.test_SOCK_CLOEXECc
C s4 t j � } | � | j | j � d � Wd QRXd S)NF)r r� r� )r r� r r r
�test_default_inheritable� s z(InheritanceTest.test_default_inheritablec C sX t j � } | �? | j � } | j � | � | j | j � d � Wd QRXWd QRXd S)NF)r r� r
r� r� )r r� Znewsockr r r
�test_dup s
zInheritanceTest.test_dupc
C sd t j � } | �K | j d � | j | j � d � | j d � | j | j � d � Wd QRXd S)NTF)r �set_inheritabler� r� )r r� r r r
�test_set_inheritable s
z$InheritanceTest.test_set_inheritableNz
need fcntlc C s� t j � } | �v | j � } | j | j � d � t j | t j � } | t j M} t j | t j | � | j | j � d � Wd QRXd S)NFT)r r� r� r� �fcntl�F_GETFD�
FD_CLOEXECZF_SETFD)r r� r~ r= r r r
� test_get_inheritable_cloexec s z,InheritanceTest.test_get_inheritable_cloexecc C s� t j � } | �m | j � } | j t j | t j � t j @t j � | j d � | j t j | t j � t j @d � Wd QRXd S)NTr )r r� r� r� r� r� r� )r r� r~ r r r
� test_set_inheritable_cloexec s
z,InheritanceTest.test_set_inheritable_cloexecrr zneed socket.socketpair()c C sb t j � \ } } | j | j � | j | j � | j | j � d � | j | j � d � d S)NF)r rr r+ r
r� r� )r r� r� r r r
�test_socketpair, s
z InheritanceTest.test_socketpair)r r r rp r� rW r r rB r� r� r� r� r� r� r� r� r� r r r r
r� � s $ $$
r� r6 zSOCK_NONBLOCK not definedc @ sC e Z d Z d d d d � Z e j d d d � d d � � Z d
S) �NonblockConstantTestTg c C sg | r6 | j | j t j @� | j | j � | � n- | j | j t j @� | j | j � d � d S)N)r� r� r r6 r� r
r* )r r ZnonblockrU r r r
�
checkNonblock9 s
z"NonblockConstantTest.checkNonblockr� r7 r8 c & C s� t j t j t j t j B� �� } | j | � | j d � | j | d � | j d � | j | � | j d � | j | d � | j d � | j | d d �| j d � | j | d � Wd QRXt j � } t j d � t j � � } | j | � Wd QRXt j d � t j � � } | j | d � Wd QRXt j d � t j � � } | j | d d �Wd QRXt j d � t j � � } | j | d � Wd QRXt j | � d S)NrB Fr g @rU g ) r r r r6 r� r� r/ r r )r r �tr r r
�test_SOCK_NONBLOCKA s6
z'NonblockConstantTest.test_SOCK_NONBLOCKN)r r r r� r rB r� r r r r
r� 6 s r� r7 zWindows specificzneed multiprocessingc @ s^ e Z d Z e d d � � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
S)�TestSocketSharingc C s^ | j � } | j � } t j | � } | j � \ } } | j | � | j � | j � d S)N)rR r � fromshareri rS r
)r� �qZsdatar; r r� rV r r r
�remoteProcessServerh s
z%TestSocketSharing.remoteProcessServerc C s� t j � } t j d | j d | f � } | j � | j j | j � } | j j � } | j j � | j
| � d } | j
| � t j | � } g } x' | j
d � } | s� P| j | � q� W| j � d j | � } | j | | � | j � d S)Nr r( s
slapmahfror: r� )�multiprocessingrI ZProcessr r! r �share�pidrf r
rZ r r� r� r~ r� r� ) r r� r� r� rl r; r �m�receivedr r r
� testSharev s(
zTestSocketSharing.testSharec C sV | j j t j � � } | j t t j | d d � � | j t t j | d � d S)NrB s foor� )r r r� �getpidr� r r r� )r r� r r r
�testShareLength� s z!TestSocketSharing.testShareLengthc C s� | j | j � d � | j | j � | j � � | j | j | j � | j | j | j � | j d k r� | j | j | j � d S)Nr )r� r
r� r� r� )r Zorg�otherr r r
�compareSockets� s z TestSocketSharing.compareSocketsc
C sP | j j t j � � } t j | � } z | j | j | � Wd | j � Xd S)N)r r r� r r r� r
r
)r r� r r r r
�testShareLocal� s
z TestSocketSharing.testShareLocalc C s� t j t j g } t j t j g } x� | D]� } x� | D]� } y t j | | � } Wn t k
rk w8 Yn XzJ | j t j � � } t j | � } z | j
| | � Wd | j � XWd | j � Xq8 Wq+ Wd S)N) r r r� r r! r r r� r r� r
r
)r Zfamilies�typesr� r� �sourcer� Zsharedr r r
� testTypes� s
zTestSocketSharing.testTypesN)
r r r r� r r r r
r r r r r r
r� c s # r� c @ s~ e Z d Z d Z d? Z d Z d Z d Z e d d � � Z e d d
� � Z
d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z d/ d0 � Z d1 d2 � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z! d9 d: � Z" d; d< � Z# d= S)@�SendfileUsingSendTestz>
Test the send() implementation of socket.sendfile().
r i i r� r� c C s� d d � } d j d d � t | j � D� � } t t j d � �4 } x* | | j | j � D] } | j | � q\ WWd QRXt t j d � � } | j � | _ Wd QRXd S)Nc s s1 x | | k r! | V| | 8} q W| r- | Vd S)Nr )Ztotal�stepr r r
�chunks� s
z0SendfileUsingSendTest.setUpClass.<locals>.chunksr� c S s( g | ] } t j t j � j � � q Sr )�randomZchoice�stringZ
ascii_lettersr} )r� r r r r
r� � s z4SendfileUsingSendTest.setUpClass.<locals>.<listcomp>ri r[ )
r� rq �BUFSIZEr� r r� �FILESIZEr| r� �FILEDATA)r� r �chunkr� Zcsizer r r
�
setUpClass� s z SendfileUsingSendTest.setUpClassc C s t j t j � d S)N)r r� r� )r� r r r
�
tearDownClass� s z#SendfileUsingSendTest.tearDownClassc C sL | j j | j � | j j � \ } } | j | j � | j | j � | S)N)r r/ �TIMEOUTri r+ r
)r rk rl r r r
� accept_conn� s
z!SendfileUsingSendTest.accept_connc C s@ g } x* | j | j � } | s" P| j | � q Wd j | � S)Nr� )r� r r~ r� )r rk r r r r r
� recv_data� s z SendfileUsingSendTest.recv_datac C s
t | d � S)NZ_sendfile_use_send)rM )r r� r r r
�meth_from_sock� s z$SendfileUsingSendTest.meth_from_sockc C s� | j j � } t t j d � } t j | � �^ } | �N } | j | � } | | � } | j | | j � | j | j
� | j � Wd QRXWd QRXd S)Nr[ ) r rf r� r r� r r� r r� r �tell)r r� �filer� �meth�sentr r r
�_testRegularFile s z&SendfileUsingSendTest._testRegularFilec C sK | j � } | j | � } | j t | � | j � | j | | j � d S)N)r r r� r@ r r )r rk r� r r r
�testRegularFile s z%SendfileUsingSendTest.testRegularFilec C s� | j j � } t j | j � } t j | � �k } | �[ } | j | � } | j | | j � | j | j
� | j � | j t j | j
| � Wd QRXWd QRXd S)N)r rf �io�BytesIOr r r� r� r� r r r� r� r� )r r� r r� r! r r r
�_testNonRegularFile s z)SendfileUsingSendTest._testNonRegularFilec C sK | j � } | j | � } | j t | � | j � | j | | j � d S)N)r r r� r@ r r )r rk r� r r r
�testNonRegularFile s z(SendfileUsingSendTest.testNonRegularFilec C s� | j j � } t j d } t | d � � | j t j | � Wd QRXt | d � } t j | � �X } | �H } | j | � } | | � } | j
| d � | j
| j � d � Wd QRXWd QRXd S)N�2ri r[ r ) r rf r r� r� r+ r� r r� r r� r )r r� �filenamer r� r r! r r r
�_testEmptyFileSend& s
z(SendfileUsingSendTest._testEmptyFileSendc C s/ | j � } | j | � } | j | d � d S)Nr� )r r r� )r rk r� r r r
�testEmptyFileSend2 s z'SendfileUsingSendTest.testEmptyFileSendc C s� | j j � } t t j d � } t j | � �h } | �X } | j | � } | | d d �} | j | | j d � | j | j
� | j � Wd QRXWd QRXd S)Nr[ �offseti� ) r rf r� r r� r r� r r� r r )r r� r r� r r! r r r
� _testOffset9 s z!SendfileUsingSendTest._testOffsetc C sY | j � } | j | � } | j t | � | j d � | j | | j d d � � d S)Ni� )r r r� r@ r r )r rk r� r r r
�
testOffsetB s z SendfileUsingSendTest.testOffsetc C s� | j j � } t t j d � } t j | d d ��d } | �T } d } | j | � } | | d | �} | j | | � | j | j � | � Wd QRXWd QRXd S)Nr[ rU r� iGKL r5 )
r rf r� r r� r r� r r� r )r r� r r� r5 r r! r r r
�
_testCountJ s !z SendfileUsingSendTest._testCountc C sX d } | j � } | j | � } | j t | � | � | j | | j d | � � d S)NiGKL )r r r� r@ r )r r5 rk r� r r r
� testCountT s
z SendfileUsingSendTest.testCountc C s� | j j � } t t j d � } t j | d d ��d } | �T } d } | j | � } | | d | �} | j | | � | j | j � | � Wd QRXWd QRXd S)Nr[ rU r� rB r5 )
r rf r� r r� r r� r r� r )r r� r r� r5 r r! r r r
�_testCountSmall] s !z%SendfileUsingSendTest._testCountSmallc C sX d } | j � } | j | � } | j t | � | � | j | | j d | � � d S)NrB )r r r� r@ r )r r5 rk r� r r r
�testCountSmallg s
z$SendfileUsingSendTest.testCountSmallc C s� | j j � } t t j d � } t j | d d ��n } | �^ } d } | j | � } | | d d d | �} | j | | � | j | j � | d � Wd QRXWd QRXd S)Nr[ rU r� i�� r, i� r5 )
r rf r� r r� r r� r r� r )r r� r r� r5 r r! r r r
�_testCountWithOffsetp s !z*SendfileUsingSendTest._testCountWithOffsetc C s\ d } | j � } | j | � } | j t | � | � | j | | j d | d � � d S)Ni�� i� )r r r� r@ r )r r5 rk r� r r r
�testCountWithOffsetz s
z)SendfileUsingSendTest.testCountWithOffsetc C s� | j j � } t t j d � } t j | � �\ } | �L } | j d � | j | � } | j t
| | � | j t
| j | � Wd QRXWd QRXd S)Nr[ F) r rf r� r r� r r� r� r r� r r� )r r� r r� r r r r
�_testNonBlocking� s
z&SendfileUsingSendTest._testNonBlockingc C s, | j � } | j d � r( | j d � d S)Ni z$was not supposed to receive any data)r r� r� )r rk r r r
�testNonBlocking� s z%SendfileUsingSendTest.testNonBlockingc C s� | j j � } t t j d � } t j | d d ��E } | �5 } | j | � } | | � } | j | | j � Wd QRXWd QRXd S)Nr[ rU r� )
r rf r� r r� r r� r r� r )r r� r r� r r! r r r
�_testWithTimeout� s ! z&SendfileUsingSendTest._testWithTimeoutc C sK | j � } | j | � } | j t | � | j � | j | | j � d S)N)r r r� r@ r r )r rk r� r r r
�testWithTimeout� s z%SendfileUsingSendTest.testWithTimeoutc C sy | j j � } t t j d � } t j | d d ��< } | �, } | j | � } | j t j | | � Wd QRXWd QRXd S)Nr[ rU g{�G�z�?)
r rf r� r r� r r� r r� rU )r r� r r� r r r r
� _testWithTimeoutTriggeredSend� s z3SendfileUsingSendTest._testWithTimeoutTriggeredSendc C s | j � } | j d � d S)Ni�X )r r� )r rk r r r
� testWithTimeoutTriggeredSend� s z2SendfileUsingSendTest.testWithTimeoutTriggeredSendc C s d S)Nr )r r r r
� _test_errors� s z"SendfileUsingSendTest._test_errorsc 1 C sl t t j d � �K } t j d t j � �, } | j | � } | j t d | | � Wd QRXWd QRXt t j d � �B } t j � �, } | j | � } | j t d | | � Wd QRXWd QRXt t j d � �� } t j � �� } | j | � } | j t d | | d d �| j t d | | d d �| j t d | | d d
�| j t d | | d d �Wd QRXWd QRXd S)
Nr[ r� r rm z binary modezpositive integerr5 r( g�������?r rB r� ) r� r r� r r! r ro r rX )r r r r r r r
� test_errors� s* z!SendfileUsingSendTest.test_errorsNi ( i � )$r r r r- r r r r r� r r r r r r" r# r& r' r* r+ r- r. r/ r0 r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< r r r r
r � s@
r r� z%os.sendfile() required for this test.c @ s" e Z d Z d Z d d � Z d S)�SendfileUsingSendfileTestzB
Test the sendfile() implementation of socket.sendfile().
c C s
t | d � S)Nr� )rM )r r� r r r
r � s z(SendfileUsingSendfileTest.meth_from_sockN)r r r r- r r r r r
r= � s r= c C s: t t t t t t t t t g } | j t
t t t
t t t t t t t t t t g � | j t � | j t � | j t � | j t t g � | j t t g � | j t t! g � | j t" t# t$ t% t& t' t( t) t* t+ t, t- t. t/ t0 t1 t2 t3 t4 t5 t6 t7 t8 t9 t: g � t; j< � } t; j= | � t; j> | � d S)N)?r� r� r& r� r� r� r� r� r� �extendr1 rC rm rz r{ r| r} r~ r� r� r� r� r� r� r~ r+ r� r� r� r� r� r� r� r� rn r� r� r� r� r� r� r� r� r� r r r r r r r r
r r r r r� r r= r Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r
� test_main� sh
r@ �__main__)�rp Ztestr r� r$ r4 r r� r~ r( r� rH r� r� r& r r� �weakrefr rP rK rr r7 r r r r r� re r} r� �_threadrN r% r� r r ZHAVE_SOCKET_CANZHAVE_SOCKET_RDSrr r� rq r r r" r. r9 r: r^ rb rc rd rh ro rs rx ry r| r} r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r9 rA rD rY r^ ra rn r{ r� r� r� r� r� r� rA r� r� r� r� r� r� r� r r r r r r r r r r
r r r
rW r r r& r+ r1 rC rm rz r{ r| r} r~ r+ r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r= r@ r r r r r
�<module> s8
3k!
� � � 6 D [ Z z 7w w2�$ P;� i� �
! !& ! 2 " �l
; D 9%)%D S ' ,E+ e � ;