File: //usr/local/lib/python3.10/test/__pycache__/test_uu.cpython-310.pyc
o
�i�# � @ s� d Z ddlZddlmZ ddlZddlZddlZddlZddlZdZ dZ
G dd� dej �Z dd d
�Z
G d d � d ej�ZG d
d� dej�ZG dd� dej�ZedkrWe�� dS dS )z%
Tests for uu module.
Nick Mathewson
� N)� os_helpers7 The symbols on top of your keyboard are !@#$%^&*()_+|~
sO M5&AE('-Y;6)O;',@;VX@=&]P(&]F('EO=7(@:V5Y8F]A<F0@87)E("% (R0E
*7B8J*"E?*WQ^"@ c s. e Zd ZdZ d
� fdd� Zdd � Z� ZS ) �FakeIOz�Text I/O implementation using an in-memory buffer.
Can be a used as a drop-in replacement for sys.stdin and sys.stdout.
� �utf-8�strict�
c sZ t t| �jt�� |||d� || _|| _|r+t|t�s t|�}| � |� | �
d� d S d S )N)�encoding�errors�newliner ) �superr �__init__�io�BytesIO� _encoding�_errors�
isinstance�str�write�seek)�selfZ
initial_valuer r r
�� __class__� �)/usr/local/lib/python3.10/test/test_uu.pyr s �
�zFakeIO.__init__c C s | � � | j�� �| j| j�S �N)�flush�buffer�getvalue�decoder r �r r r r r , s zFakeIO.getvalue)r r r r )�__name__�
__module__� __qualname__�__doc__r r �
__classcell__r r r r r s �r Fc C sH |rt d| |f d�t�dd� d }|S t d| |f d�t d }|S )Nzbegin %03o %s
�ascii� � `s
`
end
s
end
)�bytes� encodedtext�replace)�mode�filename�backtick�resr r r �encodedtextwrapped1 s
�����r/ c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�UUTestc C s� t �t�}t �� }t�||d� | �|�� tdd�� t �t�}t �� }t�||dd� | �|�� tdd�� t �t�}t �� }tj||ddd� | �|�� tddd�� | �t �� t�||ddd� W d � d S 1 sow Y d S )N�t1� � T�r- )
r
r � plaintext�uu�encode� assertEqualr r/ � assertRaises� TypeError)r �inp�outr r r � test_encode< s
"�zUUTest.test_encodec C s� dD ]<}t �tdd|d��}t �� }t�||� | �|�� t� t �dtdd|d� �}t �� }t�||� | �|�� t� qd S )N)TFr2 r1 r4 sM UUencoded files may contain many lines,
even some that have 'begin' in them.
)r
r r/ r6 r r8 r r5 )r r- r; r< r r r � test_decodeL s �� �zUUTest.test_decodec
C sl t �dt �}t �� }zt�||� | �d� W d S tjy5 } z| �t|�d� W Y d }~d S d }~ww )Ns
begin 644 t1
�No exception raisedzTruncated input file) r
r r) r6 r �fail�Errorr8 r �r r; r< �er r r �test_truncatedinput[ s ��zUUTest.test_truncatedinputc
C sh t �d�}t �� }zt�||� | �d� W d S tjy3 } z| �t|�d� W Y d }~d S d }~ww )N� r? z'No valid begin line found in input file)r
r r6 r r@ rA r8 r rB r r r �test_missingbegind s
��zUUTest.test_missingbeginc C s� d}d}d}||fD ]U}| � d��! t�|�}t�� }tj||dd� | �|�� |� W d � n1 s5w Y | � d�� dd l}|�|d�}| �||� W d � n1 sZw Y q
d S ) Ns begin 644 file
!,___
end
s begin 644 file
!,___
`
end
� 3z uu.decode()T)�quiet�uu_codecr )ZsubTestr
r r6 r r8 r �codecs) r Z encodedtext1Z encodedtext2r5 r) r; r< rJ Zdecodedr r r �test_garbage_paddingm s& ��
� ���zUUTest.test_garbage_paddingc C s< t �t�}t �� }d}d}t�|||� | �||�� � d S )Nztest.txt
overflow.txts test.txt\n\roverflow.txt)r
r r5 r6 r7 ZassertInr )r r; r< r, Z safefilenamer r r �test_newlines_escaped� s
z UUTest.test_newlines_escapedc C s. d}| � tjd�� t�t�|�� W d � n1 s w Y tjrI|�dd�}| � tjd�� t�t�|�� W d � n1 sDw Y d}| � tjd�� t�t�|�� W d � n1 sew Y tjr�|�dd�}| � tjd�� t�t�|�� W d � d S 1 s�w Y d S d S )Ns< begin 644 ../../../../../../../../tmp/test1
$86)C"@``
`
end
Z directory� /� \s% begin 644 /tmp/test2
$86)C"@``
`
end
) ZassertRaisesRegexr6 rA r r
r �os�altsepr* )r Z relative_badZrelative_bad_bsZ absolute_badZabsolute_bad_bsr r r �test_no_directory_traversal� s& � � � "��z"UUTest.test_no_directory_traversalN)
r r! r" r= r> rD rF rK rL rQ r r r r r0 : s r0 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� UUStdIOTestc C s t j| _t j| _d S r )�sys�stdin�stdoutr r r r �setUp� � zUUStdIOTest.setUpc C s | j t_ | jt_d S r )rT rS rU r r r r �tearDown� rW zUUStdIOTest.tearDownc C sJ t t�d��t_t � t_t�dddd� | �tj� � t
dd��d�� d S )Nr% �-r1 r2 ) r r5 r rS rT rU r6 r7 r8 r r/ r r r r r= � s �zUUStdIOTest.test_encodec C sZ t tdd��d��t_t � t_t�dd� tj}| jt_| jt_| �|�� t �d�� d S )Nr2 r1 r% rY )
r r/ r rS rT rU r6 r8 r r5 )r rU r r r r> � s zUUStdIOTest.test_decodeN)r r! r" rV rX r= r> r r r r rR � s
rR c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�
UUFileTestc C s< t jd | _t jd | _| �t j| j� | �t j| j� d S )N�i�o)r Z TESTFN_ASCII�tmpin�tmpout�
addCleanup�unlinkr r r r rV � s zUUFileTest.setUpc C sN t | jd��
}|�t� W d � n1 sw Y t | jd��(}t | jd��}tj||| jdd� W d � n1 s=w Y W d � n1 sLw Y t | jd�� }|�� }W d � n1 sfw Y | �|t d| j�� tj| j| j| jdd� t | jd�� }|�� }W d � n1 s�w Y | �|t d| j�� d S )N�wb�rbr3 )r+ )
�openr] r r5 r^ r6 r7 �readr8 r/ )r ZfinZfout�sr r r r= � s$ � �� �
�
�zUUFileTest.test_encodec C s� t | jd��}|�td| j�� W d � n1 sw Y t | jd��
}t�|� W d � n1 s5w Y t | jd�� }|�� }W d � n1 sOw Y | �|t � d S �Nra r3 rb �
rc r] r r/ r^ r6 r rd r8 r5 �r �fre r r r r> � s � �
�zUUFileTest.test_decodec C s� t | jd��}|�td| j�� W d � n1 sw Y t�| j� t | jd�� }|�� }W d � n1 s:w Y | �|t � d S rf rg rh r r r �test_decode_filename� s �
�z UUFileTest.test_decode_filenamec C s� t | jd��}|�td| j�� W d � n1 sw Y t | jd��
}t�|� W d � n1 s5w Y t | jd��}| �tjtj|� W d � d S 1 sUw Y d S rf ) rc r] r r/ r^ r6 r r9 rA )r ri r r r �test_decodetwice� s � �"�zUUFileTest.test_decodetwicec C s� d}t | jd��}|�t|| j�� W d � n1 s w Y | �tj| j|tj B � t | jd��
}t
� |� W d � n1 sCw Y | � t�
t�| j�j�|� d S )Ni$ ra rb )rc r] r r/ r^ r_ rO �chmod�stat�S_IWRITEr6 r r8 �S_IMODE�st_mode)r Z
expected_moderi r r r �test_decode_mode s � ��zUUFileTest.test_decode_modeN) r r! r" rV r= r> rj rk rq r r r r rZ � s
rZ �__main__)F)r# ZunittestZ test.supportr rO rm rS r6 r
r5 r) �
TextIOWrapperr r/ ZTestCaser0 rR rZ r �mainr r r r �<module> s$
vM �