shell bypass 403
o
` b% � @ s� d dl Z d dlZd dlZd dlmZmZmZ d dlmZ d dlm Z zd dl
Z
W n e y3 dZ
Y nw zd dl Z W n e yE dZ Y nw e�
d�G dd� dej��ZG dd � d ej�Ze�e
d
�G d d � d ej��Zed
krve�� dS dS )� N)�BytesIO�StringIO�
TextIOWrapper)�mock)�supportz
os.environc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�GetpassGetuserTestc C s d}||j _| �|t�� � d S )N� some_name)�get� return_value� assertEqual�getpass�getuser)�self�environ�
expected_name� r �:/usr/local/python-3.10/lib/python3.10/test/test_getpass.py�%test_username_takes_username_from_env s z8GetpassGetuserTest.test_username_takes_username_from_envc C sF d |j _zt�� W n ty Y nw | �|j jdd� dD �� d S )Nc S s g | ]}t �|��qS r )r �call)�.0�xr r r �
<listcomp>! s zMGetpassGetuserTest.test_username_priorities_of_env_values.<locals>.<listcomp>)ZLOGNAMEZUSERZLNAME�USERNAME)r r
r r
� ImportErrorr Zcall_args_list)r r r r r �&test_username_priorities_of_env_values s � �z9GetpassGetuserTest.test_username_priorities_of_env_valuesc C s� d}d |j _trQt�d��:}t�d�� }d|_|g|_| �|t�� � |�d� W d � n1 s2w Y W d � d S W d � d S 1 sJw Y d S | � t
tj� d S )Nr z os.getuidz pwd.getpwuid�* ) r r
�pwdr �patchr r r
�assert_called_once_with� assertRaisesr )r r r �uidZgetpwr r r � test_username_falls_back_to_pwd# s �� P�z2GetpassGetuserTest.test_username_falls_back_to_pwdN)�__name__�
__module__� __qualname__r r r! r r r r r s
r c @ sP e Zd Zdd� Zdd� Ze�d�dd� �Ze�d�dd � �Zd
d � Z d d
� Z
dS )�GetpassRawinputTestc C s2 t jtd�}td�}tjd||d� |j�� d S )N��spec� input_string� some_prompt��input)r �Mockr r �
_raw_input�flushr )r �streamr+ r r r � test_flushes_stream_after_prompt4 s z4GetpassRawinputTest.test_flushes_stream_after_promptc C sV t d�}d}t�d��}tj||d� |j�|� W d � d S 1 s$w Y d S )Nr( r) z
sys.stderrr* )r r r r r- �writer )r r+ �prompt�stderrr r r �test_uses_stderr_as_default; s "�z/GetpassRawinputTest.test_uses_stderr_as_default� sys.stdinc C s$ d|j _tjt� d� |j �� d S )Nr( �r/ )�readliner
r r- r r )r �
mock_inputr r r � test_uses_stdin_as_default_inputB s z4GetpassRawinputTest.test_uses_stdin_as_default_inputc C s2 t t� dd�}d|j_tjd|d� |j�� d S )N�ascii)�encodingu Hasło: )r2 r/ )r r r7 r
r r- r )r r8 r/ r r r �#test_uses_stdin_as_different_localeH s z7GetpassRawinputTest.test_uses_stdin_as_different_localec C s t d�}| jttj|d� d S )N� r* )r r �EOFErrorr r- �r r+ r r r �test_raises_on_empty_inputP s z.GetpassRawinputTest.test_raises_on_empty_inputc C s t d�}| �dtj|d�� d S )Nztest
�testr* )r r r r- r? r r r �test_trims_trailing_newlineT s z/GetpassRawinputTest.test_trims_trailing_newlineN) r" r# r$ r0 r4 r r r9 r< r@ rB r r r r r% 2 s
r% z!tests require system with termiosc @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �UnixGetpassTestc
C s� t �d��Y}t �d��;}t �d��&}d |_t�� |�dtjtjB � |�|jd� |�|j� W d � n1 s:w Y W d � n1 sIw Y W d � d S W d � d S 1 saw Y d S )N�os.open� io.FileIO�io.TextIOWrapperz/dev/ttyzw+) r r r
r � unix_getpassr �os�O_RDWR�O_NOCTTY)r �open�fileio�textior r r �test_uses_tty_directlya s" ��
� ��P z&UnixGetpassTest.test_uses_tty_directlyc C s t �d��}}t �d��_ t �d��J t �d��5}t �d�� }d|_g d�}t|�|_t�� |�dt j|� W d � n1 s@w Y W d � n1 sOw Y W d � n1 s^w Y W d � n1 smw Y W d � d S W d � d S 1 s�w Y d S )NrD rE rF �termios.tcgetattr�termios.tcsetattr� )� rR rR rR rR )r r r
�listr rG �assert_called_with�ANY)r rK Z tcgetattr� tcsetattrZ
fake_attrsr r r �test_resets_termioso s. ����
�� � �P z#UnixGetpassTest.test_resets_termiosc
C s> t �d���}t �d��r}t �d��]}t �d��H t �d��3}t �d�� }d|_t� |_tj|_t�� |� d|j� W d � n1 sDw Y W d � n1 sSw Y W d � n1 sbw Y W d � n1 sqw Y W d � n1 s�w Y W d � d S W d � d S 1 s�w Y d S ) NrD rE rF rO rP zgetpass.fallback_getpassrQ z
Password: )
r r r
r �termios�error� side_effectr rG r )r rK rL rM rV �fallbackr r r �-test_falls_back_to_fallback_if_termios_raises{ s> ������ �� � � �P z=UnixGetpassTest.test_falls_back_to_fallback_if_termios_raisesc C s t �d��y}t �d��[ t �d��F t �d��1 t �d�� d|_t jtd�}tj|d� |j�� W d � n1 s<w Y W d � n1 sKw Y W d � n1 sZw Y W d � n1 siw Y W d � d S W d � d S 1 s�w Y d S ) NrD rE rF rO rP rQ r& r6 ) r r r
r, r r rG r. rT )r rK Z mock_streamr r r � test_flushes_stream_after_input� s( ���� �� � �P z/UnixGetpassTest.test_flushes_stream_after_inputc C s t �d��r}t jdtd��R}t|_t|j_t�� �"}| � t
j �� t
� � W d � n1 s/w Y W d � n1 s>w Y |j
�� | �d|�� � | �d|�� � W d � n1 sbw Y W d � d S W d � d S 1 szw Y d S )NrD r5 r& �Warningz Password:)r r r �IOErrorrZ �AttributeError�filenor Zcaptured_stderrZ assertWarnsr ZGetPassWarningrG r7 r ZassertIn�getvalue)r Zos_open�stdinr3 r r r �test_falls_back_to_stdin� s �
�� �
P�z(UnixGetpassTest.test_falls_back_to_stdinN)r" r# r$ rN rW r\ r] rd r r r r rC ^ s rC �__main__)r rH Zunittest�ior r r r rA r rX r r r ZTestCaser r% Z
skipUnlessrC r" �mainr r r r �<module> s0 � �
,C �