$19 GRAYBYTE WORDPRESS FILE MANAGER $25

3

��2aeU�@s�dd
lm
Z

ddlmZmZ
ddlZddlmZmZm	Z	m
Z
mZ
ddlm
Z

ddlmZ
ddlmZmZmZ
ddlmZ
dd	lmZ
dd
lmZ
ddlmZ
e
jZdd
�ZGdd�dej�ZGdd�dej�Z Gdd�dej�Z!e"dkr�ej#dd�

dS)�)
�
namedtuple)�Text�TkN)�Mock�NonCallableMagicMock�patch�sentinel�ANY)
�requires)
�idleConf)�count_lines_with_wrapping�ExpandingButton�Squeezer)
�macosx)
�	view_text)
�Hovertip)
�PyShellc
s0td
�

t
���j�
�f
dd�}
|j|
�

�S)NZguic
s�j�
�j
�
dS)
N)Zupdate_idletasksZdestroy�)
�rootr�2/opt/alt/python36/lib64/python3.6/test_squeezer.py�cleanup_roots

z&get_test_tk_root.<locals>.cleanup_root)r
rZwithdraw�
addCleanup)Z
test_instancerr)
rr�get_test_tk_roots


rc@s<eZ
dZd
d�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
S)�CountLinesTestcCs|j|
t
|||��S)
N)�assertEqualr)�self�expected�text�	linewidth�tabwidthrrr�check#s


zCountLinesTest.checkc

Cs|jt
d
�
d�
dS)N�r
)rr)
rrrr�test_count_empty)szCountLinesTest.test_count_emptyc

Cs|jt
d
�
d�
dS)Nz
text�)rr)
rrrr�!test_count_begins_with_empty_line-sz0CountLinesTest.test_count_begins_with_empty_linec

Cs|jt
d
�
d�
dS)Nztext
�
)rr)
rrrr�test_count_ends_with_empty_line1sz.CountLinesTest.test_count_ends_with_empty_linec

Cs|jt
d
�
d�
dS)Nz1
2
3
�)rr)
rrrr�test_count_several_lines5sz'CountLinesTest.test_count_several_linesc

Cs
|jd
dddd�
|jd
dddd�
|jddddd�
|jddddd�
|jddddd�
|jddd
dd�
|jd	dd
dd�
|jddd	dd�
|jdddd
d�
|jdddd
d�
|jdddd
d�
|jddd
dd�
|jddddd�
|jddddd�
dS)Nr%�
	��)rrrrr#r'��
��	���
r)z		z			z				z					z
										z
										z				z										z
										z											z						z						z						)
r )
rrrr�test_tab_width9s









zCountLinesTest.test_tab_widthN)	�__name__�
__module__�__qualname__r r"r$r&r(r3rrrrr!src@s�eZ
dZd
d�Zd$dd�
Zdd�Zdd	�Zd
d�Zdd
�Zdd�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zedd�Zeejejejejfdd�
�
Zed d!��
Zd"d#�ZdS)%�SqueezerTestc
Cst�}
t
|
_|
S)
N)rr�	__class__)r�editwinrrr�make_mock_editor_windowSs
z$SqueezerTest.make_mock_editor_windowNcCs|
dkr|j�}
t
|
�
S)
N)r:r)rZ
editor_windowrrr�make_squeezer_instance[s

z#SqueezerTest.make_squeezer_instancec
Cs$xd
D]\}
}|j||
d�
qWdS)Nr+�Pr%�Or*r'�x)rr�r+r<�r%r=�r*r<�r'r>)r?r@rArB)
�_test_count_lines_helper)rrrrrr�test_count_linesas

zSqueezerTest.test_count_linesc	s�d
�d}d}||
j_
|�d|||
jj_
||dd��
�
f
dd�|
j_t�}�f
dd�}||j
_td	|�}|j�
|j	|j
�

dS)
Nr-r#r%)ZborderZpadxZfontc

s�|S)
Nr)
�opt)
�	text_optsrr�<lambda>|szDSqueezerTest._prepare_mock_editwin_for_count_lines.<locals>.<lambda>c

s|d
kr�Std�
�
dS)N�
0z%measure should only be called on '0'!)
�
ValueError)
�char)
�
CHAR_WIDTHrr�measure�s


zCSqueezerTest._prepare_mock_editwin_for_count_lines.<locals>.measurezidlelib.squeezer.Font)Zget_tk_tabwidth�return_valuerZwinfo_width�cgetrrLr�startr�stop)	rr9rrZBORDER_WIDTHZ
PADDING_WIDTHZ	mock_fontrLZpatcherr)rKrFr�%_prepare_mock_editwin_for_count_linesks








z2SqueezerTest._prepare_mock_editwin_for_count_linescCsh|j�}|j
||
|�
|j|�
}ttd
�
}d}td|��&
|j|j|�
t�
|j||
|�
WdQRXdS)N)
rM�TEXTz*idlelib.squeezer.count_lines_with_wrapping)	r:rQr;r�SENTINEL_VALUEr�assertIs�count_lines�assert_called_with)rrrr9�squeezerZmock_count_linesrrrrrC�s








z%SqueezerTest._test_count_lines_helperc
Cs2|j�}
|j
|
�
}|j|j|
�
|j|jg�
dS)
N)r:r;rTr9r�expandingbuttons)rr9rWrrr�	test_init�s



zSqueezerTest.test_initc
Cs�|j�}
xxd
dddddgD]`}t
td�
|
_}|j|
�
}|j|jj|f�t�
|j|jd�
|j|f�
|jt	|j
�
d	�
qWdS)
Nr!rRz	LONG TEXTi�zMANY_LINES
�d)
rMr%r
)r:rrS�writer;rr9�
call_countrV�lenrX)rr9r�
orig_writerWrrr�test_write_no_tags�s






zSqueezerTest.test_write_no_tagsc
Cs�x~d
dddddgD]f}
|j�}t
|j_|j}|j|�
}|j|jj|
d�t
�
|j|jd�
|j|
d�
|jt	|j
�
d	�
qWdS)
Nr!rRz	LONG TEXTi�zMANY_LINES
rZ�stderrr%r
)r:rSr[rMr;rr9r\rVr]rX)rrr9r^rWrrr�test_write_not_stdout�s








z"SqueezerTest.test_write_not_stdoutc
Cs�|j�}
|j
|
d
dd�
xndD]f}ttd�
|
_}|j|
�
}d|_|j|jj|d�t�
|j|j	d	�
|j
|d�
|jt|j�
d
�
qWxnddd
dgD]Z}ttd�
|
_}|j|
�
}d|_|j|jj|d�d�
|j|j	d
�
|jt|j�
d	�
q�WdS)Nr<r*)rrr!rR)
rM�2�stdoutr%r
z	LONG TEXTi�zMANY_LINES
rZ)r!rR)
r:rQrrSr[r;�auto_squeeze_min_linesrr9r\rVr]rX)rr9rr^rWrrr�test_write_stdout�s&

















zSqueezerTest.test_write_stdoutc
Cs�t|�
}
t
|
�
}|jd
d�
|j�}||_|j|�
}d|_tdd�
|_|j	ddd�
|j
|jdd�d	�
|j
t|j
�
d
�
dS)N�iomarkz1.0r,r1)
rMzTEXT
rc�end�

r%)rr�mark_setr:rr;rdrrUr[r�getr]rX)rr�text_widgetr9rWrrr�test_auto_squeeze�s








zSqueezerTest.test_auto_squeezec
Cs,
t|�
}
�
xdD�
]}t
|
�
}|jdd�
|j�}||_|j_|j|�
}tdd�
|_	|j
dd|�
|jdd�
|j|jdd	�d
�
|jt
|j�
d�
|jt�d�
}|j|d
�
|j|jdd	�d�
|jt
|j�
d�
|j|jdjd�
|jdjt�d�

|j|jdd	�d
�
|jt
|j�
d�
qWdS)Nrcr`rfz1.0r1)
rMz
SOME
TEXT
�insertrgzSOME
TEXT

r
)
�event�breakz

r%z	SOME
TEXT)rcr`)rrrir:r�per�bottomr;rrUrmrrjr]rX�squeeze_current_text_event�
s�expand)rrZtag_namerkr9rW�retvalrrr�test_squeeze_current_text_event�s(













z,SqueezerTest.test_squeeze_current_text_eventc
Cs�t|�
}
t
|
�
}|jd
d�
|j�}||_|j_|j|�
}tdd�
|_	|j
ddd�
|jdd�
|j|jdd�d	�
|jt
|j�
d
�
|jt�d�
}|j|d�
|j|jdd�d	�
|jt
|j�
d
�
dS)
Nrfz1.0r1)
rMz
SOME
TEXT
ZTAGrmrgzSOME
TEXT

r
)
rnro)rrrir:rrprqr;rrUrmrrjr]rXrr)rrrkr9rWrurrr�/test_squeeze_current_text_event_no_allowed_tags
s









z<SqueezerTest.test_squeeze_current_text_event_no_allowed_tagsc
Cs
t|�
}
t
|
�
}|jd
d�
|j�}||_|j_|j|�
}tdd�
|_	|j
ddd�
|jdd�
|jt�d�

|jt
|j�
d	�
|j
dd
d�
|jdd�
|jt�d�
}|j|d�
|j|jdd�d
�
|jt
|j�
d�
|j|j|jdd|jd	��

dS)Nrfz1.0r1)
rMz
SOME
TEXT
rcrm)
rnr%zMORE
STUFF
rorgz


r#r
�
<)rrrir:rrprqr;rrUrmrrrr]rXrj�
assertTrueZcompare)rrrkr9rWrurrr�/test_squeeze_text_before_existing_squeezed_text
s*
















z<SqueezerTest.test_squeeze_text_before_existing_squeezed_text�GetOptionSignaturez:configType section option default type warn_on_default rawcCs|j|
||||||�S)
N)
r{)�clsZ
configType�section�option�default�typeZwarn_on_default�rawrrr�	_make_sig>
s
zSqueezerTest._make_sigcCs|
dd�\}}|j||�
S)Nr#���)
r�)r|Z
mock_call_obj�args�kwargsrrr�get_GetOption_signatureF
s
z$SqueezerTest.get_GetOption_signaturec

Cs8|jt
jt�
tjd
ddd�
t
j�
|jt
jd�
dS)N�mainrzauto-squeeze-min-linesZ42�*)ZassertIsInstancerrd�intrZ	SetOption�reloadr)
rrrr�test_reloadK
s


zSqueezerTest.test_reload)
N)r4r5r6r:r;rDrQrCrYr_rarerlrvrwrzrr{�classmethodrZ	NOT_GIVENr�r�r�rrrrr7Qs,


"





r7c@sPeZ
dZd
d�Zeded�dd��
Zdd�Zd	d
�Zdd�Z	d
d�Z
dd�ZdS)�ExpandingButtonTestc
Cs$t|�
}
t
�}t|
�
|j_d
|_|S)Nrb)rrrr9rrd)rrrWrrr�make_mock_squeezerW
s



z&ExpandingButtonTest.make_mock_squeezerzidlelib.squeezer.Hovertip)
�autospeccCs�|j�}|j
j}td
dd|�}|j|jd
�
|j|j|�
|jd|jd�
k�

|j|j	dd�d�
|j
d	|j��
d
tj
�r�dnd}|j
||j��
|j|
jd
�
|
j|ttd�
|
jdd
}|j
d|j��
dS)NrR�TAGSrbz50 linesrz1.0rgrhz<Double-Button-1>z<Button-%s>�
2�
3r%)
Zhover_delayr
zright-click)r�r9rr
rrsZmasterryrNrjZassertInZbindrZisAquaTkr\rVr	�	call_args�lower)rZMockHovertiprWrk�expandingbuttonZright_button_codeZtooltip_textrrrrYa
s






zExpandingButtonTest.test_initc
Cs�|j�}
t
d
dd|
�}|j}|jd|d�
|j|_|jt�d�
}|j|d�
|j|jdd�d�
|j	d	�
}|j|jd|�d
�
|j|j
dd�d|f�
|j|
jjj
d
�
|
jjj|�

dS)NrRr�rbz1.0)
�window)
rnrgzTEXT
zend-1cr%)r�r
r�
window_create�	base_textrtrrrj�indexZ
tag_nextrangerX�remover\rV)rrWr�rkruZtext_end_indexrrr�test_expand~
s








zExpandingButtonTest.test_expandc
Cs�|j�}
d
d}t
|dd|
�}|j�
|j|j�

|j}|jd|d�
|j|_td�
�$}d	|j	_
d	|j_
|jt
�d
�
}WdQRX|j|d�
|j|jjdd�d
�
td�
�$}d|j	_
d|j_
|jt
�d
�
}WdQRX|j|d�
|j|jjdd�|�
dS)N�
ar-r,r�rbz1.0)
r�zidlelib.squeezer.tkMessageBoxF)
rnrozend-1cr!Ti��
)r�r
Zset_is_dangerousryZis_dangerousrr�r�rZaskokcancelrMZaskyesnortrrrj)rrWrr�rkZmock_msgboxrurrr�test_expand_dangerous_oupput�
s(












z0ExpandingButtonTest.test_expand_dangerous_oupputc
Csp|j�}
t
d
dd|
�}t�|_t�|_|jt�d�
}|j|d�
|j|jjd�
|j|jjd�
|jjd
�

dS)NrRr�rb)
rnr%)	r�r
rZclipboard_clearZclipboard_append�copyrr\rV)rrWr�rurrr�	test_copy�
s





zExpandingButtonTest.test_copyc

Csl|j�}
t
d
dd|
�}t�|_tdtd��8}|jt�d�

|j|jd�
|j|j	dd	d
�
WdQRXdS)
NrRr�rbzidlelib.squeezer.view_text)
r�)
rnr%r
r#)
r�r
rZ
selection_ownrrZviewrr\r�)rrWr�Zmock_view_textrrr�	test_view�
s


zExpandingButtonTest.test_viewc
Cs�|j�}
t
d
dd|
�}td�
�j}t�}||_t�}d|_d|_|j|d�

|j|j	j
t|j�
�
x"|jD]^
}}|j	j
|td�
qhWWdQRXdS)NrRr�rbztkinter.Menur-)
rn)�labelZcommand)r�r
rrrM�
x�
yZcontext_menu_eventrZadd_commandr\r]Zrmenu_specsZassert_any_callr	)rrWr�Z	mock_MenuZ	mock_menuZ
mock_eventr��datarrr�
test_rmenu�
s













zExpandingButtonTest.test_rmenuN)r4r5r6r�rrrYr�r�r�r�r�rrrrr�S
s
)r��__main__r#)
�	verbosity)$�collectionsrZtkinterrrZunittestZ
unittest.mockrrrrr	Ztest.supportr
Zidlelib.configrZidlelib.squeezerrr
rZidlelibrZidlelib.textviewrZidlelib.tooltiprZidlelib.pyshellrrSrZTestCaserr7r�r4r�rrrr�<module>
s(







0*