Your IP : 216.73.216.213
�
f fDj � @ 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 � n d S) � ) �TESTFN� run_unittest�
import_module�unlink�requires�_2G�_4G�
gc_collect� cpython_onlyN�mmapc @ sz 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* dI S)J� MmapTestsc C s& t j j t � r" t j t � n d S)N)�os�path�existsr r )�self� r �3/opt/alt/python34/lib64/python3.4/test/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�mZtp�match�start�endZlengthr r r �
test_basic sd
"
$
zMmapTests.test_basicc ~ C s� d } t t d � � } | j d | � Wd QXt 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
reYn 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 QXWd QXt t d � �� } y t j | j � | d � } Wn4 t k
r^t j j d � rZ| j d � n Yn- Xt j j d � s�| j d � n | j � t j j d � r�t t d � � } | j | � Wd QXn Wd QXt t d � �q } t j | j � | d t j �} d | | d d � <| j | d d � d | d � | j � | j � Wd QXt t d � � } | j
� } Wd QX| 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 QX| j t | j d | � | j � Wd QXt t d � �, } | j t t j | j � | d d �Wd QXt j d k rt t d � �A } | j t t j | j � | d t j d t j d t j �Wd QXt j t t d d � B} t t d � �[ } t j | j � | d | �} | j t | j d � | j t | j
d � | j � Wd QXn d 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: ZstuffrJ 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 QXx� t | d � D]s } xj t | | d � D]U } | | | � } | j | j | � | j | � � | j | j | d � d � q� Wqj 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 � d S)
Nzwb+s one two oness oner s ones� r r r r rY r )
r r r* r! r# r r$ r% r( r) )r r9 r] r^ 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
� d S)Nzwb+s one two oness onera s one r r r r rY r rY r )
r r r* r! r# r r$ r% r( �rfind)r r9 r] r^ 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 � rA r rB i i )r r r! r% r r$ rK )r r9 �mfr r r �test_double_close1 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 re � mzrb+r z Map size should equal file size.i i i i ) r r r! r% r r$ r( r* rN )r r9 rf r r r �test_entire_file> s
zMmapTests.test_entire_filec C s� t t d � � } | j d
d � Wd QXt t d � �M } t j | j � d d d d t j �� } | j t | j d � Wd QXWd QXd S) Nr@ i r ri r r �offsetrB i�8 i ) r r r! r r$ rK r+ r, r- )r r9 rf r r r �test_length_0_offsetM s
*z MmapTests.test_length_0_offsetc C sj t t d � � } | j d d � Wd QXt t d � �, } | j t t j | j � d d d �Wd QXd S)Nr@ i�� ri zw+br rk i �)r r r! r+ r5 r r$ )r r9 r r r �test_length_0_large_offsetZ 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
ABCDEabcder? � 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$ Zmover( r% r\ r* �max� itertools�productr5 r+ ) r r9 rf r] �dest�src�countZexpectedr: Zoffsets�sourcer8 r r r � test_movee 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 re r ra � r rG rn r � r r r rY i���) r �
addCleanupr% r! �bytesr\ r3 r( rN )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 re �foog @r r r )r r~ r% r+ rL rN )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 rY �������)
r Nr r r� r� r rY 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 rY �������)
r Nr r r� r� r rY 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 s` t t d � } | j � t t d � �2 } | j t d t j | j � d d t j �Wd QXd S)Nzw+br zcannot mmap an empty filer rB )r r r% ZassertRaisesRegexr5 r r$ rK )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 rk r r s fooi i r rY r )r r r ZALLOCATIONGRANULARITYr� r% r$ r( r5 rL �
OverflowErrorr6 r7 r* r+ r3 r4 r8 r
r r )r r9 r� r: rV rk 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_subclassrT zneeds mmap.PROT_READc C s� d } t t d � � } | j d | � Wd QXt t d � } t j | j � | d t j �} | j t | j d � | j � d S)Nr? r@ rA r rJ r� ) r r r! r r$ rT r+ rL r% )r rV rW r9 r: r r r �test_prot_readonly1 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 s. d } t t d � � } | j d t | � � Wd QXt 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
d � | j t | j d
� d S)Ns
0123456789r@ r[ zr+br r r s 345r s bars
012bar6789ra �x )r r r! r* r r$ r% r\ r( r4 rM r+ r5 r3 � read_byterN )r r] rW r9 r: �ir r r �test_io_methods> s6
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� �� rx r r )r� r� rx r )r rM r( r3 r� r% )r rz r: r r r �test_non_ascii_byte_ 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_tagnameh 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 ro r� r� r r )r r( rO � 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 QXt 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� r@ r[ r? zr+br r r s
xxxxxxxxxx)r r% r r r! r$ r6 )r r: rW r9 r r r �test_crasher_on_windows� s( $
z!MmapTests.test_crasher_on_windowsc C sP t j � } z2 | j t � � t j | j � d � } Wd QXWd | j � Xd S)Nr? )�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 QX| j | j � d S)Nr r? r )r Z assertFalse�closed�
assertTrue)r r: r r r �test_context_manager� s z MmapTests.test_context_managerc C sf | j t � �( } t j d d � �
} t � Wd QXWd QX| j | j t d � | j | j d � d S)Nr r? z)wrong exception raised in context managerzcontext manager failedr )r+ � Exceptionr r ZassertIsInstanceZ 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 re r )r �weakref�refr� r )r Zmm�wrr r r � test_weakref� s zMmapTests.test_weakrefN)+r� r� r� r r r> rX rZ r` rb rd rg �unittest�
skipUnless�hasattrr
rj rl rm rw r{ r� r� r� r� r� r� r� r� r r� r� r� r� rS r� r
r� r� r� r� r� r� r r r r r sD ^ }
''
' 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 rB t d d t d � � n t t d � } y( | j | � | j | � | j � WnP t t
f k
r� y | j � Wn t t
f k
r� Yn Xt j
d � � Yn X| S) Nr rE �darwinZ largefilez-test requires %s bytes and a long time to runl zw+bz*filesystem does not have largefile support)rO rP r r& r r r3 r! r# r r� r% r� ZSkipTest)r Z
num_zeroes�tailr9 r r r �_make_test_file� s (
z LargeMmapTests._make_test_filec C sc | j d d � �K } t j | j � d d d d t j �� } | j | d d � Wd QXWd QXd S) Nl �� � r rk l rB i���� )r� r r$ rK 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 QX| j t � �$ t j | j � d d t j �Wd QXn t j | j � d d t j �� } | j | j
� d � Wd QXWd QXd S)Nl �� r� l rB r i ) r� rO �maxsizer+ r� r r$ rK 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 | | � �K } t j | j � d d t j ��! } | j | | | � | � Wd QXWd QXd S)Ns DEARdear r r rB )r* r� r r$ rK 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� rO 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/ rq r� rO r� r r" ZTestCaser r� r� r� r r r r �<module> s @ � � �=