Your IP : 216.73.216.213
3
��2a% � @ 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 ej
d�G dd� dej��ZG dd � d ej�Zeje
d
�G d d � d ej��Zed
kr�ej� dS )� N)�BytesIO�StringIO�
TextIOWrapper)�mock)�supportz
os.environc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�GetpassGetuserTestc C s d}||j _| j|tj� � d S )N� some_name)�get� return_value� assertEqual�getpass�getuser)�self�environ�
expected_name� r �1/opt/alt/python36/lib64/python3.6/test_getpass.py�%test_username_takes_username_from_env s z8GetpassGetuserTest.test_username_takes_username_from_envc C sH d |j _y tj� W n tk
r( Y nX | j|j jdd� dD �� d S )Nc S s g | ]}t j|��qS r )r Zcall)�.0�xr r r �
<listcomp>! s zMGetpassGetuserTest.test_username_priorities_of_env_values.<locals>.<listcomp>�LOGNAME�USER�LNAME�USERNAME)r r r r )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 _trftjd��D}tjd��.}d|_|g|_| j|tj� � |jd� W d Q R X W d Q R X n| j 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� Zejd�dd� �Zejd�dd � �Zd
d � Z d d
� Z
dS )�GetpassRawinputTestc C s2 t jtd�}td�}tjd||d� |jj� 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}tjd�� }tj||d� |jj|� 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 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� |jj� d S )N�ascii)�encodingu Hasło: )r1 r. )r r r4 r
r r, r )r r5 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�}| jdtj|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/ r3 r r r6 r9 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 jd��n}t jd��X}t jd��B}d |_tj� |jdtjtjB � |j|jd� |j|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 jd���}t jd��~ t jd��h t jd��R}t jd��<}d|_dddddg}t|�|_tj� |jdt 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 rD Z tcgetattr� tcsetattrZ
fake_attrsr r r �test_resets_termioso s
z#UnixGetpassTest.test_resets_termiosc - C s� t jd���}t jd���}t jd��t}t jd��^ t jd��H}t jd��2}d|_t� |_tj|_tj� |j 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_getpassrH z
Password: )
r r r
r �termios�error� side_effectr r@ r )r rD rE rF rM �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 jd���}t jd��p t jd��Z t jd��D t jd��. d|_t jtd�}tj|d� |jj� 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.tcsetattrrH )r' )r. ) r r r
r+ r r r@ r- rK )r rD Z mock_streamr r r � test_flushes_stream_after_input� s z/UnixGetpassTest.test_flushes_stream_after_inputc C s� t jd���}t jdtd��r}t|_t|j_tj� �&}| j t
j �� t
j � W d Q R X W d Q R X |j
j� | jd|j� � | jd|j� � W d Q R X W d Q R X d S )Nzos.openz sys.stdin)r' �Warningz Password:)r r r �IOErrorrQ �AttributeError�filenor Zcaptured_stderrZ assertWarnsr ZGetPassWarningr@ r4 r ZassertIn�getvalue)r Zos_open�stdinr2 r r r �test_falls_back_to_stdin� s
z(UnixGetpassTest.test_falls_back_to_stdinN)r# r$ r% rG rN rS rT r[ r r r r r? ^ s
r? �__main__)r rA Zunittest�ior r r r r= r rO r r r ZTestCaser r& Z
skipUnlessr? r# �mainr r r r �<module> s(
,
C