File: //opt/python38/lib/python3.8/idlelib/idle_test/__pycache__/test_squeezer.cpython-38.opt-2.pyc
U
�i�_xN � @ 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 )� )�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 )NZguic 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 @ sL e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dS )�CountLinesTestc C s | � |t||��S r �� assertEqualr )�self�expected�text� linewidthr r r �check% s �zCountLinesTest.checkc C s | � td�d� d S )N� r r �r r r r �test_count_empty+ s z CountLinesTest.test_count_emptyc C s | � td�d� d S )Nz
text� r r$ r r r �!test_count_begins_with_empty_line/ s z0CountLinesTest.test_count_begins_with_empty_linec C s | � td�d� d S )Nztext
� r r$ r r r � test_count_ends_with_empty_line3 s z.CountLinesTest.test_count_ends_with_empty_linec C s | � td�d� d S )Nz1
2
3
� r 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__r" r% r' r) r+ r2 r3 r: r r r r r # s r c @ s� e Zd Zd!dd�Zd"dd�Zd#dd�Zd d
� Zd d � Zd
d� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd � Zd d � Zd d � ZdS )$� SqueezerTestFc C s: t � }d|_|r6t| �|_| j|jd�}| |_|j_|S )Nr- 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 |d kr| � � }t|�}|S r )rD 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 rC r r r r@ 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 qd S )N))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* )rM r6 � )� text_code�
line_widthr r? )rD rF ZsubTest�evalr �objectr � count_lines)r rB rE rO rP r r r r r �test_count_linesu s
�z SqueezerTest.test_count_linesc C s2 | � � }| �|�}| �|j|� | �|jg � d S r )rD rF ZassertIsrB r �expandingbuttons)r rB rE r r r � test_init� s
zSqueezerTest.test_initc C s~ | � � }dddd dfD ]`}ttd� |_}| �|�}| �|j�|d�t� | �|jd� |�|d� | �t |j
�d � qd S )
Nr# �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 ) rD r �SENTINEL_VALUE�writerF r rB �
call_count�assert_called_with�lenrU �r rB r �
orig_writerE 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 ) Nr# rW rX rY rZ �stderrr( r ) rD r] r^ r\ rF r rB r_ r` ra rU )r r rB rc rE 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 )
N)r# rW r[ �2 �stdoutr( r rX rY rZ ) rD r r] r^ rF �auto_squeeze_min_linesr rB r_ r` ra rU rb 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 ) NT�rA r4 r5 r[ z TEXT
TEXT
TEXT
TEXT
TEXT
TEXT
rh rG �endr, r( ) rD r rF ri r rS r^ r �getra rU )r rB rC rE 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 � qd S )N)rh re Trk r5 r[ rG �
SOME
TEXT
�insertrl � SOME
TEXT
r ��event�breakr/ r( z SOME
TEXT)rD r rF r rS rp rH r rm ra rU �squeeze_current_text_event�s�expand)r Ztag_namerB rC rE �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 � d S )NTrk r5 r[ rG ro ZTAGrp rl rq r rr rt ) rD r rF r rS rp rH r rm ra rU ru �r rB rC rE rx 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
�� d S )NTrk r5 r[ rG ro rh rp rr r( z MORE
STUFF
rt rl z
r& r �<)rD r rF r rS rp rH ru r ra rU rm �
assertTrueZcomparerz 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|� d S )NTrk r0 �mainr zauto-squeeze-min-lines)rD rF ri rL �strr �reloadr )r rB rE Zorig_auto_squeeze_min_linesZnew_auto_squeeze_min_linesr r r � test_reload s
��zSqueezerTest.test_reloadc C s t �� d S r )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= rD rF r@ rL rT rV rd rf rj rn ry r{ r~ r� r� r r r r r> R s
r> c @ sP e Zd Zdd� Zeded�dd� �Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dS )�ExpandingButtonTestc C s$ t | �}t� }t|�|j_d|_|S )Nrg )r r r rB r ri )r r rE 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|�� � d S )NrW �TAGSrg z50 linesr rG rl r, z<Double-Button-1>z <Button-%s>�2�3r( )Z hover_delayr z right-click)r� rB r r
r rv Zmasterr} Zcgetrm ZassertIn�bindr ZisAquaTkr_ r` r � call_args�lower)r Z MockHovertiprE rC �expandingbuttonZright_button_codeZ tooltip_textr r r rV ? 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 �|� d S ) NrW r� rg rG �Zwindowrr rl zTEXT
�end-1cr( )r� r
r �
window_create� base_textrw r r rm �indexZ
tag_nextrangerU �remover_ r` )r rE r� rC rx 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 �|� d S )N�ai�� r� rg rG r� z idlelib.squeezer.tkMessageBoxFrr rt r� r# T)r� r
Zset_is_dangerousr} Z is_dangerousr r� r� r Z askokcancelr\ Zaskyesnorw r r rm )r rE r r� rC Z mock_msgboxrx 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� d S )NrW r� rg rr r( ) r� r
r Zclipboard_clearZclipboard_append�copyr r_ r` )r rE r� rx 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 )
NrW r� rg zidlelib.squeezer.view_textr� rr r( r r& )
r� r
r Z
selection_ownr r �viewr r_ r� )r rE 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 d S )NrW r� rg z tkinter.Menur0 rr )�labelZcommand)r� r
r r r\ �x�yZcontext_menu_eventr Z add_commandr_ ra Z rmenu_specsZassert_any_callr )r rE r� Z mock_MenuZ mock_menuZ
mock_eventr� �datar r r �
test_rmenu� s
�z ExpandingButtonTest.test_rmenuN) r; r< r= r� r r rV r� r� r� r� r� r r r r r� 1 s
(r� �__main__r& )� verbosity)$�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& / ` *