Your IP : 216.73.216.213
�
f f% � @ 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 r;e j � n 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/python34/lib64/python3.4/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 �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 � �W } t j d � �? } d | _ | g | _ | j | t j � � | j d � Wd QXWd QXn | 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 sQ t d � } d } t j d � �* } t j | d | �| j j | � Wd QXd 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: r2 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 � } | 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& r0 r4 r r r7 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 � �o } t j d � �W } d | _ t j � | j d t j t j B� | j | j d � | j | j � Wd QXWd QXWd QXd 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 � �l } t j d � �T } d | _ d d d d d g } t | � | _ t j � | j d t j | � Wd QXWd QXWd QXWd QXWd QXd S)Nzos.openz io.FileIOzio.TextIOWrapperztermios.tcgetattrztermios.tcsetattr� � )r r r
�listr rA �assert_called_with�ANY)r rE 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 � �s t j d � �] } t j d � �E } d | _ t � | _ t j | _ t j � | j d | j � Wd QXWd QXWd QXWd QXWd QXWd QXd S) Nzos.openz io.FileIOzio.TextIOWrapperztermios.tcgetattrztermios.tcsetattrzgetpass.fallback_getpassrI z
Password: )
r r r
r �termios�error� side_effectr rA r! )r rE rF rG rN 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 � �i t j d � �S t j d � �= d | _ t j d t � } t j d | � | j j � Wd QXWd QXWd QXWd QXWd QXd S) Nzos.openz io.FileIOzio.TextIOWrapperztermios.tcgetattrztermios.tcsetattrrI r( r/ ) r r r
r, r r rA r. rL )r rE 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 QXWd QX| j
j � | j d | j � � | j d | j � � Wd QXWd QXd S)Nzos.openz sys.stdinr( �Warningz Password:)r r r �IOErrorrR �AttributeError�filenor Zcaptured_stderrZ assertWarnsr ZGetPassWarningrA r5 r! ZassertIn�getvalue)r Zos_open�stdinr3 r r r �test_falls_back_to_stdin� s
z(UnixGetpassTest.test_falls_back_to_stdinN)r$ r% r& rH rO rS rT r[ r r r r r@ ^ s
r@ �__main__)r rB Zunittest�ior r r r r> r rP r r r ZTestCaser r' Z
skipUnlessr@ r$ �mainr r r r �<module> s(
, C