Your IP : 3.147.79.237
B
� fxN � @ 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 �D/opt/alt/python37/lib64/python3.7/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 )N)� assertEqualr )�self�expected�text� linewidthr r r �check% s zCountLinesTest.checkc C s | � 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 | � 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 | � 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_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
� N)r 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
�
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 r r z�aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa
)r )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 r r( � �( � � � r* )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' r+ r, r3 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)r> 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)ZCourierr* ZfontZiomarkz1.0)r r �mark_set)r r r= r r r r: g s z SqueezerTest.make_text_widgetc C s6 t �|||�}t �||||� | �t j||||� d S )N)r Z GetOptionZ SetOptionr )r Z
configTypeZsectionZoption�value�prev_valr r r � set_idleconf_option_with_cleanupo s z-SqueezerTest.set_idleconf_option_with_cleanupc C sz | � � }| �|�}xbdD ]Z\}}}| j|||d��: t|�}t�|d|�� | �|�|�|� W dQ R X W dQ R X qW dS )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& )z 'aa\t' * 20r) r" )z 'aa\t' * 21r) r& )z 'aa\t' * 20r/ � )� text_code�
line_widthr r9 N)r> r@ ZsubTest�evalr �objectr � count_lines)r r< r? rF rG 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)r> r@ ZassertIsr< r �expandingbuttons)r r< r? r r r � test_init� s
zSqueezerTest.test_initc C s� | � � }xtdddd dgD ]`}ttd� |_}| �|�}| �|j�|d�t� | �|jd� |�|d� | �t |j
�d � qW d
S ) z@Test Squeezer's overriding of the EditorWindow's write() method.r �TEXTz LONG TEXTi� aL 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) r> r �SENTINEL_VALUE�writer@ r r< �
call_count�assert_called_with�lenrL )r r< r �
orig_writer? r r r �test_write_no_tags� s
z SqueezerTest.test_write_no_tagsc C s� xzdddd dgD ]f}| � � }t|j_|j}| �|�}| �|j�|d�t� | �|jd� |�|d� | �t |j
�d� qW d S )
z@Test Squeezer's overriding of the EditorWindow's write() method.r rN z LONG TEXTi� aL 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
�stderrr$ r N) r> rP rQ rO r@ r r< rR rS rT rL )r r r< rU r? r r r �test_write_not_stdout� s
z"SqueezerTest.test_write_not_stdoutc C s� | � � }xndD ]f}ttd� |_}| �|�}d|_| �|j�|d�t� | �|jd� |� |d� | �t
|j �d� qW xjdd d gD ]Z}ttd� |_}| �|�}d|_| �|j�|d�d
� | �|jd� | �t
|j �d� q�W d
S ) z@Test Squeezer's overriding of the EditorWindow's write() method.)r rN )rO �2 �stdoutr$ r z LONG TEXTi� aL 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
N) r> r rP rQ r@ �auto_squeeze_min_linesr r< rR rS rT rL )r r< r rU r? 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)r; r- r. )rO z TEXT
TEXT
TEXT
TEXT
TEXT
TEXT
rZ z1.0�endr( r$ N) r> r r@ r[ r rJ rQ r �getrT rL )r r< r= r? r r r �test_auto_squeeze� s
z SqueezerTest.test_auto_squeezec C s
�xdD ]�}| 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 � qW dS )z$Test the squeeze_current_text event.)rZ rW T)r; r. )rO z1.0z
SOME
TEXT
�insertr] z SOME
TEXT
r )�event�breakz
r$ z SOME
TEXTN)r> r r@ r rJ r` rA r r^ rT rL �squeeze_current_text_event�s�expand)r Ztag_namer< r= r? �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.T)r; r. )rO z1.0z
SOME
TEXT
ZTAGr` r] z SOME
TEXT
r )ra rb N) r> r r@ r rJ r` rA r r^ rT rL rc )r r< r= r? rf 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.T)r; r. )rO z1.0z
SOME
TEXT
rZ r` )ra r$ z MORE
STUFF
rb r] z
r" r �<N)r> r r@ r rJ r` rA rc r rT rL r^ �
assertTrueZcompare)r r< r= r? rf 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.T)r; r* �mainr zauto-squeeze-min-linesN)r> r@ r[ rD �strr �reloadr )r r< r? 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 rn )r r r r �!test_reload_no_squeezer_instances, s z.SqueezerTest.test_reload_no_squeezer_instances)F)N)N)r4 r5 r6 r7 r> r@ r: rD rK rM rV rX r\ r_ rg rh rk ro rp r r r r r8 R 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.rY )r r r r< r r[ )r r r? r r r �make_mock_squeezer5 s
z&ExpandingButtonTest.make_mock_squeezerzidlelib.squeezer.Hovertip)�autospecc 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.rN �TAGSrY z50 linesr z1.0r] r( z<Double-Button-1>z <Button-%s>�2�3r$ )Z hover_delayr z right-clickN)rr r< r r
r rd Zmasterrj Zcgetr^ ZassertInZbindr ZisAquaTkrR rS r � call_args�lower)r Z MockHovertipr? r= �expandingbuttonZright_button_codeZ tooltip_textr r r rM ? 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.rN rt rY z1.0)�window)ra Nr] zTEXT
zend-1cr$ )rr r
r �
window_create� base_textre r r r^ �indexZ
tag_nextrangerL �removerR rS )r r? ry r= rf 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 d
Q R X | �|d � | �|j�dd �d
� td��$}d|j _
d|j _
|j t
� d �}W d
Q R X | �|d
� | �|j�dd �|� d
S )z@Test that expanding very long output asks user for confirmation.�ai�� rt rY z1.0)rz z idlelib.squeezer.tkMessageBoxF)ra Nrb zend-1cr T)rr r
Zset_is_dangerousrj Z is_dangerousr r{ r| r Z askokcancelrO Zaskyesnore r r r^ )r r? r ry r= Z mock_msgboxrf 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.rN rt rY )ra Nr$ ) rr r
r Zclipboard_clearZclipboard_append�copyr rR rS )r r? ry rf r r r � test_copy� s z ExpandingButtonTest.test_copyc C sl | � � }tddd|�}t� |_tdtd��8}|jt� d� | �|jd� | �|j d d d� W d
Q R X d
S ) zTest the view event.rN rt rY zidlelib.squeezer.view_text)rs )ra r$ r r" N)
rr r
r Z
selection_ownr r Zviewr rR rw )r r? ry Zmock_view_textr r r � test_view� s z ExpandingButtonTest.test_viewc C s� | � � }tddd|�}td��j}t� }||_t� }d|_d|_|j|d� | �|j j
t |j �� x"|j D ]^}}|j j
|td� qhW W dQ R X dS ) zTest the context menu.rN rt rY z tkinter.Menur* )ra )�labelZcommandN)rr r
r r rO �x�yZcontext_menu_eventr Z add_commandrR rT Z rmenu_specsZassert_any_callr )r r? ry Z mock_MenuZ mock_menuZ
mock_eventr� �datar r r �
test_rmenu� s
z ExpandingButtonTest.test_rmenuN)
r4 r5 r6 r7 rr r r rM r r� r� r� r� r r r r rq 1 s
(rq �__main__r" )� verbosity)%r7 �textwrapr 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 rP r ZTestCaser r8 rq r4 rl r r r r �<module> s( / ` *