File: //proc/thread-self/root/usr/local/lib/python3.7/test/__pycache__/test_fileio.cpython-37.opt-1.pyc
B
��g�L � @ s d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dl
m Z m Z m
Z
mZmZmZmZ d dlmZ d dlZd dlZG dd� d�ZG d d
� d
eej�ZG d d � d eej�ZG d
d� d�ZG dd� deej�ZG dd� deej�Z dd� Z e dk�re � dS )� N)�array)�proxy)�wraps)�TESTFN�TESTFN_UNICODE�check_warnings� run_unittest� make_bad_fd� cpython_only� swap_attr)�UserListc @ sz e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Ze�ejd)kd*�d+d,� �Zd-d.� Z d/d0� Z e d1d2� �Z e d3d4� �Z e d5d6� �Z e d7d8� �Z!e d9d:� �Z"e d;d<� �Z#e d=d>� �Z$e d?d@� �Z%e dAdB� �Z&e dCdD� �Z'dEdF� Z(e dGdH� �Z)e dIdJ� �Z*e dKdL� �Z+dMS )N�
AutoFileTestsc C s | � td�| _d S )N�w)�FileIOr �f)�self� r �,/usr/local/lib/python3.7/test/test_fileio.py�setUp s zAutoFileTests.setUpc C s | j r| j �� t�t� d S )N)r �close�os�remover )r r r r �tearDown s
zAutoFileTests.tearDownc C sV t | j�}|�ttd��� | �| j�� |�� � | j�� d | _| �t t
|d� d S )N�
�tell) r r �write�bytes�range� assertEqualr r � assertRaises�ReferenceError�getattr)r �pr r r � testWeakRefs s
zAutoFileTests.testWeakRefsc C s� | j �ttd��� | �| j �� d� | j �d� | �| j �� d� | j �d� | �| j �� d� | j �dd� | �| j �� d� | j �dd� | �| j �� d� | j �dd� | �| j �� d� d S ) N� r r � � � ���� )r r r r r r �seek)r r r r � testSeekTell( s zAutoFileTests.testSeekTellc C sJ | j }| �|jd� | �|jd� x"dD ]}| �ttft||d� q(W d S )N�wbF)�mode�closedZoops)r r r- r. r �AttributeError� TypeError�setattr)r r �attrr r r �testAttributes6 s
z AutoFileTests.testAttributesc C s@ t j}ttd�r,t�| j�� �}t|d|�}| �| jj |� d S )N�fstat�
st_blksize)
�io�DEFAULT_BUFFER_SIZE�hasattrr r4 r �filenor! r �_blksize)r ZblksizeZfstr r r � testBlksizeB s
zAutoFileTests.testBlksizec C sh | j �tddddg�� | j �� td�}| �td��}|�|�}W d Q R X | �|d� | �|d� d S ) Nr& r) r � s abcdefgh�rs �efgh� ) r r r r � bytearrayr r �readintor )r Zbar �nr r r �testReadintoByteArrayL s
z#AutoFileTests.testReadintoByteArrayc C s� | j �tddddg�� | j �� ttd��}| �td��}|�|�}W d Q R X | � |d� | � |d� ttd��j
d ddgd
�}| �td��}|�|�}W d Q R X | � t|�d� | � |d� d S ) Nr& r) r r< s abcdefghr= s �efghr> �H)�shape) r r r r �
memoryviewr? r r r@ r �cast)r �mr rA r r r �_testReadintoMemoryviewV s
z%AutoFileTests._testReadintoMemoryviewc
C s$ | j �tddddg�� | j �� tdd�}| �td��}|�|�}W d Q R X | �|tddddddd d
d g�� | �|d � td
d�}| �td��}|�|�}W d Q R X | �|td
dddddd d
d g�� | �|d � tdd�}| �td��}|�|�}W d Q R X | �|tdd�� | �|d � d S )Nr& r) r r< �Bs abcdefghr= �e �f �g �h r> �b����Is �efgh) r r r r r r r r@ r )r �ar rA r r r �_testReadintoArrayf s"
"
"
z AutoFileTests._testReadintoArrayc C sF ddg}| j �|� | j �� | �td�| _ | j �� }| �|d� d S )Ns 123s 456�rbs 123456)r �
writelinesr r r �readr )r �l�bufr r r �testWritelinesList| s
z AutoFileTests.testWritelinesListc C sJ t ddg�}| j�|� | j�� | �td�| _| j�� }| �|d� d S )Ns 123s 456rS s 123456)r r rT r r r rU r )r rV rW r r r �testWritelinesUserList� s
z$AutoFileTests.testWritelinesUserListc C s@ | � t| jjdddg� | � t| jjd � | � t| jjd� d S )Nr& r) � �abc)r r0 r rT )r r r r �testWritelinesError� s z!AutoFileTests.testWritelinesErrorc C st | j �d� | j �� | �td�| _ | �| j �d �d� | j �d� | �| j �d �d� | �| j � d �ddg� d S )Ns
hi
bye
abcr= r s hi
s bye
s abc)
r r r r r r rU r* �readline� readlines)r r r r �test_none_args� s
z AutoFileTests.test_none_argsc C s | � t| jjd� d S )NzHello!)r r0 r r )r r r r � test_reject� s zAutoFileTests.test_rejectc C s| | � t| j�d| j| jj| jjf � | j`| � t| j�d| j| j�� | jjf � | j�� | � t| j�d| jf � d S )Nz(<%s.FileIO name=%r mode=%r closefd=True>z&<%s.FileIO fd=%r mode=%r closefd=True>z<%s.FileIO [closed]>)r �reprr �
modulename�namer- r9 r )r r r r �testRepr� s
zAutoFileTests.testReprc C s` t �tt j�}z@| j|ddd��&}| �t|�d| j|j|j f � W d Q R X W d t �
|� X d S )Nr= F)�closefdz)<%s.FileIO name=%r mode=%r closefd=False>) r �openr �O_RDONLYr r ra rb rc r- r )r �fdr r r r �testReprNoCloseFD� s
"z AutoFileTests.testReprNoCloseFDc
C s@ t | jd| j��& | �t�� t| j� W d Q R X W d Q R X d S )Nrc )r r r � RuntimeErrorra )r r r r �testRecursiveRepr� s z AutoFileTests.testRecursiveReprc C s� | j }| �|�� � | �|j� | �t|jd� |�� | �|j� | � t
d�}| �t |j d� | �|j� |�� | �|j� d S )Nr r= � )
r Z assertFalse�isattyr. r �
ValueErrorrU r �
assertTruer r r0 r@ )r r r r r �
testErrors� s zAutoFileTests.testErrorsc C s� ddddddddd d
d d g }| j �� | �| j j� x$|D ] }t| j |�}| �t|� q:W | �t| j j� | �t| j jt d
�� | �t| j j
� | �t| j j
d� | �t| j j � | �t| j j d� | �t| j j � | �t| j j d� d S )Nr9 rm �seekable�readable�writablerU �readallr] r^ r �truncate�flushr& r � )
r r ro r. r! r rn r0 r@ r? r* r rT )r �methodsZ
methodname�methodr r r � testMethods� s
zAutoFileTests.testMethodsc
C s^ y| � dd� W n> tk
rN } z | �|jd� | �|jd� W d d }~X Y n X | �d� d S )N�.r= r zShould have raised OSError)r �OSErrorZassertNotEqual�errnor �filename�fail)r �er r r � testOpendir� s zAutoFileTests.testOpendir�ntz&test only works on a POSIX-like systemc C sP t �dt j�}| �t��}| �|d� W d Q R X t �|� | �|jj t j
� d S )Nr{ r= ) r rf rg r r| r r r Z exceptionr} ZEISDIR)r rh �cmr r r �
testOpenDirFD� s
zAutoFileTests.testOpenDirFDc s t � �� fdd��}|S )Nc sN | j }t�|�� � z� | |� W d y| j �� W n tk
rF Y nX X d S )N)r r r r9 r| )r r )�funcr r �wrapper� s z'AutoFileTests.ClosedFD.<locals>.wrapper)r )r� r� r )r� r �ClosedFD� s zAutoFileTests.ClosedFDc s t � �� fdd��}|S )Nc s� | j }t�|�� � zPy� | |� W n2 tk
rV } z| �|jtj� W d d }~X Y n X | �d� W d y| j �� W n tk
r� Y nX X d S )NzShould have raised OSError) r r r r9 r| r r} ZEBADFr )r r r� )r� r r r� s "z-AutoFileTests.ClosedFDRaises.<locals>.wrapper)r )r� r� r )r� r �ClosedFDRaises s z AutoFileTests.ClosedFDRaisesc C s |� � d S )N)r )r r r r r �testErrnoOnClose s z AutoFileTests.testErrnoOnClosec C s |� d� d S )N� a)r )r r r r r �testErrnoOnClosedWrite s z$AutoFileTests.testErrnoOnClosedWritec C s |� d� d S )Nr )r* )r r r r r �testErrnoOnClosedSeek s z#AutoFileTests.testErrnoOnClosedSeekc C s |� � d S )N)r )r r r r r �testErrnoOnClosedTell! s z#AutoFileTests.testErrnoOnClosedTellc C s |� d� d S )Nr )ru )r r r r r �testErrnoOnClosedTruncate% s z'AutoFileTests.testErrnoOnClosedTruncatec C s |� � d S )N)rq )r r r r r �testErrnoOnClosedSeekable) s z'AutoFileTests.testErrnoOnClosedSeekablec C s |� � d S )N)rr )r r r r r �testErrnoOnClosedReadable- s z'AutoFileTests.testErrnoOnClosedReadablec C s |� � d S )N)rs )r r r r r �testErrnoOnClosedWritable1 s z'AutoFileTests.testErrnoOnClosedWritablec C s |� � d S )N)r9 )r r r r r �testErrnoOnClosedFileno5 s z%AutoFileTests.testErrnoOnClosedFilenoc C s | � |�� d� d S )NF)r rm )r r r r r �testErrnoOnClosedIsatty9 s z%AutoFileTests.testErrnoOnClosedIsattyc C sH y| j �� W n tk
r" Y nX | �td�| _ t�| j �� � | j S )Nr= )r r r| r r r r9 )r r r r �
ReopenForRead= s zAutoFileTests.ReopenForReadc C s | � � }|�d� d S )Nr& )r� rU )r r r r r �testErrnoOnClosedReadF s z#AutoFileTests.testErrnoOnClosedReadc C s | � � }|�� d S )N)r� rt )r r r r r �testErrnoOnClosedReadallK s z&AutoFileTests.testErrnoOnClosedReadallc C s | � � }tdd�}|�|� d S )NrN s
xxxxxxxxxx)r� r r@ )r r rQ r r r �testErrnoOnClosedReadintoP s
z'AutoFileTests.testErrnoOnClosedReadintoN),�__name__�
__module__� __qualname__r r r# r+ r3 r; rB rH rR rX rY r\ r_ r` rd ri rk rp rz r� �unittest�skipIfr rc r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r
sJ
r
c @ s e Zd ZejZdZdS )�CAutoFileTests�_ioN)r� r� r� r� r rb r r r r r� V s r� c @ s e Zd ZejZdZdS )�PyAutoFileTests�_pyioN)r� r� r� r� r rb r r r r r� Z s r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Ze� e
� � d kd �d
d� �Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�OtherFileTestsc C s� �z�| � td�}| �|�� d� | �|�� d� | �|�� d� |�� | � td�}| �|�� d� | �|�� d� | �|�� d� |�� | � td�}| �|�� d� | �|�� d� | �|�� d� | �|�� d� |�� tj dk�r�y| � dd�}W n t
k
�r Y npX | �|�� d� | �|�� d� tj d k�rjd
tj k�rjtj � d ��sj| �|�� d� | �|�� d� |�� W d t �
t� X d S ) Nr FTr= za+�win32z/dev/ttyrQ �darwinZbsd)ZsunosZaix)r r r rr rs rq r rm �sys�platformr| �
startswithr �unlink)r r r r r � testAblesa s> zOtherFileTests.testAblesc C sN xHdD ]@}y| � t|�}W n tk
r. Y qX |�� | �d| � qW d S )N)rl ZaUzwU+�rw�rtz%r is an invalid file mode)r r rn r r )r r- r r r r �testInvalidModeStrings� s
z%OtherFileTests.testInvalidModeStringsc
C s^ z@x:dD ]2}| � t|d ��}| �|j|d � W d Q R X qW W d tj�t�rXt�t� X d S )N) )r r, )r, r, )zwb+zrb+)zw+bzrb+)rQ �ab)r� r� )zab+zab+)za+bzab+)r= rS )rS rS )zrb+zrb+)zr+bzrb+r r& )r r r r- r �path�existsr� )r Zmodesr r r r �testModeStrings� s
$ z OtherFileTests.testModeStringsc C s& | � tt�d�}|�� t�t� d S )Nr )r �strr r r r� )r r r r r �testUnicodeOpen� s z OtherFileTests.testUnicodeOpenc C s� yt �d�}W n" tk
r0 | �dt � Y nX | �|d�}z<|�d� |�� tt d��}| �|� � d� W d Q R X W d t
� t � X d S )N�asciiz could not encode %r to asciir s abcrS ) r �encode�UnicodeEncodeError�skipTestr r r rf r rU r r� )r �fnr r r r �
testBytesOpen� s
z OtherFileTests.testBytesOpenzutf-8z%test only works for utf-8 filesystemsc C s� yt �d�}W n" tk
r0 | �dt � Y nX | �|d�}z<|�d� |�� tt d��}| �|� � d� W d Q R X W d t
� t � X d S )Nzutf-8z could not encode %r to utf-8r s abcrS ) r r� r� r� r r r rf r rU r r� )r r� r r r r �testUtf8BytesOpen� s
z OtherFileTests.testUtf8BytesOpenc C s2 d}| � t| j|d� | � t| jt|d�d� d S )Nzfoo barr r� )r rn r r )r Z fn_with_NULr r r � testConstructorHandlesNULChars� s z-OtherFileTests.testConstructorHandlesNULCharsc C sJ | � t| jd� | � t| jt� � tjdkrFdd l}| � t|jt� � d S )Ni����r� r ) r rn r r| r r� r� �msvcrtZ
get_osfhandle)r r� r r r �
testInvalidFd� s
z OtherFileTests.testInvalidFdc
C s� d}y| � t|�}W nV tk
rj } z8|jd dkrZt|�}t|ksL||krZ| �d| � W d d }~X Y nX |�� | �d| � d S )NZqwertyr z&bad error message for invalid mode: %sz no error for invalid mode: %s)r r rn �argsr� r r )r Zbad_moder �msg�sr r r �testBadModeArgument� s z"OtherFileTests.testBadModeArgumentc C s� | � td�}|�tttd���� | �|�� d� |�d� | �|�� d� | �|� dt
j �d� |�d� | �|�� d� | �|� dt
j �d� |� � d S )Nr r r% r r' )
r r r r r? r r r ru r* r6 �SEEK_ENDr )r r r r r � testTruncate� s
zOtherFileTests.testTruncatec s( � fdd�}z
|� W d t �t� X d S )Nc s� � � td�} | �ttd��� | �� � � td�} | �d�}|ttd��krZ� �d| � | �� dkrx� �d| �� � | � � | �� dkr�� �d| �� � | �� t
j � t�}|dkrȈ �d| � d S ) Nr � zr+r% z(Read on file opened for update failed %rz File pos after read wrong %dz!File pos after ftruncate wrong %dz"File size after ftruncate wrong %d)
r r r r r r rU r r ru r r� �getsize)r �data�size)r r r � bug801631� s
z7OtherFileTests.testTruncateOnWindows.<locals>.bug801631)r r� r )r r� r )r r �testTruncateOnWindows� s
z$OtherFileTests.testTruncateOnWindowsc C s� zbt td�}|�d� |�� t td�}|�d� |�� t td�}|�� }|�� | �|d� W d yt�t� W n Y nX X d S )Nr, s spamr� s eggsrS s spameggs)rf r r r rU r r r� )r r �dr r r �
testAppend s
zOtherFileTests.testAppendc C s | � t| jddd� d S )N�1r )r r0 r )r r r r �testInvalidInit" s z OtherFileTests.testInvalidInitc C sf t dd��R}| �|jg � | �t| jg � | �|jg � | �t| jdd� | �|jg � W d Q R X d S )NT)�quietz/some/invalid/namer� )r r �warningsr r0 r rn )r r r r r � testWarnings% s zOtherFileTests.testWarningsc sR G dd� dt �� G � �fdd�d| j��t�ttj�}| �� �|� t�|� d S )Nc @ s e Zd ZdS )z=OtherFileTests.testUnclosedFDOnException.<locals>.MyExceptionN)r� r� r� r r r r � MyException. s r� c s e Zd Z��� fdd�Z� ZS )z:OtherFileTests.testUnclosedFDOnException.<locals>.MyFileIOc s" |dkr� d��t �| ��||�S )Nrc zblocked setting name)�super� __setattr__)r rc �value)r� �MyFileIO� __class__r r r� 0 s zFOtherFileTests.testUnclosedFDOnException.<locals>.MyFileIO.__setattr__)r� r� r� r� �
__classcell__r )r� r� )r� r r� / s r� )� Exceptionr r rf �__file__rg r r )r rh r )r� r� r �testUnclosedFDOnException- s
z(OtherFileTests.testUnclosedFDOnExceptionN)r� r� r� r� r� r� r� r� r� r� r� �getfilesystemencodingr� r� r� r� r� r� r� r� r� r� r r r r r� _ s * r� c @ s" e Zd ZejZdZedd� �ZdS )�COtherFileTestsr� c C s8 dd l }| �t| j|jd � | �t| j|jd � d S )Nr r& )� _testcapir r0 r ZINT_MAXZINT_MIN)r r� r r r �testInvalidFd_overflow< s z&COtherFileTests.testInvalidFd_overflowN)r� r� r� r� r rb r
r� r r r r r� 8 s r� c @ s e Zd ZejZdZdS )�PyOtherFileTestsr� N)r� r� r� r� r rb r r r r r� C s r� c C s0 zt tttt� W d tj�t�r*t� t� X d S )N)
r r� r� r� r� r r� r� r r� r r r r � test_mainH s
r� �__main__) r� r r6 r} r� r �weakrefr � functoolsr Z test.supportr r r r r r
r � collectionsr r� r� r
ZTestCaser� r� r� r� r� r� r� r r r r �<module> s. $ D Z