shell bypass 403
o
` b� � @ sd d dl mZ d dlmZmZ d dlZd dlZd dlmZm Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlmZ d dlZd dlZe�d�Zd d lmZm Z m Z da d
d � Z G d d
� d
ej �Z!G dd� de!�Z"G dd� de!�Z#G dd� de!�Z$G dd� de!�Z%G dd� de!�Z&dd� Z'e(dkr�e�)� dS dS )� )�support)�
bigmemtest�_4GN)�BytesIO�DEFAULT_BUFFER_SIZE)�
import_helper)�threading_helper)�unlink�bz2)�BZ2File�
BZ2Compressor�BZ2Decompressorc C s4 t d u r tt�d��a t rtjdg| d�S t�| �S )NZbunzip2)�input)�has_cmdline_bunzip2�bool�shutil�which�
subprocess� check_outputr
�
decompress)�data� r �6/usr/local/python-3.10/lib/python3.10/test/test_bz2.py�ext_decompress s
r c @ s� e Zd ZdZg d�Zd�e�ZdZdZdZ dZ
e d�Z e
�
ej�e
�ej�e��d ��D ]*Zeed
��Ze
e�ee �e
d � �7 Z
W d � n1 sKw Y e
dkrV qWq,eje d d
�Zdd� Zdd� Z d S )�BaseTestzBase for other testcases.)s root:x:0:0:root:/root:/bin/bash
s bin:x:1:1:bin:/bin:
s daemon:x:2:2:daemon:/sbin:
s adm:x:3:4:adm:/var/adm:
s lp:x:4:7:lp:/var/spool/lpd:
s sync:x:5:0:sync:/sbin:/bin/sync
s- shutdown:x:6:0:shutdown:/sbin:/sbin/shutdown
s! halt:x:7:0:halt:/sbin:/sbin/halt
s" mail:x:8:12:mail:/var/spool/mail:
s" news:x:9:13:news:/var/spool/news:
s# uucp:x:10:14:uucp:/var/spool/uucp:
s operator:x:11:0:operator:/root:
s! games:x:12:100:games:/usr/games:
s, gopher:x:13:30:gopher:/usr/lib/gopher-data:
s( ftp:x:14:50:FTP User:/var/ftp:/bin/bash
s# nobody:x:65534:65534:Nobody:/home:
s. postfix:x:100:101:postfix:/var/spool/postfix:
s- niemeyer:x:500:500::/home/niemeyer:/bin/bash
s> postgres:x:101:102:PostgreSQL Server:/var/lib/pgsql:/bin/bash
s6 mysql:x:102:103:MySQL server:/var/lib/mysql:/bin/bash
s# www:x:103:104::/var/www:/bin/false
� s� BZh91AY&SY.�N >_� @��n ?��0�� �F��#&�F��Ц�MQ���4hh$�i4�4S�<Q�H���Շ���
��I��/E pu�]ɻDL
Q 1ߠ���2O9�� 7d�I �J�L#����/��#C��W����M��Xa<e�t�L�����˄.�����`n}�lh E,�ZVH>���9gd6
�Պ��.K�v��xw�ߡ���|/";���ձ#A���o��y\�O煚��f8�2��"%�>V,��z⟣�"�E)I�^��r�U�2�6���U��0��^��DF����ݸ����ankR���3]���f��"6l$�j@���l1..�k�ls����E��
��k��n��Q�%y�v��z��̹z n7.G�]��B@�!8`s BZh9rE8P� s this is not a valid bzip2 filer i z*.py�rbN� ��
compresslevelc C s t �� \}| _t�|� d S �N)�tempfile�mkstemp�filename�os�close)�self�fdr r r �setUpR s zBaseTest.setUpc C s t | j� d S r )r r# �r& r r r �tearDownV s zBaseTest.tearDown) �__name__�
__module__� __qualname__�__doc__�
TEXT_LINES�join�TEXT�DATA�
EMPTY_DATA�BAD_DATAZ test_size� bytearray�BIG_TEXT�globr$ �path�escape�dirname�__file__Zfname�openZfh�readinto�
memoryviewr
�compress�BIG_DATAr( r* r r r r r & s&
& �� r c @ s� e Zd ZdZd}dd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z d>d?� Z!d@dA� Z"dBdC� Z#dDdE� Z$dFdG� Z%dHdI� Z&dJdK� Z'dLdM� Z(dNdO� Z)dPdQ� Z*dRdS� Z+dTdU� Z,dVdW� Z-dXdY� Z.dZd[� Z/d\d]� Z0d^d_� Z1d`da� Z2dbdc� Z3ddde� Z4dfdg� Z5dhdi� Z6djdk� Z7dldm� Z8dndo� Z9dpdq� Z:drds� Z;dtdu� Z<dvdw� Z=dxdy� Z>dzd{� Z?d|S )~� BZ2FileTestzTest the BZ2File class.r r c C sL t | jd��}|�| j| � |�|� W d � d S 1 s w Y d S )N�wb)r< r# �writer2 )r& �streams�suffix�fr r r �createTempFile] s "�zBZ2FileTest.createTempFilec C s� | � ttd� | � tttjd� | � tttjd� | � tttjd� | j tttjdd� | j tttjdd� | � tttjdd � d S )
Ngw��/�^@�z�rxZrbtr r �
�r� )� assertRaises� TypeErrorr �
ValueErrorr$ �devnullr) r r r � testBadArgsb s zBZ2FileTest.testBadArgsc C s\ | � � t| j��}| �t|jt� � | �|�� | j� W d � d S 1 s'w Y d S r � rG r r# rM rN �read�float� assertEqualr1 �r& �bz2fr r r �testReadm �
"�zBZ2FileTest.testReadc C sN | j d| jd� t| j��}| �t|j� W d � d S 1 s w Y d S )Nr �rD rE )rG r4 r r# rM �OSErrorrS rV r r r �testReadBadFiles s "�zBZ2FileTest.testReadBadFilec C sd | j dd� t| j�� }| �t|jt� � | �|�� | jd � W d � d S 1 s+w Y d S �N� �rD rR rV r r r �testReadMultiStreamx s
"�z BZ2FileTest.testReadMultiStreamc C s� t j}t| j�t _z;| jdd� t| j�� }| �t|j t
� � | � |� � | j d � W d � n1 s4w Y W |t _d S W |t _d S |t _w r] )
� _compression� BUFFER_SIZE�lenr2 rG r r# rM rN rS rT rU r1 )r& Z buffer_sizerW r r r �testReadMonkeyMultiStream~ s �
�z%BZ2FileTest.testReadMonkeyMultiStreamc C sP | j | jd� t| j��}| �|�� | j� W d � d S 1 s!w Y d S )N)rE �rG r4 r r# rU rS r1 rV r r r �testReadTrailingJunk� s "�z BZ2FileTest.testReadTrailingJunkc C sV | j d| jd� t| j��}| �|�� | jd � W d � d S 1 s$w Y d S )Nr^ rZ re rV r r r � testReadMultiStreamTrailingJunk� s "�z+BZ2FileTest.testReadMultiStreamTrailingJunkc C s\ | � � t| j��}| �t|jt� � | �|�d�d� W d � d S 1 s'w Y d S )Nr r )rG r r# rM rN rS rT rU rV r r r � testRead0� rY zBZ2FileTest.testRead0c C sf | � � t| j�� }d} |�d�}|sq||7 }q
| �|| j� W d � d S 1 s,w Y d S )Nr TrJ �rG r r# rS rU r1 �r& rW �text�strr r r �testReadChunk10� s
�"�zBZ2FileTest.testReadChunk10c C sn | j dd� t| j��"}d} |�d�}|sq ||7 }q| �|| jd � W d � d S 1 s0w Y d S )Nr^ r_ r TrJ ri rj r r r �testReadChunk10MultiStream� s
�"�z&BZ2FileTest.testReadChunk10MultiStreamc C sT | � � t| j��}| �|�d�| jd d� � W d � d S 1 s#w Y d S )N�d )rG r r# rU rS r1 rV r r r � testRead100� s "�zBZ2FileTest.testRead100c C st | � � t| j��'}|�� }| �t|�d� | �| j�|�� | � |�
� | j� W d � d S 1 s3w Y d S �Nr ) rG r r# �peek�assertNotEqualrc �
assertTruer1 �
startswithrU rS )r& rW �pdatar r r �testPeek� s "�zBZ2FileTest.testPeekc C s� | � � t| j��J}d}t|�}| �|�|�|� | �|| jd |� � t| j�| }tt| j��}| �|�|�|� | �|d |� | j| d � � W d � d S 1 sVw Y d S )N� )rG r r# r5 rU r= r1 rc )r& rW �n�br r r � testReadInto� s ""�zBZ2FileTest.testReadIntoc C sd | � � t| j�� }| �t|jd � | jD ]
}| �|�� |� qW d � d S 1 s+w Y d S r �rG r r# rM rN �readliner/ rU �r& rW �liner r r � testReadLine� s
�"�zBZ2FileTest.testReadLinec C sl | j dd� t| j��!}| �t|jd � | jd D ]
}| �|�� |� qW d � d S 1 s/w Y d S r] r| r~ r r r �testReadLineMultiStream� s �"�z#BZ2FileTest.testReadLineMultiStreamc C sZ | � � t| j��}| �t|jd � | �|�� | j� W d � d S 1 s&w Y d S r �rG r r# rM rN � readlinesrU r/ rV r r r �
testReadLines� s
"�zBZ2FileTest.testReadLinesc C sb | j dd� t| j�� }| �t|jd � | �|�� | jd � W d � d S 1 s*w Y d S r] r� rV r r r �testReadLinesMultiStream� s
"�z$BZ2FileTest.testReadLinesMultiStreamc C sN | � � t| j��}| �tt|��| j� W d � d S 1 s w Y d S r �rG r r# rU �list�iterr/ rV r r r � testIterator� s "�zBZ2FileTest.testIteratorc C sV | j dd� t| j��}| �tt|��| jd � W d � d S 1 s$w Y d S r] r� rV r r r �testIteratorMultiStream� s "�z#BZ2FileTest.testIteratorMultiStreamc C s: | � � t| j�}|�� | �tt|� | �t|j� d S r )rG r r# r% rM rO �nextr� rV r r r �testClosedIteratorDeadlock� s
z&BZ2FileTest.testClosedIteratorDeadlockc C s� t | jd��}| �t|j� |�| j� W d � n1 s w Y t| jd��}| �t|� � �| j� W d � d S 1 s@w Y d S )N�wr �
r r# rM rN rC r1 r< rU r rS �r& rW rF r r r � testWrite� s �"�zBZ2FileTest.testWritec C s� t | jd��%}d} | j|d |d d � }|sq$|�|� |d7 }q
W d � n1 s.w Y t| jd��}| �t|�� �| j� W d � d S 1 sPw Y d S )Nr� r TrJ r r )r r# r1 rC r< rU r rS )r& rW ry rl rF r r r �testWriteChunks10� s
� �"�z BZ2FileTest.testWriteChunks10c C s� t j| jdd�}t| jddd��}|�| j� W d � n1 s!w Y t| jd��}| �|�� |� W d � d S 1 s@w Y d S )Nr^ r r� r ) r
r? r1 r r# rC r< rU rS )r& ZexpectedrW rF r r r � testWriteNonDefaultCompressLevel
s �"�z,BZ2FileTest.testWriteNonDefaultCompressLevelc C s� t | jd��}| �t|j� |�| j� W d � n1 s w Y | �t|jdg� t| jd��}| �t |�
� �| j � W d � d S 1 sIw Y d S )Nr� �ar ) r r# rM rN �
writelinesr/ rO r< rU r rS r1 r� r r r �testWriteLines s �"�zBZ2FileTest.testWriteLinesc C s� t | jd��
}|�d� W d � n1 sw Y t | jd��}| �t|jd� | �t|jdg� W d � d S 1 s>w Y d S )Nr� s abcrK � a)r r# rC rM r[ r� rV r r r � testWriteMethodsOnReadOnlyFile s �"�z*BZ2FileTest.testWriteMethodsOnReadOnlyFilec C s� t | jd��}| �t|j� |�| j� W d � n1 s w Y t | jd��}| �t|j� |�| j� W d � n1 sAw Y t| jd��}| �t|� � �| jd � W d � d S 1 sew Y d S )Nr� r� r � r� r� r r r �
testAppend# s � � "�zBZ2FileTest.testAppendc C sj | � � t| j��"}| �t|j� |�d� | �|�� | jdd � � W d � d S 1 s.w Y d S �N� ) rG r r# rM rN �seekrU rS r1 rV r r r �testSeekForward- s
"�zBZ2FileTest.testSeekForwardc C sx | j dd� t| j��'}| �t|j� |�t| j�d � | �|� � | jdd � � W d � d S 1 s5w Y d S )Nr� r_ r� )
rG r r# rM rN r� rc r1 rU rS rV r r r � testSeekForwardAcrossStreams4 s "�z(BZ2FileTest.testSeekForwardAcrossStreamsc C sh | � � t| j��!}|�d� |�dd� | �|�� | jdd � � W d � d S 1 s-w Y d S �Ni� �j���r i^ )rG r r# rS r� rU r1 rV r r r �testSeekBackwards; s
"�z BZ2FileTest.testSeekBackwardsc C s� | j dd� t| j��7}t| j�d }|dkr$|t|�|��8 }|dks|�dd� | �|�� | jdd � | j � W d � d S 1 sEw Y d S )Nr� r_ ro r r� r i���)rG r r# rc r1 rS r� rU )r& rW Zreadtor r r � testSeekBackwardsAcrossStreamsB s � ""�z*BZ2FileTest.testSeekBackwardsAcrossStreamsc C sh | � � t| j��!}|�dd� | �|�� | jt| j�d d � � W d � d S 1 s-w Y d S )Nr� r� r� )rG r r# r� rU rS r1 rc rV r r r �testSeekBackwardsFromEndK s
&"�z$BZ2FileTest.testSeekBackwardsFromEndc C sf | j dd� t| j�� }|�dd� | �|�� | jd dd � � W d � d S 1 s,w Y d S )Nr� r_ i���)rG r r# r� rU rS r1 rV r r r �%testSeekBackwardsFromEndAcrossStreamsQ s
"�z1BZ2FileTest.testSeekBackwardsFromEndAcrossStreamsc C sh | � � t| j��!}|�d� | �|�� t| j�� | �|�� d� W d � d S 1 s-w Y d S �N��I r � rG r r# r� rU �tellrc r1 rS rV r r r �testSeekPostEndW s
"�zBZ2FileTest.testSeekPostEndc C sp | j dd� t| j��#}|�d� | �|�� t| j�d � | �|�� d� W d � d S 1 s1w Y d S �Nr^ r_ r� r r� rV r r r �testSeekPostEndMultiStream^ s
"�z&BZ2FileTest.testSeekPostEndMultiStreamc C sr | � � t| j��&}|�d� |�d� | �|�� t| j�� | �|�� d� W d � d S 1 s2w Y d S r� r� rV r r r �testSeekPostEndTwicee s
"�z BZ2FileTest.testSeekPostEndTwicec C sz | j dd� t| j��(}|�d� |�d� | �|�� t| j�d � | �|�� d� W d � d S 1 s6w Y d S r� r� rV r r r � testSeekPostEndTwiceMultiStreamm s
"�z+BZ2FileTest.testSeekPostEndTwiceMultiStreamc C sd | � � t| j�� }|�d� | �|�� d� | �|�� | j� W d � d S 1 s+w Y d S )Nr� r �rG r r# r� rU r� rS r1 rV r r r �testSeekPreStartu s
"�z BZ2FileTest.testSeekPreStartc C sl | j dd� t| j��!}|�d� | �|�� d� | �|�� | jd � W d � d S 1 s/w Y d S )Nr� r_ r� r r� rV r r r �testSeekPreStartMultiStream| s
"�z'BZ2FileTest.testSeekPreStartMultiStreamc C sz | � � t| jd��"}t|�}z| �|�� |�� � W |�� n|�� w W d � n1 s/w Y | �t|j� d S )Nr ) rG r< r# r rU �filenor% rM rO )r& ZrawfrW r r r �
testFileno� s �zBZ2FileTest.testFilenoc C s� t t| j��}z| �|�� � |�� | �|�� � W |�� n|�� w | �t|j� t t� d�}z
| � |�� � W |�� n|�� w | �t|j� t| j�}dd� |_t |�}z
| � |�� � W |�� n|�� w | �t|j� d S )Nr� c S s dS )NFr r r r r �<lambda>� s z*BZ2FileTest.testSeekable.<locals>.<lambda>)
r r r2 rt �seekablerS r% rM rO � assertFalse)r& rW �srcr r r � testSeekable� s&
zBZ2FileTest.testSeekablec C � t t| j��}z| �|�� � |�� | �|�� � W |�� n|�� w | �t|j� t t� d�}z
| � |�� � W |�� n|�� w | �t|j� d S �Nr� )
r r r2 rt �readablerS r% rM rO r� rV r r r � testReadable� � zBZ2FileTest.testReadablec C r� r� )
r r r2 r� �writablerS r% rM rO rt rV r r r � testWritable� r� zBZ2FileTest.testWritablec C s& | � � td�D ]}t| j�}~qd S )Ni' )rG �ranger r# )r& �i�or r r � testOpenDel� s
�zBZ2FileTest.testOpenDelc C � | � ttd� d S )Nz
/non/existent)rM r[ r r) r r r �testOpenNonexistent� � z BZ2FileTest.testOpenNonexistentc C s� d}t | jd��
}|�|� W d � n1 sw Y t| j�� }|�� }W d � n1 s1w Y | �|dg� t| j��}t|�� �}W d � n1 sSw Y | �|dg� d S )Ns, BZh91AY&SY�b�] � !�h3M<]��BCe�%trB s Test)r< r# rC r r� rU r� )r& r rF rW �linesZxlinesr r r �testReadlinesNoNewline� s �
� �z"BZ2FileTest.testReadlinesNoNewlinec C s� d }t | jd��
}|�d� W d � n1 sw Y t | jd�}|�� z|� W d � n1 s5w Y W n tyD Y nw | �d� z t | jd�� }dd W d � n1 s`w Y W n
typ Y d S w | �d� d S )NrB s xxxr z4__enter__ on a closed file didn't raise an exceptionr r z 1/0 didn't raise an exception)r r# rC r% rO Zfail�ZeroDivisionError�r& rF r r r �testContextProtocol� s0 � �� �
�� �z BZ2FileTest.testContextProtocolc s� dd �d}t | jd��8���fdd�� � fdd�t|�D �}t�|�� W d � n1 s/w Y W d � d S W d � d S 1 sGw Y d S ) N� 1i rJ rB c s t d�D ]} ��� � qd S �Nr^ )r� rC )r� )r rF r r �comp� s �z'BZ2FileTest.testThreading.<locals>.compc s g | ]}t j� d ��qS ))�target)� threading�Thread)�.0r� )r� r r �
<listcomp>� s z-BZ2FileTest.testThreading.<locals>.<listcomp>)r r# r� r Z
start_threads)r& Znthreads�threadsr )r� r rF r �
testThreading� s ��"�zBZ2FileTest.testThreadingc C s> | � � t| jd �}|d }t| j��)}|�|� | �t|�| jd |d � � | �|�� | j|d � � W d � n1 s@w Y t| j��"}|� � | �t|�| jd � | �|� � | jd � W d � n1 sow Y t| j�� }|�
� | � t t|� | �|�
� g � W d � d S 1 s�w Y d S )Nr r� r )
rG rc r/ r r# rS rU r� r1 r} r� rM �
StopIteration)r& ZlinelenZhalflenrW r r r �testMixedIterationAndReads� s$
� � "�z&BZ2FileTest.testMixedIterationAndReadsc C s� d}d}t | jd��
}|�|� W d � n1 sw Y t | jd��
}|�|� W d � n1 s5w Y t | j��}| �|�� || � W d � d S 1 sUw Y d S )Ns� foofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoofoos� barbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarbarr� r� )r r# rC rU rS )r& Zdata1Zdata2rW r r r �testMultiStreamOrdering s � � "�z#BZ2FileTest.testMultiStreamOrderingc C s� | j }z|�d�}W n ty | �d� Y nw t|d��}|�| j� W d � n1 s/w Y t|d��}| �|�� | j� W d � n1 sMw Y t|d��}| �|�� | j� W d � d S 1 slw Y d S )N�asciiz%Temporary file name needs to be ASCIIrB r ) r# �encode�UnicodeEncodeErrorZskipTestr rC r2 rU rS )r& Z str_filenameZbytes_filenamerF r r r �testOpenBytesFilename s � � � "�z!BZ2FileTest.testOpenBytesFilenamec C s� t �| j�}t|d��}|�| j� W d � n1 s w Y t|d��}| �|�� | j� W d � d S 1 s;w Y d S )NrB r )�pathlib�Pathr# r rC r2 rU rS )r& r# rF r r r �testOpenPathLikeFilename) s � "�z$BZ2FileTest.testOpenPathLikeFilenamec C sh t jdtd� dd�}| �t|�tj� tt|��}| � |�
d�d� dt }| � |j
j�� |d� dS )z-Decompressed data buffering should be limited� g ��>A� r r z)Excessive amount of data was decompressedN)r
r? �intZ
assertLessrc ra rb r r rU rS r �assertLessEqual�_buffer�rawr� )r& Zbomb�decompZ
max_decompr r r �testDecompressLimited0 s �z!BZ2FileTest.testDecompressLimitedc C s� t | j��5}t|��}| �t|jt� � | �|�� | j� W d � n1 s'w Y | � |j
� W d � d S 1 s=w Y d S r ) r r2 r rM rN rS rT rU r1 r� �closed�r& �biorW r r r �testReadBytesIO> s
�"�zBZ2FileTest.testReadBytesIOc C s� t | j��C}t|��&}|�� }| �t|�d� | �| j�|�� | � |�
� | j� W d � n1 s3w Y W d � d S W d � d S 1 sKw Y d S rq ) r r2 r rr rs rc rt r1 ru rU rS )r& r� rW rv r r r �testPeekBytesIOE s
��"�zBZ2FileTest.testPeekBytesIOc C s� t � �<}t|d��}| �t|j� |�| j� W d � n1 s!w Y | �t|�� �| j� | � |j
� W d � d S 1 sBw Y d S r� ) r r rM rN rC r1 rU r �getvaluer� r� r� r r r �testWriteBytesIOM s �"�z BZ2FileTest.testWriteBytesIOc C s� t | j��>}t|��!}| �t|j� |�d� | �|�� | jdd � � W d � n1 s.w Y W d � d S W d � d S 1 sFw Y d S r� ) r r2 r rM rN r� rU rS r1 r� r r r �testSeekForwardBytesIOU s
��"�z"BZ2FileTest.testSeekForwardBytesIOc C s� t | j��=}t|�� }|�d� |�dd� | �|�� | jdd � � W d � n1 s-w Y W d � d S W d � d S 1 sEw Y d S r� )r r2 r rS r� rU r1 r� r r r �testSeekBackwardsBytesIO\ s
��"�z$BZ2FileTest.testSeekBackwardsBytesIOc C s� | j d d� }tt|���}| �t|j� W d � n1 s w Y tt|��� }| �|�t| j��| j� | �t|jd� W d � n1 sJw Y t d�D ]$}tt|d |� ���}| �t|jd� W d � n1 srw Y qSd S )N���r � )
r2 r r rM �EOFErrorrS rU rc r1 r� )r& Z truncatedrF r� r r r �test_read_truncatedc s � � ���z BZ2FileTest.test_read_truncatedc C sr t � dg d��}t|�|j }tt� d��}| �|�|�|� | �|�� |� W d � d S 1 s2w Y d S )N�Q)r r� rL � r^ r� )�arrayrc �itemsizer r rU rC r� )r& �qZLENGTHrF r r r �test_issue44439p s "�zBZ2FileTest.test_issue44439N)r r )@r+ r, r- r. rG rQ rX r\ r` rd rf rg rh rm rn rp rw 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� r� r� r� r� r� r� r r r r rA Z s|
rA c @ sJ e Zd Zdd� Zdd� Zdd� Zejee d dd �d
d � ��Z
d d
� Z dS )�BZ2CompressorTestc C sB t � }| �t|j� |�| j�}||�� 7 }| �t|�| j� d S r )r rM rN r? r1 �flushrU r �r& �bz2cr r r r � testCompressz s
z BZ2CompressorTest.testCompressc C s. t � }|�d�}||�� 7 }| �|| j� d S �Nr )r r? r� rU r3 r� r r r �testCompressEmptyString� s
z)BZ2CompressorTest.testCompressEmptyStringc C sj t � }d}d} | j|d |d d � }|sq$||�|�7 }|d7 }q||�� 7 }| �t|�| j� d S )Nr r TrJ r )r r1 r? r� rU r )r& r� ry r rl r r r �testCompressChunks10� s � z&BZ2CompressorTest.testCompressChunks10ro r� ��sizeZmemusec C sv t � }d| }z|�|�}||�� 7 }W d }nd }w t�|�}z| �t|�|� | �t|�d��d� W d }d S d }w )N� xr )r r? r� r
r rU rc �strip)r& r r� r �
compressedr r r �testCompress4G� s
z BZ2CompressorTest.testCompress4Gc C �P t tjd �D ] }| �t�� t�t� |� W d � n1 s w Y qd S �Nr )r� �pickle�HIGHEST_PROTOCOLrM rN �dumpsr �r& �protor r r �
testPickle� � ���z BZ2CompressorTest.testPickleN) r+ r, r- r� r r r �skip_if_pgo_taskr r r r r r r r r� y s
r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zej e
e d d d
�dd� ��Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd� Zejd d � �Zd S ) �BZ2DecompressorTestc C r� )N�* )rM rN r
r) r r r �test_Constructor� r� z$BZ2DecompressorTest.test_Constructorc C s2 t � }| �t|j� |�| j�}| �|| j� d S r )r
rM rN r r2 rU r1 �r& �bz2drk r r r �testDecompress� s z"BZ2DecompressorTest.testDecompressc C sZ t � }d}d} | j|d |d d � }|sq$||�|�7 }|d7 }q| �|| j� d S )Nr r TrJ r )r
r2 r rU r1 )r& r rk ry rl r r r �testDecompressChunks10� s �z*BZ2DecompressorTest.testDecompressChunks10c C s: t � }d}|�| j| �}| �|| j� | �|j|� d S )Ns this is unused data)r
r r2 rU r1 � unused_data)r& r r rk r r r �testDecompressUnusedData� s
z,BZ2DecompressorTest.testDecompressUnusedDatac C s6 t � }|�| j�}| �t|jd� | �t|jd� d S )Ns anythingr )r
r r2 rM r� r r r r � testEOFError� s z BZ2DecompressorTest.testEOFErrorro gffffff
@r c C sh d}t �|�}z%||| d }t�|�}t� }|�|�}| �||k� W d }d }d }d S d }d }d }w )Ni � r )�random� randbytesr
r? r
r rt )r& r � blocksize�blockr r r Z decompressedr r r �testDecompress4G� s
�z$BZ2DecompressorTest.testDecompress4Gc C r r )r� r
r rM rN r r
r
r r r r � r z BZ2DecompressorTest.testPicklec C s t � }d}g }t| j�d }|�|j| jd |� |d�� | �|j� | �t|d �|� |�|jd|d�� | �|j� | �t|d �|� |�|j| j|d � |d�� | �t|d �|� |j sz|�|jd|d�� | �t|d �|� |j rcd�
|�}| �|| j � | �|j d� d S )Nro �@ ��
max_length���r )
r
rc r@ �appendr r� � needs_inputrU r� �eofr0 r6 r )r& �bzdr# �outZlen_r r r � testDecompressorChunksMaxsize� s. � ��
z1BZ2DecompressorTest.testDecompressorChunksMaxsizec C s� t � }g }| �|j| jd d� dd�d� |�|�dd�� |�|�| jdd� d�� |�|�| jdd � �� | �d�|�| j� d S )Nro r r" r r� �i � �r
rU r r2 r% r0 r1 �r& r( r) r r r � test_decompressor_inputbuf_1 s �� z0BZ2DecompressorTest.test_decompressor_inputbuf_1c C s� t � }g }| �|j| jd d� dd�d� |�|�d�� |�|�| jdd� d�� |�|�| jdd� d�� |�|�| jdd � �� | �d�|�| j� d S )N�� r r" r i r� �, r- r. r r r � test_decompressor_inputbuf_2 s �� z0BZ2DecompressorTest.test_decompressor_inputbuf_2c C st t � }g }|�|�| jd d� d�� |�|�| jdd� d�� |�|�| jdd � �� | �d�|�| j� d S )Nr0 r^ r1 r )r
r% r r2 rU r0 r1 r. r r r � test_decompressor_inputbuf_3/ s z0BZ2DecompressorTest.test_decompressor_inputbuf_3c C s6 t � }| �t|j| jd � | �t|j| jd � d S )N� )r
rM � Exceptionr r4 )r& r( r r r � test_failure? s z BZ2DecompressorTest.test_failurec C sH t �td�}t� }|� }td�D ]}|�� q| j|� | ddd� d S )N�gettotalrefcountro r rJ )Zdelta)r Z
get_attribute�sysr
r� �__init__ZassertAlmostEqual)r& r7 r( Z refs_beforer� r r r �test_refleaks_in___init__E s
z-BZ2DecompressorTest.test_refleaks_in___init__N)r+ r, r- r r r r r r r r r r r r* r/ r2 r3 r6 Z
refcount_testr: r r r r r � s" r c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�CompressDecompressTestc C s" t �| j�}| �t|�| j� d S r )r
r? r1 rU r )r& r r r r r� P s z#CompressDecompressTest.testCompressc C s t �d�}| �|| j� d S r� )r
r? rU r3 �r& rk r r r r T s
z.CompressDecompressTest.testCompressEmptyStringc C s t �| j�}| �|| j� d S r �r
r r2 rU r1 r< r r r r X s z%CompressDecompressTest.testDecompressc C s t �d�}| �|d� d S r� )r
r rU r< r r r �testDecompressEmpty\ s
z*CompressDecompressTest.testDecompressEmptyc C s t �| j�}| �|d� d S r� )r
r r3 rU r< r r r �testDecompressToEmptyString` s z2CompressDecompressTest.testDecompressToEmptyStringc C s | � ttj| jd d� � d S )Nr� )rM rO r
r r2 r) r r r �testDecompressIncompleted s z/CompressDecompressTest.testDecompressIncompletec C s | � ttj| j� d S r )rM r[ r
r r4 r) r r r �testDecompressBadDatag s z,CompressDecompressTest.testDecompressBadDatac C s& t �| jd �}| �|| jd � d S r� r= r< r r r �testDecompressMultiStreamj s z0CompressDecompressTest.testDecompressMultiStreamc C s$ t �| j| j �}| �|| j� d S r �r
r r2 r4 rU r1 r< r r r �testDecompressTrailingJunkn s z1CompressDecompressTest.testDecompressTrailingJunkc C s, t �| jd | j �}| �|| jd � d S r� rC r< r r r �%testDecompressMultiStreamTrailingJunkr s z<CompressDecompressTest.testDecompressMultiStreamTrailingJunkN)
r+ r, r- r� r r r> r? r@ rA rB rD rE r r r r r; O s r; c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�OpenTestzTest the open function.c O s t j|i |��S r )r
r< )r& �args�kwargsr r r r< z s z
OpenTest.openc C sd dD ]�}|dkr
t | j� | �| j|��}|�| j� W d � n1 s%w Y t| jd��}t|�� �}| �|| j� W d � n1 sHw Y | �| jd��}| �|�� | j� W d � n1 shw Y | �| jd��}|�| j� W d � n1 s�w Y t| jd��}t|�� �}| �|| jd � W d � n1 s�w Y qd S )N)rB �xbrI r �abr� �r r# r< rC r1 r rS rU �r& �moderF � file_datar r r �test_binary_modes} s,
� � � � ���zOpenTest.test_binary_modesc C sd dD ]�}|dkr
t | j� | �| j|��}|�| j� W d � n1 s%w Y t| jd��}t|�� �}| �|| j� W d � n1 sHw Y | �| jd��}| �|�� | j� W d � n1 shw Y | �| jd��}|�| j� W d � n1 s�w Y t| jd��}t|�� �}| �|| jd � W d � n1 s�w Y qd S )N)r� �xrP r rK r� r� rK rL r r r �test_implicit_binary_modes� s,
� � � � ���z#OpenTest.test_implicit_binary_modesc C s� | j �d�}|�dtj�}dD ]�}|dkrt| j� | j| j|dd��
}|�|� W d � n1 s3w Y t| jd��}t |�
� ��d�}| � ||� W d � n1 sXw Y | j| jddd��}| � |�
� |� W d � n1 syw Y | j| jddd��
}|�|� W d � n1 s�w Y t| jd��}t |�
� ��d�}| � ||d � W d � n1 s�w Y qd S )
Nr� �
)�wt�xtrT ��encodingr �rtZatr� ) r1 �decode�replacer$ �linesepr r# r< rC r rS rU )r& rk �text_native_eolrM rF rN r r r �test_text_modes� s0
� � � � ���zOpenTest.test_text_modesc
C s� dD ]T}t | j� d|v rdnd }| j| j||d��}W d � n1 s%w Y | �t�� | �| j|��}W d � n1 sBw Y W d � n1 sQw Y qd S )N)rP rI rT �t�utf-8rU )r r# r< rM �FileExistsError)r& rM rV rF r r r � test_x_mode� s
� �� ���zOpenTest.test_x_modec C s� | � t| j�d��}| �|�� | j� W d � n1 s w Y | � t| j�d��}| �|�� | j� W d � n1 s?w Y | j�d�}| j t| j�ddd��}| �|�� |� W d � d S 1 siw Y d S )NrK r r� rW r^ rU )r< r r2 rU rS r1 rX )r& rF rk r r r � test_fileobj� s � � "�zOpenTest.test_fileobjc C st | � t| j| jd� | � t| j| jd� | j t| j| jddd� | j t| j| jddd� | j t| j| jddd � d S )
NZwbtZxbtr r^ rU �ignore)�errorsrR )�newline)rM rO r< r# r) r r r �test_bad_params� s
�
� � �
�zOpenTest.test_bad_paramsc C s� | j �d�}|�dtj�}| j| jddd��
}|�|� W d � n1 s&w Y t| jd��}t|� � ��d�}| �
||� W d � n1 sKw Y | j| jddd��}| �
|� � |� W d � d S 1 smw Y d S )Nr� rR rS z utf-16-lerU r rW ) r1 rX rY r$ rZ r< r# rC r rS rU )r& rk r[ rF rN r r r �
test_encoding� s � �"�zOpenTest.test_encodingc C s� | � | jd��
}|�d� W d � n1 sw Y | j | jdddd��}| �|�� d� W d � d S 1 s:w Y d S )NrB s foo�barrW r� rb )rV rc Zfoobar)r< r# rC rU rS r� r r r �test_encoding_error_handler� s �"�z$OpenTest.test_encoding_error_handlerc C s� | j �d�}| j| jdddd��
}|�|� W d � n1 s w Y | j| jdddd��}| �|�� |g� W d � d S 1 sDw Y d S )Nr� rS r^ rR )rV rd rW �
)r1 rX r< r# rC rU r� )r& rk rF r r r � test_newline� s �"�zOpenTest.test_newlineN)r+ r, r- r. r< rO rQ r\ r` ra re rf rg ri r r r r rF w s
rF c C s t �� d S r )r �
reap_childrenr r r r �tearDownModule� s rk �__main__)*�testr Z test.supportr r r� Zunittest�ior r r$ r
r7 r! r� r r r r� r r Ztest.support.os_helperr ra r8 �
import_moduler
r r r
r r ZTestCaser rA r� r r; rF rk r+ �mainr r r r �<module> sL
4 #3 $(y �