Your IP : 216.73.216.196
B
� fV# � @ 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__� �1/opt/alt/python37/lib64/python3.7/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�backtickZresr 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 Q R X d S )N�t1i� i� T)r* )
r
r � plaintext�uu�encode� assertEqualr r+ � assertRaises� TypeError)r �inp�outr r r � test_encode< s
zUUTest.test_encodec C s� x�dD ]x}t �tdd|d��}t �� }t�||� | �|�� t� t �dtdd|d� �}t �� }t�||� | �|�� t� qW d S )N)TFi� r- )r* sM UUencoded files may contain many lines,
even some that have 'begin' in them.
)r
r r+ r/ r r1 r r. )r r* r4 r5 r r r � test_decodeL s
zUUTest.test_decodec
C sj t �dt �}t �� }yt�||� | �d� W n4 tjk
rd } z| �t|�d� W d d }~X Y nX d S )Ns
begin 644 t1
zNo exception raisedzTruncated input file) r
r r& r/ r �fail�Errorr1 r )r r4 r5 �er r r �test_truncatedinput[ s zUUTest.test_truncatedinputc
C sf t �d�}t �� }yt�||� | �d� W n4 tjk
r` } z| �t|�d� W d d }~X Y nX d S )N� zNo exception raisedz'No valid begin line found in input file)r
r r/ r r8 r9 r1 r )r r4 r5 r: r r r �test_missingbegind s
zUUTest.test_missingbeginc
C s� d}d}d}x�||fD ]�}| � d��8 t�|�}t�� }tj||dd� | �|�� |� W d Q R X | � d��& dd l}|�|d�}| �||� W d Q R X qW d S ) Ns begin 644 file
!,___
end
s begin 644 file
!,___
`
end
� 3z uu.decode()T)�quiet�uu_codecr )ZsubTestr
r r/ r r1 r �codecs) r Z encodedtext1Z encodedtext2r. r& r4 r5 rA 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 r. r/ r0 ZassertInr )r r4 r5 r) Z safefilenamer r r �test_newlines_escaped� s
z UUTest.test_newlines_escapedc C s� d}| � tjd�� t�t�|�� W d Q R X tjrj|�dd�}| � tjd�� t�t�|�� W d Q R X d}| � tjd�� t�t�|�� W d Q R X tjr�|�dd�}| � tjd�� t�t�|�� W d Q R X d S )Ns< begin 644 ../../../../../../../../tmp/test1
$86)C"@``
`
end
Z directory� /� \s% begin 644 /tmp/test2
$86)C"@``
`
end
) ZassertRaisesRegexr/ r9 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 r6 r7 r; r= rB rC rH r r r r r, : s r, c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� UUStdIOTestc C s t j| _t j| _d S )N)�sys�stdin�stdout)r r r r �setUp� s zUUStdIOTest.setUpc C s | j t_ | jt_d S )N)rK rJ rL )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" �-r- i� ) r r. r rJ rK rL r/ r0 r1 r r+ )r r r r r6 � s
zUUStdIOTest.test_encodec C sZ t tdd��d��t_t � t_t�dd� tj}| jt_| jt_| �|�� t �d�� d S )Ni� r- r" rO )
r r+ r rJ rK rL r/ r1 r r. )r rL r r r r7 � s zUUStdIOTest.test_decodeN)r r r rM rN r6 r7 r r r r rI � s rI 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 ZTESTFN�tmpin�tmpout�
addCleanup�unlink)r r r r rM � s zUUFileTest.setUpc
C s� t | jd��}|�t� W d Q R X t | jd��2}t | jd��}tj||| jdd� W d Q R X W d Q R X t | jd��}|�� }W d Q R X | �|t d| j�� tj| j| j| jdd� t | jd��}|�� }W d Q R X | �|t d| j�� d S )N�wb�rbi� )r( )
�openrS r r. rT r/ r0 �readr1 r+ )r ZfinZfout�sr r r r6 � s (zUUFileTest.test_encodec C s| t | jd��}|�td| j�� W d Q R X t | jd��}t�|� W d Q R X t | jd��}|�� }W d Q R X | �|t � d S )NrW i� rX )
rY rS r r+ rT r/ r rZ r1 r. )r �fr[ r r r r7 � s zUUFileTest.test_decodec C sf t | jd��}|�td| j�� W d Q R X t�| j� t | jd��}|�� }W d Q R X | �|t � d S )NrW i� rX )
rY rS r r+ rT r/ r rZ r1 r. )r r\ r[ r r r �test_decode_filename� s z UUFileTest.test_decode_filenamec C sz t | jd��}|�td| j�� W d Q R X t | jd��}t�|� W d Q R X t | jd��}| �tjtj|� W d Q R X d S )NrW i� rX ) rY rS r r+ rT r/ r r2 r9 )r r\ r r r �test_decodetwice� s zUUFileTest.test_decodetwicec C s� d}t | jd��}|�t|| j�� W d Q R X | �tj| j|tj B � t | jd��}t
� |� W d Q R X | � t�
t�| j�j�|� d S )Ni$ rW rX )rY rS r r+ rT rU rF �chmod�stat�S_IWRITEr/ r r1 �S_IMODE�st_mode)r Z
expected_moder\ r r r �test_decode_mode s zUUFileTest.test_decode_modeN) r r r rM r6 r7 r] r^ rd r r r r rP � s
rP �__main__)F)r ZunittestZtestr rF r` rJ r/ r
r. r& �
TextIOWrapperr r+ ZTestCaser, rI rP r �mainr r r r �<module> s
vL