Your IP : 216.73.216.213
V�EV45 � @ s� 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 m Z d d l m Z d Z d d � e
d d � D� Z d Z d
d � Z
d d d
� Z d d � Z Gd d � d e j � Z d S)� N)�futures)�test)�_utilsg�������?c C s g | ] } d | d � q S)g{�G�z�?g Y@� )�.0�xr r �J/opt/alt/python35/lib/python3.5/site-packages/fasteners/tests/test_lock.py�
<listcomp>$ s r � i, c C sC d } x6 | D]. \ } } | | k r
| | k r
| d 7} q
W| S)Nr � r )�times�start�endZoverlaps�s�er r r �_find_overlaps+ s
r c
sn t j � � t j � � � � f d d � } � � f d d � } | d k rj t d | � t d | � } | d k r t j d | � �u } d } x1 t d | � D] } | j | | � | d 7} q� Wx1 t d | � D] } | j | | � | d 7} q� WWd QRXg } g } xU t � � D]G \ }
} } |
� j
k rM| j | | f � q| j | | f � qW| | f S)Nc sq � j � �^ t j � } t j t | t t � � t j � } � j � j | | f � t j t � Wd QRXd S)N)
� read_lockr �now�time�sleep�
WORK_TIMES�len�append�READER�
NAPPY_TIME)�ident�
enter_time� exit_time)�lock� start_stopsr r � read_func7 s
z#_spawn_variation.<locals>.read_funcc sq � j � �^ t j � } t j t | t t � � t j � } � j � j | | f � t j t � Wd QRXd S)N)
�
write_lockr r r r r r r �WRITERr )r r r )r r r r �
write_funcB s
z$_spawn_variation.<locals>.write_funcr � max_workersr ) � collections�deque� fasteners�ReaderWriterLock�maxr �ThreadPoolExecutor�range�submit�listr" r )
�readers�writersr$ r r# r �countZ_i� writer_times� reader_timesZ lock_typer
�stopr )r r r �_spawn_variation3 s, r4 c C s t j d | � } d | _ | S)N�targetT)� threading�Thread�daemon)r5 �tr r r �_daemon_thread` s r: c @ s� 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 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)%�ReadWriteLockTest� c s t j � � t j d d � � � j � t j � � t j � � � � � f d d � � � � f d d � } g } x= t d | j � D]) } t | � } | j
| � | j � q� Wx | r� | j � } | j � q� W| j
g t � � � | j
g t � � � d S)N�durationr
c s} � j � �j t � � d k r9 � j | � � j � � � j | � z t j t j � d � Wd � j | � XWd QRXd S)Nr �d )r! r r �extendr r �random�remove)�me)�active�dupsr r r �
acquire_checkp s
z?ReadWriteLockTest.test_no_double_writers.<locals>.acquire_checkc s- t j � } x � j � s( � | � q Wd S)N)r6 �current_thread�expired)rB )rE �watchr r �run{ s z5ReadWriteLockTest.test_no_double_writers.<locals>.runr )r' r( r � StopWatchr
r% r& r+ � THREAD_COUNTr: r �pop�join� assertEqualr- )�selfrI �threads�ir9 r )rE rC rD r rH r �test_no_double_writersi s"
z(ReadWriteLockTest.test_no_double_writersc s t j � � t j d d � � � j � t j � � t j � � � � � f d d � � � � f d d � } g } x= t d | j � D]) } t | � } | j
| � | j � q� Wx | r� | j � } | j � q� W| j
g t � � � | j
g t � � � d S)Nr= r
c s� | r � j } n � j } | � �p | sW t � � d k rW � j | � � j � � � j | � z t j t j � d � Wd � j | � XWd QRXd S)Nr r> ) r r! r r r? r r r@ rA )rB �readerZ lock_func)rC rD r r r rE � s
zKReadWriteLockTest.test_no_concurrent_readers_writers.<locals>.acquire_checkc s? t j � } x, � j � s: � | t j d d g � � q Wd S)NTF)r6 rF rG r@ �choice)rB )rE rH r r rI � s zAReadWriteLockTest.test_no_concurrent_readers_writers.<locals>.runr )r' r( r rJ r
r% r& r+ rK r: r rL rM rN r- )rO rI rP rQ r9 r )rE rC rD r rH r �"test_no_concurrent_readers_writers� s"
z4ReadWriteLockTest.test_no_concurrent_readers_writersc sU t j � � � j � j � � � f d d � } � j t | � � j � j � d S)Nc
s: � j � �' � j � j � j � t d � � Wd QRXd S)N�Broken)r! rN r" �owner� RuntimeErrorr )r rO r r �blow_up� s
z4ReadWriteLockTest.test_writer_abort.<locals>.blow_up)r' r( � assertFalserW � assertRaisesrX )rO rY r )r rO r �test_writer_abort� s
z#ReadWriteLockTest.test_writer_abortc sU t j � � � j � j � � � f d d � } � j t | � � j � j � d S)Nc
s: � j � �' � j � j � j � t d � � Wd QRXd S)NrV )r rN r rW rX r )r rO r r rY � s
z4ReadWriteLockTest.test_reader_abort.<locals>.blow_up)r' r( rZ rW r[ rX )rO rY r )r rO r �test_reader_abort� s
z#ReadWriteLockTest.test_reader_abortc s� t j � � t j � � � f d d � } � � f d d � } t j d d � �N } xD t d d � D]3 } | d d k r� | j | � qd | j | � qd WWd QRX| j d t d
d � � D� � � d S) Nc s8 � j � �% � j � � t d � � Wd QRXWd QRXd S)NrV )r rX r )r r r �double_bad_reader� s
zEReadWriteLockTest.test_double_reader_abort.<locals>.double_bad_readerc s( � j � � � j � j � Wd QRXd S)N)r! r rW r )� activatedr r r � happy_writer� s
z@ReadWriteLockTest.test_double_reader_abort.<locals>.happy_writerr$ r< r � �
c S s" g | ] } | d k r | � q S)�wr )r �ar r r r � s z>ReadWriteLockTest.test_double_reader_abort.<locals>.<listcomp>)
r' r( r% r& r r* r+ r, rN r )rO r^ r` r rQ r )r_ r r �test_double_reader_abort� s z*ReadWriteLockTest.test_double_reader_abortc s� t j � � t j � � t j � � � � � f d d � } � � f d d � } t | � } | j � � j t � | j
� j � � t | � } | j � | j � | j � | j
d t � � � | j
d d g t � � � d S)Nc sc � j � �P � j � x � j s3 t j d � q W� j � � � j � j � Wd QRXWd QRXd S)Ng����MbP?)r �setZhas_pending_writersr r r rW r )r_ rC r r r �
double_reader� s
zBReadWriteLockTest.test_double_reader_writer.<locals>.double_readerc s( � j � � � j � j � Wd QRXd S)N)r! r rW r )r_ r r r r` � s
zAReadWriteLockTest.test_double_reader_writer.<locals>.happy_writerra �rrc )r' r( r% r& r6 �Eventr: r
�wait� WAIT_TIMEOUT�
assertTrue�is_setrM rN r r- )rO rg r` rS �writerr )r_ rC r r �test_double_reader_writer� s
z+ReadWriteLockTest.test_double_reader_writerc s t j � � t j � � � � f d d � } � � f d d � } t j d d � �d } xZ t d d � D]I } | d d k r� | j | d t | d
d k � �qg | j | � qg WWd QRXd d � � D� } d
d � � D� } | j d t
| � � | j d t
| � � d S)Nc s= � j � �* | r" t d � � n � j � j � Wd QRXd S)NrV )r rX r rW )rY )r_ r r r �chaotic_reader s
z=ReadWriteLockTest.test_reader_chaotic.<locals>.chaotic_readerc s( � j � � � j � j � Wd QRXd S)N)r! r rW r )r_ r r r r` s
z;ReadWriteLockTest.test_reader_chaotic.<locals>.happy_writerr$ r< r ra rY � c S s" g | ] } | d k r | � q S)rc r )r rd r r r r s z9ReadWriteLockTest.test_reader_chaotic.<locals>.<listcomp>c S s" g | ] } | d k r | � q S)rh r )r rd r r r r s rb r
) r' r( r% r& r r* r+ r, �boolrN r )rO rp r` r rQ r/ r. r )r_ r r �test_reader_chaotic s &z%ReadWriteLockTest.test_reader_chaoticc s t j � � t j � � � � f d d � } � � f d d � } t j d d � �d } xZ t d d � D]I } | d d k r� | j | d t | d
d k � �qg | j | � qg WWd QRXd d � � D� } d
d � � D� } | j d t
| � � | j d t
| � � d S)Nc s= � j � �* | r" t d � � n � j � j � Wd QRXd S)NrV )r! rX r rW )rY )r_ r r r �chaotic_writer" s
z=ReadWriteLockTest.test_writer_chaotic.<locals>.chaotic_writerc s( � j � � � j � j � Wd QRXd S)N)r r rW r )r_ r r r � happy_reader) s
z;ReadWriteLockTest.test_writer_chaotic.<locals>.happy_readerr$ r< r ra rY rq c S s" g | ] } | d k r | � q S)rc r )r rd r r r r 4 s z9ReadWriteLockTest.test_writer_chaotic.<locals>.<listcomp>c S s" g | ] } | d k r | � q S)rh r )r rd r r r r 5 s r
rb ) r' r( r% r& r r* r+ r, rr rN r )rO rt ru r rQ r/ r. r )r_ r r �test_writer_chaotic s &z%ReadWriteLockTest.test_writer_chaoticc C s� t j � } | j � �f | j | j � � | j � �? | j | j � � | j � � | j | j � � Wd QRXWd QRXWd QRXd S)N)r' r( r! rl � is_writerr � is_reader)rO r r r r �test_writer_reader_writer9 s
z+ReadWriteLockTest.test_writer_reader_writerc C s� g } t j � } | j � �. | j | j � � | j d t | � � Wd QRX| j � �% | j d � | j | j � � Wd QRX| j � �. | j | j � � | j d t | � � Wd QRX| j
| j � � | j
| j � � d S)Nr r ) r' r( r rl rx rN r r! r rw rZ )rO �resultsr r r r �test_single_reader_writerB s
z+ReadWriteLockTest.test_single_reader_writerc
s t j � � � f d d � } � j � �( | j t | � | j � j � � Wd QRX| j � j � � | j � j � � d S)Nc s � j � � Wd QRXd S)N)r! r )r r r � writer_funcT s
z<ReadWriteLockTest.test_reader_to_writer.<locals>.writer_func)r' r( r r[ rX rZ rw rx )rO r| r )r r �test_reader_to_writerQ s
z'ReadWriteLockTest.test_reader_to_writerc s� t j � � � � f d d � } � j � �( � j | � � � j � j � � Wd QRX� j � j � � � j � j � � d S)Nc s> � j � �+ � j � j � � � j � j � � Wd QRXd S)N)r rl rw rx r )r rO r r � reader_funcb s
z<ReadWriteLockTest.test_writer_to_reader.<locals>.reader_func)r' r( r! Z assertIsNonerZ rx rw )rO r~ r )r rO r �test_writer_to_reader_ s
z'ReadWriteLockTest.test_writer_to_readerc C s� t j � } | j � �e | j | j � � | j | j � � | j � � | j | j � � Wd QRX| j | j � � Wd QRX| j | j � � | j | j � � d S)N)r' r( r! rZ rx rl rw )rO r r r r �test_double_writern s
z$ReadWriteLockTest.test_double_writerc C s� t j � } | j � �e | j | j � � | j | j � � | j � � | j | j � � Wd QRX| j | j � � Wd QRX| j | j � � | j | j � � d S)N)r' r( r rl rx rZ rw )rO r r r r �test_double_readerz s
z$ReadWriteLockTest.test_double_readerc C s� t d d � \ } } | j d t | � � | j d t | � � xL | D]D \ } } | j d t | | | � � | j d t | | | � � qH Wx0 | D]( \ } } | j d t | | | � � q� Wd S)Nrb r r )r4 rN r r )rO r1 r2 r
r3 r r r � test_multi_reader_multi_writer� s z0ReadWriteLockTest.test_multi_reader_multi_writerc C sq t d d � \ } } | j d t | � � | j d t | � � | d \ } } | j d t | | | � � d S)N� r r )r4 rN r r )rO r1 r2 r
r3 r r r � test_multi_reader_single_writer� s
z1ReadWriteLockTest.test_multi_reader_single_writerc C sx t d d � \ } } | j d t | � � | j d t | � � x0 | D]( \ } } | j d t | | | � � qH Wd S)Nr rb r )r4 rN r r )rO r1 r2 r
r3 r r r �test_multi_writer� s
z#ReadWriteLockTest.test_multi_writerN)�__name__�
__module__� __qualname__rK rR rU r\ r] re ro rs rv ry r{ r} r r� r� r� r� r� r r r r r; f s$ # +
r; )r% r@ r6 r Z
concurrentr r' r r r r+ r rk r r4 r: ZTestCaser; r r r r �<module> s -