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 s2 | j dd | jjjd d }|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 sF d}t j|dd | jjddd W d n1 s80 Y d S )N1certbot._internal.display.util.input_with_timeoutZenter return_valuer TZforce_interactive)r patchr r r input_with_timeoutr r r test_notification_pause$ s .z-FileOutputDisplayTest.test_notification_pausec C s& | | jjd | jjjd d }d S )Nr r _force_noninteractiver r r
r r r r r r r test_notification_noninteractive+ s z6FileOutputDisplayTest.test_notification_noninteractivec C sH | | jjd | jjjd d }| | jjd | jjjd d }d S )Nr r message2r* r, r r r !test_notification_noninteractive20 s
z7FileOutputDisplayTest.test_notification_noninteractive2c C sV ddl m} | jjdddd | jjjd d }| jjddd | jjjd d }d S Nr )osr F)pauseZdecorater. )r2 Zcertbot.compatr1 r r r
r r r r1 r r r r test_notification_decoration= s z2FileOutputDisplayTest.test_notification_decorationz )r+ r rD r r? rE rH r r r test_input_noninteractived s
z/FileOutputDisplayTest.test_input_noninteractivec C sB t t$ | j| jjddd W d n1 s40 Y d S )Nr --flagZcli_flag)pytestraises Exceptionr+ r rD r r r r test_input_assertion_faill s z/FileOutputDisplayTest.test_input_assertion_failc C sn t dP ttj$ | j| jjddd W d n1 sB0 Y W d n1 s`0 Y d S )Nz&certbot.display.util.assert_valid_callr= rL rM ) r r& rN rO r Errorr+ r rD r r r r test_input_assertion_fail2r s z0FileOutputDisplayTest.test_input_assertion_fail2c C s& d}t j|dd W d n1 s(0 Y t j|dd W d n1 sV0 Y t j|ddgd W d n1 s0 Y t j|dd W d n1 s0 Y t j|dd gd W d n1 s0 Y t j|d
d W d n1 s0 Y d S ) Nr" ZYesr# ymaybe side_effectZNocancelnar r& r' r r r
test_yesnow s z FileOutputDisplayTest.test_yesnoc C s d S Nr r r r r test_yesno_noninteractive s z/FileOutputDisplayTest.test_yesno_noninteractiver" c C s d|_ | jjdtdd\}}d S )Nz2 1r= Tr% r$ r checklistTAGSr
mock_inputrE Ztag_listr r r test_checklist_valid s
z*FileOutputDisplayTest.test_checklist_validc C s d|_ | jjdtdd\}}d S )N r= Tr% r_ rb r r r test_checklist_empty s z*FileOutputDisplayTest.test_checklist_emptyc C s g d|_ | jjdtdd}d S )N)10z tag1 please1r= Tr% rW r r` ra r rc r; r r r test_checklist_miss_valid s
z/FileOutputDisplayTest.test_checklist_miss_validc C s ddg|_ | jjdtdd}d S )Nrg rB r= Tr% ri rj r r r test_checklist_miss_quit s
z.FileOutputDisplayTest.test_checklist_miss_quitc C s" t }| j| jjdt |d\}}d S )Nr= r> )ra r+ r r` rJ r r r test_checklist_noninteractive s
z3FileOutputDisplayTest.test_checklist_noninteractivec C sP dgg dddgg}dhddhddhg}t |D ]\}}t| j|t}q0d S )Nrh )rh 2rh rn 3r r r ) enumeratesetr _scrub_checklist_inputra )r indicesexpilist_Zset_tagsr r r test_scrub_checklist_input_valid s z6FileOutputDisplayTest.test_scrub_checklist_input_validc C sd g d}ddgddgg dg dg}ddgddgddgddgg}t |D ]\}}| j||}qHd S )N) ZT1ZT2ZT3T4ZT5ZT6ZT7ZT8T949)rz r{ rz )r{ rz r{ rx ry )rp 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 s" g d}d}||_ | jj| }d S )N)r=
/var/www/htmlrL Tr~ )r$ r directory_select)r rc argsZ
user_inputZreturnedr r r test_directory_select s
z+FileOutputDisplayTest.test_directory_selectc C s d}| j | jjd|d\}}d S )Nr~ r= r> )r+ r r rJ 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 |rxn |S )Nz'certbot._internal.display.obj.sys.stdinFr )r skipped_interactionr r& isattyr$ )r funcr kwargsr Z
mock_stdinr r@ r r r r+ s Jz+FileOutputDisplayTest._force_noninteractivec C s* dgdgdgddgddgg}|D ]}q d S )N0rz r rh rn or )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: ra r r r r test_print_menu s z%FileOutputDisplayTest.test_print_menuc C sh d}t j|dd W d n1 s(0 Y d}t j||d W d n1 sZ0 Y d S )Nr" rh r# rn r[ )r r( ansr r r test_get_valid_int_ans_valid
s z2FileOutputDisplayTest.test_get_valid_int_ans_validc C sT ddgg ddgg}d}|D ]2}t j||d W d q 1 sD0 Y q d S )Nr rB )rz ZoneCr" rV r[ )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- r/ r5 r< rA rG rI rK rQ rS r\ r^ rd rf rk rl rm 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 r] )r r r
r ZNoninteractiveDisplayr r r r r r $ s
z NoninteractiveDisplayTest.setUpr c C s2 | j dd | jjjd d }|jdd d S )Nr
r r r r r r r r! ( s z4NoninteractiveDisplayTest.test_notification_no_pausec C sV ddl m} | jjdddd | jjjd d }| jjddd | jjjd d }d S r0 r3 r4 r r r r5 0 s z6NoninteractiveDisplayTest.test_notification_decorationc C sP d}| j jd|d}ttj | j d W d n1 sB0 Y d S )Nzan incomputable valuer r> )r rD rN rO r MissingCommandlineFlagr dr; r r r
test_input; s z$NoninteractiveDisplayTest.test_inputc C sP | j jdtdd}ttj | j dt W d n1 sB0 Y d S )Nr r6 r> )r r9 r: rN rO r r )r r; r r r r< B s z#NoninteractiveDisplayTest.test_menuc C sP d}| j jd|d}ttj | j d W d n1 sB0 Y d S )NFr r> )r ZyesnorN rO r r r r r r r\ H s z$NoninteractiveDisplayTest.test_yesnoc C sX ddg}| j jdt|d}ttj | j dt W d n1 sJ0 Y d S )Nr6 r r> )r r` ra rN rO r r r r r r test_checklistO s z(NoninteractiveDisplayTest.test_checklistc C sX d}t j|f}| jd|}ttj | jd W d n1 sJ0 Y d S )Nr~ r= )r7 r8 r r rN rO r r )r r? Zexpectedactualr r r r V s
z/NoninteractiveDisplayTest.test_directory_selectN)r r r r r r r& r! r5 r r< r\ r r r r r r r " s
r __main__r6 )r sysZunittestr rN Zcertbotr Zcertbot._internal.displayr r Zcertbot.displayr r7 r: ra ZTestCaser r r exitmainargv__file__r r r r s >