Sindbad~EG File Manager
a
��gb# � @ 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�Zedkr�e�� dS )z%
Tests for uu module.
Nick Mathewson
� N)�supports7 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 ZdZd
� 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 sV t t| �jt�� |||d� || _|| _|rRt|t�s>t|�}| � |� | �
d� 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.9/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 sF |r(t d| |f d�t�dd� d }nt 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 � n1 s�0 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 ]x}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 n6 tjyf } z | �t|�d� W Y d }~n
d }~0 0 d S )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 n6 tjyb } z | �t|�d� W Y d }~n
d }~0 0 d S )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 ]�}| � d��B t�|�}t�� }tj||dd� | �|�� |� W d � n1 sj0 Y | � d��0 dd l}|�|d�}| �||� W d � q1 s�0 Y qd 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 s80 Y tjr�|�dd�}| � tjd�� t�t�|�� W d � n1 s�0 Y d}| � tjd�� t�t�|�� W d � n1 s�0 Y tj�r(|�dd�}| � tjd�� t�t�|�� W d � n1 �s 0 Y 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� s zUUStdIOTest.setUpc C s | j t_ | jt_d S r )rT rS rU r r r r �tearDown� s 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% rX )
r r/ r rS rT rU r6 r8 r r5 )r rU r r r r> � s zUUStdIOTest.test_decodeN)r r! r" rV rW 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 sP t | jd��}|�t� W d � n1 s,0 Y t | jd��P}t | jd��$}tj||| jdd� W d � n1 sz0 Y W d � n1 s�0 Y t | jd��}|�� }W d � n1 s�0 Y | �|t d| j�� tj| j| j| jdd� t | jd��}|�� }W d � n1 �s.0 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 (P&(zUUFileTest.test_encodec C s� t | jd��"}|�td| j�� W d � n1 s40 Y t | jd��}t�|� W d � n1 sj0 Y t | jd��}|�� }W d � n1 s�0 Y | �|t � d S �Nr` r3 ra �
rb r\ r r/ r] r6 r rc r8 r5 �r �frd r r r r> � s 0(&zUUFileTest.test_decodec C s� t | jd��"}|�td| j�� W d � n1 s40 Y t�| j� t | jd��}|�� }W d � n1 st0 Y | �|t � d S re rf rg r r r �test_decode_filename� s 0 &z UUFileTest.test_decode_filenamec C s� t | jd��"}|�td| j�� W d � n1 s40 Y t | jd��}t�|� W d � n1 sj0 Y t | jd��"}| �tjtj|� W d � n1 s�0 Y d S re ) rb r\ r r/ r] r6 r r9 rA )r rh r r r �test_decodetwice� s 0(zUUFileTest.test_decodetwicec C s� d}t | jd��"}|�t|| j�� W d � n1 s80 Y | �tj| j|tj B � t | jd��}t
� |� W d � n1 s�0 Y | � t�
t�| j�j�|� d S )Ni$ r` ra )rb r\ r r/ r] r^ rO �chmod�stat�S_IWRITEr6 r r8 �S_IMODE�st_mode)r Z
expected_moderh r r r �test_decode_mode s 0(�zUUFileTest.test_decode_modeN) r r! r" rV r= r> ri rj rp r r r r rY � s
rY �__main__)F)r# Zunittest�testr rO rl rS r6 r
r5 r) �
TextIOWrapperr r/ ZTestCaser0 rR rY r �mainr r r r �<module> s
vL
Sindbad File Manager Version 1.0, Coded By Sindbad EG ~ The Terrorists