← Back
Editing: util_test.cpython-38.pyc
U �]�B � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZ ddlm Z ddl m Z ddgZ dd d gZd d d gZG dd� dej�ZG dd� dej�ZG dd� dej�ZG dd� dej�ZG dd� dej�Zedkr�e�� dS )z!Test :mod:`certbot.display.util`.� N)�errors)� interfaces)�util)ZFirstZ Description1)ZSecondZ Description2�tag1�tag2�tag3)�1r )�2r )�3r c @ sH e Zd ZdZedd� �Zdd� Zddd�Ze� d �d d � �Z d d � Z dS )�InputWithTimeoutTestz2Tests for certbot.display.util.input_with_timeout.c O s ddl m} |||�S )Nr )�input_with_timeout)�certbot.display.utilr )�cls�args�kwargsr � r �A/usr/lib/python3/dist-packages/certbot/tests/display/util_test.py�_call s zInputWithTimeoutTest._callc C sB t �d��.}tjd|d�� | �t| j� W 5 Q R X W 5 Q R X d S )Nzr+� certbot.display.util.sys.stdin)�new)�tempfileZ TemporaryFile�mock�patch� assertRaises�EOFErrorr )�self�fr r r �test_eof s z InputWithTimeoutTest.test_eofNc C sL d}t �|d �}t�d��&}|gg g f|_| �| �|�|� W 5 Q R X d S )Nzfoo bar� z!certbot.compat.misc.select.select)�six�StringIOr r � return_value� assertEqualr )r �prompt�expected�stdinZ mock_selectr r r � test_input s z InputWithTimeoutTest.test_inputz certbot.display.util.sys.stdoutc C s( d}| � |� |j�|� |j�� d S )Nz test prompt: )r& �writeZassert_called_once_with�flush)r � mock_stdoutr# r r r �test_input_with_prompt& s z+InputWithTimeoutTest.test_input_with_promptc C s\ t � t jt j�}|�d� |�d� t�d|�� | jtj | j dd� W 5 Q R X |� � d S )N)� r � r g����MbP?)Ztimeout) �socketZAF_INETZ SOCK_STREAMZbindZlistenr r r r �Errorr �close)r r% r r r � test_timeout- s z!InputWithTimeoutTest.test_timeout)N) �__name__� __module__� __qualname__�__doc__� classmethodr r r& r r r* r0 r r r r r s r c s< e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Ze � d �d d� �Z dd� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd � Zd d � Ze � d �d d!� �Ze � d �d"d#� �Ze � d �d$d%� �Ze � d �d&d'� �Zd(d)� Zd*d+� Ze � d �d,d-� �Zd.d/� Zd0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z!d<d=� Z"� Z#S )>�FileOutputDisplayTestz�Test stdout display. Most of this class has to deal with visual output. In order to test how the functions look to a user, uncomment the test_visual function. c s, t t| ��� t�� | _t�| jd�| _d S )NF) �superr6 �setUpr � MagicMockr) � display_utilZ FileDisplay� displayer�r �� __class__r r r8 > s zFileOutputDisplayTest.setUpc C s2 | j �dd� | jjjd d }| �d|k� d S )N�messageFr �r; � notificationr) r' � call_args� assertTrue�r �stringr r r �test_notification_no_pauseC s z0FileOutputDisplayTest.test_notification_no_pausec C sN d}t j|dd�� | jjddd� W 5 Q R X | �d| jjjd d k� d S )N�'certbot.display.util.input_with_timeoutZenter�r! r? T��force_interactiver )r r r; rA rC r) r' rB �r r r r r �test_notification_pauseI s z-FileOutputDisplayTest.test_notification_pausec C s4 | � | jjd� | jjjd d }| �d|k� d S )Nr? r )�_force_noninteractiver; rA r) r' rB rC rD r r r � test_notification_noninteractiveP s z6FileOutputDisplayTest.test_notification_noninteractivec C sr | � | jjd� | jjjd d }| �d|k� | �| jj� | � | jjd� | jjjd d }| �d|k� d S )Nr? r Zmessage2)rM r; rA r) r' rB rC �skipped_interactionrD r r r �!test_notification_noninteractive2U s z7FileOutputDisplayTest.test_notification_noninteractive2z3certbot.display.util.FileDisplay._get_valid_int_ansc C s4 t jdf|_| jjdtdd�}| �|t jdf� d S )Nr, r? TrI r )r: �OKr! r; �menu�CHOICESr" )r Zmock_ans�retr r r � test_menub s z FileOutputDisplayTest.test_menuc C s0 d}| j | jjdt|d�}| �|tj|f� d S )Nr �msg��default)rM r; rR rS r" r: rQ )r rX �resultr r r �test_menu_noninteractivei s �z.FileOutputDisplayTest.test_menu_noninteractivec C sD d}t j|dd�� | jjddd�\}}W 5 Q R X | �|tj� d S )NrG �crH r? TrI )r r r; �inputrC r: �CANCEL)r r �code�_r r r �test_input_cancelo s z'FileOutputDisplayTest.test_input_cancelc C sP d}t j|dd�� | jjddd�\}}W 5 Q R X | �|tj� | �|d� d S )NrG z domain.comrH r? TrI )r r r; r\ r" r: rQ )r r r^ �input_r r r �test_input_normalv s z'FileOutputDisplayTest.test_input_normalc C s: d}| j | jjd|d�\}}| �|tj� | �||� d S )NZfoor? rW )rM r; r\ r" r: rQ �r rX r^ ra r r r �test_input_noninteractive~ s � z/FileOutputDisplayTest.test_input_noninteractivec C s | j t| j| jjddd� d S )Nr? �--flag�Zcli_flag)r � ExceptionrM r; r\ r< r r r �test_input_assertion_fail� s �z/FileOutputDisplayTest.test_input_assertion_failc C s6 t �d��" | jtj| j| jjddd� W 5 Q R X d S )Nz&certbot.display.util.assert_valid_callrV re rf )r r r r r. rM r; r\ r< r r r �test_input_assertion_fail2� s �z0FileOutputDisplayTest.test_input_assertion_fail2c C s2 d}t j|dd�� | �| jjddd�� W 5 Q R X t j|dd�� | �| jjddd�� W 5 Q R X t j|ddgd �� | �| jjddd�� W 5 Q R X t j|d d�� | �| jjddd�� W 5 Q R X t j|d d gd �� | �| jjddd�� W 5 Q R X t j|d d�� | �| jjdddd�� W 5 Q R X d S )NrG ZYesrH r? TrI �y�maybe�� side_effectZNoZcancel�n�arV ZAgree)Z yes_labelrJ )r r rC r; �yesno� assertFalserK r r r � test_yesno� s@ � � � � � �z FileOutputDisplayTest.test_yesnoc C s | � | j| jjddd�� d S )Nr? TrW )rC rM r; rp r< r r r �test_yesno_noninteractive� s �z/FileOutputDisplayTest.test_yesno_noninteractiverG c C sB d|_ | jjdtdd�\}}| �|t|�ftjtddg�f� d S )Nz2 1rV TrI r r �r! r; � checklist�TAGSr" �setr: rQ �r � mock_inputr^ Ztag_listr r r �test_checklist_valid� s � �z*FileOutputDisplayTest.test_checklist_validc C sD d|_ | jjdtdd�\}}| �|t|�ftjtdddg�f� d S )Nr+ rV TrI r r r rt rx r r r �test_checklist_empty� s �z*FileOutputDisplayTest.test_checklist_emptyc C s6 dddg|_ | jjdtdd�}| �|tjdgf� d S )N�10z tag1 pleaser rV TrI r )rm r; ru rv r" r: rQ �r ry rT r r r �test_checklist_miss_valid� s z/FileOutputDisplayTest.test_checklist_miss_validc C s2 ddg|_ | jjdtdd�}| �|tjg f� d S )Nr| r[ rV TrI )rm r; ru rv r" r: r] r} r r r �test_checklist_miss_quit� s z.FileOutputDisplayTest.test_checklist_miss_quitc C s< t }| j| jjdt |d�\}}| �|tj� | �||� d S )NrV rW )rv rM r; ru r" r: rQ rc r r r � test_checklist_noninteractive� s � z3FileOutputDisplayTest.test_checklist_noninteractivec C sn dgdddgddgg}t dg�t ddg�t ddg�g}t|�D ]*\}}t | j�|t��}| �||| � q>d S )Nr r r r r r )rw � enumerater; �_scrub_checklist_inputrv r" )r �indices�exp�i�list_Zset_tagsr r r � test_scrub_checklist_input_valid� s � � �z6FileOutputDisplayTest.test_scrub_checklist_input_validc C s8 ddddg}d}||_ | jj|� }| �|tj|f� d S )NrV � /var/www/htmlre T)r! r; �directory_selectr" r: rQ )r ry r Z user_inputZreturnedr r r �test_directory_select� s z+FileOutputDisplayTest.test_directory_selectc C s: d}| j | jjd|d�\}}| �|tj� | �||� d S )Nr� rV rW )rM r; r� r" r: rQ rc r r r �$test_directory_select_noninteractive� s � z:FileOutputDisplayTest.test_directory_select_noninteractivec O sn | j j}t�d��.}d|j_t�d��}|||�}W 5 Q R X W 5 Q R X |rZ| �|jj� n| � |jj d� |S )Nr Fzcertbot.display.util.loggerr, ) r; rO r r �isattyr! rq ZwarningZcalledr" Z call_count)r �funcr r rO Z mock_stdinZ mock_loggerrY r r r rM � s z+FileOutputDisplayTest._force_noninteractivec C s@ dgdgdgddgddgg}|D ]}| � | j�|t�g � q d S )N�0�4r r r �o)r" r; r� rv )r r� r� r r r �"test_scrub_checklist_input_invalid s � �z8FileOutputDisplayTest.test_scrub_checklist_input_invalidc C s | j �dt� | j �dt� d S )NrV )r; Z _print_menurS rv r< r r r �test_print_menu s z%FileOutputDisplayTest.test_print_menuc C s* d� d�}t�|�}| �|�d�d� d S )Nz�This is just a weak test{0}This function is only meant to be for easy viewing{0}Test a really really really really really really really really really really really really long line...r � )�formatr: Z _wrap_linesr" �count)r rV �textr r r �test_wrap_lines s � z%FileOutputDisplayTest.test_wrap_linesc C sx d}t j|dd�� | �| j�d�tjdf� W 5 Q R X d}t j||d��$ | �| j�d�tjt|�f� W 5 Q R X d S )NrG r rH r, r r� )r r r" r; �_get_valid_int_ansr: rQ �int)r r �ansr r r � test_get_valid_int_ans_valid s � �z2FileOutputDisplayTest.test_get_valid_int_ans_validc C s\ ddgdddgdgg}d}|D ]8}t j||d�� | �| j�d�tjd f� W 5 Q R X q d S ) Nr� r[ r� Zone�CrG rl r� ���)r r r" r; r� r: r] )r Zanswersr r� r r r � test_get_valid_int_ans_invalid* s � �z4FileOutputDisplayTest.test_get_valid_int_ans_invalidc C sH t j�� D ]8}tjr tj}ntj}|t| j |��}| � d|j k� q d S )NrJ ) r �IDisplay�namesr �PY2�inspect� getargspec�getfullargspec�getattrr; rC r )r �namer� Zarg_specr r r �#test_methods_take_force_interactive8 s z9FileOutputDisplayTest.test_methods_take_force_interactive)$r1 r2 r3 r4 r8 rF rL rN rP r r rU rZ r` rb rd rh ri rr rs rz r{ r~ r r� r� r� r� rM r� r� r� r� r� r� � __classcell__r r r= r r6 6 sH r6 c sX e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z � Z S )�NoninteractiveDisplayTestz:Test non-interactive display. These tests are pretty easy!c s* t t| ��� t�� | _t�| j�| _d S �N) r7 r� r8 r r9 r) r: ZNoninteractiveDisplayr; r< r= r r r8 F s z NoninteractiveDisplayTest.setUpc C s2 | j �dd� | jjjd d }| �d|k� d S )Nr? � r r@ rD r r r rF K s z4NoninteractiveDisplayTest.test_notification_no_pausec C s> d}| j jd|d�}| �|tj|f� | �tj| j jd� d S )Nzan incomputable valuer? rW )r; r\ r" r: rQ r r �MissingCommandlineFlag�r �drT r r r r& Q s z$NoninteractiveDisplayTest.test_inputc C s> | j jdtdd�}| �|tjdf� | �tj| j jdt� d S )Nr? r, rW ) r; rR rS r" r: rQ r r r� )r rT r r r rU W s z#NoninteractiveDisplayTest.test_menuc C s8 d}| j jd|d�}| �||� | �tj| j jd� d S )NFr? rW )r; rp r" r r r� r� r r r rr \ s z$NoninteractiveDisplayTest.test_yesnoc C sF ddg}| j jdt|d�}| �|tj|f� | �tj| j jdt� d S )Nr, r� r? rW ) r; ru rv r" r: rQ r r r� r� r r r �test_checklistb s z(NoninteractiveDisplayTest.test_checklistc C s@ d}t j|f}| j�d|�}| �||� | �tj| jjd� d S )Nr� rV )r: rQ r; r� r" r r r� )r rX r$ �actualr r r r� h s �z/NoninteractiveDisplayTest.test_directory_selectc C s\ t j�� D ]L}t| j|�}tjr<t�|�j }| � |d k� q t� |�j }| � |d k� q d S r� ) r r� r� r� r; r r� r� r� �keywordsrq r� Zvarkw)r r� �methodrY r r r �test_methods_take_kwargsq s z2NoninteractiveDisplayTest.test_methods_take_kwargs) r1 r2 r3 r4 r8 rF r& rU rr r� r� r� r� r r r= r r� D s r� c @ sD e Zd ZdZdd� Zedd� �Zdd� Zdd � Zd d � Z d d � Z dS )�SeparateListInputTestzTest Module functions.c C s ddddg| _ d S )Nro �br[ Ztest)r� r< r r r r8 � s zSeparateListInputTest.setUpc C s ddl m} ||�S )Nr )�separate_list_input)r r� )r ra r� r r r r � s zSeparateListInputTest._callc C s | � | �d�| j� d S )Nz a,b,c,test�r" r r� r< r r r � test_commas� s z!SeparateListInputTest.test_commasc C s | � | �d�| j� d S )Nz a b c testr� r< r r r � test_spaces� s z!SeparateListInputTest.test_spacesc C s | � | �d�| j� d S )Nz a, b, c, testr� r< r r r � test_both� s z SeparateListInputTest.test_bothc C s8 | � d�| � d�| � d�g}|D ]}| �|| j� q d S )Nz a , b c testz,a, ,, , b c test z,,,,, , a b,,, , c,test)r r" r� )r r� Zactr r r � test_mess� s �z SeparateListInputTest.test_messN) r1 r2 r3 r4 r8 r5 r r� r� r� r� r r r r r� � s r� c @ s( e Zd Zedd� �Zdd� Zdd� ZdS )�PlaceParensTestc C s ddl m} ||�S )Nr )�_parens_around_char)r r� )r Zlabelr� r r r r � s zPlaceParensTest._callc C s | � d| �d�� d S )Nz(a)ro �r" r r< r r r �test_single_letter� s z"PlaceParensTest.test_single_letterc C s( | � d| �d�� | � d| �d�� d S )Nz(L)abelZLabelz (y)es pleasez yes pleaser� r< r r r � test_multiple� s z PlaceParensTest.test_multipleN)r1 r2 r3 r5 r r� r� r r r r r� � s r� �__main__)r4 r� r- r Zunittestr r Zcertbotr r Zcertbot.displayr r: rS rv Z TAGS_CHOICESZTestCaser r6 r� r� r� r1 �mainr r r r �<module> s* # ?
Save File
Cancel