3
feU @ 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 )z*Helper for tests: Create a root Tk object.Zguic s j j d S )N)Zupdate_idletasksZdestroy )rootr D/opt/alt/python36/lib64/python3.6/idlelib/idle_test/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 ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dS )CountLinesTestz1Tests for the count_lines_with_wrapping function.c C s | j |t|||S )N) assertEqualr )selfexpectedtext linewidthtabwidthr r r check# s zCountLinesTest.checkc C s | j tdd dS )zTest with an empty string. r N)r r )r r r r test_count_empty) s z CountLinesTest.test_count_emptyc C s | j tdd dS )z/Test with a string which begins with a newline.z
text N)r r )r r r r !test_count_begins_with_empty_line- s z0CountLinesTest.test_count_begins_with_empty_linec C s | j tdd dS )z-Test with a string which ends with a newline.ztext
N)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 dS )z Test with several lines of text.z1
2
3
N)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 dd d dd | j dd ddd dS ) z-Test with various tab widths and line widths.r% )r r r r r# r'
Nr) 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____doc__r r" r$ r& r( r3 r r r r r ! s r c @ s e Zd ZdZdd Zd%ddZdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dd Z
dd Zdd Zdd Zed d Zeejejejejfd d Zed!d" Zd#d$ ZdS )& SqueezerTestz Tests for the Squeezer class.c C s t }t|_|S )z$Create a mock EditorWindow instance.)r r __class__)r editwinr r r make_mock_editor_windowS s z$SqueezerTest.make_mock_editor_windowNc C s |dkr| j }t|S )z r* r= r' r? )r@ rA rB rC )_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_||ddfdd|j_t } fdd }||j_td
|}|j | j |j
dS ) z| 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 r: 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 dQ R X dS )z Helper for test_count_lines.)rM TEXTz*idlelib.squeezer.count_lines_with_wrappingN) r; rQ r< r SENTINEL_VALUEr assertIs count_linesassert_called_with)r r r r: squeezerZmock_count_linesr r r r rD s
z%SqueezerTest._test_count_lines_helperc C s2 | j }| j|}| j|j| | j|jg dS )z(Test the creation of Squeezer instances.N)r; r< rT r: r expandingbuttons)r r: 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 ) z@Test Squeezer's overriding of the EditorWindow's write() method.r! rR z LONG TEXTi z MANY_LINES
d )rM r% r N) r; r rS writer< r r:
call_countrV lenrX )r r: 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 ) z@Test Squeezer's overriding of the EditorWindow's write() method.r! rR z LONG TEXTi z MANY_LINES
rZ stderrr% r N) r; rS r[ rM r< r r: r\ rV r] rX )r r r: 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 dS )z@Test Squeezer's overriding of the EditorWindow's write() method.r= r* )r r r! rR )rM 2 stdoutr% r z LONG TEXTi z MANY_LINES
rZ N)r! rR )
r; rQ r rS r[ r< auto_squeeze_min_linesr r: r\ rV r] rX )r r: 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 ) zATest that the auto-squeezing creates an ExpandingButton properly.iomarkz1.0r, r1 )rM zTEXT
rc end
r% N)r r mark_setr; r r< rd r rU r[ r getr] rX )r r text_widgetr: 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 dS )z$Test the squeeze_current_text event.rc r` rf z1.0r1 )rM z
SOME
TEXT
insertrg z SOME
TEXT
r )eventbreakz
r% z SOME
TEXTN)rc r` )r r ri r; r perbottomr< r rU rm r rj r] rX squeeze_current_text_eventsexpand)r r Ztag_namerk r: 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 )z@Test that the event doesn't squeeze text without a relevant tag.rf z1.0r1 )rM z
SOME
TEXT
ZTAGrm rg z SOME
TEXT
r )rn ro N)r r ri r; r rp rq r< r rU rm r rj r] rX rr )r r rk r: 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 dS )z Test the reload() class-method.mainr zauto-squeeze-min-linesZ42* N)ZassertIsInstancer rd intr Z SetOptionreloadr )r r r r test_reloadK s zSqueezerTest.test_reload)N)r4 r5 r6 r7 r; r< rE rQ rD rY r_ ra re rl rv rw rz r r{ classmethodr Z NOT_GIVENr r r r r r r r8 Q s.
" r8 c @ sT e Zd ZdZdd Zededdd Zdd Zd
d Z d d
Z
dd Z dd Z dS )ExpandingButtonTestz$Tests for the ExpandingButton class.c C s$ t | }t }t||j_d|_|S )z0Helper for tests: Create a mock Squeezer object.rb )r r r r: 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 dS )z1Test the simplest creation of an ExpandingButton.rR TAGSrb z50 linesr z1.0rg rh zz 23r% )Z hover_delayr z right-clickN)r r: 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| dS ) zTest the expand event.rR r rb z1.0)window)rn Nrg 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 )z@Test that expanding very long output asks user for confirmation.ar- r, r rb z1.0)r z idlelib.squeezer.tkMessageBoxF)rn Nro 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 dS )zTest the copy event.rR r rb )rn Nr% ) 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 ) zTest the view event.rR r rb zidlelib.squeezer.view_text)r )rn r% r r# N)
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 dQ R X dS ) zTest the context menu.rR r rb z tkinter.Menur- )rn )labelZcommandN)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 r7 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 r8 r r4 r r r r r s( 0 *