shell bypass 403
o
` b� � @ s� d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dlm Z m
Z
dZ G dd� dej �Z
G dd� dej�Zed krDe�� dS dS )
� N)�findfile)�TESTFN�unlink)
)�
python.png�png)z
python.gifZgif)z
python.bmpZbmp)z
python.ppmZppm)z
python.pgmZpgm)z
python.pbmZpbm)z
python.jpgZjpeg)z
python.rasZrast)z
python.sgiZrgb)z python.tiffZtiff)z
python.xbmZxbm)z python.webpZwebp)z
python.exrZexrc @ s e Zd Zdd� Zdd� ZdS )� UnseekableIOc C � t j��N��io�UnsupportedOperation��self� r �9/usr/local/python-3.10/lib/python3.10/test/test_imghdr.py�tell � zUnseekableIO.tellc O r r r
)r �args�kwargsr r r �seek r zUnseekableIO.seekN)�__name__�
__module__� __qualname__r r r r r r r s r c @ sx e Zd Zedd� �Zdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� ZdS ) �
TestImghdrc C sJ t ddd�| _t| jd��}|�� | _W d � d S 1 s w Y d S )Nr �
imghdrdata�Zsubdir�rb)r �testfile�open�read�testdata)�cls�streamr r r �
setUpClass# s "�zTestImghdr.setUpClassc C s t t� d S r )r r r
r r r �tearDown) s zTestImghdr.tearDownc C s� t D ]`\}}t|dd�}| �t�|�|� t|d��}| �t�|�|� W d � n1 s.w Y t|d�� }|�� }W d � n1 sGw Y | �t�d |�|� | �t�d t|��|� qd S )Nr r r )�
TEST_FILESr � assertEqual�imghdr�whatr r � bytearray)r �filename�expectedr" �datar r r � test_data, s �
��zTestImghdr.test_datac C sb t D ],\}}| j|d�� t|dd�}| �t�t�|��|� W d � n1 s)w Y qd S )N)r* r r )r% ZsubTestr r& r' r( �pathlib�Path)r r* r+ r r r �test_pathlike_filename7 s ���z!TestImghdr.test_pathlike_filenamec C s: dd� }t j�|� | �t jj� | �t �d d�d� d S )Nc S s | � d�rdS d S )N� eggs�ham)�
startswith)�h�filer r r �
test_jumbo> s
�z1TestImghdr.test_register_test.<locals>.test_jumbor1 r2 )r' Ztests�appendZ
addCleanup�popr& r( )r r6 r r r �test_register_test= s z TestImghdr.test_register_testc C s� t td��}|�d� |�� }|�| j� W d � n1 s w Y t td�� }|�|� | �t�|�d� | �|�� |� W d � d S 1 sKw Y d S )N�wbs
ababagalamagar r ) r r �writer r r r& r' r( )r r" �posr r r �
test_file_posE s
�
"�zTestImghdr.test_file_posc C sT | � t�� t�� W d � n1 sw Y | � t��
t�d � W d � n1 s.w Y | � t�� t�| jd� W d � n1 sJw Y | � t�� t�t�| j�� W d � n1 shw Y t| jd��-}| � t�� t�|� � � W d � n1 s�w Y W d � d S W d � d S 1 s�w Y d S )N� r )
� assertRaises� TypeErrorr' r( �AttributeErrorr �os�fsencoder �fileno)r �fr r r �
test_bad_argsO s&
� � � � ��"�zTestImghdr.test_bad_argsc C s dD ] }| � t�d |�� qd S )N)s �PN
s �s Y�s cutecats 000000JFIs GIF80)Z assertIsNoner' r( )r �headerr r r �test_invalid_headers\ s �z TestImghdr.test_invalid_headersc
C s� t �� �p t �dt� tD ]^\}}t|dd�}t|d��}|�� �d�}W d � n1 s.w Y | � t
�� t � t
�|�� W d � n1 sKw Y | � t
�� t � d |� W d � n1 sfw Y q
W d � d S 1 sww Y d S )N�ignorer r r �latin1)�warnings�catch_warnings� simplefilter� BytesWarningr% r r r �decoder? r@ r' r( r �StringIO)r r* �_r" r, r r r �test_string_datae s
� � ���"�zTestImghdr.test_string_datac C s: | � t�� t�d� W d � d S 1 sw Y d S )N�missing)r? �FileNotFoundErrorr' r( r
r r r �test_missing_fileq s "�z TestImghdr.test_missing_filec C s� t | jd�}|�� | �t��
}t�|� W d � n1 s w Y t�| j �}|�� | �t��}t�|� W d � d S 1 sDw Y d S )Nr )
r r �closer? �
ValueErrorr' r( r �BytesIOr �r r" �cmr r r �test_closed_fileu s � "�zTestImghdr.test_closed_filec C s� t td��}|�| j� W d � n1 sw Y ttd��,}| �tj��
t� |� W d � n1 s7w Y W d � d S W d � d S 1 sOw Y d S )Nr: r )
r r r; r r r? r r r' r( )r r" r r r �test_unseekable s � ��"�zTestImghdr.test_unseekablec C s� t td��6}|�| j� |�d� | �t��
}t�|� W d � n1 s&w Y W d � d S W d � d S 1 s>w Y d S )Nr: r ) r r r; r r r? �OSErrorr' r( rY r r r �test_output_stream� s
��"�z TestImghdr.test_output_streamN)r r r � classmethodr# r$ r- r0 r9 r= rF rH rR rU r[ r\ r^ r r r r r " s
r �__main__)r' r rB r. ZunittestrK Z test.supportr Ztest.support.os_helperr r r% �FileIOr ZTestCaser r �mainr r r r �<module> s k �