Your IP : 216.73.216.213
��Yf% � @ s< d d l Z d d l Z d d l Z d d l m Z m Z m Z d d l m Z d d l m Z y d d l
Z
Wn e k
r� d Z
Yn Xy d d l Z Wn e k
r� d Z Yn Xe j
d � Gd d � d e j � � Z Gd d � d e j � Z e j e
d
� Gd d � d e j � � Z e d
k r8e j � d S)� N)�BytesIO�StringIO�
TextIOWrapper)�mock)�supportz
os.environc @ s4 e Z d Z d d � Z d d � Z d d � Z d S)�GetpassGetuserTestc C s, d } | | j _ | j | t j � � d S)N� some_name)�get� return_value� assertEqual�getpass�getuser)�self�environ�
expected_name� r �6/opt/alt/python35/lib64/python3.5/test/test_getpass.py�%test_username_takes_username_from_env s z8GetpassGetuserTest.test_username_takes_username_from_envc C sV d | j _ y t j � Wn t k
r. Yn X| j | j j d d � d D� � d S)Nc S s g | ] } t j | � � q Sr )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 s� d } d | j _ t r� t j d � �X } t j d � �? } d | _ | g | _ | j | t j � � | j d � Wd QRXWd QRXn | j t
t j � 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 @ sv e Z d Z d d � Z d d � Z e j d � d d � � Z e j d � d d � � Z d
d � Z d d
� Z
d S)�GetpassRawinputTestc C sE t j d t � } t d � } t j d | d | �| j 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 sR t d � } d } t j d � �* } t j | d | �| j j | � Wd QRXd 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_defaultz sys.stdinc C s0 d | j _ t j d t � � | 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 sH t t � d d �} d | j _ t j d d d | � | j j � d S)N�encoding�asciiu 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 � } | j t t j 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 � } | j d t j d | � � d S)Nztest
�testr* )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 @ sL e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d S) �UnixGetpassTestc C s� t j d � �� } t j d � �p } t j d � �W } d | _ t j � | j d t j t j B� | j | j d � | j | j � Wd QRXWd QRXWd QRXd 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 j d � �� } t j d � �� t j d � �� t j d � �m } t j d � �T } d | _ d d d d d g } t | � | _ t j � | j d t j | � Wd QRXWd QRXWd QRXWd QRXWd QRXd 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 j d � �� } t j d � �� } t j d � �� } t j d � �u t j d � �^ } t j d � �E } d | _ t � | _ t j | _ t j � | j d | j � Wd QRXWd QRXWd QRXWd QRXWd QRXWd QRXd 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 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 j d � �� } t j d � �� t j d � �k t j d � �T t j d � �= d | _ t j d t � } t j d | � | j j � Wd QRXWd QRXWd QRXWd QRXWd QRXd 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 j d � �� } t j d d t ��� } t | _ t | j _ t j � �+ } | j t
j � � t
j � Wd QRXWd QRX| j
j � | j d | j � � | j d | j � � Wd QRXWd QRXd S)Nzos.openz sys.stdinr' �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 rR rS rZ 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