Your IP : 216.73.216.213
�
g f� � + @ 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 y d d l Z Wn e k
r.d Z Yn Xy d d l Z Wn e k
rYd 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- � Z. Gd d � d e) e- � Z/ Gd d � d e+ e- � Z0 Gd d � d e, e- � Z1 Gd d! � d! e. � Z2 Gd" d# � d# e j' e- � Z3 Gd$ d% � d% e j' � Z4 Gd& d' � d' e4 � Z5 Gd( d) � d) e* e4 e- � Z6 Gd* d+ � d+ e5 e6 � Z7 Gd, d- � d- e4 � Z8 Gd. d/ � d/ e8 � Z9 Gd0 d1 � d1 e4 � 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> � Z? d< d= � Z@ d> d? � ZA d@ dA � ZB GdB dC � dC e j' � ZC e jD e# dD � GdE dF � dF e j' � � ZE e jD e# dD � e jD e dG � GdH dI � dI e0 � � � ZF e jD e$ dJ � GdK dL � dL e j' � � ZG e jD e$ dJ � e jD e dG � GdM dN � dN e1 � � � ZH e jD e dG � GdO dP � dP e2 � � ZI e jD e dG � GdQ dR � dR e/ � � ZJ GdS dT � dT e* � ZK GdU dV � dV eK � ZL GdW dX � dX eK � ZM GdY dZ � dZ eK � ZN Gd[ d\ � d\ eK � ZO Gd] d^ � d^ eK � ZP Gd_ d` � d` eK � ZQ Gda db � db eQ � ZR Gdc dd � dd eR � ZS Gde df � df eR � ZT Gdg dh � dh eK � ZU Gdi dj � dj eU � ZV Gdk dl � dl eU � ZW Gdm dn � dn eL eU � ZX Gdo dp � dp e j' � ZY Gdq dr � dr eQ � ZZ Gds dt � dt eQ � Z[ Gdu dv � dv eM eO e6 e< � Z\ eA e j dw � e jD e dG � Gdx dy � dy eT e\ � � � Z] eA e j dz � e jD e dG � Gd{ d| � d| eW e\ � � � Z^ eA e j d} � e jD e dG � Gd~ d � d eX e\ � � � Z_ Gd� d� � d� eM eO e6 e? � Z` eA e j dw � e jD e ja d� � eB d� d� � e jD e dG � Gd� d� � d� eT e` � � � � � Zb eA e j dz � e jD e ja d� � eB d� d� � e jD e dG � Gd� d� � d� eW e` � � � � � Zc eA e j d} � e jD e ja d� � eB d� d� � e jD e dG � Gd� d� � d� eX e` � � � � � Zd eA e j dz � e jD e ja d� � eA e d� � eB d� d� � e jD e dG � Gd� d� � d� e[ e` � � � � � � Ze eA e j d} � e jD e ja d� � eA e d� � eB d� d� � e jD e dG � Gd� d� � d� eL e[ e` � � � � � � Zf Gd� d� � d� eP e7 e; � Zg eA e j dw � e jD e dG � Gd� d� � d� eS eg � � � Zh eA e j dz � e jD e dG � Gd� d� � d� eW eV eg � � � Zi eA e j d} � e jD e dG � Gd� d� � d� eX eV eg � � � Zj Gd� d� � d� eN eP e7 e= � Zk eA e j dw � eB d� d� d� � e jD e dG � Gd� d� � d� eS ek � � � � Zl eA e j dz � eB d� d� d� � e jD e dG � Gd� d� � d� eW eV ek � � � � Zm eA e j d} � eB d� d� d� � e jD e dG � Gd� d� � d� eX eV ek � � � � Zn Gd� d� � d� eP e7 e9 � Zo eA e j dw � eA e d� � e jD e dG � Gd� d� � d� eS eo � � � � Zp eA e j dz � eA e d� � e jD e dG � Gd� d� � d� eW eV eo � � � � Zq eA e j d} � eA e d� � e jD e dG � Gd� d� � d� eX eV eo � � � � Zr eA e j dw dz � eA e d� d� d� � e jD e dG � Gd� d� � d� eZ eo � � � � Zs eA e j dw d} � eA e d� d� d� � e jD e dG � Gd� d� � d� eL eZ eo � � � � Zt Gd� d� � d� e j' � Zu eA e d� � e jD ev e d� � pev e d� � d� � Gd� d� � d� eu e< � � � Zw eA e d� � e jD ev e d� � pTev e d� � d� � e jD e dG � Gd� d� � d� eu e* e5 e; � � � � Zx e jD e dG � Gd� d� � d� e. � � Zy e jD ev e d� � d� � e jD e dG � Gd� d� � d� e3 � � � Zz e jD e dG � Gd� d� � d� e. � � Z{ e jD e dG � Gd� d� � d� e2 � � Z| Gd� d� � d� e j' � 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� eI � Z� Gd� d� � d� e j' � Z� e jD e dG � Gd� d� � d� e( e- � � Z� e jD 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 jD e j d� k d� � Gd� d� � d� e j' � � Z� e jD ev e d� � d� � Gd� d� � d� e j' � � Z� e jD e dG � Gd� d� � d� e2 � � Z� d� Z� d� Z� d� Z� d� d� � Z� e jD e� � d� � Gd� d� � d� e j' � � Z� e jD e� � d� � Gd� d� � d� e j' e- � � Z� e jD e dG � Gd� d� � d� e. � � Z� Gd� d� � d� e j' � Z� e jD ev e d� � d� � Gd� d� � d� e j' � � Z� e jD e
j� d� k d � e jD e d� Gdd� de( � � � Z� dd� Z� e� dk r�e� � n 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/python34/lib64/python3.4/test/test_socket.py�_have_socket_can/ 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_rds9 s
r �ic @ s( e Z d Z d d � Z d d � Z d S)�
SocketTCPTestc C sD t j t j t j � | _ t j | j � | _ | j j d � d S)N� )r �AF_INET� SOCK_STREAM�servr � bind_port�port�listen)�selfr r r
�setUpL s zSocketTCPTest.setUpc C s | j j � d | _ d S)N)r r
)r r r r
�tearDownQ s
zSocketTCPTest.tearDownN)�__name__�
__module__� __qualname__r r r r r r
r J 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 W 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! U s r! c sU e Z d Z d Z e rQ � f d d � Z � f d d � Z � f d d � Z n � 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% g s z"ThreadSafeCleanupTestCase.__init__c
s' | j � t � j | | � SWd QXd S)N)r( r$ �
addCleanup)r r) r* )r+ r r
r, k s
z$ThreadSafeCleanupTestCase.addCleanupc
s' | j � t � j | | � SWd QXd S)N)r( r$ �
doCleanups)r r) r* )r+ r r
r- o s
z$ThreadSafeCleanupTestCase.doCleanups)r r r �__doc__r& r% r, r- r r )r+ r
r# _ 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. r4 �bufsize�
can_frame_fmt�structZcalcsizeZcan_frame_size�bcm_cmd_msg_fmtr r r r r
r/ s 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 r5 )r r r r
r � s
zSocketRDSTest.setUpN)r r r r. r6 r r r r r
r: � s r: 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 preceeded 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)Nr F�.�_T)r& �Eventr@ � client_ready�done�queue�Queue�server_crashed�id�rfind�getattr�thread�start_new_thread� clientRunZ
client_threadr< rA �wait)r Z
methodnamer Z test_methodr r r
r> � s$ zThreadableTest._setUpc C sB | j � | j j � | j j � r> | j j � } | � n 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 � sY t d � � n 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 rA 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 rA rN �exit)r r r r
rV s
z ThreadableTest.clientTearDownN) r r r r. r% rB 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% 1 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 5 s z!ThreadedUDPSocketTest.clientSetUpc C s' | j j � d | _ t j | � d S)N)ra r
r; rV )r r r r
rV 8 s
z$ThreadedUDPSocketTest.clientTearDownN)r r r r% rU rV r r r r
rb / 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% ? 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 r3 r4 r )r r r r
rU C s
!
z!ThreadedCANSocketTest.clientSetUpc C s' | j j � d | _ t j | � d S)N)ra r
r; rV )r r r r
rV L 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` )r: r% r; )r r` r r r
r% S 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 r3 �HOST� getsockname�cli_addrr )r r r r
rU W s
z!ThreadedRDSSocketTest.clientSetUpc C s' | j j � d | _ t j | � d S)N)ra r
r; rV )r r r r
rV b s
z$ThreadedRDSSocketTest.clientTearDownN)r r r r% rU rV r r r r
rd Q 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% n s z SocketConnectedTest.__init__c C s9 t j | � | j � | j j � \ } } | | _ d S)N)r^ r rB r �accept�cli_conn)r �conn�addrr r r
r q s
zSocketConnectedTest.setUpc C s' | j j � d | _ t j | � d S)N)rj r
r^ r )r r r r
r y 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 g 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 � d S)Nr )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 rB 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 r3 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)Nr )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 1 s zUDPTestBase.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)�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 7 s zSCTPStreamBase.newSocketN)r r r r. rt r r r r
r� 4 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 D s zUDP6TestBase.newSocketN)r r r r. rt r r r r
r� A 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_passS s z%skipWithClientIf.<locals>.client_passc s@ t j � � | � } t | t � s< � f d d � | _ n | S)Nc s � S)Nr )�f)r� r r
�<lambda>X s z3skipWithClientIf.<locals>.skipdec.<locals>.<lambda>)rp �skip�
isinstance�type� client_skip)�objZretval)r� �reasonr r
�skipdecU s z!skipWithClientIf.<locals>.skipdecc S s4 t | t � p t | d � s0 d d � | _ n | 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
� noskipdecZ s z#skipWithClientIf.<locals>.noskipdecr )Z conditionr� r� r� r )r� r� r
�skipWithClientIfJ 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>f s z requireAttrs.<locals>.<listcomp>z don't have z, c s s | ] } | Vq d S)Nr )r� r� r r r
� <genexpr>h s z requireAttrs.<locals>.<genexpr>)r� �join)r� Z
attributes�missingr )r� r
� requireAttrsa 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� s s z!requireSocket.<locals>.<listcomp>z don't have z, c s s | ] } | Vq d S)Nr )r� r� r r r
r� v 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� x 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
�
requireSocketk 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 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 � Z/ d[ d\ � Z0 d] d^ � Z1 d_ d` � Z2 da db � Z3 dc dd � Z4 de df � Z5 e+ j6 dg dh � � Z7 e j e+ j8 di � dj dk � � Z9 dl dm � Z: e j; e& j' dK k dn � do dp � � Z< d S)q�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 QX| 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� assertNotInr3 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 QX| j t d | d �� t j � Wd QX| j t d | d �� t j � Wd QXd 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 B C s� t j t j t j � } | j | j � | j d � | j � } | j t � � } | j d | � Wd QX| j
t | j � d � | j t � � } | j d | � Wd QX| j
t | j � d � | j t � � } | j d d � Wd QX| j
d t | j � � | j t � � } | j d d | � Wd QX| j
t | j � d � | j t � � } | j d d | � Wd QX| j
t | j � d � | j t � � } | j d d d � Wd QX| j
d t | j � � | j t � � } | j d d | � Wd QX| j
d
t | j � � | j t � � } | j d d d � Wd QX| j
d
t | j � � | j t � � } | j d � Wd QX| j
d t | j � � | j t � � } | j d d | d � Wd QX| j
d
t | j � � d S)N� r u ☠z+'str' does not support the buffer interfacey @z/'complex' does not support the buffer interfaces fooz not NoneType�barzan integer is requiredz (1 given)� z (4 given))r� r )r r r" r, r
r3 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 � n d S)Nzname lookup failurerC r zError resolving host to ip.z=Error testing host resolution mechanisms. (fqdn: %s, all: %s)) r � gethostname�
gethostbynamer r5 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 �errnoZEPERMr5 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
�testInterfaceNameIndex1 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
� testInvalidInterfaceNameIndexA s z0GeneralModuleTests.testInvalidInterfaceNameIndex� getrefcountz test needs sys.getrefcount()c C sa y# t j t � } t j t d � Wn7 t k
r\ t j t � | k rX | j d � n Yn Xd S)Nr z$socket.getnameinfo loses a reference)�sysr� r r � getnameinforX r� )r Zorigr r r
�testRefCountGetNameInfoK s
z*GeneralModuleTests.testRefCountGetNameInfoc C s- y t j d d � Wn t k
r( Yn Xd S)Nr1 r )r1 r r r )r r� r )r r r r
�testInterpreterCrashW s
z'GeneralModuleTests.testInterpreterCrashc C s� i d t j 6d t j 6d t j 6d t j 6} x� | j � D]� \ } } d | >d } x6 d D]. } | j | | @| | | | @� � | @� q\ W| | � } | j | | @| � | j t | d � q; Wd S)
N� � r r � r� �gE#�2Tv�" � ��)r r r� r� r� r� r� l ) r �htonl�ntohl�htons�ntohs�itemsr� r� �
OverflowError)r Zsizes�func�size�maskr Zswappedr r r
�testNtoH_ 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)
Nr r� � ���������r� r� r� )r r� r� r� r� r� r� )r Z good_valuesZ
bad_values�kr r r
�testNtoHErrorsm s
z!GeneralModuleTests.testNtoHErrorsc C s] | 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 � | � n | j t t j d � | j t t j d
� d S)N�freebsd�netbsd� gnukfreebsd�linux�darwin�daytime�qotd�domain�echoZtcpZudpr i )r� r� r )zlinuxzdarwin)r r �domain)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�
r 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 sL y6 d d l m � m � m } | s5 � j d � n Wn t k
rW � j d � Yn Xt j d k r� y � � d � Wq� t k
r� } z$ | j d k r� � j d � n WYd d } ~ Xq� Xn � � 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 Br2 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 r5 � 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 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
� � | | d � | | d � | | d � � j d | d � � � j d | d � � � j d
| d � � | | d � | | d � | | 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� )r r� r r r )r r r r
�testStringToIPv4# s
z#GeneralModuleTests.testStringToIPv4c sP y6 d d l m � m � m } | s5 � j d � n Wn t k
rW � j d � Yn Xt j d k r� y � � d � Wq� t k
r� } z$ | j d k r� � j d � n WYd d } ~ Xq� Xn � � f d
d � � � � f d d � } � j
d
� d � � � j
d � d � � � j
d � 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� L s z5GeneralModuleTests.testStringToIPv6.<locals>.<lambda>c s � j t t f � | � S)N)r� r r )r )r� r r r
r� M s z::z::1� s z!aef:b01:506:1001:ffff:9997:55:170s
� ���� Ups � r� s s s s s s s ) r r r� r r5 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 invalidr )r �find_unused_portr r r r, r
r3 rf r� r� r r5 r� r� )r r r� r� Z
my_ip_addrr r r
� testSockName^ 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
�testGetSockOptp 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)Nr r zfailed to set reuse mode)
r r r r, r
�
setsockoptr� r� r) r* )r r� r+ r r r
�testSetSockOptx s
z!GeneralModuleTests.testSetSockOptc C sI t j t j t j � } | j d � | j � | j t | j d � d S)Nr 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]v } t j � } y | j t | f � WnI t k
r� } z) | j t j
k s� | d k r� � n WYd d } ~ Xq� XPq� Wd S)Ni r" )r r, r
r r' r� r� r3 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_VALSr �d r� )r 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 s� y t j d d � WnM t j k
rc } z* | j t j k rN | j d � n � WYd d } ~ Xn Xx0 t j t d � D] } | j t | � d � qw Wt j d d � t j d d � t j d d � t j
r� t j d d � n 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 � q8Wt 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 � ry# t j d d d
d
d
t j � Wqt j k
r{YqXn d 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_SERVICEr5 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 sc t j d � � t j d � Wd QXd } 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 sQ t t d � s t t d � r0 � j d � n � f d d � } � f d d � } t j � \ } } t j t j | � } z� | r� | j d � n � j t � �& t j d � | j
d
t j � Wd QX| rt j t j | � t j d � � j t j
| j
d
t j � n 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 r )r� r rK rL )r) )r r r
�raising_handler s zEGeneralModuleTests.check_sendall_interrupted.<locals>.raising_handlerg �?r � xr )rW �signalr r5 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_interrupted4 s z+GeneralModuleTests.test_sendall_interruptedc C s | j d � d S)NT)rX )r r r r
�%test_sendall_interrupted_with_timeout7 s z8GeneralModuleTests.test_sendall_interrupted_with_timeoutc C s� t j t j t j � } t | � } | j t � � } d } t j � Wd QX| j | t | j
j d � � t j t j t j � } | j d � } t | � } d } t j � | j t � � d } t j � Wd QXd 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 sT t j t j t j � �6 } | j d � } | j � | j t | � d � Wd QXd S)Nr[ z <_io.BufferedReader name=-1>)r r r r] r
r� r� )r r� �fpr r r
�test_name_closed_socketioK 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 QXd S)Nr[ � bufferingr )
r r] r� �readabler* �writable�seekabler
r� r )r r� r` r r r
� test_unusable_closed_socketioQ s
z0GeneralModuleTests.test_unusable_closed_socketioc
C s� t j � } | �< x4 t t j d � D] } | j t t j | | � q' WWd QXx} t t j d � D]h } t j t j t j | � � } | j | t j � t j t j t j
| � � } | j | t j
� qd Wd S)Nr ) r �range�pickleZHIGHEST_PROTOCOLr� rX �dumps�loadsr r� r )r r� Zprotocolr� r� r r r
� test_pickle\ s # z GeneralModuleTests.test_picklec C sW xP d D]H } t j t j t j � } | j t d f � | j | � | j � q Wd S)Nr r r� )r r� )r r r r3 re r r
)r Zbacklog�srvr r r
�test_listen_backlogg 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 r ) � _testcapir r r r3 re r� r� r �INT_MAXr
)r rn rl r r r
� test_listen_backlog_overflown s
z/GeneralModuleTests.test_listen_backlog_overflowz IPv6 required for this test.c
C sl | 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 QXd S)Nr l �� r i����) r� r� r r� r r� r� r r3 )r r r r r
�
test_flowinfow s z GeneralModuleTests.test_flowinfoc
C sW t j t j t j � �9 } | j t | j � d � | j t | j � d � Wd QXd 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 sc t j � \ } } t j d d d d d | � �- } | j | j d � | j | j d � Wd QXd S)Nr� i�� r� i4 r� )r~ �mkstempr r� r� r� )r �fdrD r r r r
� test_uknown_socket_family_repr� s
!z1GeneralModuleTests.test_uknown_socket_family_repr)=r r r r� r� rp �
skipUnlessr� r� r� r� r� r� r� r� rW r r� r� r� r� r� r� r� r� r r r r r r# r& r( r, r. r1 r3 r6 r� r� r; rF rH r Zis_resource_enabledrI rX rY rZ r_ ra rf rk rm � cpython_onlyrp rA rq rr �skipIfru r r r r
r� � sn $
, * ' =" ' 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 QXd 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 QXd S)N)r r r" r{ )r r r r r
�testCreateBCMSocket� s !z BasicCANTest.testCreateBCMSocketc C s8 t j t j t j t j � � } | j d � Wd QXd S)Nr� )r� )r r r r r3 )r r r r r
� testBindAny� s !zBasicCANTest.testBindAnyc
C sK t j t j t j t j � �' } | j t d | j d d f � Wd QXd S)Nzinterface name too longr1 i )r r r r ZassertRaisesRegexr r3 )r r r r r
�testTooLongInterfaceName� s ! z%BasicCANTest.testTooLongInterfaceName�CAN_RAW_LOOPBACKz/socket.CAN_RAW_LOOPBACK required for this test.c
C sw 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 QXd S)Nr r )r r ) 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 � �E } | j t j t j | � | j | | j
t j t j d � � Wd QXd S)N� � z=IIr2 )r� r� ) r8 �packr 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 rv rW r r} r~ r r� r� r� r� r r r r
ry � s ry 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.r2 s )r@ �ljustr8 r� r7 )�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)r8 Zunpackr7 )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 r )r �recvfromr6 r� �cfr4 r rz )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� r6 r� r� )r r� rl r r r
�testSendMaxFrame� s zCANTest.testSendMaxFramec C s, | j d d � | _ | j j | j � d S)Nr s r2 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� r6 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
�_testSendMultiFrames 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� r6 r� r� r� r� r� )r r� rl r� r� r� r r r
�_testBCM 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" r )r r r" r{ r, r
rm r4 r� �bytesr� r� r� r| r8 r� r9 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
�testBCM s2
zCANTest.testBCMN)r r r r% � classmethodr� r� r� r� r� r� r� r� rp rv 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� 8 s z!BasicRDSTest.testCrucialConstantsc C s( t j t j t j d � � } Wd QXd S)Nr )r r r )r r r r r
r~ < s z BasicRDSTest.testCreateSocketc C s` d } t j t j t j d � �9 } | j t j t j | � | j t j t j | � Wd QXd S)Ni @ r )r r r r- r� � SO_RCVBUFZ SO_SNDBUF)r r6 r r r r
�testSocketBufferSize@ s z!BasicRDSTest.testSocketBufferSizeN)r r r r� r~ r� r r r r
r� 5 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% K s zRDSTest.__init__c s t � j � t j � | _ d S)N)r$ r r& rE �evt)r )r+ r r
r N s
z
RDSTest.setUpc C sE | j j | j � \ } } | j | j | � | j | j | � d S)N)r r� r6 r� r� rg )r r� rl r r r
�testSendAndRecvR 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
�_testSendAndRecvW s zRDSTest._testSendAndRecvc C sf | j j | j t j � \ } } | j | j | � | j j | j � \ } } | j | j | � d S)N)r r� r6 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
� _testPeeka s zRDSTest._testPeek�recvmsgc C s8 | j j | j � \ } } } } | j | j | � d S)N)r r� r6 r� r� )r r� �ancdataZ msg_flagsrl r r r
�testSendAndRecvMsge 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
�_testSendAndRecvMsgj s
zRDSTest._testSendAndRecvMsgc C s` | j j | j � \ } } | j | j | � | j j | j � \ } } | j | j | � d S)N)r r� r6 r� �data1�data2)r r� rl r r r
�testSendAndRecvMultio 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
�_testSendAndRecvMultiv 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� r6 r� r� )r r^ �wr1 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 QX| 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� rA r� r� r� ZENOBUFSr� r r� )r r� r^ r� r1 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� G 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 sK d } x* | j j d � } | s% Pn | | 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 rt 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 r r� ) rn rn r0 r� r� r� r� ro �UINT_MAX)r rn 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
�
testDetach s
zBasicTCPTest.testDetachc C s | j j t � d S)N)rn r0 r� )r r r r
� _testDetach s zBasicTCPTest._testDetachN)r r r r% r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r rw 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% ! 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
�testSendtoAndRecv$ 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
�_testSendtoAndRecv) 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� , 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� 1 s zBasicUDPTest._testRecvFromc C s | j t | j j d � d S)Nr r� )r� r r r� )r r r r
�testRecvFromNegative4 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
�_testRecvFromNegative8 s z"BasicUDPTest._testRecvFromNegativeN)
r r r r% r� r� r� r� r� r� r r r r
r� 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
� sendToServera s z SendrecvmsgBase.sendToServerc G s' | j j | | j t | � d � � S)N)r� r� �sendmsg_to_server_defaultsr@ )r r) r r r
�sendmsgToServeri s z SendrecvmsgBase.sendmsgToServerc G s# | j | | � } | j | � | S)N)r� �registerRecvmsgResult)r r� r6 r) �resultr r r
� doRecvmsgq 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} n | | M} | | M} | | O} | | O} | | @| @} | r� t d j | � � � n | | 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)z
MSG_CTRUNCr� )r r r � fail_timeoutr r� r� r� r� r� r� r� r� rM r r� r� r� r� r r )r+ r
r� V 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 r )� bytearray� recvmsg_intor� �assertGreaterEqual�assertLessEqualr� )r r� r6 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
s z"SendrecvmsgConnectedBase.serv_sockc C s | j S)N)rn )r r r r
r� s z!SendrecvmsgConnectedBase.cli_sockc C s d S)Nr )r r� r� r r r
r� 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 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
� testSendmsg s zSendmsgTests.testSendmsgc C s& | j | j t g � t t � � d S)N)r� r� r� r@ )r r r r
� _testSendmsg" s zSendmsgTests._testSendmsgc C s& | j | j j t t � � t � d S)N)r� r
r� r@ r� )r r r r
�testSendmsgDataGenerator% 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� * s z9SendmsgTests._testSendmsgDataGenerator.<locals>.<genexpr>)r� r� r� r@ )r r r r
�_testSendmsgDataGenerator) s "z&SendmsgTests._testSendmsgDataGeneratorc C s& | j | j j t t � � t � d S)N)r� r
r� r@ r� )r r r r
� testSendmsgAncillaryGenerator- 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� 2 s z>SendmsgTests._testSendmsgAncillaryGenerator.<locals>.<genexpr>)r� r� r� r@ )r r r r
� _testSendmsgAncillaryGenerator1 s %z+SendmsgTests._testSendmsgAncillaryGeneratorc C s& | j | j j t t � � t � d S)N)r� r
r� r@ r� )r r r r
�testSendmsgArray5 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
�_testSendmsgArray9 s !z SendmsgTests._testSendmsgArrayc C s& | j | j j t t � � t � d S)N)r� r
r� r@ r� )r r r r
�testSendmsgGather= 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
�_testSendmsgGatherA s z SendmsgTests._testSendmsgGatherc C s | j | j j d � d � d S)Ni� s done)r� r
r� )r r r r
�testSendmsgBadArgsD 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
�_testSendmsgBadArgsH s
z SendmsgTests._testSendmsgBadArgsc C s | j | j j d � d � d S)Ni� s done)r� r
r� )r r r r
�testSendmsgBadCmsgZ 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
�testSendmsgBadMultiCmsgm 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
�_testSendmsgBadMultiCmsgs 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 sg t t d � sV | j t � � } | j t g d d g � Wd QX| j | j j � n | j d � d S)Nr! 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 r r� r r" r� r# r$ r& r' r( r r r r
r
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] zH | j j d � | j t j � � x | j d d g � q) WWd QXWd | j j � Xd S)Ng� ��Q��?r, i )r� r/ r� r rU r� r� rA )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 sw zb | j t � �+ } x! | j d d g g t j � q WWd QX| 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� rA )r r� r r r
�_testSendmsgDontWait� s 'z'SendmsgStreamTests._testSendmsgDontWaitN> �linux)
r r 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 � 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 r3 r4 r r r r
r2 � s r2 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._testRecvmsgShorterr2 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
�_testRecvmsgTrunc s z%RecvmsgGenericTests._testRecvmsgTruncc C st | 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
� testRecvmsgShortAncillaryBuf 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
�testRecvmsgLongAncillaryBuf s z/RecvmsgGenericTests.testRecvmsgLongAncillaryBufc C s | j t � d S)N)r� r� )r r r r
� _testRecvmsgLongAncillaryBuf s z0RecvmsgGenericTests._testRecvmsgLongAncillaryBufc C s- | j j � | j t | j | j d � d S)Ni )r
r
r� r r� )r r r r
�testRecvmsgAfterClose# s
z)RecvmsgGenericTests.testRecvmsgAfterClosec C s d S)Nr )r r r r
�_testRecvmsgAfterClose( 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� rA )r r r r
�testRecvmsgTimeout+ s
z&RecvmsgGenericTests.testRecvmsgTimeoutc C s# | j | j j d | j � � d S)NrU )r� r� rQ r� )r r r r
�_testRecvmsgTimeout4 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
�testRecvmsgPeek7 s( 'z#RecvmsgGenericTests.testRecvmsgPeekc C s | j t � d S)N)r� r� )r r r r
�_testRecvmsgPeekZ 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
�_testRecvmsgFromSendmsgi s z+RecvmsgGenericTests._testRecvmsgFromSendmsgN) r r r r6 r7 r8 r9 r: r; r Zrequires_freebsd_versionr< r= r> r? r@ rA rB rC rD rE r� r rF r� rG rH rI r r r r
r5 � s(
# r5 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
�testRecvmsgEOFr 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_skipInRpmBuildrK rL rM rN r r r r
rJ n s rJ 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)Nr 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 rP rQ r r r r
rO � s rO 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 writabler 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 rS 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 rS 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 Zb1Zb2Zb3rS 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 rT rU rV rW rY rZ r] r^ r r r r
rR � s
rR 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 r i r r� r� )�socklen_t_limitr r` �listrg r� r �itemsizer� r r� r� r� �maxsize)r �toobig�values�n�retr r r
� testCMSG_LEN
s ))
zCmsgMacroTests.testCMSG_LENr! 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)Nr i r r r� r� )ra r r! rb rg r� r rc r r` r r� r� r� rd )r re rf �lastrg rh r r r
�testCMSG_SPACE!
s )
z CmsgMacroTests.testCMSG_SPACE) r r r rn r ra �minro r� r ri rk r r r r
r_ � s
r_ c @ s� e Z d Z dd 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 d/ d0 � Z d1 d2 � Z d3 d4 � Z! d5 d6 � Z" e e d � d7 d8 � � Z# e# j d9 d: � � Z$ d
d; d< � Z% d= d> � Z& d? d@ � Z' dA dB � Z( dC dD � Z) dE dF � Z* dG dH � Z+ dI dJ � Z, dK dL � Z- dM dN � Z. dO dP � Z/ d
dQ dR � Z0 dS dT � Z1 dU dV � Z2 dW dX � Z3 dY dZ � Z4 d[ d\ � Z5 d] d^ � Z6 d_ d` � Z7 da db � Z8 dc S)e�
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) rg r~ rs r, r� r� r
�writer� �encode�append)r rg �fdsr rt r� r r r
�newFDsB
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� ro )r rq rg rt r r r
�checkFDsN
s zSCMRightsTest.checkFDsc C s | j | j | � d S)N)r, �closeRecvmsgFDs)r r� r r r
r� U
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 q Wd S)Nr r ) r r� �
SCM_RIGHTSr � frombytesr@ rc r� r
)r Zrecvmsg_result�
cmsg_level� cmsg_type� cmsg_datarq rt r r r
rw X
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� rx r rr r@ )r rg r r r
�createAndSendFDsd
s !z SCMRightsTest.createAndSendFDsr 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� rb r r@ r �tupler� rx r� �
SIZEOF_INTry rv )r Znumfdsr� �maxcmsgs� ignoreflagsr� r� r= rl rq �itemrz r{ r| r r r
�checkRecvmsgFDsn
s$
z SCMRightsTest.checkRecvmsgFDsc C s, | j d | j | j t t � d � � d S)Nr 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 r )
r� r� r� r r� rx r rr rX 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._testMultipleFDPassr! c C s9 | j d | j | j t t � t j d t � � � d S)Nr� )r� r� r
r@ r� r r! 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)Nr r� r� ) r� r� r
r@ r� r r` r r� )r r r r
�testFDPassCMSG_LEN�
s z SCMRightsTest.testFDPassCMSG_LENc C s | j d � d S)Nr )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 ) rr r� r� r� r r� rx 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 r! r r` 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 ) rr r� r� r� r r� rx 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� rS r[ r r r
�sendAncillaryIfPossible�
s %z%SCMRightsTest.sendAncillaryIfPossiblec 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� rx )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� r )r� r
r@ r� r� r� rg r� r r� r r� rx �
assertLessr )r r� r� r= rl rz 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 r r� )r� r� r r� rx r �badfdrX )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� rx ry rc rv ) r r� r� r= rl rq rz r{ r| r r r
� testFDPassPartialIntInMiddle 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 r r� ) rr r� r� r r� rx r r� rX )r r� r� r r r
� _testFDPassPartialIntInMiddle' 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
�checkTruncatedHeader3 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
�testCmsgTruncNoBufSize= s !z$SCMRightsTest.testCmsgTruncNoBufSizec C s | j d � d S)Nr )r} )r r r r
�_testCmsgTruncNoBufSizeF 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
�testCmsgTrunc0I s $z SCMRightsTest.testCmsgTrunc0c C s | j d � d S)Nr )r} )r r r r
�_testCmsgTrunc0N s z SCMRightsTest._testCmsgTrunc0c C s) | j | j | j t t � d � � d S)Nr )r� r� r
r@ r� )r r r r
�testCmsgTrunc1T s z SCMRightsTest.testCmsgTrunc1c C s | j d � d S)Nr )r} )r r r r
�_testCmsgTrunc1W 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
�testCmsgTrunc2IntZ s z SCMRightsTest.testCmsgTrunc2Intc C s | j d � d S)Nr )r} )r r r r
�_testCmsgTrunc2Inta s z SCMRightsTest._testCmsgTrunc2Intc C s6 | j | j | j t t � t j d � d � � d S)Nr r )r� r� r
r@ r� r r` )r r r r
�testCmsgTruncLen0Minus1d s z%SCMRightsTest.testCmsgTruncLen0Minus1c C s | j d � d S)Nr )r} )r r r r
�_testCmsgTruncLen0Minus1h 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 r r )r� r
r@ r� r� r� rg r� r r� r� rx r r r ry rc rv ) r �ancbuf�maxdataZmindatar� r� r= rl rz r{ r| rq r r r
�checkTruncatedArrayn s" "z!SCMRightsTest.checkTruncatedArrayc C s# | j d t j d � d d � d S)Nr� r r� )r� r r` )r r r r
�testCmsgTruncLen0� s z SCMRightsTest.testCmsgTruncLen0c C s | j d � d S)Nr )r} )r r r r
�_testCmsgTruncLen0� s z SCMRightsTest._testCmsgTruncLen0c C s' | j d t j d � d d d � d S)Nr� r r r� )r� r r` )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 r` 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� r r� )r� r r` 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� rr rv r� rw r} r� r� r� r� r� r� r r� r� r� r� r� rp rx 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
rm :
sl
!0 !
0
rm 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� r c C sJ i } x= | D]5 \ } } } | j | | f | � | | | | f <q
W| S)N)r� )r r� �drz 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)Nr r� Tr� r� r r )r
r- r � IPPROTO_IPV6�IPV6_RECVHOPLIMITr� rA r� r@ r� r� r� rg r� r� r� r~ �
IPV6_HOPLIMITr� r r ry r r ) r �
ancbufsize�maxhopr� r� r� r= rl rz 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._testRecvHopLimitr! c C s | j d t j t � � d S)Nr� )r� r r! 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)
Nr r� Tr� r� r� r r r� )r
r- r r� r� �IPV6_RECVTCLASSr� rA r� r@ r� r� r� rg r� r� r� � IPV6_TCLASSr r ry r r r� ) r r� r� r� r� r� r= rl ZancmapZtcdatar Zhldatar r r
� checkTrafficClassAndHopLimit s2
z1RFC3542AncillaryTest.checkTrafficClassAndHopLimitr� r� c C s | j d d � d S)Nr� i ( )r� )r r r r
� testRecvTrafficClassAndHopLimit( 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 r! r )r r r r
�)testRecvTrafficClassAndHopLimitCMSG_SPACE3 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_SPACE; s z?RFC3542AncillaryTest._testRecvTrafficClassAndHopLimitCMSG_SPACEc C s | j d d d | j � d S)Nr� i ( r� )r� r� )r r r r
� testSetTrafficClassAndHopLimit@ 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
� _testSetTrafficClassAndHopLimitI s z4RFC3542AncillaryTest._testSetTrafficClassAndHopLimitc C s | j d d d | j � d S)Nr� i ( r� )r� r� )r r r r
�testOddCmsgSizeT 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� rX r� r� r r� r� r� r� r@ )r rS 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)Nr r� Tr� r� )r
r- r r� r� r� rA r� r@ r� r� r� rg r� r� )r r� r� r) r� r� r= rl r r r
� checkHopLimitTruncatedHeadert 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� r )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 r )r� r r` )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 � n d S)Nr r� Tr� r )r
r- r r� r� r� rA r� r@ r� r` r r� r� rg r� r� r r� r� )r r� r� r= rl rz 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) Nr r� Tr� r� r r r� )r
r- r r� r� r� r� rA r� r@ r� r� r� rg r� r� r� r� r� r r ry r r ) r r� r� r� r� r= rl rz 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 r! 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
�_testSecondCmsgTrunc0
s z*RFC3542AncillaryTest._testSecondCmsgTrunc0c C s | j t j t � d � d S)Nr )r� r r! r )r r r r
�testSecondCmsgTrunc1
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 r! 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
�_testSecondCmsgTrunc2Int
s z-RFC3542AncillaryTest._testSecondCmsgTrunc2Intc C s+ | j t j t � t j d � d � d S)Nr r )r� r r! r r` )r r r r
� testSecondCmsgTruncLen0Minus1
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
� _testSecondCmsgTruncLen0Minus1#
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 � n | j | g � d S)Nr r� Tr� r r r� ) r
r- r r� r� r� r� rA r� r@ r� r! r r` r� r� rg r� r� r� r� �pop�remover ry r r r� )
r r� r� r= rl Z
cmsg_typesrz r{ r| r r r r
�testSecomdCmsgTruncInData(
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
�_testSecomdCmsgTruncInDataM
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� U
s r� r� c @ s e Z d Z d S)�SendmsgUDPTestN)r r r r r r r
r� Z
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� d
s r� c @ s e Z d Z d d � Z d S)�SendrecvmsgUDP6TestBasec C s( | j | d d � | d d � � d S)Nr r� r� )r� )r r� r� r r r
r� n
s z+SendrecvmsgUDP6TestBase.checkRecvmsgAddressN)r r r r� r r r r
r� j
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
s r� c @ s e Z d Z d S)�RecvmsgUDP6TestN)r r r r r r r
r� z
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 sh y t t | � j � WnJ t k
rc } z* | j t j k rD � n | j d � WYd d } ~ Xn Xd S)Nz4sporadic ENOTCONN (kernel issue?) - see issue #13876)r$ r� rK r r� �ENOTCONNr5 )r r[ )r+ r r
rK �
s z$RecvmsgSCTPStreamTest.testRecvmsgEOF)r r r rK r r )r+ r
r� �
s r� c s" e Z d Z � f d d � Z � S)�RecvmsgIntoSCTPStreamTestc sh y t t | � j � WnJ t k
rc } z* | j t j k rD � n | j d � WYd d } ~ Xn Xd S)Nz4sporadic ENOTCONN (kernel issue?) - see issue #13876)r$ r� rK r r� r� r5 )r r[ )r+ r r
rK �
s z(RecvmsgIntoSCTPStreamTest.testRecvmsgEOF)r r r rK 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� rx 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 S)Nr )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
s
z InterruptedTimeoutBase.setUpg @� setitimerg�������?c C s t j t j | � d S)N)rP r � ITIMER_REAL)r �secondsr r r
r s z InterruptedTimeoutBase.setAlarmr� c C s t j | � d S)N)rP rJ )r r r r r
r s ) r r r r rU rW rP �
alarm_timer r r )r+ r
r� 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 - s
z InterruptedRecvTimeoutTest.setUpc
O sh | j | j � | j t � � } | | | � Wd QX| j | j t j � | j | j j t j
� d S)N) r r r� r �assertNotIsInstancer� r rU r� r� �EINTR)r r� r) r* r� r r r
�checkInterruptedRecv1 s
z/InterruptedRecvTimeoutTest.checkInterruptedRecvc C s | j | j j d � d S)Ni )r r r� )r r r r
�testInterruptedRecvTimeout: s z5InterruptedRecvTimeoutTest.testInterruptedRecvTimeoutc C s | j | j j t d � � d S)Ni )r r � recv_intor )r r r r
� testInterruptedRecvIntoTimeout= s z9InterruptedRecvTimeoutTest.testInterruptedRecvIntoTimeoutc C s | j | j j d � d S)Ni )r r r� )r r r r
� testInterruptedRecvfromTimeout@ s z9InterruptedRecvTimeoutTest.testInterruptedRecvfromTimeoutc C s | j | j j t d � � d S)Ni )r r �
recvfrom_intor )r r r r
�"testInterruptedRecvfromIntoTimeoutC s z=InterruptedRecvTimeoutTest.testInterruptedRecvfromIntoTimeoutr� c C s | j | j j d � d S)Ni )r r r� )r r r r
� testInterruptedRecvmsgTimeoutF s z8InterruptedRecvTimeoutTest.testInterruptedRecvmsgTimeoutr c C s# | j | j j t d � g � d S)Ni )r r r r )r r r r
�!testInterruptedRecvmsgIntoTimeoutJ s z<InterruptedRecvTimeoutTest.testInterruptedRecvmsgIntoTimeout)
r r r r r r
r r
r r� r r r r r )r+ r
r & 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 i s z$InterruptedSendTimeoutTest.doConnectc
O so | j t � �+ } x! | j | j � | | | � q WWd QX| j | j t j � | j | j j t j
� d S)N) r� r r r r r� r rU r� r� r )r r� r) r* r� r r r
�checkInterruptedSendl s z/InterruptedSendTimeoutTest.checkInterruptedSendr r� c C s | j | j j d d � d S)Nr, i )r rn r0 )r r r r
�testInterruptedSendTimeoutx 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 Q 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 �?r r� )r ri r
ra r� r� r� )r rk rl Zsdr� rn 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 rr z test needs socket.socketpair()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# r$ 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� r% 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 � n | j j d � | j | j j � d � | j j | j d � | j | j j � � d S)Nr zneeds UINT_MAX < ULONG_MAXFg r ) rn r� Z ULONG_MAXr5 r r� r� r r% )r rn 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 d � t
j
� } y | j j � Wn t k
r� Yn Xt
j
� } | j
| | d k d � d S)Nr g �?z&Error creating with non-blocking mode.)r r
r r r r* r r r r r ri r r� )r r r& 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
�_testInheritFlags 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� r% r r
)r rk rl r� rn r� r r r
�
testAccept 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
� _testAccept s
z NonBlockingTCPTests._testAcceptc C s# | j j � \ } } | j � d S)N)r ri r
)r rk rl r r r
� testConnect$ 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
� _testConnect) 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� rn r� r r r
r� - 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� ? s
z NonBlockingTCPTests._testRecvN)r r r r% r' r( r rw r) Z_testSetBlocking_overflowrp rv rW r �requires_linux_versionr- r. r0 r1 r2 r3 r4 r5 r� r� r r r r
r% � s" $ r% 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.
r zutf-8�strictNr[ �wbr_ 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)rg �evt1�evt2�
serv_finished� cli_finishedrh r rj r] � read_moder6 r: r; r< � 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)r? rA rB r
r� �closedrh r )r r r r
r h s
z FileObjectClassTestCase.tearDownc C sJ t j | � | j j | j | j d | j d | j d | j �| _ d S)Nr: r; r< )
rh rU rn r] �
write_moder6 r: r; r< �
write_file)r r r r
rU o s
z#FileObjectClassTestCase.clientSetUpc C sG | j j � | j j � | j | j j � d | _ t j | � d S)N)r@ rA rE r
r� rC rh rV )r r r r
rV w s
z&FileObjectClassTestCase.clientTearDownc
C s� | j j d � | j j d � | j t j | j j d � | j t � � } | j j d � Wd QX| j d t | j
� � d S)Nr r� z!cannot read from timed out object) rj r/ rB 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� )rE rn � write_msg�flushr? 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� )rB 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)rE rn rG rH )r r r r
�_testSmallRead� s z&FileObjectClassTestCase._testSmallReadc C s& | j j � } | j | | j � d S)N)rB r� r� rJ )r r� r r r
� testFullRead� s z$FileObjectClassTestCase.testFullReadc C s$ | j j | j � | j j � d S)N)rE rn rG r
)r r r r
�
_testFullRead� s z%FileObjectClassTestCase._testFullReadc C sV t | j � � } x* | j j d � } | s1 Pn | | 7} q W| j | | j � d S)Nr )r� rJ rB r� r� )r r �charr r r
�testUnbufferedRead� s z*FileObjectClassTestCase.testUnbufferedReadc C s$ | j j | j � | j j � d S)N)rE rn rG rH )r r r r
�_testUnbufferedRead� s z+FileObjectClassTestCase._testUnbufferedReadc C s& | j j � } | j | | j � d S)N)rB �readliner� rJ )r �liner r r
� testReadline� s z$FileObjectClassTestCase.testReadlinec C s$ | j j | j � | j j � d S)N)rE rn rG rH )r r r r
�
_testReadline� s z%FileObjectClassTestCase._testReadlinec C s3 | j j � | j j � } | j | | j � d S)N)rj r
rB r� r� rJ )r r� r r r
�testCloseAfterMakefile� s
z.FileObjectClassTestCase.testCloseAfterMakefilec C s$ | j j | j � | j j � d S)N)rE rn rG rH )r r r r
�_testCloseAfterMakefile� s z/FileObjectClassTestCase._testCloseAfterMakefilec C s] | j j � | j j t t � � } t | j t � rF | j � } n | j
| | j � d S)N) rB r
rj r� r@ r� r� rJ r� �decoder� )r r� r r r
� testMakefileAfterMakefileClose� s
z6FileObjectClassTestCase.testMakefileAfterMakefileClosec C s$ | j j | j � | j j � d S)N)rE rn rG rH )r r r r
� _testMakefileAfterMakefileClose� s z7FileObjectClassTestCase._testMakefileAfterMakefileClosec C s | j | j j � d S)N)r� rB rC )r r r r
�testClosedAttr� s z&FileObjectClassTestCase.testClosedAttrc C s | j | j j � d S)N)r� rE rC )r r r r
�_testClosedAttr� s z'FileObjectClassTestCase._testClosedAttrc C s< | j | j j | j � | j | j j | j j � � d S)N)r� rB �moderA 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� rE r^ rD 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)rB 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. r6 r: r; r< rA r� rJ rD rG r% r r rU rV rF rI rL rM rN rO rQ rR rU rV rW rX rZ r[ r\ r] r_ r` ra rb r r r r
r7 D sD
r7 c @ s� e Z d Z d Z Gd d � d e � Z e d d � � Z d d d � Z d d d d
� Z d d d d � Z
d
d � Z d d � Z d d d � Z
d d � Z d S) � FileObjectInterruptedTestCasez=Test that the file object correctly handles EINTR internally.c @ sF e Z d Z f d d � Z d d � Z d d � Z d d d � Z d
S) z(FileObjectInterruptedTestCase.MockSocketc C s t | � | _ d S)N)�iter�
_recv_step)r �
recv_funcsr r r
r% � s z1FileObjectInterruptedTestCase.MockSocket.__init__c C s2 t | j � � } | | d t | � � <t | � S)N)�nextre r@ )r �bufferr� r r r
r � s z2FileObjectInterruptedTestCase.MockSocket.recv_intoc C s d S)Nr )r r r r
�_decref_socketios� s z:FileObjectInterruptedTestCase.MockSocket._decref_socketiosr c C sn t j | d � } | d k r* t j } n | d k r: | St j | | � } t j | d d � } d | _ | S)Nr^ r r[ )r �SocketIO�io�DEFAULT_BUFFER_SIZE�BufferedReader�
TextIOWrapperr^ )r rb �rawrh �textr r r
�_textiowrap_for_test� s z=FileObjectInterruptedTestCase.MockSocket._textiowrap_for_testNr� )r r r r% r ri rq r r r r
�
MockSocket� s rr c C s t t j d � � d S)NZ interrupted)r r� r r r r r
� _raise_eintr s z*FileObjectInterruptedTestCase._raise_eintrr c C sn t j | d � } | d k r* t j } n | d k r: | St j | | � } t j | d d � } d | _ | S)Nr^ r r[ )r rj rk rl rm rn r^ )r Zmockrb ro rh rp r r r
�_textiowrap_mock_socket s z5FileObjectInterruptedTestCase._textiowrap_mock_socketc C s� | j d d d � | j d d � d d � d d � g � } | j d | � } | j | j | � d � | j | j | � d � d S)
Nrf c S s d S)Ns" This is the first line
And the secr r r r r
r� s z>FileObjectInterruptedTestCase._test_readline.<locals>.<lambda>c S s d S)Ns ond line is here
r r r r r
r� s c S s d S)Nr� r r r r r
r� s c S s d S)Nr� r r r r r
r� s rb zThis is the first line
z And the second line is here
)rr rs rq r� rS )r r� rb � mock_sock�for r r
�_test_readline s z,FileObjectInterruptedTestCase._test_readlinec C s� | j d d d � | j d d � d d � d d � g � } d } | j d | � } | d k ri d
} n d } | j d � } x? t | � t | � k r� | j | � } | s� Pn | | 7} q� W| j | | � d S)
Nrf c S s d S)Ns" This is the first line
And the secr r r r r
r� " s z:FileObjectInterruptedTestCase._test_read.<locals>.<lambda>c S s d S)Ns ond line is here
r r r r r
r� $ s c S s d S)Nr� r r r r r
r� % s c S s d S)Nr� r r r r r
r� & s s3 This is the first line
And the second line is here
rb r r� r� zutf-8)rr rs rq rY r@ r� r� )r r� rb ru Z expectingrv r� �partr r r
�
_test_read s$ z(FileObjectInterruptedTestCase._test_readc C s8 | j � | j d d � | j � | j d d � d S)Nr� r: )rw ry )r r r r
� test_default7 s
z*FileObjectInterruptedTestCase.test_defaultc C sP | j d d � | j d d d d � | j d d � | j d d d d � d S)Nrb i r� r: )rw ry )r r r r
�test_with_1k_buffer= s z1FileObjectInterruptedTestCase.test_with_1k_bufferc C s� | j d d d � d d � d d � | j d d � d d � g � } | j d d � } | j | j | � d
� | j | j | � d � d S) Nrf c S s d S)Nr, r r r r r
r� E s zHFileObjectInterruptedTestCase._test_readline_no_buffer.<locals>.<lambda>c S s d S)Ns
r r r r r
r� F s c S s d S)N� Br r r r r
r� G s c S s d S)N� br r r r r
r� I s c S s d S)Nr� r r r r r
r� J s rb r s a
s Bb)rr rs rq r� rS )r r� ru rv r r r
�_test_readline_no_bufferC s z6FileObjectInterruptedTestCase._test_readline_no_bufferc C sD | j � | j d d � | j d d � | j d d d d � d S)Nr� r� rb r r: )r~ ry )r r r r
�test_no_bufferP s
z,FileObjectInterruptedTestCase.test_no_bufferNr� r� r� r� r� r� )r r r r. r rr � staticmethodrs rt rw ry rz r{ r~ r r r r r
rc � s
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. )rB rS r� rG rj r] )r rT r r r
�testUnbufferedReadlinec s
z8UnbufferedFileObjectClassTestCase.testUnbufferedReadlinec C s? | j j d | j � | j j d | j � | j j � d S)Ns A. s B. )rE rn rG rH )r r r r
�_testUnbufferedReadlinek 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� rJ rB r� r )r r� r r r
�testMakefileClosep s
z3UnbufferedFileObjectClassTestCase.testMakefileClosec C s$ | j j | j � | j j � d S)N)rE rn rG rH )r r r r
�_testMakefileClosey s z4UnbufferedFileObjectClassTestCase._testMakefileClosec C sI t j | j � } | j j � t j | j � } | j | d | � d S)Nr )r� r� rj rB r
r� )r Zrefcount_beforeZrefcount_afterr r r
� testMakefileCloseSocketDestroy} s
z@UnbufferedFileObjectClassTestCase.testMakefileCloseSocketDestroyc C s d S)Nr )r r r r
� _testMakefileCloseSocketDestroy� s zAUnbufferedFileObjectClassTestCase._testMakefileCloseSocketDestroyc C si | 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 � } n 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� r )rj r� r� rB �readintor r� r@ rJ r= rA r> rQ r r )r rK r rg r� r r r
�testSmallReadNonBlocking� s ")
" "z:UnbufferedFileObjectClassTestCase.testSmallReadNonBlockingc C sQ | j j d � | j j | j � | j j � | j j � | j j d � d S)Ng �?g @) r= rQ rE rn rG rH r> rA r? )r r r r
�_testSmallReadNonBlocking� s
z;UnbufferedFileObjectClassTestCase._testSmallReadNonBlockingc C s0 | j j d � | j r, | j | j � n d S)Ng @)r@ rQ � serv_skippedr5 )r r r r
�testWriteNonBlocking� s z6UnbufferedFileObjectClassTestCase.testWriteNonBlockingc C s� d | _ | j j d � d t j } d } | j j | � } | j | d � xO t | � D]8 } | j j | � } | d k r� Pn | j | d � q[ Wd | _ d S)NFrO r r z$failed to saturate the socket buffer) r� rn r� r rT rE rn r� rg )r ZBIGZLIMITrg r r r r
�_testWriteNonBlocking� s
z7UnbufferedFileObjectClassTestCase._testWriteNonBlockingN)r r r r. r6 r� r� r� r� r� r� r� r� r� r� r r r r
r� W s r� c @ s e Z d Z d Z d S)�#LineBufferedFileObjectClassTestCaser N)r r r r6 r r r r
r� � s r� c @ s e Z d Z d Z d S)�$SmallBufferedFileObjectClassTestCaser� N)r r r r6 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-8r9 r� N) r r r r. rA r� rY rJ rD rG r< 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[ r� zutf-8r� N) r r r r. rA r� rJ rD rY rG r< 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-8r� r� N) r r r r. rA r� rY rJ rD rG r< 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
rr � s rr c c s/ t j } | j t _ z d VWd | t _ Xd S)z*Return a socket which times out on connectN)r rr )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 QX| 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 QXt j g } t t d � re | j
t j � n | j | j
j | � d S)N� ENETUNREACH)r r' r� r r r� re r� r� rW rp r� r� r� )r r r� Zexpected_errnosr r r
�test_create_connection s z0NetworkConnectionNoServer.test_create_connectionc C sC | j � �1 | j t j � � t j t d f � Wd QXWd QXd 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 rr �
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% 6 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
� _justAcceptB 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
� _testFamilyG 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� r )
r r� re r r� ra r, r
r� rf )r r r r
�_testSourceAddressM 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
�_testTimeoutDefaultV 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
�_testTimeoutNoneb 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
�_testTimeoutValueNamedn 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
�_testTimeoutValueNonameds s z8NetworkConnectionAttributesTest._testTimeoutValueNonamedN)r r r r% rU rV r� Z
testFamilyr� ZtestSourceAddressr� ZtestTimeoutDefaultr� ZtestTimeoutNoner� ZtestTimeoutValueNamedr� ZtestTimeoutValueNonamedr� r r r r
r� 3 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% { s z'NetworkConnectionBehaviourTest.__init__c C s d S)Nr )r r r r
rU 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
�_testInsideTimeout� s z1NetworkConnectionBehaviourTest._testInsideTimeoutc sH t j t | j f d d �| _ � | j t j � f d d � � d S)NrU r c s
� j d � S)Nr" )r� r )r� r r
r� � s zDNetworkConnectionBehaviourTest._testOutsideTimeout.<locals>.<lambda>)r r� re r ra r� rU )r r )r� r
�_testOutsideTimeout� s %z2NetworkConnectionBehaviourTest._testOutsideTimeoutN)
r r r r% rU rV r� ZtestOutsideTimeoutr� r� r r r r
r� x 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_timeout� s z4TCPTimeoutTest.testTCPTimeout.<locals>.raise_timeoutz*Error generating a timeout exception (TCP))r� r rU )r r� r )r r
�testTCPTimeout� 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 � n 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
�testTimeoutZero� 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
�Alarm� s r� c s
� � d S)Nr )rP r� )r� r r
�
alarm_handler� 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
�testInterruptedTimeout� s(
*
z%TCPTimeoutTest.testInterruptedTimeoutN)
r r r r� r� rp rv rW rP r� r r r r
r� � 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 � n 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 @ sF e Z d Z 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 d � t j t j t j � �_ } | j | j � � | j � d �3 } | j | j � | � | j | j � | � Wd QXWd QXWd QXd S)Ns python-test-hello �r r )
r r� r r3 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 QXd S)Ns � hr )�
UNIX_PATH_MAXr r� r r3 r� rf )r r� r r r r
� testMaxName s
z&TestLinuxAbstractNamespace.testMaxNamec C sL d d | j } t j t j t j � � } | j t | j | � Wd QXd S)N� �h)r� r r� r r� r r3 )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 r3 r� rf r
)r r r r r
� testStrName s
z&TestLinuxAbstractNamespace.testStrNameN)r r r r� r� r� r� r� r r r r
r� � s
r� ztest needs socket.AF_UNIXc @ 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)�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 r5 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)r3 r r� r5 r� )r r� r� r[ r r r
r3 ( s
zTestUnixDomain.bindc C sX t j j t j � } | j | j | � | j t j | � | j | j j
� | � d S)N) r� r� �abspathr �TESTFNr3 r� r, r� r� rf )r r� r r r
� testStrAddr4 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� r3 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� r3 r� rY r, r� r� rf )r r� rE r r r
�testSurrogateescapeBindB s
z&TestUnixDomain.testSurrogateescapeBindc C sw t j d k r | j d � n t j j t j � } | j | j | � | j t j | � | j
| j j � | � d S)Nz!No unencodable filename available) r ZTESTFN_UNENCODABLEr5 r� r� r� r3 r� r, r� r� rf )r r� r r r
�testUnencodableAddrK s z"TestUnixDomain.testUnencodableAddrN) r r r r r r� r3 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% Z s zBufferIOTest.__init__c 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 rS 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
�_testRecvIntoArrayd 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 rS r� r r r
�testRecvIntoBytearrayh 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 r\ r� r@ r� )r r rS r� r r r
�testRecvIntoMemoryviewq s
z#BufferIOTest.testRecvIntoMemoryviewc 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 rS rl r� r r r
�testRecvFromIntoArrayz s
z"BufferIOTest.testRecvFromIntoArrayc C s t t � } | j j | � d S)N)r� r� rn r0 )r r r r r
�_testRecvFromIntoArray� 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 rS rl r� r r r
�testRecvFromIntoBytearray� 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 r\ r� r@ r� )r r rS rl r� r r r
�testRecvFromIntoMemoryview� s
z'BufferIOTest.testRecvFromIntoMemoryviewc C s, t d � } | j t | j j | d � d S)Nr2 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
�testRecvFromIntoEmptyBuffer� 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� U s$ r� i� �� �� c
C sf t t d � s d St j j d � s) d St d � �+ } x! | D] } | j d � r? d Sq? WWd QXd 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� r� �isfile�openr )r� rT r r r
�isTipcAvailable� 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)Nr r� r i )r r� r� r, r
r- r� r� �TIPC_ADDR_NAMESEQ�
TIPC_STYPE�
TIPC_LOWER�
TIPC_UPPERr3 �TIPC_ADDR_NAMEr� r� r� r� r� rf )r rl 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 d � | j � | j j � \ | _ | _ | j | j j � d S)Nr r" )r r� r rl r, r
r- r� r� r� r� r� r� r3 r rB 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 QX| j | j � t j � � } | j � Wd QX| j | j � t j � � } | j t | j d � Wd QX| 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 sq | j j � } t j | � �= } | j | j � | j d � | j | j d � d � Wd QX| 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 QX| 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� r+ r, c
C sV t j t j t j t j B� �1 } | j | j t j @� | j | j � � Wd QXd 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 s3 t j � } | � | j | j � d � Wd QXd S)NF)r r� r� )r r� r r r
�test_default_inheritable3 s z(InheritanceTest.test_default_inheritablec C sV t j � } | �> | j � } | j � | � | j | j � d � Wd QXWd QXd S)NF)r r� r
r� r� )r r� Znewsockr r r
�test_dup8 s
zInheritanceTest.test_dupc
C sc t j � } | �K | j d � | j | j � d � | j d � | j | j � d � Wd QXd 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 QXd S)NFT)r r� r� r� �fcntl�F_GETFD�
FD_CLOEXECZF_SETFD)r r� rt r= r r r
� test_get_inheritable_cloexecI 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 QXd S)NTr )r r� r� r r r r )r r� rt r r r
� test_set_inheritable_cloexecW 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_socketpaird s
z InheritanceTest.test_socketpair)r r r rp rv rW r r r6 r� r� r r rx r r r r r r r r
r� ) s $ $$
r� r* 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 r* r� r r* )r r ZnonblockrU r r r
�
checkNonblockq s
z"NonblockConstantTest.checkNonblockr� r+ r, 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 QXt j � } t j d � t j � � } | j | � Wd QXt j d � t j � � } | j | d � Wd QXt j d � t j � � } | j | d d �Wd QXt j d � t j � � } | j | d � Wd QXt j | � d S)Nr Fr g @rU g ) r r r r* r
r� r/ r
r )r r �tr r r
�test_SOCK_NONBLOCKy s6
z'NonblockConstantTest.test_SOCK_NONBLOCKN)r r r r
r r6 r r r r r
r n 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
�remoteProcessServer� 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� Pn | 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� rp r� r� ) r r r� r� rl r/ r �mZreceivedr r r
� testShare� s(
zTestSocketSharing.testSharec C sV | j j t j � � } | j t t j | d d � � | j t t j | d � d S)Nr 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 � n 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
� s # r
c C s7 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 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/ t0 t1 t2 t3 t4 t5 t6 t7 t8 t9 g � t: j; � } t: j< | � t: j= | � d S)N)>r� r� r r� r� r� r� r� r� �extendr% r7 rc r� r� r� r� r� r� r� r� r� r� r� r rp r r� r� r� r� 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 Zthreading_setupZ run_unittestZthreading_cleanup)Ztests� thread_infor r r
� test_main sf
r �__main__)�rp Ztestr r� rk r4 r r� r~ r r� rH r� r� r r r� �weakrefr rP rK rh r8 r r r re ro r� �_threadrN r& r� r r ZHAVE_SOCKET_CANZHAVE_SOCKET_RDSrc r rq r r! r# r/ r: r; r^ rb rc rd rh ro rs rx ry r| r} r� r� r� r� r� r� r� r� r� r� r� rv ry r� r� r� r� r� r� r� r r r r r r
r) r2 r5 rJ rO rR r_ rm 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 r% r7 rc 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> s,
3k!
� � �
C [ Z z 7w w2�$ P;� h� �
! !( ! 4 " �ol
; D 9%#%@ Q ' ,E+ e :