Your IP : 216.73.216.213
�
g f � � - @ s� 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 Z d d l Z d d l Z d d l Z e j j d � Z e j j d � 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 y d d l m Z e j Z! Wn e" k
r�d Z! Yn Xy d d l# m$ Z$ m% Z% d Z& Wn e" k
r�d Z& Yn Xy d d l' Z' Wn e" k
r�d Z' Yn Xd d
� Z( e j) Z* d Z+ d Z, e, r2d d
d f \ Z- Z. Z/ n d d d f \ Z- Z. Z/ e0 e d d � Z1 e j2 d k Z3 d d l m4 Z4 d d � Z5 y e j6 d � Z7 Wn d Z7 Yn Xd d g Z8 y d d l9 m: Z: m; Z; m< Z< Wn" e" k
r�e= Z: d Z; Z< Yn Xd d � Z> Gd d � d e= � Z? Gd d � d e= � Z@ d d � ZA Gd! d" � d" e@ � ZB Gd# d$ � d$ e jC � ZD Gd% d& � d& e@ � ZE d' d( � ZF d) d* � ZG Gd+ d, � d, e@ � ZH Gd- d. � d. e@ � ZI Gd/ d0 � d0 e@ � ZJ Gd1 d2 � d2 e@ � ZK Gd3 d4 � d4 e@ � ZL Gd5 d6 � d6 e= � ZM d7 d8 � ZN Gd9 d: � d: e= � ZO Gd; d< � d< e= � ZP Gd= d> � d> e@ � ZQ Gd? d@ � d@ e@ � ZR GdA dB � dB e@ � ZS GdC dD � dD e@ � ZT dE dF dG � ZU dH dI � ZV GdJ dK � dK eW � ZX dL dM � ZY GdN dO � dO e@ � ZZ dP dQ � Z[ dR dS � Z\ GdT dU � dU e@ � Z] GdV dW � dW e@ � Z^ d dX l m_ Z_ m` Z` ma Za GdY dZ � dZ e= � Zb d[ d\ � Zc Gd] d^ � d^ e` � Zd Gd_ d` � d` e_ � Ze ee jf da db eb �ee jf dc db eb dd de df f �ee jf d\ db ec dg ed �Gdh di � di e@ � Zg e jh � Zi dj dk � Zj Gdl dm � dm e_ � Zk ek jf dk db ej �Gdn do � do e_ � Zl el jf dk � dp Zm Gdq dr � dr e@ � Zn Gds dt � dt e@ � Zo e( du � Zp Gdv dw � dw e@ � Zq Gdx dy � dy e@ � Zr Gdz d{ � d{ e@ � Zs Gd| d} � d} e@ � Zt e ju e! d~ � Gd d� � d� e@ � � Zv Gd� d� � d� e@ � Zw Gd� d� � d� e: � Zx Gd� d� � d� e@ � Zy Gd� d� � d� e@ � Zz Gd� d� � d� e j{ � Z| Gd� d� � d� e@ � Z} Gd� d� � d� e@ � Z~ Gd� d� � d� e j{ � Z Gd� d� � d� e j{ � Z� d� d� � Z� Gd� d� � d� e j{ � Z� d� d� � Z� d� d� � Z� d� d� � Z� d� d� � Z� Gd� d� � d� e= � 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� 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� e j� e j2 d k d� � Gd� d� � d� e j{ � � Z� Gd� d� � d� e= � Z� Gd� d� � d� e= � Z� Gd� d� � d� e= � Z� d� d� � Z� d S)�� N�_multiprocessingzmultiprocessing.synchronize)�util)� reductionF)�Value�copyTc C s
| j d � S)N�latin)�encode)�s� r
�?/opt/alt/python34/lib64/python3.4/test/_test_multiprocessing.pyr ? s r g�������?g=
ףp=�?gffffff�?gffffff�?�HAVE_BROKEN_SEM_GETVALUE�win32)�waitc C s1 | d k r! | d k r! d } n t | g | � S)Ng )r )Zhandle�timeoutr
r
r �wait_for_handleZ s r � SC_OPEN_MAX� �__main__z$test.test_multiprocessing_forkserver)� Structure�c_int�c_doublec C sk d } y t j d � } Wn t t f k
r7 d SYn X| d k sP | | k rT d St j d | � � d S)zACheck that the system supports enough semaphores to run the test.r �SC_SEM_NSEMS_MAXN� zHThe OS doesn't support enough semaphores to run the test (required: %d).���)�os�sysconf�AttributeError�
ValueError�unittest�SkipTest)Z nsems_minZnsemsr
r
r �check_enough_semaphoresr s r c @ s( e Z d Z d d � Z d d � Z d S)�
TimingWrapperc C s | | _ d | _ d S)N)�func�elapsed)�selfr" r
r
r �__init__� s zTimingWrapper.__init__c O s; t j � } z | j | | � SWd t j � | | _ Xd S)N)�timer" r# )r$ �args�kwds�tr
r
r �__call__� s zTimingWrapper.__call__N)�__name__�
__module__� __qualname__r% r* r
r
r
r r! � s r! c @ s@ e Z d Z d Z d d � Z d d � Z d d � Z e Z d
S) � BaseTestCase� processes�manager�threadsc C s t r | j | | d � n d S)Nr )�
CHECK_TIMINGS�assertAlmostEqual)r$ �a�br
r
r �assertTimingAlmostEqual� s z$BaseTestCase.assertTimingAlmostEqualc G s9 y | | � } Wn t k
r$ Yn X| j | | � Sd S)N)�NotImplementedError� assertEqual)r$ �valuer" r' �resr
r
r �assertReturnsIfImplemented� s
z'BaseTestCase.assertReturnsIfImplementedc G s t d � � d S)Nz#shouldn't try to pickle a test case)r7 )r$ r' r
r
r �
__reduce__� s zBaseTestCase.__reduce__N)z processeszmanagerr1 )r+ r, r- �
ALLOWED_TYPESr6 r; r< �
__reduce_ex__r
r
r
r r. � s
r. c C sm y | j � SWnX t k
rh y | j SWn8 t k
rc y | j SWn t k
r^ t � Yn XYn XYn Xd S)N)� get_valuer Z_Semaphore__valueZ_valuer7 )r$ r
r
r r? � s
r? c @ s� e Z d Z d Z d d � Z d d � Z e d d � � Z d d
� Z e d d � � Z d
d � Z
d d � Z d d � Z e d d � � Z
d d � Z e d d � � Z d d � Z d S) � _TestProcessr/ r1 c C s� | j d k r+ | j d j | j � � n | j � } | j } | j | j � � | j | j � | j | t � | j t
| � d k � | j | j t
j � � | j | j d � d S)Nr1 ztest not appropriate for {}r )�TYPE�skipTest�format�current_process�authkey�
assertTrue�is_alive�daemon�assertIsInstance�bytes�lenr8 Zidentr �getpid�exitcode)r$ �currentrE r
r
r � test_current� s z_TestProcess.test_currentc C s� | j d k r+ | j d j | j � � n | j d | j � } | j | j | j � j � | j d | j d d � } | j | j � | j d | j d d � } | j | j � d S)Nr1 ztest not appropriate for {}�targetrH TF)
rA rB rC �Process�_testr8 rH rD rF � assertFalse)r$ Zproc0Zproc1Zproc2r
r
r �test_daemon_argument� s z!_TestProcess.test_daemon_argumentc O sr | j � } | j | � | j | � | j | j � | j d k rn | j t | j � � | j | j � n d S)Nr1 )rD �put�namerA rJ rE �pid)�cls�qr' r( rN r
r
r rR � s
z_TestProcess._testc C sB | j d � } | j � } | d d f } i d d 6d d 6} d } | j d | j d | d
| d | � } d | _ | j � } | j d
k r� | j | j | j � n | j | j � d � | j | j d � | j
| | j � � | j t
| j � � t k � | j | j d � | j � | j | j d � | j | j � d � | j | | j � � | j | j � | d d � � | j | j � | � | j | j � | j � | j d
k r�| j | j � | j � | j | j � | j � n | j � | j | j d � | j | j � d � | j
| | j � � d S)Nr � � �hellogR� ��Q@ZbyeZ SomeProcessrP r' �kwargsrV Tr1 Fr )�Queue�EventrQ rR rH rD rA r8 rE rG � assertNotIn�active_childrenrF �type�listrM �start�assertIn�getrV rW �join)r$ rY �er' r] rV �prN r
r
r � test_process� s<
z_TestProcess.test_processc C s t j d � d S)N�d )r& �sleep)rX r
r
r �_test_terminate s z _TestProcess._test_terminatec s | j d k r+ | j d j | j � � n | j d | j � � d � _ � j � | j � j � d � | j � | j
� � | j � j d � t � j
� } | j | d � d � | j | j d � | j � j � d � | j | d
� d � | j | j d � | j � j � d � t j d � � j � t t d � r�� f d d
� } t j t j | � } z$ t j d � | j | � d � Wd t j d � t j t j | � Xn | j | � d � | j | j d � | j � j � d � | j � | j
� � � j
� d S)Nr1 ztest not appropriate for {}rP Tr g r �alarmc s t d � � � d S)Nzjoin took too long: %s)� RuntimeError)r' )ri r
r �handler2 s z,_TestProcess.test_terminate.<locals>.handler�
Fr )rA rB rC rQ rm rH rd r8 rG re ra rM r! rg r6 r# r& rl � terminate�hasattr�signal�SIGALRMrn r` )r$ rg rp Z old_handlerr
)ri r �test_terminate s<
z_TestProcess.test_terminatec C s[ y t j � } Wn t k
r* d } Yn X| j t | � t k � | j | d k � d S)Nr )�multiprocessing� cpu_countr7 rF rb �int)r$ Zcpusr
r
r �test_cpu_countH s
z_TestProcess.test_cpu_countc C s� | j t | j � � t � | j d t j d t f � } | j | | j � � d | _ | j
� | j | | j � � | j � | j | | j � � d S)NrP r' T)
r8 rb ra rc rQ r& rl �DELTAr` rH rd re rg )r$ ri r
r
r �test_active_childrenP s
z!_TestProcess.test_active_childrenc C sy | j | � t | � d k ru xS t d � D]B } | j d | j d | | | g f � } | j � | j � q, Wn d S)NrZ rP r' )�sendrK �rangerQ �_test_recursionrd rg )rX �wconn�id�iri r
r
r r ] s
z _TestProcess._test_recursionc C s� | j d d � \ } } | j | g � t j t � g } x# | j � r` | j | j � � q> Wg d g d d g d d g d g d d g d d g g } | j | | � d S)N�duplexFr r ) �Piper r& rl r{ �poll�append�recvr8 )r$ �rconnr� �result�expectedr
r
r �test_recursionh s
z_TestProcess.test_recursionc C s | j d � d S)Ng $@)r )rX �eventr
r
r �_test_sentinel| s z_TestProcess._test_sentinelc C s� | j d k r+ | j d j | j � � n | j � } | j d | j d | f � } | j t � � | j Wd QX| j � | j
| j � | j } | j | t
� | j t | d d �� | j � | j � | j t | d d �� d S)Nr1 ztest not appropriate for {}rP r' r g r )rA rB rC r_ rQ r� � assertRaisesr �sentinelrd �
addCleanuprg rI ry rS r �setrF )r$ r� ri r� r
r
r �
test_sentinel� s
z_TestProcess.test_sentinelN)z processeszthreads)r+ r, r- r= rO rT � classmethodrR rj rm rv rz r| r r� r� r� r
r
r
r r@ � s
' 4
r@ c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
� _UpperCaserc C s, t j j | � t j � \ | _ | _ d S)N)rw rQ r% r� �
child_conn� parent_conn)r$ r
r
r r% � s z_UpperCaser.__init__c C sT | j j � x3 t | j j d � D] } | j j | j � � q# W| j j � d S)N)r� �close�iterr� r� r} �upper)r$ r r
r
r �run� s
z_UpperCaser.runc C s | j j | � | j j � S)N)r� r} r� )r$ r r
r
r �submit� s z_UpperCaser.submitc C s. | j j d � | j j � | j j � d S)N)r� r} r� r� )r$ r
r
r �stop� s
z_UpperCaser.stopN)r+ r, r- r% r� r� r� r
r
r
r r� � s r� c @ s^ e Z d Z d
Z d d � Z d d � Z e d d � � Z e d d � � Z d
d � Z d S)�_TestSubclassingProcessr/ c C sf t � } d | _ | j � | j | j d � d � | j | j d � d � | j � | j � d S)NTr\ ZHELLOZworldZWORLD)r� rH rd r8 r� r� rg )r$ Z
uppercaserr
r
r �test_subclassing� s
z(_TestSubclassingProcess.test_subclassingc
C s� | j d k r+ | j d j | j � � n t j j } | j t j j | � | j d | j d | f � } | j
� | j � t | d � �C } | j
� } | j d | � | j d | � | j d | � Wd QXd S) Nr1 ztest not appropriate for {}rP r' �r�ZeroDivisionErrorztest_multiprocessing.pyz 1/0 # MARKER)rA rB rC �test�support�TESTFNr� �unlinkrQ �_test_stderr_flushrd rg �open�readre )r$ �testfn�proc�f�errr
r
r �test_stderr_flush� s
z)_TestSubclassingProcess.test_stderr_flushc C s t | d � t _ d d d S)N�wr r )r� �sys�stderr)rX r� r
r
r r� � s z*_TestSubclassingProcess._test_stderr_flushc C s# t | d � t _ t j | � d S)Nr� )r� r� r� �exit)rX �reasonr� r
r
r �_test_sys_exit� s z&_TestSubclassingProcess._test_sys_exitc C sm | j d k r+ | j d j | j � � n t j j } | j t j j | � x� d d d g d f d f D]� \ } } | j d | j d | | f � } d | _
| j � | j d
� | j
| j | � t | d � �) } | j
| j � j � t | � � Wd QXqi Wx_ d D]W } | j d t j d | f � } d | _
| j � | j d
� | j
| j | � qWd S)Nr1 ztest not appropriate for {}r rZ � � ignore thisrP r' T� r� F� )r� r )TFr� )rA rB rC r� r� r� r� r� rQ r� rH rd rg r8 rM r� r� �rstrip�strr� r� )r$ r� r� �coderi r� r
r
r �
test_sys_exit� s$ (!
,
z%_TestSubclassingProcess.test_sys_exitN)z processes)
r+ r, r- r= r� r� r� r� r� r� r
r
r
r r� � s r� c C s- t | d � r | j � S| j � d k Sd S)N�emptyr )rs r� �qsize)rY r
r
r � queue_empty� s
r� c C s- t | d � r | j � S| j � | k Sd S)N�full)rs r� r� )rY �maxsizer
r
r �
queue_full� s
r� c @ s� e Z d Z e d d � � Z d d � Z e d d � � Z d d � Z e d d
� � Z d d � Z d
d � Z
e d d � � Z d d � Z d d � Z
d d � Z d S)�
_TestQueuec C s9 | j � x t d � D] } | j � q W| j � d S)N� )r r~ rf r� )rX �queue�child_can_start�parent_can_continuer� r
r
r � _test_put s
z_TestQueue._test_putc C s� d } | j d | � } | j � } | j � } | j d | j d | | | f � } d | _ | j � | j t | � d � | j t | | � d � | j d � | j d d � | j d d d � | j d
d � | j d d d � | j
d � t j t
� | j t | � d � | j t | | � d � t | j � } t | j
� } | j t j | d d � | j | j d
� | j t j | d d d � | j | j d
� | j t j | d � | j | j d
� | j t j | d d t � | j | j t � | j t j | d d t � | j | j d
� | j t j | d d d t �| j | j t � | j � | j � | j t | � d � | j t | | � d � | j � d S)Nr� r� rP r' TFr rZ r� � r� � r r )r^ r_ rQ r� rH rd r8 r� r� rU �
put_nowaitr& rl r{ r! r� �pyqueueZFullr6 r# �TIMEOUT1�TIMEOUT2�TIMEOUT3r� r rg )r$ ZMAXSIZEr� r� r� r� rU r� r
r
r �test_put sN
z_TestQueue.test_putc C sL | j � | j d � | j d � | j d � | j d � | j � d S)NrZ r� r� r� )r rU r� )rX r� r� r� r
r
r � _test_getA s
z_TestQueue._test_getc C sP | j � } | j � } | j � } | j d | j d | | | f � } d | _ | j � | j t | � d � | j � | j � t
j t � | j t | � d � | j | j
d d � d � | j | j
d � d � | j | j
d d � d � | j | j � d
� | j t | � d � t | j
� } t | j � } | j t j | d � | j | j d � | j t j | d d � | j | j d � | j t j | � | j | j d � | j t j | d t � | j | j t � | j t j | d t � | j | j d � | j t j | d t �| j | j t � | j � d S) NrP r' TFrZ r� r r r� r� r )r^ r_ rQ r� rH rd r8 r� r� r r& rl r{ rf �
get_nowaitr! r� r� �Emptyr6 r# r� r� r� rg )r$ r� r� r� r� rf r� r
r
r �test_getK sB
z_TestQueue.test_getc C s+ x$ t d d � D] } | j | � q Wd S)Nrq � )r~ rU )rX r� r� r
r
r �
_test_fork s z_TestQueue._test_forkc C s� | j � } x! t d � D] } | j | � q Wt j t � | j d | j d | f � } d | _ | j � x* t d � D] } | j
| j � | � q{ W| j t
j | j d � | j � d S)Nrq rP r' Tr� F)r^ r~ rU r& rl r{ rQ r� rH rd r8 rf r� r� r� rg )r$ r� r� ri r
r
r � test_fork� s
z_TestQueue.test_forkc C s� | j � } y | j | j � d � Wn t k
rG | j d � Yn X| j d � | j | j � d � | j d � | j | j � d � | j � | j | j � d � | j � | j | j � d � d S)Nr z qsize method not implementedr r� rZ )r^ r8 r� r7 rB rU rf )r$ rY r
r
r �
test_qsize� s
z_TestQueue.test_qsizec C s8 x1 t | j d � D] } t j t � | j � q Wd S)N)r� rf r& rl r{ Z task_done)rX rY �objr
r
r �_test_task_done� s
z_TestQueue._test_task_donec s� � j � � � � f d d � t d � D� } x! | D] } d | _ | j � q5 Wx! t d � D] } � j | � q_ W� j � x | D] } � j d � q� Wx | D] } | j � q� Wd S)Nc s. g | ]$ } � j d � j d � f � � q S)rP r' )rQ r� )�.0r� )r� r$ r
r �
<listcomp>� s z-_TestQueue.test_task_done.<locals>.<listcomp>r� Trq )�
JoinableQueuer~ rH rd rU rg )r$ Zworkersri r� r
)r� r$ r �test_task_done� s
z_TestQueue.test_task_donec C s� t j j � �� d } t | d d � � } | j d � Wd QXt j j t j � � �8 y t | � Wn" t j
k
r� | j d � Yn XWd QXWd QXd S)NZ imported_by_an_imported_modulez.pyr� z�if 1:
import multiprocessing
q = multiprocessing.Queue()
q.put('knock knock')
q.get(timeout=3)
q.close()
del q
z?Probable regression on import lock contention; see Issue #22853) r� r� Ztemp_cwdr� �writeZ
DirsOnSysPathr �getcwd�
__import__r� r� �fail)r$ Z module_namer� r
r
r � test_no_import_lock_contention� s
z)_TestQueue.test_no_import_lock_contentionc C sX t j � } t j � } | j t j | j d d � t j � | } | j | d � d S)NTg�������?g�(\���?)rw r^ r& r� r� r� rf �assertGreaterEqual)r$ rY rd �deltar
r
r � test_timeout� s
z_TestQueue.test_timeoutN)r+ r, r- r� r� r� r� r� r� r� r� r� r� r� r� r
r
r
r r� � s :
4 r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)� _TestLockc C sq | j � } | j | j � d � | j | j d � d � | j | j � d � | j t t j f | j � d S)NTF)�Lockr8 �acquire�releaser� r � threadingZ ThreadError)r$ �lockr
r
r � test_lock� s
z_TestLock.test_lockc C s� | j � } | j | j � d � | j | j � d � | j | j � d � | j | j � d � | j | j � d � | j | j � d � | j t t f | j � d S)NT)�RLockr8 r� r� r� �AssertionErrorro )r$ r� r
r
r �
test_rlock� s z_TestLock.test_rlockc C s | j � � Wd QXd S)N)r� )r$ r
r
r �test_lock_context s
z_TestLock.test_lock_contextN)r+ r, 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 S)
�_TestSemaphorec C s� | j d t | � | j | j � d � | j d t | � | j | j � d � | j d t | � | j | j d � d � | j d t | � | j | j � d � | j d t | � | j | j � d � | j d t | � d S)NrZ Tr r F)r; r? r8 r� r� )r$ �semr
r
r �_test_semaphore s z _TestSemaphore._test_semaphorec C sr | j d � } | j | � | j | j � d � | j d t | � | j | j � d � | j d t | � d S)NrZ r� r� )� Semaphorer� r8 r� r; r? )r$ r� r
r
r �test_semaphore s
z _TestSemaphore.test_semaphorec C s | j d � } | j | � d S)NrZ )�BoundedSemaphorer� )r$ r� r
r
r �test_bounded_semaphore s z%_TestSemaphore.test_bounded_semaphorec C s& | j d k r+ | j d j | j � � n | j d � } t | j � } | j | d � d � | j | j d � | j | d d � d � | j | j d � | j | d t � d � | j | j d � | j | d t
� d � | j | j t
� | j | d t � d � | j | j t � d S)Nr/ ztest not appropriate for {}r Fg Tr ) rA rB rC r� r! r� r8 r6 r# r� r� r� )r$ r� r� r
r
r r� % s z_TestSemaphore.test_timeoutN)r+ r, r- r� r� r� r� r
r
r
r r� s
r� c @ s� e Z d Z e d d d � � Z d d � Z d d � Z d d � Z d
d � Z e d d
� � Z e
j e d � d d � � Z
e d d � � Z e
j e d � d d � � Z e d d � � Z d d � Z d S)�_TestConditionNc C s9 | j � | j � | j | � | j � | j � d S)N)r� r� r )rX �cond�sleeping�wokenr r
r
r r� > s
z_TestCondition.fc C st | j d k rp yI | j j � | j j � } | j | d � | j | j j � d � Wqp t k
rl Yqp Xn d S)Nr/ r )rA Z_sleeping_countr? Z _woken_countr8 Z_wait_semaphorer7 )r$ r� Zsleepersr
r
r �check_invariantF s
z _TestCondition.check_invariantc C sc | j � } | j d � } | j d � } | j d | j d | | | f � } d | _ | j � t j d | j d | | | f � } d | _ | j � | j � | j � t j
t � | j d t
| � | j � | j � | j � t j
t � | j d t
| � | j � | j � | j � t j
t � | j d t
| � | j | � | j � d S)Nr rP r' Tr rZ )� Conditionr� rQ r� rH rd r� �Threadr� r& rl r{ r; r? �notifyr� r� rg )r$ r� r� r� ri r
r
r � test_notifyQ s2 $
$
z_TestCondition.test_notifyc C ss | j � } | j d � } | j d � } x� t d � D]z } | j d | j d | | | t f � } d | _ | j � t j d | j d | | | t f � } d | _ | j � q7 Wx t d � D] } | j
� q� Wx t d � D] } | j
� q� W| j d t | � | j
| � x� t d � D]t } | j d | j d | | | f � } d | _ | j � t j d | j d | | | f � } d | _ | j � q$Wx t d � D] } | j
� q�Wt j t � | j d t | � | j
� | j � | j � xQ t d � D]C } y t | � d k r'Pn Wn t k
r=PYn Xt j t � qW| j d t | � | j
| � d S)Nr r� rP r' Tr� rq )r� r� r~ rQ r� r� rH rd r� r� r� r; r? r� r& rl r{ Z
notify_allr� r7 )r$ r� r� r� r� ri r) r
r
r �test_notify_all| sR
$
$
z _TestCondition.test_notify_allc C sb | j � } t | j � } | j � | t � } | j � | j | d � | j | j t � d S)NF) r� r! r r� r� r� r8 r6 r# )r$ r� r r: r
r
r r� � s
z_TestCondition.test_timeoutc se | �Y d � _ | j � | j � f d d � � } | sK � j d k r[ t j d � n Wd QXd S)Nr c s
� j d k S)Nr� )r9 r
)�stater
r �<lambda>� s z0_TestCondition._test_waitfor_f.<locals>.<lambda>r� r )r9 r� �wait_forr� r� )rX r� r r� r
)r r �_test_waitfor_f� s
z _TestCondition._test_waitfor_fzneeds sharedctypesc s | j � } | j d d � � | j d | j d | � f � } d | _ | j � | �@ | j � f d d � � } | j | � | j � j d � Wd QXxG t
d � D]9 } t j d
� | � � j d 7_ | j
� Wd QXq� W| j d � | j | j � � | j | j d � d S)
Nr� r rP r' Tc s
� j d k S)Nr )r9 r
)r r
r r � s z-_TestCondition.test_waitfor.<locals>.<lambda>r r� g{�G�z�?r� r )r� r rQ r rH rd r rF r8 r9 r~ r& rl r� rg rS rG rM )r$ r� ri r� r� r
)r r � test_waitfor� s" !
z_TestCondition.test_waitforc s� | j � | � d } t j � } | j � f d d � d | �} t j � | } | r� | d | k oz | d k n r� d | _ n Wd QXd S)Ng�������?c s
� j d k S)Nr� )r9 r
)r r
r r � s z8_TestCondition._test_waitfor_timeout_f.<locals>.<lambda>r g333333�?g $@T)r� r& r r9 )rX r� r �successr� r� Zdtr� r
)r r �_test_waitfor_timeout_f� s
!+z&_TestCondition._test_waitfor_timeout_fc C s� | j � } | j d d � } | j d d � } | j d � } | j d | j d | | | | f � } d | _ | j � | j | j d d � � xG t d � D]9 } t
j d
� | � | j d 7_ | j
� Wd QXq� W| j d � | j | j � d S)
Nr� r FrP r' Tr rq r� g{�G�z�?r r� )r� r r� rQ r rH rd rF r� r~ r& rl r9 r� rg )r$ r� r r r� ri r� r
r
r �test_waitfor_timeout� s
z#_TestCondition.test_waitfor_timeoutc C sJ | � | j � Wd QXt j d � | d k rF t j | t j � n d S)Nr )r� r& rl r �killrt �SIGINT)rX �crW r
r
r �_test_wait_result s
z _TestCondition._test_wait_resultc
C s� t | t � r- t j d k r- t j � } n d } | j � } | �� | j | j d � � | j | j d � � | j d | j
d | | f � } | j � | j | j d � � | d k r� | j
t | j d � n | j � Wd QXd S)Nr
r g�������?rP r' rq )�
isinstance�ProcessesMixinr� �platformr rL r� rS r rQ r rd rF r� �KeyboardInterruptrg )r$ rW r ri r
r
r �test_wait_result s !
z _TestCondition.test_wait_result)r+ r, r- r� r� r� r� r r� r r �
skipUnless�HAS_SHAREDCTYPESr r r r r r
r
r
r r� < s + A r� c @ s. e Z d Z e d d � � Z d d � Z d S)�
_TestEventc C s t j t � | j � d S)N)r& rl r� r� )rX r� r
r
r � _test_event+ s
z_TestEvent._test_eventc C sD | j � } t | j � } | j | j � d � | j | d � d � | j | j d � | j | t � d � | j | j t � | j � | j | j � d � | j | � d � | j | j d � | j | t � d � | j | j d � | j � | j
d | j d | f � } d | _ | j
� | j | � d � d S)NFg TrP r' )r_ r! r r8 Zis_setr6 r# r� r� �clearrQ r rH rd )r$ r� r ri r
r
r �
test_event0 s$
z_TestEvent.test_eventN)r+ r, r- r� 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) �
_DummyListc C sK t j j t j d � � } t j � } | j | | f � d | j d <d S)Nr� r )rw �heap�
BufferWrapper�structZcalcsizer� � __setstate__�
_lengthbuf)r$ �wrapperr� r
r
r r% ] s z_DummyList.__init__c C s1 | \ | _ | _ | j j � j d � | _ d S)Nr� )�_wrapper�_lockZcreate_memoryview�castr )r$ r r
r
r r c s z_DummyList.__setstate__c C s | j | j f S)N)r r )r$ r
r
r � __getstate__g s z_DummyList.__getstate__c C s% | j � | j d d 7<Wd QXd S)Nr r )r r )r$ �_r
r
r r� j s
z_DummyList.appendc C s | j � | j d SWd QXd S)Nr )r r )r$ r
r
r �__len__n s
z_DummyList.__len__N)r+ r, r- r% r r" r� r$ r
r
r
r r [ s
r c C s t j d � d S)Ng{�G�z�?)r& rl r
r
r
r �_waitr s r% 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)�Bunchz
A bunch of threads.
Fc C s� | | _ | | _ | | _ | j � | _ | j � | _ | j � | _ | s^ | j j � n x< t | � D]. } | j
d | j � } d | _ | j
� qk Wd S)z�
Construct a bunch of `n` threads running the same function `f`.
If `wait_before_exit` is True, the threads won't terminate until
do_finish() is called.
rP TN)r� r' �n� DummyList�started�finishedr_ � _can_exitr� r~ rQ �taskrH rd )r$ � namespacer� r' r' Zwait_before_exitr� ri r
r
r r% { s zBunch.__init__c C sX t j � } | j j | � z | j | j � Wd | j j | � | j j d � Xd S)N� ) r rL r) r� r� r' r* r+ r )r$ rW r
r
r r, � s z
Bunch.taskc C s* x# t | j � | j k r% t � q Wd S)N)rK r) r' r% )r$ r
r
r �wait_for_started� s zBunch.wait_for_startedc C s* x# t | j � | j k r% t � q Wd S)N)rK r* r' r% )r$ r
r
r �wait_for_finished� s zBunch.wait_for_finishedc C s | j j � d S)N)r+ r� )r$ r
r
r � do_finish� s zBunch.do_finishN) r+ r, r- �__doc__r% r, r/ r0 r1 r
r
r
r r& w s
r& c @ s( e Z d Z d d � Z d d � Z d S)�
AppendTruec C s
| | _ d S)N)r� )r$ r� r
r
r r% � s zAppendTrue.__init__c C s | j j d � d S)NT)r� r� )r$ r
r
r r* � s zAppendTrue.__call__N)r+ r, r- r% r* r
r
r
r r3 � s r3 c @ s{ e Z d Z d Z d Z d Z d d � Z d d � Z d d � Z d
d � Z e
d d
� � Z d d d � Z d d � Z
e
d d � � Z d d � Z e
d d � � Z d d � Z e
d d � � Z d d � Z e
d d � � Z d! d" � Z e
d# d$ � � Z d% d& � Z e
d' d( � � Z d) d* � Z e
d+ d, � � Z d- d. � Z d/ d0 � Z e
d1 d2 � � Z d3 d4 � Z d5 S)6� _TestBarrierz$
Tests for Barrier objects.
r� g >@c C s" | j | j d | j �| _ d S)Nr )�Barrier�N�defaultTimeout�barrier)r$ r
r
r �setUp� s z_TestBarrier.setUpc C s | j j � d | _ d S)N)r8 �abort)r$ r
r
r �tearDown� s
z_TestBarrier.tearDownc C s: | j d k r g S| j d k r/ | j j � St � Sd S)Nr1 r0 )rA r0 rc r )r$ r
r
r r( � s
z_TestBarrier.DummyListc C s4 t | | | | j d � } | | � | j � d S)Nr )r& r6 r0 )r$ r� r' r5 r
r
r � run_threads� s
z_TestBarrier.run_threadsc C ss | j } xJ t | � D]< } | d j d � | j � | d j d � | j � q Wy Wn t k
rn Yn Xd S)Nr Tr )�partiesr~ r� r r7 )rX r8 �resultsr' �mr� r
r
r � multipass� s
z_TestBarrier.multipassr c C s; | j � | j � g } | j | j | j | | f � d S)z;
Test that a barrier is passed in lockstep
N)r( r< r@ r8 )r$ �passesr> r
r
r � test_barrier� s z_TestBarrier.test_barrierc C s
| j d � S)zC
Test that a barrier works for 10 consecutive runs
rq )rB )r$ r
r
r �test_barrier_10� s z _TestBarrier.test_barrier_10c C s | j � } | j | � d S)N)r rU )rX r8 r� r: r
r
r �_test_wait_return_f� s z _TestBarrier._test_wait_return_fc sg | j � � | j | j | j � f � � f d d � t | j � D� } | j | j d � d � d S)z9
test the return value from barrier.wait
c s g | ] } � j � � q Sr
)rf )r� r� )r� r
r r� � s z1_TestBarrier.test_wait_return.<locals>.<listcomp>r r N)r^ r< rD r8 r~ r6 r8 �count)r$ r> r
)r� r �test_wait_return� s "z _TestBarrier.test_wait_returnc C s) | j � t | � d k r% t � n d S)Nr )r rK ro )rX r8 r> r
r
r �_test_action_f� s
z_TestBarrier._test_action_fc C s] | j � } | j | j d t | � �} | j | j | | f � | j t | � d � d S)z,
Test the 'action' callback
�actionr N)r( r5 r6 r3 r< rG r8 rK )r$ r> r8 r
r
r � test_action� s z_TestBarrier.test_actionc C s� yC | j � } | | j d k r+ t � n | j � | j d � Wn= t j k
rg | j d � Yn t k
r� | j � Yn Xd S)NrZ T)r r6 ro r� r� �BrokenBarrierErrorr: )rX r8 �results1�results2r� r
r
r �
_test_abort_f s
z_TestBarrier._test_abort_fc C s� | j � } | j � } | j | j | j | | f � | j t | � d � | j t | � | j d � | j | j j � d S)zK
Test that an abort will put the barrier in a broken state
r r N) r( r< rM r8 r8 rK r6 rF Zbroken)r$ rK rL r
r
r �
test_abort s z_TestBarrier.test_abortc C s� | j � } | | j d k rV x' | j | j d k rH t j d � q" W| j � n@ y | j � | j d � Wn" t j k
r� | j d � Yn X| j � | j d � d S)NrZ r g����MbP?T) r r6 Z n_waitingr& rl �resetr� r� rJ )rX r8 rK rL �results3r� r
r
r �
_test_reset_f s
z_TestBarrier._test_reset_fc C s� | j � } | j � } | j � } | j | j | j | | | f � | j t | � d � | j t | � | j d � | j t | � | j � d S)zL
Test that a 'reset' on a barrier frees the waiting threads
r r N)r( r< rQ r8 r8 rK r6 )r$ rK rL rP r
r
r �
test_reset, s z_TestBarrier.test_resetc C s� yC | j � } | | j d k r+ t � n | j � | j d � Wn= t j k
rg | j d � Yn t k
r� | j � Yn X| j � | j d k r� | j � n | j � | j � | j d � d S)NrZ T)r r6 ro r� r� rJ r: rO )rX r8 �barrier2rK rL rP r� r
r
r �_test_abort_and_reset_f9 s
z$_TestBarrier._test_abort_and_reset_fc C s� | j � } | j � } | j � } | j | j � } | j | j | j | | | | f � | j t | � d � | j t | � | j d � | j t | � | j � d S)zF
Test that a barrier can be reset after being broken.
r r N)r( r5 r6 r< rT r8 r8 rK )r$ rK rL rP rS r
r
r �test_abort_and_resetO s z!_TestBarrier.test_abort_and_resetc C si | j � } | | j d k r/ t j d � n y | j d � Wn" t j k
rd | j d � Yn Xd S)NrZ g �?g �?T)r r6 r& rl r� rJ r� )rX r8 r> r� r
r
r �_test_timeout_f^ s z _TestBarrier._test_timeout_fc C sH | j � } | j | j | j | f � | j t | � | j j � d S)z$
Test wait(timeout)
N)r( r< rV r8 r8 rK r= )r$ r> r
r
r r� i s z_TestBarrier.test_timeoutc C sl | j | j � } | | j d k r5 t j d � n y | j � Wn" t j k
rg | j d � Yn Xd S)NrZ g �?T)r r7 r6 r& rl r� rJ r� )rX r8 r> r� r
r
r �_test_default_timeout_fq s z$_TestBarrier._test_default_timeout_fc C sZ | j | j d d �} | j � } | j | j | | f � | j t | � | j � d S)z4
Test the barrier's default timeout
r g �?N)r5 r6 r( r< rW r8 rK r= )r$ r8 r> r
r
r �test_default_timeout| s z!_TestBarrier.test_default_timeoutc C s' | j d � } | j � | j � d S)Nr )r5 r )r$ r5 r
r
r �test_single_thread� s
z _TestBarrier.test_single_threadc
C s? x8 t | � D]* } | j � | � | j | � Wd QXq
Wd S)N)r~ r r} )rX r8 rA �connr� r� r
r
r �_test_thousand_f� s
z _TestBarrier._test_thousand_fc C s� | j d k r+ | j d j | j � � n d } | j � } | j d � \ } } xK t | j � D]: } | j d | j d | j | | | f � } | j
� qb WxD t | � D]6 } x- t | j � D] } | j | j � | � q� Wq� Wd S)Nr0 ztest not appropriate for {}i� FrP r' )
rA rB rC r� r� r~ r6 rQ r[ r8 rd r8 r� )r$ rA r� rZ r� �jri r� r
r
r �
test_thousand� s z_TestBarrier.test_thousandN) r+ r, r- r2 r6 r7 r9 r; r( r< r� r@ rB rC rD rF rG rI rM rN rQ rR rT rU rV r� rW rX rY r[ r] r
r
r
r r4 � s6
r4 c @ s� e Z d Z d Z d d d d e d � e d
� f g Z d d � Z e d d � � Z d d d � Z d d � Z
d d � Z d S) �
_TestValuer/ r� � �^ �d�
@g @�h�� �� r �x�yc C s t s | j d � n d S)Nz%requires multiprocessing.sharedctypes)r rB )r$ r
r
r r9 � s z_TestValue.setUpc C s4 x- t | | j � D] \ } } | d | _ q Wd S)NrZ )�zip� codes_valuesr9 )rX �values�sv�cvr
r
r rR � s z_TestValue._testFc s� | r% � f d d � � j D� } n � f d d � � j D� } x7 t | � j � D]# \ } } � j | j | d � qT W� j d � j d | f � } d | _ | j � | j � x7 t | � j � D]# \ } } � j | j | d � q� Wd S) Nc s+ g | ]! \ } } } � j | | � � q Sr
)�RawValue)r� r� r9 r# )r$ r
r r� � s z)_TestValue.test_value.<locals>.<listcomp>c s+ g | ]! \ } } } � j | | � � q Sr
)r )r� r� r9 r# )r$ r
r r� � s r rP r' TrZ ) ri rh r8 r9 rQ rR rH rd rg )r$ �rawrj rk rl r� r
)r$ r �
test_value� s
z_TestValue.test_valuec C s | j d d � d S)Nrn T)ro )r$ r
r
r �
test_rawvalue� s z_TestValue.test_rawvaluec
C sK | j d d � } | j � } | j � } | j d d d d �} | j � } | j � } | j � } | j d d d | �} | j � } | j � }
| j | | � | j d d d d �} | j t | d � � | j t | d � � | j t | j d d d d �| j d d � } | j t | d � � | j t | d � � d S)Nr� r� r� F�get_lock�get_objZnavalue)
r rq rr r� r8 rS rs r� r rm )
r$ Zval1�lock1�obj1Zval2�lock2�obj2r� Zval3�lock3�obj3�arr4�arr5r
r
r �test_getobj_getlock� s$ z _TestValue.test_getobj_getlockN)z processes)r� r_ r` � �)ra rb r| ����)rc r} re ) r+ r, r- r= r ri r9 r� rR ro rp r{ r
r
r
r r^ � s r^ c @ s� e Z d Z d Z e d d � � Z e j e d k d � d d d � � Z e j e d k d � d d
� � Z
e j e d k d � d d � � Z e j e d k d � d
d � � Z d S)�
_TestArrayr/ c C s: x3 t d t | � � D] } | | | | d 7<q Wd S)Nr )r~ rK )rX �seqr� r
r
r r� � s z _TestArray.fNzrequires _ctypesFc
C sw d d d d d d d d d d
g
} | r? | j d | � } n | j d | � } | j t | � t | � � | j | d | d � | j t | d
d � � t | d
d � � � t j d d d
d d g � | d d � <| d d � <| j t | d d � � | � | j | � | j d | j d | f � } d | _ | j � | j
� | j t | d d � � | � d S)Ni� ir i� i5 � �� i$ i� i� i? r� r� rZ r� r r� r� rP r' T) �RawArray�Arrayr8 rK rc �arrayr� rQ rH rd rg )r$ rn r �arrri r
r
r �
test_array� s $ 06
z_TestArray.test_arrayc C s� d } x� t d � D]� } | j d | � } | j t | � | � | j t | � d g | � t d � | d d � <| j t | � t t d � � � ~ q Wd S)Nrq r� r� r )r~ r� r8 rK rc )r$ �sizer# r� r
r
r �test_array_from_size s "z _TestArray.test_array_from_sizec C s | j d d � d S)Nrn T)r� )r$ r
r
r �
test_rawarray s z_TestArray.test_rawarrayc
C s� | j d t t d � � � } | j � } | j � } | j d t t d � � d d �} | j � } | j � } | j � } | j d t t d � � d | �} | j � } | j � }
| j | | � | j d t d � d d �} | j t | d � � | j t | d � � | j t
| j d t d � d d �| j d t d � � } | j t | d � � | j t | d � � d S)Nr� rq r� Frq rr Znotalock) r� rc r~ rq rr r� r8 rS rs r� r r� )
r$ Zarr1rs rt Zarr2ru rv r� Zarr3rw rx ry rz r
r
r �test_getobj_getlock_obj s& $ $ z"_TestArray.test_getobj_getlock_obj)z processes)
r+ r, r- r= r� r� r �skipIfr r� r� r� r� r
r
r
r r~ � s $$r~ c @ s: e Z d Z d Z d d � Z d d � Z d d � Z d S)
�_TestContainersr0 c C s[ | j t t d � � � } | j | d d � t t d � � � | j � } | j | d d � g � | j t t d � � � | j | d d � t t d � � � | j | d d � | j | d d � d d d g � | d 9} | j | d d � d d d d d d d d d d g
� | j | d d g d d d d d d d d d d d d g � | j | d d � t t d � � � | | g } | j | � } | j | d d � d d d d d d d d d
d g
d d d d d d d d d d g
g � | j | g � } | j d � | j | d d � d d d d d d d d d
d d g g � d S)
Nrq r� rZ r� r� r r r� r� r� � r\ )rc r~ r8 �extendr� )r$ r4 r5 ra rh r� r
r
r � test_list@ s( & &#
8>&
I
z_TestContainers.test_listc C s� | j � } t t d d � � } x | D] } t | � | | <q( W| j | j � t d d � | D� � � | j t | j � � | � | j t | j � � d d � | D� � | j t | j � � d d � | D� � d S)N�A �F c s s! | ] } | t | � f Vq d S)N)�chr)r� r� r
r
r � <genexpr>d s z,_TestContainers.test_dict.<locals>.<genexpr>c S s g | ] } t | � � q Sr
)r� )r� r� r
r
r r� f s z-_TestContainers.test_dict.<locals>.<listcomp>c S s" g | ] } | t | � f � q Sr
)r� )r� r� r
r
r r� g s )
�dictrc r~ r� r8 r �sorted�keysrj �items)r$ ra �indicesr� r
r
r � test_dict_ s
) )z_TestContainers.test_dictc C s� | j � } d | _ d | _ d | _ | j | j | j f d � | ` | j t | � d � | j t | d � � | j t | d � � d S)N�Bob�BuilderZhiddenzNamespace(name='Bob')rV �job)r� r� )� NamespacerV r� Z_hiddenr8 r� rF rs )r$ r' r
r
r �test_namespacei s z _TestContainers.test_namespaceN)zmanager)r+ r, r- r= r� r� r� r
r
r
r r� < s
r� g c C s t j | � | | S)N)r& rl )rf r r
r
r �sqrx s
r� c C s | | S)Nr
)rf rg r
r
r �mul| s r� c @ s e Z d Z d S)� SayWhenErrorN)r+ r, r- r
r
r
r r� s r� c c s; x4 t | � D]& } | | k r. t d � � n | Vq
Wd S)NzSomebody said when)r~ r� )ZtotalZwhenr� r
r
r � exception_throwing_generator� s r� c sT e Z d Z e � f d d � � Z e � f 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 e d) d* � � Z d+ d, � Z e d- d. � � Z d/ d0 � Z � S)1� _TestPoolc s# t � j � | j d � | _ d S)Nr� )�super�
setUpClass�Pool�pool)rX )� __class__r
r r� � s
z_TestPool.setUpClassc s4 | j j � | j j � d | _ t � j � d S)N)r� rr rg r� �
tearDownClass)rX )r� r
r r� � s
z_TestPool.tearDownClassc C s[ | j j } | j | t d � t d � � | j | t f i d d 6� t d d � � d S)Nr� r� rf )r� )r� �applyr8 r� )r$ Zpapplyr
r
r �
test_apply� s z_TestPool.test_applyc C s� | j j } | j | t t t d � � � t t t t t d � � � � � | j | t t t d � � d d �t t t t t d � � � � � d S)Nrq rk � chunksizer� )r� �mapr8 r� rc r~ )r$ Zpmapr
r
r �test_map� s @$z_TestPool.test_mapc C s� | j j } t t t d � t d d d � � � } | j | t | � t t j t | � � � t t t d � t d d
d � � � } | j | t | d d �t t j t | � � � d S) Nrq r� r rk �c r� r� r r r r )r� �starmaprc rh r~ r8 r� � itertools)r$ Zpsmap�tuplesr
r
r � test_starmap� s ''z_TestPool.test_starmapc C sb t t t d � t d d d � � � } | j | j j t | � j � t t j t | � � � d S)Nrk r� r r r )
rc rh r~ r8 r� Z
starmap_asyncr� rf r� r� )r$ r� r
r
r �test_starmap_async� s ' z _TestPool.test_starmap_asyncc C sP | j | j j t t t d � � � j � t t t t t d � � � � � d S)Nrq )r8 r� � map_asyncr� rc r~ rf r� )r$ r
r
r �test_map_async� s *z_TestPool.test_map_asyncc C s� | j d k r | j j � n g } | j j t d g d | j d | j �j � | j d t | � � | j d g | d � | j j t d g d | j d | j �j � | j d t | � � | j
| d t � d S) Nr0 �1�callback�error_callbackr r r4 rZ ) rA r0 rc r� r� ry r� r r8 rK rI r )r$ � call_argsr
r
r �test_map_async_callbacks� s $ z"_TestPool.test_map_async_callbacksc C sx | j d k r+ | j d j | j � � n Gd d � d t � } | j t � �" | j j t | � g d � Wd QXd S)Nr1 ztest not appropriate for {}c @ s e Z d Z d d � Z d S)z*_TestPool.test_map_unplicklable.<locals>.Ac S s t d � � d S)Nz
cannot pickle)ro )r$ r
r
r r< � s z5_TestPool.test_map_unplicklable.<locals>.A.__reduce__N)r+ r, r- r< r
r
r
r �A� s r� rq ) rA rB rC �objectr� ro r� r� r� )r$ r� r
r
r �test_map_unplicklable� s
z _TestPool.test_map_unplicklablec C sR y) | j j t g d d �j d t � Wn" t j k
rM | j d � Yn Xd S)Nr� r r z2pool.map_async with chunksize stalled on null list)r� r� r� rf r� rw � TimeoutErrorr� )r$ r
r
r �test_map_chunksize� s )z _TestPool.test_map_chunksizec C sT | j j t d t f � } t | j � } | j | � d � | j | j t � d S)Nr� �1 ) r� � apply_asyncr� r� r! rf r8 r6 r# )r$ r: rf r
r
r �
test_async� s z_TestPool.test_asyncc C s^ | j j t d t d f � } t | j � } | j t j | d t �| j | j
t � d S)Nr� g �?r ) r� r� r� r� r! rf r� rw r� r6 r# )r$ r: rf r
r
r �test_async_timeout� s z _TestPool.test_async_timeoutc C s& | j j t t t d � � � } | j t | � t t t t t d � � � � � | j j t t t d � � � } x. t d � D] } | j t | � | | � q� W| j t | j
� | j j t t t d � � d d �} x. t d � D] } | j t | � | | � q� W| j t | j
� d S)Nrq i� r� rk ) r� �imapr� rc r~ r8 r� �nextr� �
StopIteration�__next__)r$ �itr� r
r
r � test_imap� s !1! ' z_TestPool.test_imapc C s^ | j d k r+ | j d j | j � � n | j j t t d d � d � } x. t d � D] } | j t | � | | � qY W| j
t | j � | j j t t d d � d � } x. t d � D] } | j t | � | | � q� W| j
t | j � | j j t t d d � d
� } x. t d
� D] } | j t | � | | � q#W| j
t | j � d S) Nr0 ztest not appropriate for {}rq r� r r� r� rZ r� r� )
rA rB rC r� r� r� r� r~ r8 r� r� r� r� )r$ r� r� r
r
r �#test_imap_handle_iterable_exception� s ! ! ! z-_TestPool.test_imap_handle_iterable_exceptionc C s� | j j t t t d � � � } | j t | � t t t t t d � � � � � | j j t t t d � � d d �} | j t | � t t t t t d � � � � � d S)Ni� r� �5 )r� �imap_unorderedr� rc r~ r8 r� r� )r$ r� r
r
r �test_imap_unordered� s !1'z _TestPool.test_imap_unorderedc C s_ | j d k r+ | j d j | j � � n | j j t t d d � d � } t t t t t d � � � � } | j
t � �E x= t d � D]/ } t | � } | j
| | � | j | � q� WWd QX| j j t t d d � d � } t t t t t d � � � � } | j
t � �E x= t d � D]/ } t | � } | j
| | � | j | � q"WWd QXd S) Nr0 ztest not appropriate for {}rq r� r r� r� rZ )rA rB rC r� r� r� r� rc r� r~ r� r� r� re �remove)r$ r� Zexpected_valuesr� r9 r
r
r �-test_imap_unordered_handle_iterable_exception s( ! ! z7_TestPool.test_imap_unordered_handle_iterable_exceptionc C sl | j t t j d � | j t t j d � t j d � } | j d t | j � � | j � | j � d S)Nr r r� r ) r� r rw r� r8 rK �_poolr� rg )r$ ri r
r
r �test_make_pool s
z_TestPool.test_make_poolc C sn | j j t j d d � t d � D� d d �} | j j � t | j j � } | � | j | j d � d S)Nc S s g | ] } d � q S)g�������?r
)r� r� r
r
r r� ' s z,_TestPool.test_terminate.<locals>.<listcomp>i' r� r g �?)
r� r� r& rl r~ rr r! rg �
assertLessr# )r$ r� rg r
r
r rv % s (
z_TestPool.test_terminatec C s� | j d � } | j | j t g � g � | j t | j t g � � g � | j t | j t g � � g � | j | j t g � j � g � | j � | j
� d S)Nr ) r� r8 r� r� rc r� r� r� rf r� rg )r$ ri r
r
r �test_empty_iterable. s """
z _TestPool.test_empty_iterablec C s� | j d k r� t t d � � } d d � | D� } t j d � �/ } | j t | � } | j | j � | � Wd QX| j t
| j t | � n d S)Nr/ rq c S s g | ] } t | � � q Sr
)r� )r� r� r
r
r r� = s z*_TestPool.test_context.<locals>.<listcomp>rZ ) rA rc r~ rw r� r� r� r8 rf r� r )r$ �Lr� ri r� r
r
r � test_context: s z_TestPool.test_contextc C s t d � � d S)N�{ )ro )rX r
r
r �_test_tracebackC s z_TestPool._test_tracebackc & C s= | j d k r9| j d � �U } y | j | j � Wn+ t k
rb } z | } WYd d } ~ Xn
Xt d � � Wd QX| j t | � t � | j | j
d � | j } | j t | � t j
j � | j d | j � t j j � �9 } y
| � Wn% t k
rt j t j � � Yn XWd QX| j d | j � � n d S)Nr/ r zexpected RuntimeErrorr� z&raise RuntimeError(123) # some comment)r� )rA r� r� r� � Exceptionr� �assertIsrb ro r8 r' � __cause__rw r� ZRemoteTracebackre �tbr� r� Zcaptured_stderrr� �
excepthook�exc_info�getvalue)r$ ri rh �exc�cause�f1r
r
r �test_tracebackG s&
z_TestPool.test_tracebackc C s t d � � d S)N�foo)ro )rX r
r
r �_test_wrapped_exception` s z!_TestPool._test_wrapped_exceptionc C sB | j d � �- } | j t � � | j | j � Wd QXWd QXd S)Nr )r� r� ro r� r� )r$ ri r
r
r �test_wrapped_exceptiond s z _TestPool.test_wrapped_exception) r+ r, r- r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rv r� r� r� r� r� r� r
r
)r� r r� � s0
r� c C s t d � � d S)N�key)�KeyErrorr
r
r
r �raisingk s r� c C s
d d � S)Nc S s d S)N�* r
r
r
r
r r o s z%unpickleable_result.<locals>.<lambda>r
r
r
r
r �unpickleable_resultn s r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�_TestPoolWorkerErrorsr/ c s� t j d � } d g � � f d d � } | j t d | �} | j t | j � | j � d � | j � d t � | j � | j
� d S)NrZ c s | � d <d S)Nr r
)r� )�
scratchpadr
r �errbackx s z@_TestPoolWorkerErrors.test_async_error_callback.<locals>.errbackr� r ) rw r� r� r� r� r� rf rF rI r� rg )r$ ri r� r: r
)r� r �test_async_error_callbackt s
z/_TestPoolWorkerErrors.test_async_error_callbackc s� d d l m } t j d � } x� t d � D]� } d g � � f d d � } | j t d | �} | j | | j � � d } | j | � | j
� d | � | j | j � | j | j
� q, W| j � | j � d S)Nr )�MaybeEncodingErrorrZ r� c s | � d <d S)Nr r
)r� )r� r
r r� � s z?_TestPoolWorkerErrors.test_unpickleable_result.<locals>.errbackr� )�multiprocessing.poolr� rw r� r~ r� r� r� rf rF rI ZassertIsNotNoner� r9 r� rg )r$ r� ri Z iterationr� r: �wrappedr
)r� r �test_unpickleable_result� s
z._TestPoolWorkerErrors.test_unpickleable_resultN)z processes)r+ r, r- r= r� r� r
r
r
r r� q s r� c @ s. e Z d Z d Z d d � Z d d � Z d S)�_TestPoolWorkerLifetimer/ c C sq t j d d d �} | j d t | j � � d d � | j D� } g } x0 t d � D]" } | j | j t | f � � qW Wx6 t | � D]( \ } } | j | j
� t | � � q� W| j � d } x> | rt d d � | j D� � r| d
8} t
j t � q� Wd d � | j D� } | j d | � | j d | � | j t | � t | � � | j � | j � d S) Nr� �maxtasksperchildrq c S s g | ] } | j � q Sr
)rW )r� r� r
r
r r� � s zE_TestPoolWorkerLifetime.test_pool_worker_lifetime.<locals>.<listcomp>rk �2 c s s | ] } | j � Vq d S)N)rG )r� r� r
r
r r� � s zD_TestPoolWorkerLifetime.test_pool_worker_lifetime.<locals>.<genexpr>r c S s g | ] } | j � q Sr
)rW )r� r� r
r
r r� � s )rw r� r8 rK r� r~ r� r� r� � enumeraterf Z_repopulate_pool�allr& rl r{ r` ZassertNotEqualr� r� rg ) r$ ri Zorigworkerpidsr> r� r\ r: Z countdownZfinalworkerpidsr
r
r �test_pool_worker_lifetime� s&
&
z1_TestPoolWorkerLifetime.test_pool_worker_lifetimec C s� t j d d d �} g } x3 t d � D]% } | j | j t | d f � � q( W| j � | j � x6 t | � D]( \ } } | j | j
� t | � � qr Wd S)Nr� r� r r� g333333�?) rw r� r~ r� r� r� r� rg r� r8 rf )r$ ri r> r� r\ r: r
r
r �%test_pool_worker_lifetime_early_close� s #
z=_TestPoolWorkerLifetime.test_pool_worker_lifetime_early_closeN)z processes)r+ r, r- r= r� r� r
r
r
r r� � s r� )� BaseManager� BaseProxy� RemoteErrorc @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�FooBarc C s d S)Nzf()r
)r$ r
r
r r� � s zFooBar.fc C s
t � d S)N)r )r$ r
r
r �g� s zFooBar.gc C s d S)Nz_h()r
)r$ r
r
r �_h� s z FooBar._hN)r+ r, r- r� r� r r
r
r
r r� � s r� c c s$ x t d � D] } | | Vq
Wd S)Nrq )r~ )r� r
r
r �baz� s r c @ s. e Z d Z d Z d d � Z d d � Z d S)�
IteratorProxyr� c C s | S)Nr
)r$ r
r
r �__iter__� s zIteratorProxy.__iter__c C s
| j d � S)Nr� )� _callmethod)r$ r
r
r r� � s zIteratorProxy.__next__N)z__next__)r+ r, r- Z _exposed_r r� r
r
r
r r � s r c @ s e Z d Z d S)� MyManagerN)r+ r, r- r
r
r
r r � s r �Foo�callable�BarZexposedr� r Z proxytypec @ sF e Z d Z d Z d d � Z d d � Z d d � Z d d � Z d
S) �_TestMyManagerr0 c C sD t � } | j � | j | � | j � | j | j j d � d S)Nr )r rd �common�shutdownr8 �_processrM )r$ r0 r
r
r �test_mymanager� s
z _TestMyManager.test_mymanagerc C s9 t � � } | j | � Wd QX| j | j j d � d S)Nr )r r
r8 r rM )r$ r0 r
r
r �test_mymanager_context� s z%_TestMyManager.test_mymanager_contextc C sG t � } | j � | � | j | � Wd QX| j | j j d � d S)Nr )r rd r
r8 r rM )r$ r0 r
r
r �!test_mymanager_context_prestarted� s
z0_TestMyManager.test_mymanager_context_prestartedc se | j � � | j � � | j � } � f d d � d D� } � f d d � d D� } | j | d d g � | j | d d g � | j � j � d � | j t � j � | j � j d � d � | j t � j d � | j � j � d � | j � j
� d � | j � j d � d � | j � j d � d � | j t | � d d � t d
� D� � d S)
Nc s% g | ] } t � | � r | � q Sr
)rs )r� rV )r� r
r r� s z)_TestMyManager.common.<locals>.<listcomp>r� r� r c s% g | ] } t � | � r | � q Sr
)rs )r� rV )�barr
r r� s zf()z_h()c S s g | ] } | | � q Sr
r
)r� r� r
r
r r� s rq )r� r� �_h)r� r� r )
r r r r8 r� r� r r� r r� r rc r~ )r$ r0 r Z foo_methodsZ bar_methodsr
)r r� r r
s z_TestMyManager.commonN)zmanager)r+ r, r- r= r
r r r
r
r
r
r r � s
r c C s t S)N)�_queuer
r
r
r � get_queue s r c @ s e Z d Z d Z d S)� QueueManagerz$manager class used by server processN)r+ r, r- r2 r
r
r
r r " s r c @ s e Z d Z d Z d S)�
QueueManager2z@manager class which specifies the same interface as QueueManagerN)r+ r, r- r2 r
r
r
r r & s r Z xmlrpclibc @ s_ e Z d Z d
Z d d d d d d d g Z e d d � Z e d d
� � Z d d � Z d S)�_TestRemoteManagerr0 z hello worldNTg @u hallå världenu привіт світs
hall� v�rldenc C sK t d | d | d t � } | j � | j � } | j t | j � � d S)N�addressrE �
serializer)r �
SERIALIZER�connectr rU �tuplerj )rX r rE r0 r� r
r
r �_putter7 s
z_TestRemoteManager._putterc C s� t j d � } t d t j j d f d | d t � } | j � | j d | j d | j
| f � } d | _ | j � t d | j
d | d t � } | j
� | j � } | j | j � | j � | j t | j t j � ~ | j � d S) N� r r rE r rP r' T)r �urandomr r� r� �HOSTr rd rQ r r rH r r r r8 rf r� r� r� rU r& rl r )r$ rE r0 ri Zmanager2r� r
r
r � test_remoteA s !
$
z _TestRemoteManager.test_remote)zmanager) r+ r, r- r= rj r� r� r r r
r
r
r r - s
r c @ s. e Z d Z e d d � � Z d d � Z d S)�_TestManagerRestartc C sB t d | d | d t � } | j � | j � } | j d � d S)Nr rE r z hello world)r r r r rU )rX r rE r0 r� r
r
r r _ s
z_TestManagerRestart._putterc C sg t j d � } t d t j j d f d | d t � } | j � } | j } | j j
� | j � | j d | j
d | j | f � } d | _ | j � | j � } | j | j � d � ~ | j � t d | d | d t � } y | j � Wne t k
rX} zE | j t j k r � n t j d
� t d | d | d t � } WYd d } ~ Xn X| j � d S) Nr r r rE r rP r' Tz hello worldg �?)r r r r� r� r r Z
get_serverr Zlistenerr� rd rQ r rH r r8 rf r �OSError�errnoZ
EADDRINUSEr& rl )r$ rE r0 Zsrvr�addrri r� rh r
r
r �test_rapid_restartg s2 !
$
(z&_TestManagerRestart.test_rapid_restartN)r+ r, r- r� r r% r
r
r
r r! ] s r! � c @ s` e Z d Z d% Z e d d � � Z d d � Z d d � Z d d
� Z d d � Z e d
d � � Z
e d d d � � Z e j
e d � d d � � Z e j
e d � e j e j d k d � e j e d k d � e j
e e d � d � d d � � � � � Z e d d � � Z e j
e d � e j e j d k d � d d! � � � Z d" d# � Z d$ S)&�_TestConnectionr/ r1 c C s8 x' t | j t � D] } | j | � q W| j � d S)N)r� �
recv_bytes�SENTINEL�
send_bytesr� )rX rZ �msgr
r
r �_echo� s z_TestConnection._echoc C s� | j � \ } } | j d | j d | f � } d | _ | j � d d d g } t d � } | d } t j d t t d � � � } | j d
k r� | j
t | j � � t
� n | j
| j | � d � | j
| j � | � | j
| j | � d � | j
| j � | � | j d
k r�t j d d g d � } t | � d g d t | � } | j
| j | � d � | j
| j | � t | � | j � | j
t | � | � t j d d g d � } d g d t | � d g d t | � } | j
| j | � d � | j
| j | d | j � t | � | j � | j
t | � | � t t d
d � � } | j
| j | � d � y | j | � }
Wn> t j k
r�} z | j
| j | f � WYd d } ~ Xq�X| j d |
� n t | j � } | j
| � d � | j | j d � | j
| d � d � | j | j d � | j
| t � d � | j | j t � | j d � t j d � | j
| t � d � | j | j d � | j
| j � d � t d � d }
| j |
� | j
| j � |
� | j t! � | j" � | j d
k rs| j
| j# d � | j
| j$ d � | j% t& | j � | j% t& | j � n | j' � d S)NrP r' Tr g @z hello worldrq r� r� r/ r r� � �( z expected BufferTooShort, got %sFg�������?�Xi � r� r i i )(r� rQ r, rH rd r r� rc r~ rA r8 rb �filenory r} r� r* r( rK Zrecv_bytes_into�itemsize� bytearrayrw ZBufferTooShortr' r� r! r� r6 r# r� r& rl r) r� �readable�writabler� �EOFErrorrg )r$ rZ r� ri r r+ Zlongmsgr� �bufferr� r: rh r� Zreally_big_msgr
r
r �test_connection� sr
!, )
z _TestConnection.test_connectionc C s� | j d d � \ } } | j | j d � d � | j | j � d � | j d k r� | j | j d � | j | j d � | j | j d � | j | j d � | j t | j d � | j t | j � | j t | j � n d S)Nr� Fr r/ TrZ )
r� r8 r} r� rA r4 r5 r� r" r� )r$ �reader�writerr
r
r �test_duplex_false� s z!_TestConnection.test_duplex_falsec C s� | j � \ } } | j d | j d | f � } d | _ | j � | j � t d � } | j | � | j | j � | � | j t
� | j � | j � d S)NrP r' Tr\ ) r� rQ r, rH rd r� r r* r8 r( r) rg )r$ rZ r� ri r+ r
r
r �test_spawn_close� s
z _TestConnection.test_spawn_closec C s� | j d k r+ | j d j | j � � n t d � } | j � \ } } | j | � | j | j � | � | j | d � | j | j � | d d � � | j | d d � | j | j � | d d � � | j | d � | j | j � t d � � | j | d d � | j | j � t d � � | j t | j | d
� | j t | j | d d � | j t | j | d d � | j t | j | d � | j t | j | d
d � d S)Nr/ ztest not appropriate for {}Zabcdefghijklmnopqrstuvwxyzr� r� r� � r& r � � r r� � r r )
rA rB rC r r� r* r8 r( r� r )r$ r+ r4 r5 r
r
r �test_sendbytes s&
z _TestConnection.test_sendbytesc C sZ y t j | � Wn> t k
rQ } z | j t j k r< d S� WYd d } ~ Xn Xd Sd S)NFT)r �fstatr" r# �EBADF)rX �fdrh r
r
r �_is_fd_assigned( s z _TestConnection._is_fd_assignedFc C s� | rK xB t d d � D]. } | j | � s t j | j � | � q q Wn t j | � } t rx t j | t j � } n t j
| | � t j | � d S)Nr r ) r~ rE r �dup2r1 r � recv_handle�msvcrtZopen_osfhandle�O_WRONLYr� r� )rX rZ �dataZcreate_dummy_fdsr� rD r
r
r �_writefd3 s z_TestConnection._writefdz$test needs multiprocessing.reductionc C s! | j d k r | j d � n | j d d � \ } } | j d | j d | d f � } d | _ | j � | j t j j
t j j � t t j j d � �A } | j
� } t r� t j | � } n t j | | | j � Wd QX| j � t t j j d � � } | j | j � d � Wd QXd S)
Nr/ z only makes sense with processesr� TrP r' s foo�wb�rb)rA rB r� rQ rK rH rd r� r� r� r� r� r� r1 rH Z
get_osfhandler � send_handlerW rg r8 r� )r$ rZ r� ri r� rD r
r
r �test_fd_transfer? s !
z _TestConnection.test_fd_transferr
z*test semantics don't make sense on Windowsr z)largest assignable fd number is too smallrF ztest needs os.dup2()c C sk | j d k r | j d � n | j d d � \ } } | j d | j d | d d f � } d | _ | j � | j t j j
t j j � t t j j d � �� } | j
� } x7 t d t � D] } | j | � s� Pq� q� W| j d
� t j | | � z t j | | | j � Wd t j | � XWd QX| j � t t j j d � � } | j | j � d � Wd QXd S) Nr/ z only makes sense with processesr� TrP r' s barrL r z2could not find an unassigned large file descriptorrM )rA rB r� rQ rK rH rd r� r� r� r� r� r� r1 r~ �MAXFDrE r� r rF r rN rW r� rg r8 r� )r$ rZ r� ri r� rD Znewfdr
r
r �test_large_fd_transferR s( $
z&_TestConnection.test_large_fd_transferc C s t j | j � d � d S)Ns )r r� r1 )r$ rZ r
r
r �_send_data_without_fds s z%_TestConnection._send_data_without_fdz doesn't make sense on Windowsc C s� | j d k r | j d � n | j d d � \ } } | j d | j d | f � } d | _ | j � | j t t j
| � | j � d S)Nr/ z only makes sense with processesr� TrP r' ) rA rB r� rQ rR rH rd r� ro r rG rg )r$ rZ r� ri r
r
r �test_missing_fd_transferw s
z(_TestConnection.test_missing_fd_transferc C s� | j � \ } } | �g | �Z | j d � | j | j � d � | j d k ru | j | j � | j | j � n Wd QXWd QX| j d k r� | j | j � | j | j � | j t | j � | j t | j � n d S)Ni� r/ )
r� r} r8 r� rA rS �closedrF r� r" )r$ r4 r5 r
r
r r� � s
z _TestConnection.test_contextN)z processeszthreads)r+ r, r- r= r� r, r8 r; r< rA rE rK r r �
HAS_REDUCTIONrO r� r� r rP rs r rQ rR rS r� r
r
r
r r' � s* O
*r' c @ s. e Z d Z d Z d d � Z d d � Z d S)�
_TestListenerr/ c C s_ xX | j j D]J } | j j d | � } | j | j � | j t | j j | j | � q
Wd S)N�family)�
connection�families�Listenerr� r� r� r" r )r$ rW �lr
r
r �test_multiple_bind� s
z _TestListener.test_multiple_bindc C s� | j j � �] } | j j | j � �? } | j � �* } | j d � | j | j � d � Wd QXWd QXWd QX| j d k r� | j t
| j � n d S)Ni� r/ ) rX rZ �Clientr �acceptr} r8 r� rA r� r" )r$ r[ r ra r
r
r r� � s
(z_TestListener.test_contextN)z processes)r+ r, r- r= r\ r� r
r
r
r rV � s rV c @ sL e Z d Z d Z e d d � � Z d d � Z d d � Z d d
� Z d S)
�_TestListenerClientr/ r1 c C s- | j j | � } | j d � | j � d S)Nr\ )rX r] r} r� )rX r rZ r
r
r rR � s
z_TestListenerClient._testc C s� x� | j j D]� } | j j d | � } | j d | j d | j f � } d | _ | j � | j � } | j | j
� d � | j � | j � q
Wd S)NrW rP r' Tr\ )
rX rY rZ rQ rR r rH rd r^ r8 r� rg r� )r$ rW r[ ri rZ r
r
r �test_listener_client� s !
z(_TestListenerClient.test_listener_clientc C s� | j j � } | j d | j d | j f � } d | _ | j � t j d � | j � } | j
| j � d � | j � | j
� | j � d S)NrP r' Tr r\ )rX rZ rQ rR r rH rd r& rl r^ r8 r� r� rg )r$ r[ ri rZ r
r
r �test_issue14725� s !
z#_TestListenerClient.test_issue14725c C s� x� | j j D]} } | j j d | � } | j j | j � } | j � } | j d � | j | j d � � | j � | j � | j � q
Wd S)NrW s hellor )
rX rY rZ r] r r^ r* rF r� r� )r$ �famr[ r r4 r
r
r �test_issue16955� s
z#_TestListenerClient.test_issue16955N)z processeszthreads) r+ r, r- r= r� rR r` ra rc r
r
r
r r_ � s
r_ c @ s| e Z d Z d Z d d � Z e d d � � Z d d � Z e d d
� � Z d d � Z e d
d � � Z
d d � Z d S)� _TestPollr/ r1 c C se | j � \ } } | j | j � d � | j d � | j | j � d � | j | j � d � d S)NF� T)r� r8 r� r* )r$ r4 r5 r
r
r �test_empty_string� s
z_TestPoll.test_empty_stringc C s9 x( | D] } t j d � | j | � q W| j � d S)Ng�������?)r& rl r* r� )rX rZ �stringsr r
r
r �_child_strings� s
z_TestPoll._child_stringsc C s� d } | j � \ } } | j d | j d | | f � } | j � xT | D]L } x' t d � D] } | j d
� r] Pq] q] W| j � } | j | | � qJ W| j � d S) N� hellore � a� b� bye� loprP r' �� g{�G�z�?)ri re rj rk re rl re rm ) r� rQ rh rd r~ r� r( r8 rg )r$ rg r4 r5 ri r r� rf r
r
r � test_strings� s !
z_TestPoll.test_stringsc C s | j d � d S)Nr� )r� )rX r� r
r
r �_child_boundaries� s z_TestPoll._child_boundariesc C s� | j d � \ } } | j d | j d | f � } | j � t j d � d d g } x | D] } | j | � q] W| j � | j � | j | j
� | � d S)NFrP r' rZ s firsts second) r� rQ rp rd r& rl r* r� rg re r( )r$ r� r� ri r� r� r
r
r �test_boundaries
s
z_TestPoll.test_boundariesc C s+ | j d � | j d � | j d � d S)Nrj rk s cd)r* )rX r5 r
r
r �_child_dont_merge
s
z_TestPoll._child_dont_mergec C s9 | j � \ } } | j | j d � d � | j | j d � d � | j d | j d | f � } | j � | j | j � d � | j | j d � d � | j | j d � d � | j | j � d � | j | j d � d � | j | j d � d � | j | j d � d � | j | j � d
� | j � d S) Ng Fg�������?rP r' rj g �?Trk s cd)r� r8 r� rQ rr rd r( rg )r$ r4 r5 ri r
r
r �test_dont_merge
s
z_TestPoll.test_dont_mergeN)z processeszthreads) r+ r, r- r= rf r� rh ro rp rq rr rs r
r
r
r rd � s rd z$test needs multiprocessing.reductionc @ sv e Z d Z d Z e d d � � Z e d d � � Z e d d � � Z d d � Z e d
d � � Z d d
� Z
d S)�_TestPicklingConnectionsr/ c C s$ d d l m } | j d d � d S)Nr )�resource_sharerr r� )rw ru r� )rX ru r
r
r r� 2
s z&_TestPicklingConnections.tearDownClassc C s� x` | D]X } | j j d | � } | j | j � | j � } | j | � | j � | j � q Wt j � } | j t j j
d f � | j d � | j | j � � | j � \ } } | j | � | j � | j � | j
� d S)NrW r r )rX rZ r} r r^ r� �socket�bindr� r� r �listen� getsocknamer� )rX rZ rY rb r[ �new_connr$ r
r
r � _listener7
s
z"_TestPicklingConnections._listenerc C s� xO t | j d � D]; \ } } | j j | � } | j | j � � | j � q W| j � \ } } t j � } | j | � | j | j � � | j � | j � d S)N)
r� r� rX r] r} r� r� rv r �sendall)rX rZ r r+ �clientr
r
r �_remoteL
s
z _TestPicklingConnections._remotec C s� | j j } | j � \ } } | j d | j d | | f � } d | _ | j � | j � | j � \ } } | j d | j d | f � } d | _ | j � | j � xh | D]` } d | j d � } | j
� }
| j |
| f � | j
� } | j | j
� | j
� � q� W| j d � t d � } | j
� }
| j |
| f � | j
� } g } x* | j
d � }
|
szPn | j |
� qaWd j | � } | j | | j
� � | j � | j d � | j � | j � | j � | j � d S) NrP r' Tz This connection uses family %s�asciiz$This connection uses a normal socketrk re )rX rY r� rQ r{ rH rd r� r~ r r� r} r8 r� r r� rg )r$ rY ZlconnZlconn0Zlpr� Zrconn0Zrprb r+ r rz Zbufr r
r
r �
test_pickling[
sH !
z&_TestPicklingConnections.test_picklingc C sZ | j � } | j d � | j � | j � } | j � } | j | d � | j � d S)Nz all is wellrZ )r� r} r� )rX rZ r� r� r+ r
r
r � child_access�
s
z%_TestPicklingConnections.child_accessc C s� | j � \ } } | j d | j d | f � } d | _ | j � | j � | j d d � \ } } | j | � | j � | j | j � d � | j � | j d d � \ } } | j | � | j � | j d � | j � | j | j � d � d S)
NrP r' Tr� Fz all is wellZfoobarrZ Z foobarfoobar) r� rQ r� rH rd r� r} r8 r� )r$ rZ r� ri r� r� r
r
r � test_access�
s
z$_TestPicklingConnections.test_accessN)z processes) r+ r, r- r= r� r� r{ r~ r� r� r� r
r
r
r rt -
s . rt c @ s. e Z d Z d Z d d � Z d d � Z d S)� _TestHeapr/ c C s d } d } g } xz t | � D]l } t t j d d � d � } t j j | � } | j | � t | � | k r t j | � } | | =q q Wt j j j
} g } d } | j j � | j
| j j � xc t | j j � � D]L }
xC |
D]; \ } } }
| j | j j | � | |
|
| d f � q� Wq� WxT | j D]I \ } } }
| j | j j | � | |
|
| d f � | |
| 7} q:W| j � x� t t | � d � D]o } | | d d � \ } } }
| | d d d � \ } } } | j | | k r
| d k p|
| k � q�Wd S) Ni� r� r r i� Zfree�occupiedr� )r~ ry �randomZlognormvariaterw r r r� rK Z randrangeZ_heapr r� r� r� rc Z _len_to_seqrj Z_arenas�indexZ_allocated_blocks�sortrF )r$ Z
iterationsZ maxblocksZblocksr� r� r5 r r� r� r� Zarenard r� ZnarenaZnstartZnstopr
r
r � test_heap�
s:
! z_TestHeap.test_heapc C s� t j � s) t j � | j t j � n t j � } | j t j | � t j d � xJ t d � D]< } t j j
d � } t j j
d � } | | _ | | _ qb Wd S)Nrq i� r ) �gc� isenabled�enabler� �disableZ
get_thresholdZ
set_thresholdr~ rw r r Zbuddy)r$ Z
thresholdsr� r4 r5 r
r
r �test_free_from_gc�
s
z_TestHeap.test_free_from_gcN)z processes)r+ r, r- r= r� r� r
r
r
r r� �
s 'r� c @ s( e Z d Z d e f d e f g Z d S)�_Foorf rg N)r+ r, r- r r Z_fields_r
r
r
r r� �
s r� c @ s[ e Z d Z d Z d d � Z e d d � � Z d d d � Z d d
� Z d d � Z d
S)�_TestSharedCTypesr/ c C s t s | j d � n d S)Nz%requires multiprocessing.sharedctypes)r rB )r$ r
r
r r9 s z_TestSharedCTypes.setUpc C sz | j d 9_ | j d 9_ | j d 9_ | j d 9_ | j d 9_ x( t t | � � D] } | | d 9<q^ Wd S)NrZ )r9 rf rg r~ rK )rX rf rg r� r� �stringr� r
r
r �_double s z_TestSharedCTypes._doubleFc C sl t d d d | �} t t d d | �} t t d d d | �} | j d t t d � � d | �} | j d
d d | �} t d � | _ | j d
| j d | | | | | f � } d | _
| j � | j � | j
| j d � | j | j d � | j
| j d � | j | j d � x, t d � D] } | j | | | d � q-W| j
| j t d � � d S)Nr� r� r� g �?g @r� rZ ra rq r r� r\ rP r' T� g @r� g @Z
hellohellogUUUUUU�?gUUUUUU�?)r r r� r� rc r~ r r9 rQ r� rH rd rg r8 r3 rf rg ) r$ r� rf rg r� r� r� ri r� r
r
r �test_sharedctypes s" $*
z#_TestSharedCTypes.test_sharedctypesc C s | j d d � d S)Nr� T)r� )r$ r
r
r �test_synchronize& s z"_TestSharedCTypes.test_synchronizec C sW t d d � } t | � } d | _ d | _ | j | j d � | j | j d � d S)NrZ g @r )r� r rf rg r8 r3 )r$ r� r r
r
r � test_copy) s z_TestSharedCTypes.test_copyN)z processes)
r+ r, r- r= r9 r� r� r� r� r� r
r
r
r r� �
s
r� c @ s4 e Z d Z d Z e d d � � Z d d � Z d S)�
_TestFinalizer/ c C s� Gd d � d t � } | � } t j | | j d d �~ | � } t j | | j d d �} | � | � ~ | � } t j | | j d d �| � } t j | | j d d d d �| � } t j | | j d d d d �| � } t j | | j d d d d �| � }
t j |
| j d d d d �t j d | j d d d d �t j d | j d d d d �t j � | j � t j d � d S) Nc @ s e Z d Z d S)z)_TestFinalize._test_finalize.<locals>.FooN)r+ r, r- r
r
r
r r ; s r r' r4 r5 r �d10Z exitpriorityr �d01r �d02�d03rh rq �STOPrk )r4 )r5 )r )zd10)zd01)zd02)zd03)rh i����)r� i����)r� r ZFinalizer} Z_exit_functionr� r �_exit) rX rZ r r4 r5 Zclose_br r� r� r� r� r
r
r �_test_finalize9 s0
z _TestFinalize._test_finalizec C s� | j � \ } } | j d | j d | f � } d | _ | j � | j � d d � t | j d � D� } | j | d d d d
d d d
g � d S)NrP r' Tc S s g | ] } | � q Sr
r
)r� r� r
r
r r� g s z/_TestFinalize.test_finalize.<locals>.<listcomp>r� r4 r5 r� r� r� r� rh ) r� rQ r� rH rd rg r� r� r8 )r$ rZ r� ri r� r
r
r �
test_finalize_ s
z_TestFinalize.test_finalizeN)z processes)r+ r, r- r= r� r� r� r
r
r
r r� 5 s &r� c @ s( e Z d Z d d � Z d d � Z d S)�_TestImportStarc C s� d d l } t j j t j � } t j j | d � } | j | � } d d � | D� } d d � | D� } | j d � | j d � | S)Nr z*.pyc S s6 g | ], } t j j t j j | � d � d � q S)r r )r �path�splitext�split)r� r� r
r
r r� u s z4_TestImportStar.get_module_names.<locals>.<listcomp>c S s g | ] } d | � q S)zmultiprocessing.r
)r� r? r
r
r r� v s zmultiprocessing.__init__rw ) �globr r� �dirnamerw �__file__rg r� r� )r$ r� Zfolder�pattern�files�modulesr
r
r �get_module_namesp s
z _TestImportStar.get_module_namesc C s | j � } t j d k rE | j d � | j d � | j d � n# | j d � t sh | j d � n t d k r� | j d � n xu | D]m } t | � t j | } | j t | d � | � x4 | j
D]) } | j t | | � d | | f � q� Wq� Wd S) Nr
zmultiprocessing.popen_forkz multiprocessing.popen_forkserverz!multiprocessing.popen_spawn_posixz!multiprocessing.popen_spawn_win32z multiprocessing.sharedctypes�__all__z %r does not have attribute %r) r� r� r r� rU r r� r� rF rs r� )r$ r� rV �mod�attrr
r
r � test_import{ s$
z_TestImportStar.test_importN)r+ r, r- r� r� r
r
r
r r� n s r� c @ s@ e Z d Z d Z d d � Z e d d � � Z d d � Z d S)
� _TestLoggingr/ c C sZ t j � } | j t j � | j | d k � | j d � | j d � | j t � d S)Nzthis will not be printedz
nor will this) rw �
get_logger�setLevelr �
SUBWARNINGrF �debug�info� LOG_LEVEL)r$ �loggerr
r
r �test_enable_logging� s
z _TestLogging.test_enable_loggingc C s# t j � } | j | j � � d S)N)rw r� r} ZgetEffectiveLevel)rX rZ r� r
r
r � _test_level� s z_TestLogging._test_levelc C s d } d } t j � } t j � } | j } t j d d � \ } } | j | � | j d | j d | f � } d | _ | j
� | j | | j � � | j t j
� | j | � | j d | j d | f � } d | _ | j
� | j | | j � � | j | � | j d t � d S) Nr �% r� FrP r' T�level)rw r� �loggingZ getLoggerr� r� r� rQ r� rH rd r8 r� ZNOTSETr� ) r$ ZLEVEL1ZLEVEL2r� Z root_loggerZ
root_levelr9 r: ri r
r
r �
test_level� s&
z_TestLogging.test_levelN)z processes)r+ r, r- r= r� r� r� r� r
r
r
r r� � s r� c @ sO e Z d Z d Z e d d � � Z e j e e d � d � d d � � Z
d S)
�_TestPollEintrr/ c C s$ t j d � t j | t j � d S)Ng�������?)r& rl r r rt �SIGUSR1)rX rW r
r
r �_killer� s
z_TestPollEintr._killerr� zrequires SIGUSR1c s� d g � � f d d � } t j � } t j t j | � } z� | j d | j d | f � } | j � z3 | j d t j d d � } | j � | j � Wd | j � X| j
� d � | j | j d � Wd t j t j | � Xd S) NFc s d � d <d S)NTr r
)r' )�
got_signalr
r �record� s z._TestPollEintr.test_poll_eintr.<locals>.recordrP r' rZ r )rZ )
r rL rt r� rQ r� rd r& rl rg rF r8 rM )r$ r� rW Z
oldhandlerZkillerri r
)r� r �test_poll_eintr� s
z _TestPollEintr.test_poll_eintrN)z processes) r+ r, r- r= r� r� r r rs rt r� r
r
r
r r� � s r� c @ s. e Z d Z e j e d � d d � � Z d S)�TestInvalidHandlezskipped on Windowsc C so t j j d � } z- y | j � Wn t t f k
r= Yn XWd d | _ X| j t t f t j j d � d S)Ni�M�r r )rw rX Z
Connectionr� r r" Z_handler� )r$ rZ r
r
r �test_invalid_handles s
z&TestInvalidHandle.test_invalid_handlesN)r+ r, r- r r� �WIN32r� r
r
r
r r� � s r� c @ s( e Z d Z d d � Z d d � Z d S)� OtherTestc C s< Gd d � d t � } | j t j t j j | � d � d S)Nc @ s( e Z d Z d d � Z d d � Z d S)zFOtherTest.test_deliver_challenge_auth_failure.<locals>._FakeConnectionc S s d S)Ns something bogusr
)r$ r� r
r
r r( s zQOtherTest.test_deliver_challenge_auth_failure.<locals>._FakeConnection.recv_bytesc S s d S)Nr
)r$ rJ r
r
r r* s zQOtherTest.test_deliver_challenge_auth_failure.<locals>._FakeConnection.send_bytesN)r+ r, r- r( r* r
r
r
r �_FakeConnection s r� s abc)r� r� rw �AuthenticationErrorrX Zdeliver_challenge)r$ r� r
r
r �#test_deliver_challenge_auth_failure s z-OtherTest.test_deliver_challenge_auth_failurec C s< Gd d � d t � } | j t j t j j | � d � d S)Nc @ s4 e Z d Z d d � Z d d � Z d d � Z d S)zEOtherTest.test_answer_challenge_auth_failure.<locals>._FakeConnectionc S s
d | _ d S)Nr )rE )r$ r
r
r r% s zNOtherTest.test_answer_challenge_auth_failure.<locals>._FakeConnection.__init__c S s? | j d 7_ | j d k r( t j j S| j d k r; d Sd S)Nr rZ s something bogusre )rE rw rX Z CHALLENGE)r$ r� r
r
r r( ! s
zPOtherTest.test_answer_challenge_auth_failure.<locals>._FakeConnection.recv_bytesc S s d S)Nr
)r$ rJ r
r
r r* ( s zPOtherTest.test_answer_challenge_auth_failure.<locals>._FakeConnection.send_bytesN)r+ r, r- r% r( r* r
r
r
r r� s r� s abc)r� r� rw r� rX Zanswer_challenge)r$ r� r
r
r �"test_answer_challenge_auth_failure s z,OtherTest.test_answer_challenge_auth_failureN)r+ r, r- r� r� r
r
r
r r� s
r� c C s | j d 7_ d S)Nr )r� )�nsr
r
r � initializer2 s r� c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
�TestInitializersc C s1 t j � | _ | j j � | _ d | j _ d S)Nr )rw �Manager�mgrr� r� r� )r$ r
r
r r9 6 s zTestInitializers.setUpc C s | j j � | j j � d S)N)r� r rg )r$ r
r
r r; ; s
zTestInitializers.tearDownc C si t j j � } | j t | j d � | j t | j f � | j | j j d � | j
� | j � d S)Nr ) rw ZmanagersZ SyncManagerr� � TypeErrorrd r� r� r8 r� r rg )r$ r? r
r
r �test_manager_initializer? s
z)TestInitializers.test_manager_initializerc C sb | j t t j d d �t j d t | j f � } | j � | j � | j | j j d � d S)Nr� r )
r� r� rw r� r� r� r� rg r8 r� )r$ ri r
r
r �test_pool_initializerG s
z&TestInitializers.test_pool_initializerN)r+ r, r- r9 r; r� r� r
r
r
r r� 5 s r� c C s2 y | j d d � } Wn t j k
r- Yn Xd S)N�blockF)rf r� r� )rY �itemr
r
r �_this_sub_processS s r� c C sH t j � } t j d t d | f � } d | _ | j � | j � d S)NrP r' T)rw r^ rQ r� rH rd rg )rY r� ZsubProcr
r
r �
_test_processY s
r� c C s | | S)Nr
)rf r
r
r �_afunc` s r� c C sQ t j d d � } | j t d d d d d d d g � } | j � | j � d S) Nr/ r� r rZ r� r� r� r� )rw r� r� r� r� rg )r� rf r
r
r �pool_in_processc s '
r� c @ sF e Z d Z d d � Z e d d � � Z d d � Z d d � Z d S)
�
_file_likec C s | | _ d | _ d S)N)� _delegate�_pid)r$ Zdelegater
r
r r% j s z_file_like.__init__c C s7 t j � } | | j k r0 | | _ g | _ n | j S)N)r rL r� �_cache)r$ rW r
r
r �cachen s
z_file_like.cachec C s | j j | � d S)N)r� r� )r$ rJ r
r
r r� w s z_file_like.writec C s) | j j d j | j � � g | _ d S)Nr& )r� r� rg r� r� )r$ r
r
r �flushz s z_file_like.flushN)r+ r, r- r% �propertyr� r� r� r
r
r
r r� i s r� c @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�TestStdinBadfiledescriptorc C s? t j � } t j d t d | f � } | j � | j � d S)NrP r' )rw r^ rQ r� rd rg )r$ r� r� r
r
r �test_queue_in_process� s
z0TestStdinBadfiledescriptor.test_queue_in_processc C s* t j d t � } | j � | j � d S)NrP )rw rQ r� rd rg )r$ ri r
r
r �test_pool_in_process� s
z/TestStdinBadfiledescriptor.test_pool_in_processc sQ t j � } t | � � � j d � t j d � f d d � � } � j � d S)Nr� rP c s
� j � S)N)r� r
)�fliker
r r � s z:TestStdinBadfiledescriptor.test_flushing.<locals>.<lambda>)�io�StringIOr� r� rw rQ r� )r$ Zsior� r
)r� r �
test_flushing� s
z(TestStdinBadfiledescriptor.test_flushingN)r+ r, r- r� r� r� r
r
r
r r� ~ s r� c @ s� e Z d Z e d d � � Z d d d � Z e d d � � Z d d d � Z d
d � Z d d
� Z d d � Z
e d d � � Z d d � Z d d � Z
d S)�TestWaitc C s^ xM t d � D]? } | r3 t j t j � d � n | j | t j � f � q
W| j � d S)Nrq g�������?)r~ r& rl r� r} r rL r� )rX r� �slowr� r
r
r �_child_test_wait� s
zTestWait._child_test_waitFc si d d l m } g } g � g } x� t d � D]� } t j d d � \ } } t j d | j d | | f � } d | _ | j � | j � | j
| � � j
| � | j | j � q/ Wxj | r%x] | | � D]O } y | j
� } Wn) t k
r| j | � | j � Yq� X| j
| � q� Wq� W| j � t � f d d
� t d � D� � }
| j | |
� d S) Nr )r r� r� FrP r' Tc 3 s+ | ]! } � D] } | | j f Vq
q d S)N)rW )r� r� ri )�procsr
r r� � s z%TestWait.test_wait.<locals>.<genexpr>rq )�multiprocessing.connectionr r~ rw r� rQ r� rH rd r� r� r� rg r� r6 r� r� r� r8 ) r$ r� r �readersZmessagesr� r� r� ri r+ r� r
)r� r � test_wait� s0 !
%zTestWait.test_waitc C sx t j � } | j | � xN t d � D]@ } | rL t j t j � d � n | j d | j d � � q& W| j � d S)Nrq g�������?z%s
r ) rv r r~ r& rl r� r| r r� )rX r r� r r� r
r
r �_child_test_wait_socket� s
z TestWait._child_test_wait_socketc C s� d d l m } t j � } | j t j j d f � | j d � | j � } g } g } i } xe t d � D]W } t
j d | j d | | f � } d | _
| j � | j | � | j | j � qm Wx= t d � D]/ } | j � \ }
} | j |
� g | |
<q� W| j � xa | ruxT | | � D]F }
|
j d � } | s]| j |
� |
j � q(| |
j | � q(WqWd j d d
� t d � D� � j d � }
x- | j � D] } | j d
j | � |
� q�Wd S)Nr )r r� rP r' Tr r& c s s | ] } d | Vq d S)z%s
Nr
)r� r� r
r
r r� � s z,TestWait.test_wait_socket.<locals>.<genexpr>rq r re )r� r rv rw r� r� r rx ry r~ rw rQ r� rH rd r� r� rg r^ r� r� r� r rj r8 )r$ r� r r[ r$ r� r� Zdicr� ri r� r# r+ r� �vr
r
r �test_wait_socket� s<
+zTestWait.test_wait_socketc C s | j d � d S)NT)r� )r$ r
r
r �test_wait_slow� s zTestWait.test_wait_slowc C s | j d � d S)NT)r� )r$ r
r
r �test_wait_socket_slow� s z TestWait.test_wait_socket_slowc C s� d d l m } d } t j � \ } } t j � } | | | g | � } t j � | } | j | g � | j | | d � | j | | d � | j d � t j � } | | | g d � } t j � | } | j | | g � | j | d � d S)Nr )r r� rZ g �?r� g�������?) r� r rw r� r& r8 r� �
assertGreaterr} )r$ r r� r4 r5 rd r: r� r
r
r �test_wait_timeout� s
zTestWait.test_wait_timeoutc C s | j � t j | � d S)N)r� r& rl )rX r� Zperiodr
r
r �signal_and_sleep
s
zTestWait.signal_and_sleepc C s d d l m } d } d d � } t j d � } t j � \ } } t j d | j d | | f � } | j � | j | j t
� | j | j d d � � t
j
� } | | | j | g | d � } t
j
� | }
| j | | j g � | j |
| d
� | j |
| d
� | j d � t
j
� } | | | j | g d � } t
j
� | }
| j | | � | | j | g � � | j |
d � | j d � t
j
� } | | | j | g d � } t
j
� | }
| j | | � | | | j | g � � | j |
d � | j � | j � d S) Nr )r r� c S s t | d d d � �S)Nr� c S s
t | � S)N)r� )rf r
r
r r
s z>TestWait.test_wait_integer.<locals>.<lambda>.<locals>.<lambda>)r� )r[ r
r
r r
s z,TestWait.test_wait_integer.<locals>.<lambda>rP r' r r� rZ g�������?)r� r rw r� r� rQ r� rd rI r� ry rF r� r& r8 r� r� r} rr rg ) r$ r r� Zsorted_r� r4 r5 ri rd r: r� r
r
r �test_wait_integer
s<
%
(
zTestWait.test_wait_integerc C s� d d l m } t j � \ } } t j � } | | g d d �} t j � | } | j | g � | j | d � | j � | j � d S)Nr )r r r r )r� r rw r� r& r8 r� r� )r$ r r4 r5 r) r: r
r
r �test_neg_timeout8
s
zTestWait.test_neg_timeoutN)r+ r, r- r� r� r� r� r� r� r� r� r� r� r� r
r
r
r r� � s
% +r� c @ sL e Z d Z e j e d � d d � � Z e j e d � d d � � Z d S)�TestInvalidFamilyzskipped on Windowsc C s* | j t � � t j j d � Wd QXd S)Nz\\.\test)r� r rw rX rZ )r$ r
r
r �test_invalid_familyI
s z%TestInvalidFamily.test_invalid_familyz skipped on non-Windows platformsc C s* | j t � � t j j d � Wd QXd S)Nz/var/test.pipe)r� r rw rX rZ )r$ r
r
r �test_invalid_family_win32N
s z+TestInvalidFamily.test_invalid_family_win32N) r+ r, r- r r� r� r� r r� r
r
r
r r� G
s r� c @ s@ e Z d Z e d d � � Z e d d � � Z d d � Z d S)� TestFlagsc C s | j t t j � � d S)N)r} r r� �flags)rX rZ r
r
r �run_in_grandchildX
s zTestFlags.run_in_grandchildc C s� d d l } t j d d � \ } } t j d | j d | f � } | j � | j � } | j � | j � | j � t t
j � | f } t | j
| � � d S)Nr r� FrP r' )�jsonrw r� rQ r� rd r� rg r� r r� r� �print�dumps)rX r� r� r� ri �grandchild_flagsr� r
r
r � run_in_child\
s
zTestFlags.run_in_childc C sx d d l } d d l } d d } | j t j d d d d | g � } | j | j d � � \ } } | j | | � d S) Nr z2from test._test_multiprocessing import TestFlags; zTestFlags.run_in_child()z-Ez-Sz-Oz-cr )r� �
subprocessZ check_outputr� �
executable�loads�decoder8 )r$ r� r �progrJ Z child_flagsr r
r
r �
test_flagsi
s zTestFlags.test_flagsN)r+ r, r- r� r� r r r
r
r
r r� W
s
r� c @ s. e Z d Z e d d � � Z d d � Z d S)� TestTimeoutsc C sQ t j d � | j d � | j � t j j | � } | j d � | j � d S)Nr r� i� )r& rl r} r� rw rX r] )rX �childr rZ r
r
r �
_test_timeoutx
s
zTestTimeouts._test_timeoutc C s� t j � } z� t j d � t j d d � \ } } t j j d d � } t j d | j d | | j f � } | j
� | j � | j | j
� d � | j � | j � } | j | j
� d � | j � | j � | j d
� Wd t j | � Xd S) Ng�������?r� TrW �AF_INETrP r' r� i� rq )rv ZgetdefaulttimeoutZsetdefaulttimeoutrw r� rX rZ rQ r
r rd r� r8 r� r^ rg )r$ Z old_timeout�parentr r[ ri rZ r
r
r r� �
s"
zTestTimeouts.test_timeoutN)r+ r, r- r� r
r� r
r
r
r r w
s r c @ s e Z d Z d d � Z d S)�TestNoForkBombc C s� t j � } t j j t j j t � d � } | d k rz t j j | | � \ } } } | j
| d � | j d | � nD t j j | | � \ } } } | j
| j
� d � | j
| d � d S)Nzmp_fork_bomb.py�forkre s RuntimeErrors 123)rw �get_start_methodr r� rg r� r� r� Z
script_helperZassert_python_failurer8 re Zassert_python_okr� )r$ ZsmrV �rc�outr� r
r
r �test_noforkbomb�
s ! z TestNoForkBomb.test_noforkbombN)r+ r, r- r r
r
r
r r
�
s r
c @ s. e Z d Z e d d � � Z d d � Z d S)�TestForkAwareThreadLockc C s| | d k rX t j d | j d | d | f � } | j � | j � | j d d � n | j t t j � � | j � d S)Nr rP r' r r� )
rw rQ r rd r� rg r} rK r �_afterfork_registry)rX r' rZ ri r
r
r r �
s %
z TestForkAwareThreadLock.childc C s� t j d � \ } } t j � } t t j � } t j d | j d d | f � } | j � | j � | j
� } | j d d � | j | | � d S)NFrP r' r� r )
rw r� r ZForkAwareThreadLockrK r rQ r rd r� r� rg ZassertLessEqual)r$ r� r� r[ Zold_sizeri Znew_sizer
r
r r� �
s !
z!TestForkAwareThreadLock.test_lockN)r+ r, r- r� r r� r
r
r
r r �
s r c @ sF e Z d Z d d � Z d d � Z e d d � � Z d d � Z d S)
� TestCloseFdsc C s� t r t j � j � St j � j � } g } x, | d k r\ | j | � t j | � } q1 Wx | D] } t j | � qd W| Sd S)Nr� )r� rv �detachr� r �dupr� )r$ rD Zto_closerf r
r
r �get_high_socket_fd�
s
z TestCloseFds.get_high_socket_fdc C s0 t r t j d | � j � n
t j | � d S)Nr1 )r� rv r� r )r$ rD r
r
r r� �
s zTestCloseFds.closec C so y t j | t j t j � } Wn2 t k
rS } z | j | � WYd d } ~ Xn X| j � | j d � d S)N)rv Zfromfdr Z SOCK_STREAMr� r} r� )rX rZ rD r rh r
r
r �_test_closefds�
s
zTestCloseFds._test_closefdsc C s t s t j d � � n t j � \ } } | j � } zU t j d | j d | | f � } | j � | j � | j
� } | j d d � Wd | j | � | j � | j � Xt j � d k r� | j
| d � nA d } | j | t � | j | j t j k p| j | k | � d S)Nzrequires fd picklingrP r' r r� r i6' )rU r r rw r� r rQ r rd r� r� rg r r� rI r" rF r# rC Zwinerror)r$ r9 r: rD ri rh Z WSAENOTSOCKr
r
r � test_closefd�
s(
zTestCloseFds.test_closefdN)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 j e e d � d � d d � � Z e d d � � Z
e j e e d � d � d d
� � Z d S) �TestIgnoreEINTRc C sZ d d � } t j t j | � | j d � | j � } | j | � | j d d � d S)Nc S s d S)Nr
)�signum�framer
r
r rp s z-TestIgnoreEINTR._test_ignore.<locals>.handler�ready� xi i )rt r� r} r� r* )rX rZ rp rf r
r
r � _test_ignore
s
z TestIgnoreEINTR._test_ignorer� zrequires SIGUSR1c C s t j � \ } } z� t j d | j d | f � } d | _ | j � | j � | j | j � d � t j
d � t j | j
t j � t j
d � | j d � | j | j � d � t j
d � t j | j
t j � | j | j � d d � t j
d � | j � Wd | j � Xd S)
NrP r' Tr g�������?i� r i i )rw r� rQ r rH rd r� r8 r� r& rl r r rW rt r� r} r( rg )r$ rZ r� ri r
r
r � test_ignore s&
zTestIgnoreEINTR.test_ignorec C sd d d � } t j t j | � t j j � �0 } | j | j � | j � } | j d � Wd QXd S)Nc S s d S)Nr
)r r r
r
r rp 0 s z6TestIgnoreEINTR._test_ignore_listener.<locals>.handler�welcome)rt r� rw rX rZ r} r r^ )rX rZ rp r[ r4 r
r
r �_test_ignore_listener. s z%TestIgnoreEINTR._test_ignore_listenerc C s� t j � \ } } z� t j d | j d | f � } d | _ | j � | j � | j � } t j d � t
j | j t
j � t j d � t j j | � } | j | j � d � | j � Wd | j � Xd S)NrP r' Tg�������?r" )rw r� rQ r# rH rd r� r� r& rl r r rW rt r� rX r] r8 rg )r$ rZ r� ri r r} r
r
r �test_ignore_listener8 s
z$TestIgnoreEINTR.test_ignore_listenerN) r+ r, r- r� r r r rs rt r! r# r$ r
r
r
r r s
'
r c @ sR e Z d Z e d d � � Z d d � Z d d � Z d d � Z d d
� Z d S) �TestStartMethodc C s | j t j � � d S)N)r} rw r )rX rZ r
r
r �_check_contextL s z TestStartMethod._check_contextc C s� | j d d � \ } } | j d | j d | f � } | j � | j � | j � } | j � | j � | j | | j � � d S)Nr� FrP r' ) r� rQ r& rd r� r� rg r8 r )r$ �ctxr� r� ri Z child_methodr
r
r �
check_contextP s
z TestStartMethod.check_contextc
C s� x� d D]� } y t j | � } Wn t k
r7 w Yn X| j | j � | � | j | j � | � | j t | j d � | j t | j d � | j | � q Wd S)Nr �spawn�
forkserver)zforkr) r* ) rw � get_contextr r8 r r� r� �set_start_methodr( )r$ �methodr' r
r
r r� Z s
z TestStartMethod.test_contextc C s$ t j t � d } t j � } z� x� d D]� } y t j | d d �Wn t k
r] w) Yn X| j t j � | � t j � } | j | j � | � | j t | � j
j � j | � � | j | j
j
j � j | � � | j t � | d 7} q) WWd t j | d d �X| j | d � d S) Nr r r) r* �forceTr )zforkzspawnz
forkserver)rw �set_forkserver_preload�PRELOADr r, r r8 r+ rF rb r+ �lower�
startswithrQ r( r� )r$ rE Z
old_methodr- r' r
r
r � test_set_getf s&
%
z TestStartMethod.test_set_getc C sc t j � } t j d k r1 | j | d g � n. | j | d d g k p[ | d d d g k � d S)Nr
r) r r* )rw Zget_all_start_methodsr� r r8 rF )r$ �methodsr
r
r � test_get_all| s
z TestStartMethod.test_get_allN) r+ r, r- r� r& r( r� r3 r5 r
r
r
r r% K s
r% z*test semantics don't make sense on Windowsc @ s e Z d Z d d � Z d S)�TestSemaphoreTrackerc C s� d d l } d } t j � \ } } | j t j d | | | f g d | g d | j �} t j | � t | d d d ��= } | j � j
� j d � } | j � j
� j d � } Wd QXt j
| � | j � | j � t j d
� | j t � � } t j
| � Wd QX| j | j j t j t j f � | j j � j d � }
| j j � d } | j |
| � | j |
d
| � d S)Nr aM if 1:
import multiprocessing as mp, time, os
mp.set_start_method("spawn")
lock1 = mp.Lock()
lock2 = mp.Lock()
os.write(%d, lock1._semlock.name.encode("ascii") + b"\n")
os.write(%d, lock2._semlock.name.encode("ascii") + b"\n")
time.sleep(10)
z-cZpass_fdsr� rM �closefdTr g @zutf-8z9semaphore_tracker: There appear to be 2 leaked semaphoresz semaphore_tracker: %r: \[Errno) r r �pipe�Popenr� r �PIPEr� r� �readliner� r r Z
sem_unlinkrr r r& rl r� r" re Z exceptionr# �ENOENTZEINVALr� r� Z assertRegex) r$ r �cmdr� r� ri r� Zname1Zname2r' r� r� r
r
r �test_semaphore_tracker� s.
!
"
z+TestSemaphoreTracker.test_semaphore_trackerN)r+ r, r- r> r
r
r
r r6 � s r6 c @ s' e Z d Z d Z e j Z e j Z e e j � Z e e j � Z e e j
� Z
e e j � Z e e j � Z e e j
� Z
e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z e e j � Z d S)r r/ N)r+ r, r- rA rw rQ rX � staticmethodrD ra r� r� r^ r� r� r� r� r� r� r_ r5 r r� rm r� r
r
r
r r � s( r c @ s{ e Z d Z d Z e j Z e e j d � � Z e e j d � � Z
e e j d � � Z e e j d � � Z e e j d � � Z
e e j d � � Z e e j d � � Z e e j d � � Z e e j d
� � Z e e j d � � Z e e j d � � Z e e j d
� � Z e e j d � � Z e e j d � � Z e d d � � Z e d d � � Z e d d � � Z d S)� ManagerMixinr0 z
manager.Queuezmanager.JoinableQueuez manager.Lockz
manager.RLockzmanager.Semaphorezmanager.BoundedSemaphorezmanager.Conditionz
manager.Eventzmanager.Barrierz
manager.Valuez
manager.Arrayz manager.listz manager.dictzmanager.Namespacec O s | j j | | � S)N)r0 r� )rX r' r( r
r
r r� � s zManagerMixin.Poolc C s t j � | _ d S)N)rw r� r0 )rX r
r
r r� � s zManagerMixin.setUpClassc C s� d } x? t t j � � d k rG | d k rG t j | � | d 9} q Wt j � | j j � d k r� t d � t | j j
� � n | j j � | j j � d | _ d S)Ng{�G�z�?r r� rZ r z5Shared objects which still exist at manager shutdown:)
rK rw ra r& rl r� �collectr0 Z_number_of_objectsr� Z _debug_infor rg )rX r) r
r
r r� � s '
zManagerMixin.tearDownClassN)r+ r, r- rA rw rQ r� �operator�
attrgetterr^ r� r� r� r� r� r� r_ r5 r r� rc r� r� r� r� r� r� r
r
r
r r@ � s& r@ c @ s9 e Z d Z d Z e j j Z e j j Z e e j j � Z e e j j
� Z
e e j � Z e e j j � Z e e j j
� Z
e e j j � Z e e j j � Z e e j j � Z e e j j � Z e e j j � Z e e j j � Z e e j j � Z e e j j � Z e e j j � Z e e j j � Z d S)� ThreadsMixinr1 N)r+ r, r- rA rw ZdummyrQ rX r? rD ra r� r� r^ r� r� r� r� r� r� r_ r5 r r� r
r
r
r rD � s$ rD c
s� | d } t � } d d d h } x,| j � D] \ } } t | t � sP q/ n t | t � r� | t k rq q/ n x� | j D]z } d | j � | d d � } | | j � d } Gd d � d | | t j � }
| |
_
|
_ | |
_ |
| | <q{ Wq/ t | t j � r/ Gd
d � d | t
� }
| |
_
|
_ | |
_ |
| | <q/ q/ Wd d g � d g � � � � f d d � } � � f d
d � } | | d <| | d <d S)Nr+ r/ r1 r0 ZWithr �Mixinc @ s e Z d Z d S)z*install_tests_in_module_dict.<locals>.TempN)r+ r, r- r
r
r
r �Temp s rF c @ s e Z d Z d S)z*install_tests_in_module_dict.<locals>.TempN)r+ r, r- r
r
r
r rF $ s c s t j t � t j j � t j j j � � d <t j j � � d <t j d d � � d <y t j � d d �Wn% t
k
r� t j � d � � Yn Xt
j j d � r� y t j � } Wq� t k
r� t j d � � Yq� Xn t � t j � t j � j t � d S) Nr r Z
allow_noneTr. z start method not supported�linuxz1OSError raises on RLock creation, see issue 3111!)rw r/ r0 �process�_cleanup� _danglingr r� r r, r r r r� r r2 r� r" r r Z get_temp_dirr� r� r� )r� )�dangling�old_start_method� start_methodr
r � setUpModule- s$
z1install_tests_in_module_dict.<locals>.setUpModulec s� t j � d d d �t j d � t j j � t j � t t j j � t � d � } | rz t
d | d t j �n ~ t t
j � t � d � } | r� t
d | d t j �n d S) Nr r. Tg �?zDangling processes:�filer zDangling threads:)rw r, r& rl rH rI r� rA r� rJ r� r� r� r� )Ztmp)rK rL r
r �tearDownModuleC s
z4install_tests_in_module_dict.<locals>.tearDownModulerN rP )�globalsr� r
rb �
issubclassr. r= �
capitalizer �TestCaser+ r- r, r� )
Z remote_globsrM r, Z local_globsZ ALL_TYPESrV �baseZtype_ZnewnamerE rF rN rP r
)rK rL rM r � install_tests_in_module_dict s6
rV )�r r� r� r& r� r� r� r r� r# rt r� rv r� r� r rB Z test.supportr� Ztest.script_helperr� �
import_moduler r� Zmultiprocessing.dummyrw r� Zmultiprocessing.managersZmultiprocessing.heapr� r r ZHAVE_SEND_HANDLErU � ImportErrorZ multiprocessing.sharedctypesr r r rH r r� r� r{ r2 r� r� r� �getattrZ
HAVE_GETVALUEr r� r r r rP r0 Zctypesr r r r� r r! r. r? r@ rQ r� r� r� r� r� r� r� r� r r r% r& r3 r4 r^ r~ r� r� r� r r� r� r� r� r� r� r� r� r� r� r� r r r �registerr r^ r r r r r r r! r) r' rV r_ rd r rt r� r� r� r� rT r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r
r r r r% r� r6 r r@ rD rV r
r
r
r �<module> s0
�F �6�2 -�ES< � (1 7
0. �
0R �E69+C"! � " B@='/