Your IP : 216.73.216.60
���]% � @ 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 �1/opt/alt/python35/lib64/python3.5/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 �call)�.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 �uidZgetpwr 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* r0 )�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: r3 r0 )r r r6 r
r r. r! )r r7 r0 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' r1 r5 r r r8 r; 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 rB �assert_called_with�ANY)r rF 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_getpassrJ z
Password: )
r r r
r �termios�error� side_effectr rB r! )r rF rG rH rO �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 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.tcsetattrrJ r) r0 ) r r r
r- r r rB r/ rM )r rF 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 �IOErrorrS �AttributeError�filenor Zcaptured_stderrZ assertWarnsr ZGetPassWarningrB r6 r! ZassertIn�getvalue)r Zos_open�stdinr4 r r r �test_falls_back_to_stdin� s
z(UnixGetpassTest.test_falls_back_to_stdinN)r% r& r' rI rP rU rV r] r r r r rA ^ s
rA �__main__)r rC Zunittest�ior r r r r? r rQ r r r ZTestCaser r( Z
skipUnlessrA r% �mainr r r r �<module> s(
, C