Your IP : 18.223.170.253
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"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 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.<locals>.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 )�self�expected�text� linewidth�tabwidthr r r �check# s zCountLinesTest.checkc C s | j td�d� 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 td�d� 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 td�d� 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 td�d� 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%dd�Zdd� 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<Create an actual Squeezer instance with a mock EditorWindow.N)r; r )r Z
editor_windowr r r �make_squeezer_instance[ s z#SqueezerTest.make_squeezer_instancec C s$ x dD ]\}}| j ||d� qW d S )z�Test Squeezer.count_lines() with various inputs.
This checks that Squeezer.count_lines() calls the
count_lines_with_wrapping() function with the appropriate parameters.
r+ �P r% �O r* r' �x )r r N�r+ r= �r% r> �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_||dd���fdd�|j_t� }� fdd �}||j_td
|�}|j� | j |j
� dS ) z<Prepare a mock EditorWindow object for Squeezer.count_lines.r- r# r% N)ZborderZpadxZfontc s � | S )Nr )Zopt)� text_optsr r �<lambda>| s zDSqueezerTest._prepare_mock_editwin_for_count_lines.<locals>.<lambda>c s | dkr � S t d��d S )N�0z%measure should only be called on '0'!)�
ValueError)�char)�
CHAR_WIDTHr r �measure� s zCSqueezerTest._prepare_mock_editwin_for_count_lines.<locals>.measurezidlelib.squeezer.Font) Zget_tk_tabwidth� return_valuer Z winfo_width�cgetr 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_lines�assert_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|d�t� | 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|d�t� | j|j d � |j
|d� | jt |j �d
� q W xnd d d
d gD ]Z}ttd� |_}| j|�}d|_| j|jj|d�d� | j|j d
� | jt |j �d � q�W 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 dd�d � | 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
|j�d � |jt� d �}| j |d
� | j |j dd �d� | j t
|j�d� | j |jd jd� |jd jt� d � | j |j dd �d
� | j t
|j�d � qW dS )z$Test the squeeze_current_text event.rc r` rf z1.0r1 )rM z
SOME
TEXT
�insertrg z SOME
TEXT
r )�event�breakz
r% z SOME
TEXTN)rc r` )r r ri r; r �per�bottomr< r rU rm r rj r] rX �squeeze_current_text_event�s�expand)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 dd�d � | j t
|j�d
� |jt� d �}| j |d � | j |j dd�d � | j t
|j�d
� 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<SqueezerTest.test_squeeze_current_text_event_no_allowed_tagsc C s t | �}t|�}|jdd� | j� }| |_|j_| j|�}tdd�|_ |j
ddd� |jdd� |j t� d� | j t
|j�d � |j
dd
d� |jdd� |j t� d�}| j |d � | j |jdd �d
� | j t
|j�d� | j|j|jd d|jd �� dS )z2Test squeezing text before existing squeezed text.rf z1.0r1 )rM z
SOME
TEXT
rc rm )rn r% z MORE
STUFF
ro rg z
r# r �<N)r r ri r; r rp rq r< r rU rm rr r r] rX rj �
assertTrueZcompare)r r rk r: rW ru r r r �/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 rawc C s | j |||||||�S )N)r{ )�clsZ
configTypeZsectionZoption�default�typeZwarn_on_default�rawr r r � _make_sig> s zSqueezerTest._make_sigc C s |dd � \}}| j ||�S )Nr# ���)r� )r| Z
mock_call_obj�args�kwargsr 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 SetOption�reloadr )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� Zeded�dd� �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|jd�k� | j|j dd�d� | j
d |j � � d
t j
� r�d 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 z<Double-Button-1>z <Button-%s>�2�3r% )Z hover_delayr z right-clickN)r� r: r r
r rs Zmasterry rN rj ZassertInZbindr ZisAquaTkr\ rV r � call_args�lower)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|jdd�d � |j d
�}| j|jd|�d� | j|j
dd�d|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_append�copyr r\ rV )r rW r� ru r r r � test_copy� s z ExpandingButtonTest.test_copyc
C sl | j � }tddd|�}t� |_tdtd��8}|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|�}td��j}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 �x�yZcontext_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 �<module> s( 0 *