File: //lib64/python3.6/test/__pycache__/test_mmap.cpython-36.pyc
3
\o � @ s� d dl mZmZmZmZmZmZmZmZm Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZed�ZejZG dd� de
j�ZG dd� de
j�Zdd � Zed
kr�e� dS ) � ) �TESTFN� run_unittest�
import_module�unlink�requires�_2G�_4G�
gc_collect� cpython_onlyN�mmapc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z e j
eed�d�dd� �Ze j
eed�d�dd� �Ze j
eed�d�dd� �Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Z e j
ee d/�d0�d1d2� �Z d3d4� Z d5d6� Z d7d8� Z!e j
ej"d9kd:�d;d<� �Z#e$e j
ej"d9kd:�d=d>� ��Z%e j
ej"d9kd:�d?d@� �Z&e j
ej"d9kd:�dAdB� �Z'dCdD� Z(dEdF� Z)dGdH� Z*dIdJ� Z+e j,ej"d9kdK�dLdM� �Z-dNdO� Z.dPS )Q� MmapTestsc C s t jjt�rt jt� d S )N)�os�path�existsr r )�self� r �!/usr/lib64/python3.6/test_mmap.py�setUp s zMmapTests.setUpc C s( yt jt� W n tk
r" Y nX d S )N)r
r r �OSError)r r r r �tearDown s zMmapTests.tearDownc C s� t td�}zJ|jdt � |jd� |jdtd � |j� tj|j� dt �}W d |j� X tt |��}| j
|j d�t� | j
t |�dt � | j
|d d� | j
|dd� d� | j
t|jt |�� | j
t|jt |�d� d|d< d |td td d �< | j
|d d� | j
|dd� d
� | j
|td td � d
� |j� tjd|�}|d k�rj| jd� n2|jd�\}}|| }| j
|t� | j
|td � |jdd� | j
|j� d� |jdd � | j
|j� d� |jdd� | j
|j� t |�� | j
t|jd� | j
t|jd d� | j
t|jt |� d d� y|jd� W n tk
�rV Y njX | j
t |�d� | j
t|jdd� t td�}z |jdd� | j
|j� d� W d |j� X | j
|j� d� |j� d S )Nzbw+� 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�endZlengthr r r �
test_basic sd
zMmapTests.test_basicc � C s� d}t td��}|jd| � W d Q R X t td����}tj|j� |tjd�}| j|d d � d| d� yd| |d d �< W n tk
r� Y n X | jd� y d|d < W n tk
r� Y n X | jd� y|j d d � |jd
� W n tk
r� Y n X | jd� y|j d d � |j
d � W n tk
�r0 Y n X | jd� y|j d | � W n* t k
�rd Y n tk
�rx Y n X | jd
� t td�� }| j|j
� d| d� W d Q R X W d Q R X t td���}ytj|j� |d �}W n. tk
�r tjjd��r| jd� Y n"X tjjd��s&| jd� |j� tjjd��r\t td��}|j|� W d Q R X W d Q R X t td��V}tj|j� |tjd�}d| |d d �< | j|d d � d| d� |j� |j� W d Q R X t td��}|j
� }W d Q R X | j|d| d� t td���}tj|j� |tjd�}d | |d d �< | j|d d � d | d� |j� t td�� }| j|j
� d| d� W d Q R X | jt|j d | � |j� W d Q R X t td�� }| jttj|j� |dd� W d Q R X tjdk�r|t td��*}| jttj|j� |tjtjtjd� W d Q R X tjt td d �B }t td��B}tj|j� ||d �}| jt|jd � | jt|j
d � |j� W d Q R X 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�protrD Z PROT_EXEC)rL s abcdef) r! r r" r r% � ACCESS_READr) � TypeErrorr2 r4 �
write_byter7 r8 �readr6 �sys�platform�
startswithr& �truncateZ ACCESS_WRITEr$ Z ACCESS_COPYr, r
�nameZ MAP_PRIVATE� PROT_READ�getattr)r �mapsize�fpr: r; ZstuffrL r r r �test_access_parameterz s�
$ z MmapTests.test_access_parameterc C s | j ttjdd� 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 td��4}d}t|�}|j|� |j� tj|j� |�}W d Q R X xft|d �D ]V}xPt||d �D ]>}|||� }| j|j |�|j |�� | j|j |d �d� qfW qRW |j
� d S )Nzwb+s aabaac deef aa r � xr ) r! r r+ r" r$ r r% �ranger) r* r& )r r: �data�nr; r> Zfinish�slicer r r �test_tougher_find� s
zMmapTests.test_tougher_findc C s� t td�}d}t|�}|j|� |j� tj|j� |�}|j� | j|j d�d� | j|j d�d� | j|j ddd �d� | j|j dd�d� | j|j ddd
�d� | j|j ddd �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 r: r_ r` r; r r r �
test_find_end
s
zMmapTests.test_find_endc C s� t td�}d}t|�}|j|� |j� tj|j� |�}|j� | j|j d�d� | j|j d�d� | j|j ddd �d� | j|j ddd
�d� | j|j ddd �d� | j|j ddd �d
� | j|j t
d��d� d S )Nzwb+s one two oness onerc s one r r r r r[ r r[ r ) r! r r+ r" r$ r r% r& r) �rfindrd )r r: r_ r` r; r r r �
test_rfind s
zMmapTests.test_rfindc C s\ t td�}|jdd � |j� t td�}tj|j� dtjd�}|j� |j� |j� d S ) Nzwb+r � rC r )rD i i )r! r r" r& r r% rM )r r: �mfr r r �test_double_close3 s
zMmapTests.test_double_close�statzneeds os.stat()c C sv t td�}|jdd � |j� t td�}tj|j� d�}| jt|�d d� | j|jd
�d d � |j� |j� d S ) Nzwb+r rh � mzrb+r z Map size should equal file size.i i i i ) r! r r" r& r r% r) r+ rP )r r: ri r r r �test_entire_file@ s
zMmapTests.test_entire_filec C sr t td��}|jd d � W d Q R X t td��:}tj|j� ddtjd��}| jt|jd� W d Q R X W d Q R X d S )
NrB i r rl r r )�offsetrD i�8 i ) r! r r" r r% rM r, r- r. )r r: ri r r r �test_length_0_offsetO s
z MmapTests.test_length_0_offsetc C sX t td��}|jdd � W d Q R X t td�� }| jttj|j� ddd� W d Q R X d S )NrB i�� rl zw+br i �)rn )r! r r" r, r6 r r% )r r: r r r �test_length_0_large_offset\ s
z$MmapTests.test_length_0_large_offsetc C s2 t td�}|jd� |j� tj|j� d�}|jddd� | j|d d � dd� |j� |j� d}x�t t
|��D ]�}x�t t
|��D ]�}x�t t
|�t ||� �D ]v}|d |� |||| � ||| d � }tjd t
|��}||d d �< |j|||� | j|d d � |� |j� q�W q�W qrW tjd
d
�}dddd d
g} xHt j
| | | �D ]6\}
}} y|j|
|| � W n tk
�r~ Y nX �qLW dddd d d d"g} x(| D ] \}
}} | jt|j|
|| � �q�W |j� tjd#d �}| jt|jddd � | jt|jd dd � | jt|jdd d � |jddd � |jddd� 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% Zmover) r& r^ r+ �max� itertools�productr6 r, ) r r: ri r_ �dest�src�countZexpectedr; Zoffsets�sourcer9 r r r � test_moveg sH
,
zMmapTests.test_movec C sf t j dt�}x"tt�D ]}| j|| dd� qW x0tt�D ]$}|d@ }|||< | j|| |� q:W d 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� |jttd��� |jd� | j|j� ttd��� |jd� | j|j� ttdd��� |jd� | j|j� d� |jd� | j|jd �ttdd��� |jd� | j|jd
�ttdd��� |jd� | j|jd�ttdd��� |jd
� | j|jd�ttd
d��� d S )Nr rh r rc � r rI rq r � r r r r[ i���) r �
addCleanupr& r" �bytesr^ r4 r) rP )r r; r r r �
test_read_all� s"
zMmapTests.test_read_allc C sR t j dd�}| j|j� | jt|jd� | jt|jd� | jt|jdddg� d S )Nr rh �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 ttd���}tjd t|��}||d d �< | j|d d � |� d}xP|D ]H}xB|D ]:}x4|dd � D ]$}| j||||� ||||� � qfW qTW qJW d 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 ttd���}tjd t|��}d}x�|D ]�}x�|D ]�}x�|dd � D ]z}||d d �< | j|d d � |� t|�}||||� }t t|��}|||||�< |||||�< | j|d d � t |�� qFW q4W q*W d 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 sB |j d| � |j d� |j d|d � |j� tj|j� d�S )Nr s foor r )r" r$ r r% )r r: �halfsizer r r �make_mmap_file� s
zMmapTests.make_mmap_filec C sJ t td�}|j� t td��$}| jtdtj|j� dtjd� W d Q R X d S )Nzw+br zcannot mmap an empty filer )rD )r! r r& ZassertRaisesRegexr6 r r% rM )r r: r r r �test_empty_file� s
zMmapTests.test_empty_filec C s� t td�}�z�tj}| j||�}|j� |j� |d }t td�}xXdD ]P}y$tj|j� ||d�}| jdd� W n tt t
fk
r� Y qHX | jdd� qHW |j� t td�}tj|j� || |d�}| j|dd� d� |j� y|j d � W n t k
�r Y n~X | jt
|�d � | jt|jd
d� | j|dd� d� t td �}|jdd� | j|j� |d � |j� | j|j� |d � |j� W d |j� ytjt� W n tk
�r� Y nX X d S )Nzw+br zr+br )rn r r s fooi i r r[ r )r[ r N)r! r r ZALLOCATIONGRANULARITYr� r& r% r) r6 rN �
OverflowErrorr7 r8 r+ r, r4 r5 r9 r
r r )r r: r� r; rX rn r r r � test_offset� sL
zMmapTests.test_offsetc C s G dd� dt j �}|t� d S )Nc @ s e Zd Zdd� ZdS )z*MmapTests.test_subclass.<locals>.anon_mmapc _ s t j j| df|�|�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 sd d}t td��}|jd| � W d Q R X t td�}tj|j� |tjd�}| jt|jd� |j� d S )NrA rB rC r )rL r� ) r! r r" r r% rV r, rN r& )r rX rY r: r; r r r �test_prot_readonly3 s
z MmapTests.test_prot_readonlyc C s | j tjt� d S )N)�assertIsr �errorr )r r r r �
test_error= s zMmapTests.test_errorc C s� d}t td��}|jdt|� � W d Q R X t td�}tj|j� t|��}|j� xFtt|��D ]6}| j|j � |� |j
|| � | j|j � |d � q`W | j t |j
d� | j|d d � |� |j
d� xLtt|��D ]<}| j|j � |� | j|j� || � | j|j � |d � q�W | j t |j� |j
d� | j|jd�d� | j|j � d � |j
d� |jd
� | j|j � d � | j|d d � d � |jtd �� | j|j � d
� | j|d d � d� | j t |jd� 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) r5 rO r, r6 r4 � read_byterP rd )r r_ rY r: r; �ir r r �test_io_methods@ s:
zMmapTests.test_io_methodsc C sZ xTdD ]L}t j dd�}|j|� | j|d |� |jd� | j|j� |� |j� qW d S )N� �� r{ r r )r� r� r{ r )r rO r) r4 r� r& )r r} r; r r r �test_non_ascii_bytec s
z MmapTests.test_non_ascii_byte�ntzrequires Windowsc C s d}d}t |�t |�ks t�tjdt |�dd�}||d d �< tjdt |�dd�}||d d �< | j|d d � |� | j|d d � |� |j� |j� tjd t |�dd�}||d d �< tjd
t |�dd�}||d d �< | j|d d � |� | j|d d � |� |j� |j� d S ) Ns
0123456789s
abcdefghijr r� )�tagnameZboor r r r )r+ �AssertionErrorr r) r& )r Zdata1Zdata2�m1�m2r r r � test_tagnamel s& zMmapTests.test_tagnamec C sH t j dd�}d}t j dd|d�}| jtj|�tj|�t|� d � d S )Nr rr 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 ddd�}y t j d ddd�d d � W n Y nX |j� ttd��}|jd
� W d Q R X ttd �}t j |j� d
�}|j� y|jd
� W n Y nX y|d d � W n Y nX |j� d S )Nr i� r� )r� i� rB r] rA zr+br r r s
xxxxxxxxxx)r r& r! r r" r% r7 )r r; rY r: r r r �test_crasher_on_windows� s(
z!MmapTests.test_crasher_on_windowsc C sB t j � }z*| jt�� tj|j� d�}W d Q R X W d |j� X d S )NrA )�socketr, r r r% r& )r r� r; r r r �test_invalid_descriptor� s
z!MmapTests.test_invalid_descriptorc C s4 t j dd��}| j|j� W d Q R X | 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 sT | j t��"}tjdd��
}t�W d Q R X W d Q R X | j|jtd� | j|jd� d S )Nr rA 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 s> t j dd�}tj|�}| j|� |� ~t� | j|� d � d S )Nr rh r )r �weakref�refr� r )r �mm�wrr r r � test_weakref� s
zMmapTests.test_weakrefc C sX t j d
d�}| j|jd�d� | j|jd�d� | j|jd�d� | j|jd�d � d S ) Nr rh r r r] s yzr s pythonr r )r r) r" )r r� r r r �0test_write_returning_the_number_of_bytes_written� s
z:MmapTests.test_write_returning_the_number_of_bytes_writtenz(cannot resize anonymous mmaps on Windowsc C s� t j d
d�}| j|j� |jd� y|jd� W n tk
rN | jd� Y nX | j|jd�d� | jt |j
� | jt |j d� | jt |j d � d S ) Nr i i� i zresizing not supported� r r s abcr )
r r� r& rP r7 r8 ZskipTestr) r, r6 r� rO r" )r r; r r r �test_resize_past_pos� s
z MmapTests.test_resize_past_posc C sL t j dd�}| jt�� || W d Q R X | jt�� |d W d Q R X d S )Nr rh r r )r r, rN )r r; r r r � test_concat_repeat_exception� s
z&MmapTests.test_concat_repeat_exceptionN)/r� r� r� r r r@ rZ r\ rb re rg rj �unittest�
skipUnless�hasattrr
rm ro rp rz r~ r� r� r� r� r� r� r� r� r r� r� r� r� rU r� r
r� r� r� r� r� r� r� ZskipIfr� r� r r r r r sJ ^}
1 9
# r c @ sp e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze j
e j e
kd
�dd� �Ze j
e j e
kd
�dd� �ZdS )�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 jd d� dks t jdkr.tddtd� � ttd�}y |j|� |j|� |j� W nJ t t
t fk
r� y |j � W n t t
fk
r� Y nX t
jd��Y nX |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 r4 r" r$ r r� r6 r& r� ZSkipTest)r Z
num_zeroes�tailr: r r r �_make_test_file� s
z LargeMmapTests._make_test_filec C sP | j dd��:}tj|j� ddtjd��}| j|d d� W d Q R X W d Q R X d S )Nl �� � r l )rn rD i���� )r� r r% rM r) )r r: r; r r r �test_large_offset s z LargeMmapTests.test_large_offsetc C s� | j dd���}tjdk rp| jt�� tj|j� dtjd� W d Q R X | jt�� tj|j� dtjd� W d Q R X tj|j� dtjd��}| j |j
� d� W d Q R X W d Q R X d S )Nl �� r� l )rD r i ) r� rQ �maxsizer, r� r r% rM r6 r) r9 )r r: r; r r r �test_large_filesize s
z"LargeMmapTests.test_large_filesizec C sr d}|t |�d }|t |� }| j||��<}tj|j� dtjd��}| j|||� |� W d Q R X W d Q R X d S )Ns DEARdear r r )rD )r+ r� r r% rM r) )r �boundaryr� r> r? r: 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 tt� 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
r0 rt r� rQ r� r r# ZTestCaser r� r� r� r r r r �<module> s& , ^=