3
2aeU @ s d dl mZ d dlmZmZ d dlZd dlmZmZm Z m
Z
m Z d dl m
Z
d dlmZ d dlmZmZmZ d dlmZ d d lmZ d d
lmZ d d lmZ e
j Z d d
Z G dd dej Z G dd dej Z G dd dej Z!e"dkrej#dd dS ) )
namedtuple)TextTkN)MockNonCallableMagicMockpatchsentinelANY)requires)idleConf)count_lines_with_wrappingExpandingButtonSqueezer)macosx) view_text)Hovertip)PyShellc s0 t d t j fdd}| j| S )NZguic s j j d S )N)Zupdate_idletasksZdestroy )rootr 2/opt/alt/python36/lib64/python3.6/test_squeezer.py cleanup_root s z&get_test_tk_root..cleanup_root)r
r Zwithdraw
addCleanup)Z
test_instancer r )r r get_test_tk_root s
r c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )CountLinesTestc C s | j |t|||S )N) assertEqualr )selfexpectedtext linewidthtabwidthr r r check# s zCountLinesTest.checkc C s | j tdd d S )N r )r r )r r r r test_count_empty) s z CountLinesTest.test_count_emptyc C s | j tdd d S )Nz
text )r r )r r r r !test_count_begins_with_empty_line- s z0CountLinesTest.test_count_begins_with_empty_linec C s | j tdd d S )Nztext
)r r )r r r r test_count_ends_with_empty_line1 s z.CountLinesTest.test_count_ends_with_empty_linec C s | j tdd d S )Nz1
2
3
)r r )r r r r test_count_several_lines5 s z'CountLinesTest.test_count_several_linesc C s | j ddddd | j ddddd | j ddddd | j ddddd | j ddddd | j ddd
dd | j d dd
dd | j ddd dd | j ddddd | j ddddd | j ddddd | j ddd
dd | j ddd dd | j dd ddd d S ) Nr% )r r r r r# r'
r) z z z z z
z
z z z
z z z z )r )r r r r test_tab_width9 s z CountLinesTest.test_tab_widthN) __name__
__module__ __qualname__r r" r$ r& r( r3 r r r r r ! s r c @ s e Zd Zdd Zd$ddZdd Zdd Zd
d Zd d
Zdd Z dd Z
dd Z dd Z dd Z
dd Zdd Zed d Zeejejejejfd d Zed d! Zd"d# ZdS )% SqueezerTestc C s t }t|_|S )N)r r __class__)r editwinr r r make_mock_editor_windowS s z$SqueezerTest.make_mock_editor_windowNc C s |d kr| j }t|S )N)r: r )r Z
editor_windowr r r make_squeezer_instance[ s z#SqueezerTest.make_squeezer_instancec C s$ x d
D ]\}}| j ||d qW d S )Nr+ P r% O r* r' x )r r r+ r< r% r= r* r< r' r> )r? r@ rA rB )_test_count_lines_helper)r r r r r r test_count_linesa s z SqueezerTest.test_count_linesc s d d}d}||j _| d|| |jj_||d dfdd|j_t } fdd}||j_td |}|j | j |j
d S )
Nr- r# r% )ZborderZpadxZfontc s | S )Nr )opt) text_optsr r | s zDSqueezerTest._prepare_mock_editwin_for_count_lines..c s | dkr S t dd S )N0z%measure should only be called on '0'!)
ValueError)char)
CHAR_WIDTHr r measure s zCSqueezerTest._prepare_mock_editwin_for_count_lines..measurezidlelib.squeezer.Font) Zget_tk_tabwidth return_valuer Z winfo_widthcgetr rL r startr stop) r r9 r r Z BORDER_WIDTHZ
PADDING_WIDTHZ mock_fontrL Zpatcherr )rK rF r %_prepare_mock_editwin_for_count_linesk s
z2SqueezerTest._prepare_mock_editwin_for_count_linesc C sh | j }| j||| | j|}ttd}d}td|& | j|j|t |j||| W d Q R X d S )N)rM TEXTz*idlelib.squeezer.count_lines_with_wrapping) r: rQ r; r SENTINEL_VALUEr assertIs count_linesassert_called_with)r r r r9 squeezerZmock_count_linesr r r r rC s
z%SqueezerTest._test_count_lines_helperc C s2 | j }| j|}| j|j| | j|jg d S )N)r: r; rT r9 r expandingbuttons)r r9 rW r r r test_init s
zSqueezerTest.test_initc C s | j }xxdddd dd gD ]`}ttd |_}| j|}| j|jj|f t | j|jd |j|f | jt |j
d q W d S )
Nr! rR z LONG TEXTi z MANY_LINES
d )rM r% r ) r: r rS writer; r r9
call_countrV lenrX )r r9 r
orig_writerW r r r test_write_no_tags s
z SqueezerTest.test_write_no_tagsc C s x~dddd dd gD ]f}| j }t|j_|j}| j|}| j|jj|dt | j|jd |j|d | jt |j
d qW d S )
Nr! rR z LONG TEXTi z MANY_LINES
rZ stderrr% r ) r: rS r[ rM r; r r9 r\ rV r] rX )r r r9 r^ rW r r r test_write_not_stdout s
z"SqueezerTest.test_write_not_stdoutc C s | j }| j|ddd xndD ]f}ttd |_}| j|}d|_| j|jj|dt | j|j d |j
|d | jt |j d
q W xnd d d
d gD ]Z}ttd |_}| j|}d|_| j|jj|dd | j|j d
| jt |j d qW d S )Nr< r* )r r r! rR )rM 2 stdoutr% r z LONG TEXTi z MANY_LINES
rZ )r! rR )
r: rQ r rS r[ r; auto_squeeze_min_linesr r9 r\ rV r] rX )r r9 r r^ rW r r r test_write_stdout s&
z SqueezerTest.test_write_stdoutc C s t | }t|}|jdd | j }||_| j|}d|_tdd|_|j dd d | j
|j ddd | j
t |j
d
d S ) Niomarkz1.0r, r1 )rM zTEXT
rc end
r% )r r mark_setr: r r; rd r rU r[ r getr] rX )r r text_widgetr9 rW r r r test_auto_squeeze s
z SqueezerTest.test_auto_squeezec C s, t | }x dD ]}t|}|jdd | j }| |_|j_| j|}tdd|_ |j
dd| |jdd | j |j dd d
| j t
|jd |jt d }| j |d
| j |j dd d | j t
|jd | j |jd jd |jd jt d | j |j dd d
| j t
|jd qW d S )Nrc r` rf z1.0r1 )rM z
SOME
TEXT
insertrg z SOME
TEXT
r )eventbreakz
r% z SOME
TEXT)rc r` )r r ri r: r perbottomr; r rU rm r rj r] rX squeeze_current_text_eventsexpand)r r Ztag_namerk r9 rW retvalr r r test_squeeze_current_text_event s(
z,SqueezerTest.test_squeeze_current_text_eventc C s t | }t|}|jdd | j }| |_|j_| j|}tdd|_ |j
ddd |jdd | j |j ddd | j t
|jd
|jt d }| j |d | j |j ddd | j t
|jd
d S )
Nrf z1.0r1 )rM z
SOME
TEXT
ZTAGrm rg z SOME
TEXT
r )rn ro )r r ri r: r rp rq r; r rU rm r rj r] rX rr )r r rk r9 rW ru r r r /test_squeeze_current_text_event_no_allowed_tags s
z s zSqueezerTest._make_sigc C s |dd \}}| j ||S )Nr# )r )r| Z
mock_call_objargskwargsr r r get_GetOption_signatureF s z$SqueezerTest.get_GetOption_signaturec C s8 | j tjt tjdddd tj | jtjd d S )Nmainr zauto-squeeze-min-linesZ42* )ZassertIsInstancer rd intr Z SetOptionreloadr )r r r r test_reloadK s zSqueezerTest.test_reload)N)r4 r5 r6 r: r; rD rQ rC rY r_ ra re rl rv rw rz r r{ classmethodr Z NOT_GIVENr r r r r r r r7 Q s,
" r7 c @ sP e Zd Zdd Zededdd Zdd Zd d
Zd d Z d
d Z
dd Z dS )ExpandingButtonTestc C s$ t | }t }t||j_d|_|S )Nrb )r r r r9 r rd )r r rW r r r make_mock_squeezerW s
z&ExpandingButtonTest.make_mock_squeezerzidlelib.squeezer.Hovertip)autospecc C s | j }|jj}tddd|}| j|jd | j|j| | jd|jdk | j|j ddd | j
d |j d
t j
rd nd }| j
||j | j|jd
|j|ttd |jd d
}| j
d|j d S )NrR TAGSrb z50 linesr z1.0rg rh zz 23r% )Z hover_delayr z right-click)r r9 r r
r rs Zmasterry rN rj ZassertInZbindr ZisAquaTkr\ rV r call_argslower)r Z MockHovertiprW rk expandingbuttonZright_button_codeZ tooltip_textr r r rY a s z ExpandingButtonTest.test_initc C s | j }tddd|}|j}|jd|d |j|_|jt d}| j|d | j|jddd |j d }| j|jd|d | j|j
ddd|f | j|j j j
d
|j j j| d S ) NrR r rb z1.0)window)rn rg zTEXT
zend-1cr% )r r
r
window_create base_textrt r r rj indexZ
tag_nextrangerX remover\ rV )r rW r rk ru Ztext_end_indexr r r test_expand~ s
z ExpandingButtonTest.test_expandc C s | j }dd }t|dd|}|j | j|j |j}|jd|d |j|_td$}d |j _
d |j _
|j t
d
}W d Q R X | j|d | j|jjdd d
td$}d|j _
d|j _
|j t
d
}W d Q R X | j|d | j|jjdd | d S )Nar- r, r rb z1.0)r z idlelib.squeezer.tkMessageBoxF)rn ro zend-1cr! Ti )r r
Zset_is_dangerousry Z is_dangerousr r r r Z askokcancelrM Zaskyesnort r r rj )r rW r r rk Z mock_msgboxru r r r test_expand_dangerous_oupput s(
z0ExpandingButtonTest.test_expand_dangerous_oupputc C sp | j }tddd|}t |_t |_|jt d}| j|d | j|jjd | j|jjd |jjd d S )NrR r rb )rn r% ) r r
r Zclipboard_clearZclipboard_appendcopyr r\ rV )r rW r ru r r r test_copy s z ExpandingButtonTest.test_copyc
C sl | j }tddd|}t |_tdtd8}|jt d | j|jd | j|j d d d W d Q R X d S )
NrR r rb zidlelib.squeezer.view_text)r )rn r% r r# )
r r
r Z
selection_ownr r Zviewr r\ r )r rW r Zmock_view_textr r r test_view s z ExpandingButtonTest.test_viewc C s | j }tddd|}tdj}t }||_t }d|_d|_|j|d | j|j j
t |j x"|j D ]^}}|j j
|td qhW W d Q R X d S )NrR r rb z tkinter.Menur- )rn )labelZcommand)r r
r r rM xyZcontext_menu_eventr Z add_commandr\ r] Z rmenu_specsZassert_any_callr )r rW r Z mock_MenuZ mock_menuZ
mock_eventr datar r r
test_rmenu s
z ExpandingButtonTest.test_rmenuN) r4 r5 r6 r r r rY r r r r r r r r r r S s
)r __main__r# ) verbosity)$ collectionsr Ztkinterr r ZunittestZ
unittest.mockr r r r r Z test.supportr
Zidlelib.configr Zidlelib.squeezerr r
r Zidlelibr Zidlelib.textviewr Zidlelib.tooltipr Zidlelib.pyshellr rS r ZTestCaser r7 r r4 r r r r r s( 0 *