Your IP : 216.73.217.32
���]5m � @ s� d d l m Z m Z m Z m Z m Z m Z m Z m Z 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 e d � Z e j Z Gd d � d e
j � Z Gd d � d e
j � Z d d � Z e d
k r� e � d S) � ) �TESTFN� run_unittest�
import_module�unlink�requires�_2G�_4G�
gc_collect� cpython_onlyN�mmapc @ 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 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 d% d& � Z d' d( � Z d) d* � Z d+ d, � Z d- d. � Z e j
e e d/ � d0 � d1 d2 � � Z d3 d4 � Z d5 d6 � Z d7 d8 � Z! e j
e j" d9 k d: � d; d< � � Z# e$ e j
e j" d9 k d: � d= d> � � � Z% e j
e j" d9 k d: � d? d@ � � Z& e j
e j" d9 k d: � dA dB � � Z' dC dD � Z( dE dF � Z) dG dH � Z* e j+ e j" d9 k dI � dJ dK � � Z, dL S)M� MmapTestsc C s# t j j t � r t j t � d S)N)�os�path�existsr r )�self� r �./opt/alt/python35/lib64/python3.5/test_mmap.py�setUp s zMmapTests.setUpc C s* y t j t � Wn t k
r% Yn Xd S)N)r
r r �OSError)r r r r �tearDown s
zMmapTests.tearDownc C s� t t d � } z] | j d t � | j d � | j d t d � | j � t j | j � d t � } Wd | j � Xt t | � � } | j
| j d � t � | j
t | � d t � | j
| d d � | j
| d d � d � | j
t | j t | � � | j
t | j t | � d � d | d <d | t d t d d � <| j
| d d � | j
| d d � d
� | j
| t d t d � d
� | j � t j d | � } | d k r�| j d � nC | j d � \ } } | | } | j
| t � | j
| t d � | j d d � | j
| j � d � | j d d � | j
| j � d � | j d d � | j
| j � t | � � | j
t | j d � | j
t | j d d � | j
t | j t | � d d � y | j d � Wn t k
r Yn� X| j
t | � d � | j
t | j d d � t t d � } z* | j d d � | j
| j � d � Wd | j � X| j
| j � d � | j � d S)Nzbw+s s foo� � r s � 3s bars 3 � � s foobar s [A-Za-z]+zregex match on mmap failed!� �* i i �rb�3 r ���)�openr �write�PAGESIZE�flushr �fileno�close�str�type� assertEqual�find�len� assertRaises�
IndexError� __getitem__� __setitem__�re�search�fail�span�seek�tell�
ValueError�resize� SystemError�size)r �f�m�tp�match�start�end�lengthr r r �
test_basic sd
"
$
zMmapTests.test_basicc � C s� d } t t d � � } | j d | � Wd QRXt t d � ��} t j | j � | d t j �} | j | d d � d | d � y d | | d d � <Wn t k
r� Yn X| j d � y d | d <Wn t k
r� Yn X| j d � y! | j d d � | j d
� Wn t k
r#Yn X| j d � y! | j d d � | j
d � Wn t k
rfYn X| j d � y | j d | � Wn# t k
r�Yn t k
r�Yn X| j d
� t t d � �$ } | j | j
� d | d � Wd QRXWd QRXt t d � �� } y t j | j � | d � } Wn1 t k
r^t j j d � rZ| j d � Yn* Xt j j d � s~| j d � | j � t j j d � r�t t d � � } | j | � Wd QRXWd QRXt t d � �q } t j | j � | d t j �} d | | d d � <| j | d d � d | d � | j � | j � Wd QRXt t d � � } | j
� } Wd QRX| j | d | d � t t d � �� } t j | j � | d t j �} d | | d d � <| j | d d � d | d � | j � t t d � �$ } | j | j
� d | d � Wd QRX| j t | j d | � | j � Wd QRXt t d � �, } | j t t j | j � | d d �Wd QRXt j d k rt t d � �A } | j t t j | j � | d t j d t j d t j �Wd QRXt j t t d d � B} t t d � �[ } t j | j � | d | �} | j t | j d � | j t | j
d � | j � Wd QRXd S) N�
�wb� ar �accessz#Readonly memory map data incorrect.� bz$Able to write to readonly memory mapr s abc� dr z"Able to resize readonly memory mapz*Readonly memory map data file was modifiedzr+br �winz0Opening mmap with size+1 should work on Windows.z1Opening mmap with size+1 should raise ValueError.� cz5Write-through memory map memory not updated properly.z8Write-through memory map data file not updated properly.z4Copy-on-write memory map data not written correctly.z4Copy-on-write test data file should not be modified.� �posix�flags�protZ PROT_EXECs abcdef) r r r! r r$ � ACCESS_READr( � TypeErrorr1 r3 �
write_byter6 r7 �readr5 �sys�platform�
startswithr% �truncateZ ACCESS_WRITEr# Z ACCESS_COPYr+ r
�nameZ MAP_PRIVATE� PROT_READ�getattr)r �mapsize�fpr9 r: �stuffrL r r r �test_access_parameterz s� !!
!
!
, z MmapTests.test_access_parameterc C s | j t t j d d � d S)Nr i ���)r+ r r )r r r r �test_bad_file_desc� s z MmapTests.test_bad_file_descc
C s� t t d � �H } d } t | � } | j | � | j � t j | j � | � } Wd QRXx� t | d � D]s } xj t | | d � D]U } | | | � } | j | j | � | j | � � | j | j | d � d � q� Wqk W| j
� d S)Nzwb+s aabaac deef aa r � xr ) r r r* r! r# r r$ �ranger( r) r% )r r9 �data�nr: r= Zfinish�slicer r r �test_tougher_find� s
"%zMmapTests.test_tougher_findc C s( t t d � } d } t | � } | j | � | j � t j | j � | � } | j � | j | j d � d � | j | j d � d � | j | j d d d � d � | j | j d d � d � | j | j d d d
� d � | j | j d d d � d � | j | j t
d � � d � d S)
Nzwb+s one two oness oner s ones� r r r r r\ r ) r r r* r! r# r r$ r% r( r) � bytearray)r r9 r` ra r: r r r �
test_find_end
s
zMmapTests.test_find_endc C s+ t t d � } d } t | � } | j | � | j � t j | j � | � } | j � | j | j d � d � | j | j d � d � | j | j d d d � d � | j | j d d d
� d � | j | j d d d � d � | j | j d d d � d
� | j | j t
d � � d � d S)Nzwb+s one two oness onerd s one r r r r r\ r r\ r ) r r r* r! r# r r$ r% r( �rfindre )r r9 r` ra r: r r r �
test_rfind s
zMmapTests.test_rfindc C s| t t d � } | j d d � | j � t t d � } t j | j � d d t j �} | j � | j � | j � d S) Nzwb+r � rC r rD i i )r r r! r% r r$ rM )r r9 �mfr r r �test_double_close3 s
!
zMmapTests.test_double_close�statzneeds os.stat()c C s� t t d � } | j d d � | j � t t d � } t j | j � d � } | j t | � d d � | j | j d
� d d � | j � | j � d S) Nzwb+r ri � mzrb+r z Map size should equal file size.i i i i ) r r r! r% r r$ r( r* rP )r r9 rj r r r �test_entire_file@ s
zMmapTests.test_entire_filec C s� t t d � � } | j d
d � Wd QRXt t d � �N } t j | j � d d d d t j �� } | j t | j d � Wd QRXWd QRXd S) NrB i r rm r r �offsetrD i�8 i ) r r r! r r$ rM r+ r, r- )r r9 rj r r r �test_length_0_offsetO s
*z MmapTests.test_length_0_offsetc C sl t t d � � } | j d d � Wd QRXt t d � �, } | j t t j | j � d d d �Wd QRXd S)NrB i�� rm zw+br ro i �)r r r! r+ r5 r r$ )r r9 r r r �test_length_0_large_offset\ s
z$MmapTests.test_length_0_large_offsetc C s� t t d � } | j d � | j � t j | j � d � } | j d d d � | j | d d � d d � | j � | j � d } x� t t
| � � D]� } x� t t
| � � D]� } x� t t
| � t | | � � D]� } | d | � | | | | � | | | d � } t j d t
| � � } | | d d � <| j | | | � | j | d d � | � | j � q� Wq� Wq� Wt j d
d
� } d d d d d
g } xR t j
| | | � D]; \ }
} } y | j |
| | � Wq�t k
r�Yq�Xq�Wd d d d d d d" g } x3 | D]+ \ }
} } | j t | j |
| | � q W| j � t j d# d � } | j t | j d d d � | j t | j d d d � | j t | j d d d � | j d d d � | j d d d � d S)$Nzwb+s
ABCDEabcderA � r s
ABCDEABCDEz'Map move should have duplicated front 5s
0123456789r �d r r r i����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$ �mover( r% r_ r* �max� itertools�productr5 r+ ) r r9 rj r` �dest�src�countZexpectedr: Zoffsets�sourcer8 r r r � test_moveg sH
&4%
zMmapTests.test_movec C s� t j d t � } x+ t t � D] } | j | | d d � q Wx< t t � D]. } | d @} | | | <| j | | | � qM Wd S)Nr r z+anonymously mmap'ed contents should be zero� r )r r" r_ r( )r r: �x�br r r �test_anonymous� s
zMmapTests.test_anonymousc C s� t j d d � } | j | j � | j t t d � � � | j d � | j | j � t t d � � � | j d � | j | j � t t d d � � � | j d � | j | j � d � | j d � | j | j d � t t d d � � � | j d � | j | j d
� t t d d � � � | j d � | j | j d � t t d d � � � | j d
� | j | j d � t t d
d � � � d S)Nr ri r rd � r rI rr r � r r r r\ i���) r �
addCleanupr% r! �bytesr_ r3 r( rP )r r: r r r �
test_read_all� s"
"
%
(
(
(
zMmapTests.test_read_allc C sq t j d d � } | j | j � | j t | j d � | j t | j d � | j t | j d d d g � d S)Nr ri �foog @r r r )r r� r% r+ rN rP )r r: r r r �test_read_invalid_arg� s
z MmapTests.test_read_invalid_argc
C s� t t t d � � � } t j d t | � � } | | d d � <| j | d d � | � d } xd | D]\ } xS | D]K } xB | d d � D]0 } | j | | | | � | | | | � � q� Wqt Wqg Wd S)N� r r r � �, r � r r r\ �������)
r Nr r r� r� r r\ r� r� )r� �reversedr_ r r* r( )r �sr: �indicesr= �stop�stepr r r �test_extended_getslice� s
z MmapTests.test_extended_getslicec
C s t t t d � � � } t j d t | � � } d } x� | D]� } x� | D]� } x� | d d � D]� } | | d d � <| j | d d � | � t | � } | | | | � } t t | � � } | | | | | � <| | | | | � <| j | d d � t | � � qa WqJ Wq= Wd S)Nr� r r r r� �, r r� r r r\ �������)
r Nr r r� r� r r\ r� r� )r� r� r_ r r* r( �list) r r� r: r� r= r� r� �Lr` r r r �test_extended_set_del_slice� s
z%MmapTests.test_extended_set_del_slicec C sS | j d | � | j d � | j d | d � | j � t j | j � d � S)Ns s foor r )r! r# r r$ )r r9 �halfsizer r r �make_mmap_file� s
zMmapTests.make_mmap_filec C sa t t d � } | j � t t d � �2 } | j t d t j | j � d d t j �Wd QRXd S)Nzw+br zcannot mmap an empty filer rD )r r r% ZassertRaisesRegexr5 r r$ rM )r r9 r r r �test_empty_file� s
zMmapTests.test_empty_filec C sB t t d � } z�t j } | j | | � } | j � | j � | d } t t d � } xw d d
d g D]f } y2 t j | j � | d | �} | j d d � Wn t t t
f k
r� Yqj X| j d d � qj W| j � t t d � } t j | j � | | d | �} | j | d d � d � | j � y | j d � Wn t k
rXYn� X| j t
| � d � | j t | j d
d � | j | d d � d � t t d � } | j d d � | j | j � | d � | j � | j | j � | d � | j � Wd | j � y t j t � Wn t k
r<Yn XXd S)Nzw+br zr+br ro r r s fooi i r r\ r )r r r ZALLOCATIONGRANULARITYr� r% r$ r( r5 rN �
OverflowErrorr6 r7 r* r+ r3 r4 r8 r
r r )r r9 r� r: rX ro r r r � test_offset� sL
"
zMmapTests.test_offsetc C s' Gd d � d t j � } | t � d S)Nc @ s e Z d Z d d � Z d S)z*MmapTests.test_subclass.<locals>.anon_mmapc _ s t j j | d | | � S)Nr r )r �__new__)�klass�args�kwargsr r r r� / s z2MmapTests.test_subclass.<locals>.anon_mmap.__new__N)�__name__�
__module__� __qualname__r� r r r r � anon_mmap. s r� )r r" )r r� r r r �
test_subclass- s zMmapTests.test_subclassrV zneeds mmap.PROT_READc C s� d } t t d � � } | j d | � Wd QRXt t d � } t j | j � | d t j �} | j t | j d � | j � d S)NrA rB rC r rL r� ) r r r! r r$ rV r+ rN r% )r rX rY r9 r: r r r �test_prot_readonly3 s !z MmapTests.test_prot_readonlyc C s | j t j t � d S)N)�assertIsr �errorr )r r r r �
test_error= s zMmapTests.test_errorc C se d } t t d � � } | j d t | � � Wd QRXt t d � } t j | j � t | � � } | j � x[ t t | � � D]G } | j | j � | � | j
| | � | j | j � | d � q� W| j t | j
d � | j | d d � | � | j
d � xd t t | � � D]P } | j | j � | � | j | j � | | � | j | j � | d � qW| j t | j � | j
d � | j | j d � d � | j | j � d � | j
d � | j d
� | j | j � d � | j | d d � d � | j t d � � | j | j � d
� | j | d d � d � | j t | j d � d S)Ns
0123456789rB r^ zr+br r r s 345r s bars
012bar6789s bazr� s
012barbaz9s ba�x )r r r! r* r r$ r% r_ r( r4 rO r+ r5 r3 � read_byterP re )r r` rY r9 r: �ir r r �test_io_methods@ s:
zMmapTests.test_io_methodsc C su xn d D]f } t j d d � } | j | � | j | d | � | j d � | j | j � | � | j � q Wd S)N� �� r} r r )r� r� r} r )r rO r( r3 r� r% )r r r: r r r �test_non_ascii_bytec s
z MmapTests.test_non_ascii_byte�ntzrequires Windowsc C sX d } d } t j d t | � d d �} | | d d � <t j d t | � d d �} | | d d � <| j | d d � | � | j | d d � | � | j � | j � t j d t | � d d �} | | d d � <t j d
t | � d d �} | | d d � <| j | d d � | � | j | d d � | � | j � | j � d S) Ns
0123456789s
abcdefghijr �tagnamer� Zboor r r r )r r* r( r% )r Zdata1Zdata2�m1�m2r r r � test_tagnamel s$
zMmapTests.test_tagnamec C sd t j d d � } d } t j d d d | �} | j t j | � t j | � t | � d � d S)Nr rs r� r� r r )r r( rQ � getsizeofr* )r r� r� r� r r r � test_sizeof� s
zMmapTests.test_sizeofc C s� t j d d d d �} y$ t j d d d d �d d � Wn Yn X| j � t t d � � } | j d
� Wd QRXt t d � } t j | j � d
� } | j � y | j d
� Wn Yn Xy | d d � Wn Yn X| j � d S)Nr i� r� r� i� rB r^ rA zr+br r r s
xxxxxxxxxx)r r% r r r! r$ r6 )r r: rY r9 r r r �test_crasher_on_windows� s( $
z!MmapTests.test_crasher_on_windowsc C sQ t j � } z3 | j t � � t j | j � d � } Wd QRXWd | j � Xd S)NrA )�socketr+ r r r$ r% )r r� r: r r r �test_invalid_descriptor� s
#z!MmapTests.test_invalid_descriptorc C s@ t j d d � � } | j | j � Wd QRX| j | j � d S)Nr rA r )r Z assertFalse�closed�
assertTrue)r r: r r r �test_context_manager� s z MmapTests.test_context_managerc C sh | j t � �) } t j d d � �
} t � Wd QRXWd QRX| j | j t d � | j | j d � d S)Nr rA z)wrong exception raised in context managerzcontext manager failedr )r+ � Exceptionr r ZassertIsInstance� exceptionr� r� )r �excr: r r r � test_context_manager_exception� s z(MmapTests.test_context_manager_exceptionc C sU t j d d � } t j | � } | j | � | � ~ t � | j | � d � d S)Nr ri r )r �weakref�refr� r )r Zmm�wrr r r � test_weakref� s zMmapTests.test_weakrefz(cannot resize anonymous mmaps on Windowsc C s� t j d
d � } | j | j � | j d � y | j d � Wn t k
ra | j d � Yn X| j | j d � d � | j t | j
� | j t | j d � | j t | j d � d S) Nr i i� i zresizing not supported� r� r s abcr )
r r� r% rP r6 r7 ZskipTestr( r+ r5 r� rO r! )r r: r r r �test_resize_past_pos� s
z MmapTests.test_resize_past_posN)-r� r� r� r r r@ r[ r] rc rf rh rk �unittest�
skipUnless�hasattrr
rn rp rq r| r� r� r� r� r� r� r� r� r� r r� r� r� r� rU r� r
r� r� r� r� r� r� ZskipIfr� r r r r r sF ^ }
''
' 1 9 '
# '*'' 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 e j
e j e
k d
� d d � � Z e j
e j e
k d
� d d � � Z d S)�LargeMmapTestsc C s t t � d S)N)r r )r r r r r � s zLargeMmapTests.setUpc C s t t � d S)N)r r )r r r r r � s zLargeMmapTests.tearDownc C s� t j d d � d k s( t j d k r? t d d t d � � t t d � } y( | j | � | j | � | j � WnS t t
t f k
r� y | j � Wn t t
f k
r� Yn Xt
j d � � Yn X| S) Nr rG �darwinZ largefilez-test requires %s bytes and a long time to runl zw+bz*filesystem does not have largefile support)rQ rR r r& r r r3 r! r# r r� r5 r% r� ZSkipTest)r Z
num_zeroes�tailr9 r r r �_make_test_file� s (
z LargeMmapTests._make_test_filec C se | j d d � �L } t j | j � d d d d t j �� } | j | d d � Wd QRXWd QRXd S) Nl �� � r ro l rD i���� )r� r r$ rM r( )r r9 r: r r r �test_large_offset� s *z LargeMmapTests.test_large_offsetc C s� | j d d � �� } t j d k r� | j t � �$ t j | j � d d t j �Wd QRX| j t � �$ t j | j � d d t j �Wd QRXt j | j � d d t j �� } | j | j
� d � Wd QRXWd QRXd S)Nl �� r� l rD r i ) r� rQ �maxsizer+ r� r r$ rM r5 r( r8 )r r9 r: r r r �test_large_filesize� s &&$z"LargeMmapTests.test_large_filesizec C s� d } | t | � d } | t | � } | j | | � �L } t j | j � d d t j ��! } | j | | | � | � Wd QRXWd QRXd S)Ns DEARdear r r rD )r* r� r r$ rM r( )r �boundaryr� r= r> r9 r: r r r �_test_around_boundary s $z$LargeMmapTests._test_around_boundaryz!test cannot run on 32-bit systemsc C s | j t � d S)N)r� r )r r r r �test_around_2GB s z LargeMmapTests.test_around_2GBc C s | j t � d S)N)r� r )r r r r �test_around_4GB s z LargeMmapTests.test_around_4GBN)r� r� r� r r r� r� r� r� r� r� rQ r� r r� r� r r r r r� � s 'r� c C s t t t � d S)N)r r r� r r r r � test_main s r� �__main__)Z test.supportr r r r r r r r r
r� r
r/ rv r� rQ r� r r" ZTestCaser r� r� r� r r r r �<module> s @ � � �=