a
}|g5 @ s d Z ddlZddlZddlmZ ddlZddlmZ ddlmZ ddl
m Z ddgZ
g d ZG d
d d ejZG d d
d
ejZedkreeejdd eg dS )z*Test :mod:`certbot._internal.display.obj`. N)mock)errors)obj)util)ZFirstZ Description1)ZSecondZ Description2)tag1tag2tag3c sF e Zd ZdZ fddZeddd Zdd Zd d
Z d d Z
d
d Z eddd Z dd Z
dd Zdd Zdd Zdd Zd d Zd d Zd d! Zed"d#d$ Zed"d%d& Zed"d'd( Zed"d)d* Zd+d, Zd-d. Zd/d0 Zed"d1d2 Z d3d4 Z d5d6 Z d7d8 Z d9d: Z d;d< Z!d=d> Z" Z#S )?FileOutputDisplayTestzTest 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| jd| _d S )NF)supersetUpr MagicMock mock_stdout display_objZ FileDisplay displayerself __class__ L/usr/lib/python3.9/site-packages/certbot/_internal/tests/display/obj_test.pyr s
zFileOutputDisplayTest.setUp$certbot._internal.display.obj.loggerc C s> | j dd | jjjd d }d|v s,J |jdd d S )NmessageFr Notifying user: %sr notificationr
write call_argsdebugZassert_called_withr mock_loggerstringr r r test_notification_no_pause s z0FileOutputDisplayTest.test_notification_no_pausec C s` d}t j|dd | jjddd W d n1 s80 Y d| jjjd d v s\J d S )N1certbot._internal.display.util.input_with_timeoutZenter return_valuer Tforce_interactiver )r patchr r r
r r r input_with_timeoutr r r test_notification_pause$ s .z-FileOutputDisplayTest.test_notification_pausec C s2 | | jjd | jjjd d }d|v s.J d S )Nr r )_force_noninteractiver r r
r r r r r r r test_notification_noninteractive+ s z6FileOutputDisplayTest.test_notification_noninteractivec C sl | | jjd | jjjd d }d|v s.J | jjs:J | | jjd | jjjd d }d|v shJ d S )Nr r message2)r+ r r r
r r skipped_interactionr, r r r !test_notification_noninteractive20 s z7FileOutputDisplayTest.test_notification_noninteractive2c C s ddl m} | jjdddd | jjjd d }|d|j ksBJ | jjddd | jjjd d }d|v spJ d|j |v sJ d S Nr )osr F)pauseZdecorater. )r3 z- - - Zcertbot.compatr2 r r r
r r linesepr r2 r r r r test_notification_decoration= s z2FileOutputDisplayTest.test_notification_decorationz d|_ | jjdtdd\}}|t|ftjddhfks:J d S )Nz2 1r? Tr% r r r$ r checklistTAGSsetr9 r: r
mock_inputrG Ztag_listr r r test_checklist_valid s
z*FileOutputDisplayTest.test_checklist_validc C s> d|_ | jjdtdd\}}|t|ftjh dfks:J d S )N r? Tr% > r r r r` rd r r r test_checklist_empty s z*FileOutputDisplayTest.test_checklist_emptyc C s4 g d|_ | jjdtdd}|tjdgfks0J d S )N)10z tag1 please1r? Tr% r )rY r ra rb r9 r: r re r= r r r test_checklist_miss_valid s
z/FileOutputDisplayTest.test_checklist_miss_validc C s2 ddg|_ | jjdtdd}|tjg fks.J d S )Nri rD r? Tr% )rY r ra rb r9 rF rk r r r test_checklist_miss_quit s
z.FileOutputDisplayTest.test_checklist_miss_quitc C s< t }| j| jjdt |d\}}|tjks,J ||ks8J d S )Nr? r@ )rb r+ r ra r9 r: rL r r r test_checklist_noninteractive s
z3FileOutputDisplayTest.test_checklist_noninteractivec C s` dgg dddgg}dhddhddhg}t |D ]*\}}t| j|t}||| ks0J q0d S )Nrj )rj 2rj ro 3r r r ) enumeraterc r _scrub_checklist_inputrb )r indicesexpilist_Zset_tagsr r r test_scrub_checklist_input_valid s z6FileOutputDisplayTest.test_scrub_checklist_input_validc C st g d}ddgddgg dg dg}ddgddgddgddgg}t |D ]&\}}| j||}||| ksHJ qHd S )N) ZT1ZT2ZT3T4ZT5ZT6ZT7ZT8T949)rz r{ rz )r{ rz r{ rx ry )rq r rr )r Z source_tagsrs rt ru rv tagsr r r +test_scrub_checklist_maintain_indices_order s zAFileOutputDisplayTest.test_scrub_checklist_maintain_indices_orderc C s4 g d}d}||_ | jj| }|tj|fks0J d S )N)r?
/var/www/htmlrN Tr~ )r$ r directory_selectr9 r: )r re argsZ
user_inputZreturnedr r r test_directory_select s
z+FileOutputDisplayTest.test_directory_selectc C s: d}| j | jjd|d\}}|tjks*J ||ks6J d S )Nr~ r? r@ )r+ r r r9 r: rL r r r $test_directory_select_noninteractive s
z:FileOutputDisplayTest.test_directory_select_noninteractivec O s | j j}tdP}d|j_td }||i |}W d n1 sJ0 Y W d n1 sh0 Y |r|jjdu sJ n|jjdksJ |S )Nz'certbot._internal.display.obj.sys.stdinFr r8 ) r r/ r r' isattyr$ ZwarningZcalledZ
call_count)r funcr kwargsr/ Z
mock_stdinr rB r r r r+ s Jz+FileOutputDisplayTest._force_noninteractivec C s@ dgdgdgddgddgg}|D ]}| j |tg ks J q d S )N0rz r rj ro o)r rr rb )r rs rv r r r "test_scrub_checklist_input_invalid s z8FileOutputDisplayTest.test_scrub_checklist_input_invalidc C s | j dt | j dt d S )Nr? )r Z _print_menur< rb r r r r test_print_menu s z%FileOutputDisplayTest.test_print_menuc C s d}t j|dd* | jdtjdfks.J W d n1 sB0 Y d}t j||d. | jdtjt|fks~J W d n1 s0 Y d S )Nr" rj r# r8 ro )r r' r _get_valid_int_ansr9 r: int)r r) ansr r r test_get_valid_int_ans_valid
s 8
z2FileOutputDisplayTest.test_get_valid_int_ans_validc C sn ddgg ddgg}d}|D ]L}t j||d* | jdtjdfksJJ W d q 1 s^0 Y q d S )Nr rD )rz ZoneCr" rX r )r r' r r r9 rF )r Zanswersr) r r r r test_get_valid_int_ans_invalid s
z4FileOutputDisplayTest.test_get_valid_int_ans_invalid)$__name__
__module__ __qualname____doc__r r r' r! r* r- r0 r7 r> rC rI rK rM rS rU r^ r_ rf rh rl rm rn rw r} r r r+ r r r r
__classcell__r r r r r sJ
r c @ sZ e Zd ZdZdd Zeddd Zdd Zd d
Z d d Z
d
d Z dd Z dd Z
dS )NoninteractiveDisplayTestz:Test non-interactive display. These tests are pretty easy!c C s t | _t| j| _d S )N)r r r
r ZNoninteractiveDisplayr r r r r r $ s
z NoninteractiveDisplayTest.setUpr c C s> | j dd | jjjd d }d|v s,J |jdd d S )Nr
r r r r r r r r! ( s z4NoninteractiveDisplayTest.test_notification_no_pausec C s ddl m} | jjdddd | jjjd d }|d|j ksBJ | jjddd | jjjd d }d|v spJ d|j |v sJ d S r1 r4 r6 r r r r7 0 s z6NoninteractiveDisplayTest.test_notification_decorationc C sb d}| j jd|d}|tj|fks&J ttj | j d W d n1 sT0 Y d S )Nzan incomputable valuer r@ )r rE r9 r: rP rQ r MissingCommandlineFlagr dr= r r r
test_input; s
z$NoninteractiveDisplayTest.test_inputc C sb | j jdtdd}|tjdfks$J ttj | j dt W d n1 sT0 Y d S )Nr r8 r@ ) r r; r< r9 r: rP rQ r r )r r= r r r r> B s z#NoninteractiveDisplayTest.test_menuc C s\ d}| j jd|d}||ks J ttj | j d W d n1 sN0 Y d S )NFr r@ )r r] rP rQ r r r r r r r^ H s
z$NoninteractiveDisplayTest.test_yesnoc C sj ddg}| j jdt|d}|tj|fks,J ttj | j dt W d n1 s\0 Y d S )Nr8 r r r@ ) r ra rb r9 r: rP rQ r r r r r r test_checklistO s
z(NoninteractiveDisplayTest.test_checklistc C sd d}t j|f}| jd|}||ks(J ttj | jd W d n1 sV0 Y d S )Nr~ r? )r9 r: r r rP rQ r r )r rA Zexpectedactualr r r r V s
z/NoninteractiveDisplayTest.test_directory_selectN)r r r r r r r' r! r7 r r> r^ r r r r r r r " s
r __main__r8 )r sysZunittestr rP Zcertbotr Zcertbot._internal.displayr r Zcertbot.displayr r9 r< rb ZTestCaser r r exitmainargv__file__r r r r s >