File: //usr/local/lib/python3.7/test/__pycache__/test_getpass.cpython-37.opt-2.pyc
B
��g% � @ s� d dl Z d dlZd dlZd dlmZmZmZ d dlmZ d dlm Z y d dl
Z
W n e k
rh dZ
Y nX y d dl Z W n e k
r� dZ Y nX 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 �-/usr/local/lib/python3.7/test/test_getpass.py�%test_username_takes_username_from_env s z8GetpassGetuserTest.test_username_takes_username_from_envc C sH d |j _y t�� W n tk
r( Y nX | �|j jdd� dD �� d S )Nc S s g | ]}t �|��qS r )r Zcall)�.0�xr r r �
<listcomp>! s 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 sx d}d |j _trft�d��D}t�d��.}d|_|g|_| �|t�� � |�d� W d Q R X W d Q R X 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
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 s@ t d�}d}t�d�� }tj||d� |j�|� W d Q R X d S )Nr$ r% z
sys.stderr)r& )r r r r r( �writer )r r& �prompt�stderrr r r �test_uses_stderr_as_default; s
z/GetpassRawinputTest.test_uses_stderr_as_defaultz 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: )r- r* )r r r0 r
r r( r )r r1 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
�test)r& )r r r r( )r r& r r r �test_trims_trailing_newlineT s z/GetpassRawinputTest.test_trims_trailing_newlineN) r r r! r+ r/ r r r2 r5 r8 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��n}t �d��X}t �d��B}d |_t�� |�dtjtjB � |�|jd� |�|j� W d Q R X W d Q R X W d Q R X d S )Nzos.openz io.FileIOzio.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��h t �d��R}t �d��<}d|_dddddg}t|�|_t�� |�dt j|� W d Q R X W d Q R X W d Q R X W d Q R X W d Q R X d S )Nzos.openz io.FileIOzio.TextIOWrapperztermios.tcgetattrztermios.tcsetattr� � )r r r
�listr r< �assert_called_with�ANY)r r@ Z tcgetattr� tcsetattrZ
fake_attrsr r r �test_resets_termioso s
z#UnixGetpassTest.test_resets_termiosc C s� t �d���}t �d���}t �d��t}t �d��^ t �d��H}t �d��2}d|_t� |_tj|_t�� |� d|j� W d Q R X W d Q R X W d Q R X W d Q R X W d Q R X W d Q R X d S ) Nzos.openz io.FileIOzio.TextIOWrapperztermios.tcgetattrztermios.tcsetattrzgetpass.fallback_getpassrD z
Password: )
r r r
r �termios�error� side_effectr r< r )r r@ rA rB rI �fallbackr 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��p t �d��Z t �d��D t �d��. d|_t jtd�}tj|d� |j�� W d Q R X W d Q R X W d Q R X W d Q R X W d Q R X d S ) Nzos.openz io.FileIOzio.TextIOWrapperztermios.tcgetattrztermios.tcsetattrrD )r# )r* ) r r r
r' r r r< r) rG )r r@ Z mock_streamr r r � test_flushes_stream_after_input� s z/UnixGetpassTest.test_flushes_stream_after_inputc C s� t �d���}t jdtd��r}t|_t|j_t�� �&}| � t
j �� t
� � W d Q R X W d Q R X |j
�� | �d|�� � | �d|�� � W d Q R X W d Q R X d S )Nzos.openz sys.stdin)r# �Warningz Password:)r r r �IOErrorrM �AttributeError�filenor Zcaptured_stderrZ assertWarnsr ZGetPassWarningr< r0 r ZassertIn�getvalue)r Zos_open�stdinr. r r r �test_falls_back_to_stdin� s
z(UnixGetpassTest.test_falls_back_to_stdinN)r r r! rC rJ rO rP rW r r r r r; ^ s
r; �__main__)r r= Zunittest�ior r r r r9 r rK r r r ZTestCaser r" Z
skipUnlessr; r �mainr r r r �<module> s(
,
C