Your IP : 216.73.216.213
a
XC?h% � @ s� d dl Z d dlZd dlZd dlmZmZmZ d dlmZ d dlm Z z d dl
Z
W n e yf dZ
Y n0 z d dl Z W n e y� dZ Y n0 e�
d�G dd� dej��ZG dd � d ej�Ze�e
d
�G d d � d ej��Zed
kr�e�� 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 �6/opt/alt/python39/lib64/python3.9/test/test_getpass.py�%test_username_takes_username_from_env s z8GetpassGetuserTest.test_username_takes_username_from_envc C sF d |j _z t�� W n ty& Y n0 | �|j jdd� dD �� d S )Nc S s g | ]}t �|��qS r )r Zcall)�.0�xr r r �
<listcomp>! � zMGetpassGetuserTest.test_username_priorities_of_env_values.<locals>.<listcomp>)ZLOGNAMEZUSERZLNAMEZUSERNAME)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 _tr�t�d��b}t�d��8}d|_|g|_| �|t�� � |�d� W d � n1 sd0 Y W d � q�1 s�0 Y n| � t
tj� d S )Nr z os.getuidz pwd.getpwuid�* ) r r
�pwdr �patchr r r
�assert_called_once_with� assertRaisesr )r r r ZuidZgetpwr r r � test_username_falls_back_to_pwd# s ��Hz2GetpassGetuserTest.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 sT t d�}d}t�d��*}tj||d� |j�|� W d � n1 sF0 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: )r0 r- )r r r5 r
r r+ r )r r6 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" r. r2 r r r7 r: r> r@ 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���}t �d��v}t �d��L}d |_t�� |�dtjtjB � |�|jd� |�|j� W d � n1 st0 Y W d � n1 s�0 Y W d � n1 s�0 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 ��
�z&UnixGetpassTest.test_uses_tty_directlyc C s t �d���}t �d��� t �d��� t �d��j}t �d��@}d|_g d�}t|�|_t�� |�dt j|� W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y d S )NrB rC rD �termios.tcgetattr�termios.tcsetattr� )� rP rP rP rP )r r r
�listr rE �assert_called_with�ANY)r rI Z tcgetattr� tcsetattrZ
fake_attrsr r r �test_resets_termioso s ����
z#UnixGetpassTest.test_resets_termiosc
C s2 t �d���}t �d���}t �d���}t �d��� t �d��f}t �d��<}d|_t� |_tj|_t�� |� d|j� W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 �s0 Y W d � n1 �s$0 Y d S ) NrB rC rD rM rN zgetpass.fallback_getpassrO z
Password: )
r r r
r �termios�error� side_effectr rE r )r rI rJ rK rT Zfallbackr r r �-test_falls_back_to_fallback_if_termios_raises{ s, ������z=UnixGetpassTest.test_falls_back_to_fallback_if_termios_raisesc C s� t �d���}t �d��� t �d��� t �d��b t �d��8 d|_t jtd�}tj|d� |j�� W d � n1 sx0 Y W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y W d � n1 s�0 Y d S ) NrB rC rD rM rN rO r$ r4 ) r r r
r* r r rE r, rR )r rI Z mock_streamr r r � test_flushes_stream_after_input� s ���� z/UnixGetpassTest.test_flushes_stream_after_inputc C s� t �d���}t jdtd���}t|_t|j_t�� �D}| � t
j �� t
� � W d � n1 s^0 Y W d � n1 s|0 Y |j
�� | �d|�� � | �d|�� � W d � n1 s�0 Y W d � n1 s�0 Y d S )NrB r3 r$ �Warningz Password:)r r r �IOErrorrX �AttributeError�filenor Zcaptured_stderrZ assertWarnsr ZGetPassWarningrE r5 r ZassertIn�getvalue)r Zos_open�stdinr1 r r r �test_falls_back_to_stdin� s �
D
z(UnixGetpassTest.test_falls_back_to_stdinN)r r! r" rL rU rY rZ ra r r r r rA ^ s
rA �__main__)r rF Zunittest�ior r r r r? r rV r r r ZTestCaser r# Z
skipUnlessrA r �mainr r r r �<module> s(
,
C