File: //opt/python38/lib/python3.8/idlelib/idle_test/__pycache__/test_squeezer.cpython-38.pyc
U
�i�_xN � @ s� d Z ddlmZ ddlmZmZ ddlZddlmZm Z m
Z
m Z m Z ddl
mZ ddlmZ ddlmZmZmZ dd lmZ dd
lmZ dd lmZ dd lm Z e j Z d
d� Z G dd� dej �Z G dd� dej �Z!G dd� dej �Z"e#dkr�ej$dd� dS )zTest squeezer, coverage 95%� )�dedent)�Text�TkN)�Mock�NonCallableMagicMock�patch�sentinel�ANY)�requires)�idleConf)�count_lines_with_wrapping�ExpandingButton�Squeezer)�macosx)� view_text)�Hovertip)�PyShellc s0 t d� t� � � �� � fdd�}| �|� � S )z*Helper for tests: Create a root Tk object.Zguic s � � � � �� d S �N)Zupdate_idletasksZdestroy� ��rootr �>/opt/python38/lib/python3.8/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 @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�CountLinesTestz1Tests for the count_lines_with_wrapping function.c C s | � |t||��S r �� assertEqualr )�self�expected�text� linewidthr r r �check% s �zCountLinesTest.checkc C s | � td�d� dS )zTest with an empty string.� r Nr �r r r r �test_count_empty+ s z CountLinesTest.test_count_emptyc C s | � td�d� dS )z/Test with a string which begins with a newline.z
text� Nr r$ r r r �!test_count_begins_with_empty_line/ s z0CountLinesTest.test_count_begins_with_empty_linec C s | � td�d� dS )z-Test with a string which ends with a newline.ztext
� Nr r$ r r r � test_count_ends_with_empty_line3 s z.CountLinesTest.test_count_ends_with_empty_linec C s | � td�d� dS )z Test with several lines of text.z1
2
3
� Nr r$ r r r �test_count_several_lines7 s z'CountLinesTest.test_count_several_linesc C s4 | j dddd� | j dddd� | j dddd� d S ) Nr( �
�P �r r r! r& �
�
z
�r" r$ r r r �test_empty_lines; s z CountLinesTest.test_empty_linesc C s$ | j dddd� | j dddd� d S )Nr* Z�aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar- r. z�aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
r1 r$ r r r �test_long_line@ s z CountLinesTest.test_long_linec C s` t d�}| jd|dd� | jd|d dd� | jd|dd� | jd|d d� | jd
|d d� d S ) Nz� 13 characters
43 is the number of characters on this line
7 chars
13 characters� r- r. r, � �( � � � r0 )r r" )r r r r r �$test_several_lines_different_lengthsD s z3CountLinesTest.test_several_lines_different_lengthsN) �__name__�
__module__� __qualname__�__doc__r" r% r' r) r+ r2 r3 r: r r r r r # s r c @ s� e Zd ZdZd"dd�Zd#dd�Zd$dd �Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� ZdS )%� SqueezerTestz Tests for the Squeezer class.Fc C s: t � }d|_|r6t| �|_| j|jd�}| |_|j_|S )z$Create a mock EditorWindow instance.r- r )r �widthr r �make_text_widgetr ZperZbottom)r �with_text_widget�editwin� text_widgetr r r �make_mock_editor_windowT s
z$SqueezerTest.make_mock_editor_windowNc C s |dkr| � � }t|�}|S )z<Create an actual Squeezer instance with a mock EditorWindow.N)rE r )r Z
editor_window�squeezerr r r �make_squeezer_instance` s z#SqueezerTest.make_squeezer_instancec C s0 |d krt | �}t|�}d|d< |�dd� |S )N)ZCourierr0 ZfontZiomark�1.0)r r �mark_set)r r rD r r r rA g s z SqueezerTest.make_text_widgetc C s6 t �|||�}t �||||� | �t j||||� d S r )r Z GetOptionZ SetOptionr )r Z
configType�section�option�valueZprev_valr r r � set_idleconf_option_with_cleanupo s �z-SqueezerTest.set_idleconf_option_with_cleanupc C sv | � � }| �|�}dD ]Z\}}}| j|||d��: t|�}t�|d|�� | �|�|�|� W 5 Q R X W 5 Q R X qdS )z0Test Squeezer.count_lines() with various inputs.))z'\n'r- r( )z'\n' * 3r- r* )z'a' * 40 + '\n'r- r( )z'a' * 80 + '\n'r- r( )z'a' * 200 + '\n'r- r* )� 'aa\t' * 20r- r& )z 'aa\t' * 21r- r* )rN r6 � )� text_code�
line_widthr r@ N)rE rG ZsubTest�evalr �objectr � count_lines)r rC rF rP rQ r r r r r �test_count_linesu s
�z SqueezerTest.test_count_linesc C s2 | � � }| �|�}| �|j|� | �|jg � dS )z(Test the creation of Squeezer instances.N)rE rG ZassertIsrC r �expandingbuttons)r rC rF r r r � test_init� s
zSqueezerTest.test_initc C s~ | � � }dddd dfD ]`}ttd� |_}| �|�}| �|j�|d�t� | �|jd� |�|d� | �t |j
�d � qd
S ) �@Test Squeezer's overriding of the EditorWindow's write() method.r# �TEXT� LONG TEXT�� �L MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
MANY_LINES
�� return_valuer r( r N) rE r �SENTINEL_VALUE�writerG r rC �
call_count�assert_called_with�lenrV �r rC r �
orig_writerF r r r �test_write_no_tags� s
z SqueezerTest.test_write_no_tagsc C s| dddd dfD ]f}| � � }t|j_|j}| �|�}| �|j�|d�t� | �|jd� |�|d� | �t |j
�d� qd S )
rX r# rY rZ r[ r\ �stderrr( r N) rE r_ r` r^ rG r rC ra rb rc rV )r r rC re rF r r r �test_write_not_stdout� s
� z"SqueezerTest.test_write_not_stdoutc C s� | � � }dD ]f}ttd� |_}| �|�}d|_| �|j�|d�t� | �|jd� |� |d� | �t
|j �d� q dd d fD ]Z}ttd� |_}| �|�}d|_| �|j�|d�d
� | �|jd� | �t
|j �d� q�d
S ) rX )r# rY r] �2 �stdoutr( r rZ r[ r\ N) rE r r_ r` rG �auto_squeeze_min_linesr rC ra rb rc rV rd r r r �test_write_stdout� s$
�
z SqueezerTest.test_write_stdoutc C sd | j dd�}|j}| �|�}d|_tdd�|_|�dd� | �|�dd �d
� | �t |j
�d � d S )
zATest that the auto-squeezing creates an ExpandingButton properly.T�rB r4 r5 r] z TEXT
TEXT
TEXT
TEXT
TEXT
TEXT
rj rH �endr, r( N) rE r rG rk r rT r` r �getrc rV )r rC rD rF r r r �test_auto_squeeze� s
z SqueezerTest.test_auto_squeezec C s dD ]�}| j dd�}|j}| �|�}tdd�|_|�dd|� |�dd� | �|�dd �d
� | �t |j
�d � |j t� d �}| �|d
� | �|�dd �d� | �t |j
�d� | �|j
d j d� |j
d j
t� d � | �|�dd �d
� | �t |j
�d � qdS )z$Test the squeeze_current_text event.)rj rg Trm r5 r] rH �
SOME
TEXT
�insertrn � SOME
TEXT
r ��event�breakr/ r( z SOME
TEXTN)rE r rG r rT rr rI r ro rc rV �squeeze_current_text_event�s�expand)r Ztag_namerC rD rF �retvalr r r � test_squeeze_current_text_event� s"
z,SqueezerTest.test_squeeze_current_text_eventc C s� | j dd�}|j}| �|�}tdd�|_|�ddd� |�dd� | �|�dd �d
� | �t |j
�d � |j t� d �}| �|d
� | �|�dd �d
� | �t |j
�d � dS )z@Test that the event doesn't squeeze text without a relevant tag.Trm r5 r] rH rq ZTAGrr rn rs r rt rv N) rE r rG r rT rr rI r ro rc rV rw �r rC rD rF rz r r r �/test_squeeze_current_text_event_no_allowed_tags� s
z<SqueezerTest.test_squeeze_current_text_event_no_allowed_tagsc C s� | j dd�}|j}| �|�}tdd�|_|�ddd� |�dd� |jt� d � | �t |j
�d
� |�dd d� |�dd� |jt� d �}| �|d � | �|� dd
�d� | �t |j
�d� | � |�
|j
d d|j
d
�� dS )z2Test squeezing text before existing squeezed text.Trm r5 r] rH rq rj rr rt r( z MORE
STUFF
rv rn z
r& r �<N)rE r rG r rT rr rI rw r rc rV ro �
assertTrueZcomparer| r r r �/test_squeeze_text_before_existing_squeezed_text s&
�z<SqueezerTest.test_squeeze_text_before_existing_squeezed_textc C sR | j dd�}| �|�}|j}|d }| �dddt|�� t�� | �|j|� dS )z Test the reload() class-method.Trm r0 �mainr zauto-squeeze-min-linesN)rE rG rk rM �strr �reloadr )r rC rF Zorig_auto_squeeze_min_linesZnew_auto_squeeze_min_linesr r r � test_reload s
��zSqueezerTest.test_reloadc C s t �� dS )z@Test that Squeezer.reload() runs without any instances existing.N)r r� r$ r r r �!test_reload_no_squeezer_instances, s z.SqueezerTest.test_reload_no_squeezer_instances)F)N)N)r; r< r= r> rE rG rA rM rU rW rf rh rl rp r{ r} r� r� r� r r r r r? R s
r? 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.ri )r r r rC r rk )r r rF r r r �make_mock_squeezer5 s
z&ExpandingButtonTest.make_mock_squeezerzidlelib.squeezer.Hovertip�Zautospecc C s� | � � }|jj}tddd|�}| �|jd� | �|j|� | �d|�d�k� | �|� dd�d� | �
d |� � � d
t �
� r�d nd }| �
||� � � | �|jd
� |j|ttd� |jd d
}| �
d|�� � dS )z1Test the simplest creation of an ExpandingButton.rY �TAGSri z50 linesr rH rn r, z<Double-Button-1>z <Button-%s>�2�3r( )Z hover_delayr z right-clickN)r� rC r r
r rx Zmasterr Zcgetro ZassertIn�bindr ZisAquaTkra rb r � call_args�lower)r Z MockHovertiprF rD �expandingbuttonZright_button_codeZ tooltip_textr r r rW ? s z ExpandingButtonTest.test_initc C s� | � � }tddd|�}|j}|jd|d� |j|_|jt� d�}| �|d� | �|�dd�d � |� d
�}| �|�d|�d� | �|�
dd�d|f� | �|j j j
d � |j j �|� dS ) zTest the expand event.rY r� ri rH �Zwindowrt Nrn zTEXT
�end-1cr( )r� r
r �
window_create� base_textry r r ro �indexZ
tag_nextrangerV �removera rb )r rF r� rD rz Ztext_end_indexr r r � test_expand\ s
�z ExpandingButtonTest.test_expandc C s� | � � }dd }t|dd|�}|�� | �|j� |j}|jd|d� |j|_td��$}d|j _
d|j _
|j t
� d �}W 5 Q R X | �|d
� | �|j�dd �d � td��$}d
|j _
d
|j _
|j t
� d �}W 5 Q R X | �|d� | �|j�dd �|� dS )z@Test that expanding very long output asks user for confirmation.�ai�� r� ri rH r� z idlelib.squeezer.tkMessageBoxFrt rv r� r# TN)r� r
Zset_is_dangerousr Z is_dangerousr r� r� r Z askokcancelr^ Zaskyesnory r r ro )r rF r r� rD Z mock_msgboxrz r r r � test_expand_dangerous_oupput| s(
z0ExpandingButtonTest.test_expand_dangerous_oupputc C sp | � � }tddd|�}t� |_t� |_|jt� d�}| �|d� | �|jjd� | �|jjd� |j�d� dS )zTest the copy event.rY r� ri rt Nr( ) r� r
r Zclipboard_clearZclipboard_append�copyr ra rb )r rF r� rz r r r � test_copy� s z ExpandingButtonTest.test_copyc C sl | � � }tddd|�}t� |_tdtd��8}|jt� d� | �|jd� | �|j d d d� W 5 Q R X d
S ) zTest the view event.rY r� ri zidlelib.squeezer.view_textr� rt r( r r& N)
r� r
r Z
selection_ownr r �viewr ra r� )r rF r� Zmock_view_textr r r � test_view� s z ExpandingButtonTest.test_viewc C s� | � � }tddd|�}td��f}t� }||_t� }d|_d|_|j|d� | �|j j
t |j �� |j D ]^}}|j j
|td� qfW 5 Q R X dS ) zTest the context menu.rY r� ri z tkinter.Menur0 rt )�labelZcommandN)r� r
r r r^ �x�yZcontext_menu_eventr Z add_commandra rc Z rmenu_specsZassert_any_callr )r rF r� Z mock_MenuZ mock_menuZ
mock_eventr� �datar r r �
test_rmenu� s
�z ExpandingButtonTest.test_rmenuN)
r; r< r= r> r� r r rW r� r� r� r� r� r r r r r� 1 s
(r� �__main__r& )� verbosity)%r> �textwrapr �tkinterr r ZunittestZ
unittest.mockr r r r r Z test.supportr
Zidlelib.configr Zidlelib.squeezerr r
r Zidlelibr Zidlelib.textviewr Zidlelib.tooltipr Zidlelib.pyshellr r_ r ZTestCaser r? r� r; r� r r r r �<module> s( / ` *