B � fxN�@s�dZddlmZddlmZmZddlZddlmZm Z m Z mZmZddl mZddlmZddlmZmZmZdd lmZdd lmZddlmZddlmZejZd d�ZGdd�dej�Z Gdd�dej�Z!Gdd�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)�PyShellcs0td�t��fdd�}|�|��S)z*Helper for tests: Create a root Tk object.Zguics��dS)N)Zupdate_idletasksZdestroy�)�rootr�D/opt/alt/python37/lib64/python3.7/idlelib/idle_test/test_squeezer.py�cleanup_rootsz&get_test_tk_root..cleanup_root)r rZwithdraw� addCleanup)Z test_instancerr)rr�get_test_tk_roots rc@sPeZdZdZdd�Zdd�Zdd�Zdd �Zd d�Zdd �Z dd�Z dd�ZdS)�CountLinesTestz1Tests for the count_lines_with_wrapping function.cCs|�|t||��S)N)�assertEqualr)�self�expected�text� linewidthrrr�check%szCountLinesTest.checkcCs|�td�d�dS)zTest with an empty string.�rN)rr)rrrr�test_count_empty+szCountLinesTest.test_count_emptycCs|�td�d�dS)z/Test with a string which begins with a newline.z text�N)rr)rrrr�!test_count_begins_with_empty_line/sz0CountLinesTest.test_count_begins_with_empty_linecCs|�td�d�dS)z-Test with a string which ends with a newline.ztext �N)rr)rrrr�test_count_ends_with_empty_line3sz.CountLinesTest.test_count_ends_with_empty_linecCs|�td�d�dS)z Test with several lines of text.z1 2 3 �N)rr)rrrr�test_count_several_lines7sz'CountLinesTest.test_count_several_linescCs4|jdddd�|jdddd�|jdddd�dS) Nr$� �P)rrrr"z � z )r)rrrr�test_empty_lines;szCountLinesTest.test_empty_linescCs$|jdddd�|jdddd�dS)Nr&Z�aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaar))rrrz�aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa )r)rrrr�test_long_line@szCountLinesTest.test_long_linecCs`td�}|jd|dd�|jd|ddd�|jd|dd�|jd|d d�|jd |dd�dS)Nz� 13 characters 43 is the number of characters on this line 7 chars 13 characters�r))rrrr(��(��r*)rr)rrrrr�$test_several_lines_different_lengthsDsz3CountLinesTest.test_several_lines_different_lengthsN)�__name__� __module__�__qualname__�__doc__rr!r#r%r'r+r,r3rrrrr#src@s�eZdZdZd"dd�Zd#dd�Zd$dd �Zd d�Zdd �Zdd�Z dd�Z dd�Zdd�Zdd�Z dd�Zdd�Zdd�Zdd�Zd d!�ZdS)%�SqueezerTestzTests for the Squeezer class.FcCs:t�}d|_|r6t|�|_|j|jd�}||_|j_|S)z$Create a mock EditorWindow instance.r))r)r�widthrr�make_text_widgetrZperZbottom)r�with_text_widget�editwin�text_widgetrrr�make_mock_editor_windowTs z$SqueezerTest.make_mock_editor_windowNcCs|dkr|��}t|�}|S)zr)rZ editor_window�squeezerrrr�make_squeezer_instance`sz#SqueezerTest.make_squeezer_instancecCs0|dkrt|�}t|�}d|d<|�dd�|S)N)ZCourierr*ZfontZiomarkz1.0)rr�mark_set)rrr=rrrr:gszSqueezerTest.make_text_widgetcCs6t�|||�}t�||||�|�tj||||�dS)N)rZ GetOptionZ SetOptionr)rZ configTypeZsectionZoption�value�prev_valrrr� set_idleconf_option_with_cleanuposz-SqueezerTest.set_idleconf_option_with_cleanupcCsz|��}|�|�}xbdD]Z\}}}|j|||d��:t|�}t�|d|��|�|�|�|�WdQRXWdQRXqWdS)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_widthrr9N)r>r@ZsubTest�evalr�objectr�count_lines)rr<r?rFrGrrrrr�test_count_linesus zSqueezerTest.test_count_linescCs2|��}|�|�}|�|j|�|�|jg�dS)z(Test the creation of Squeezer instances.N)r>r@ZassertIsr<r�expandingbuttons)rr<r?rrr� test_init�s zSqueezerTest.test_initcCs�|��}xtdddddgD]`}ttd�|_}|�|�}|�|j�|d�t�|�|jd�|�|d�|�t |j �d �qWd S)z@Test Squeezer's overriding of the EditorWindow's write() method.r �TEXTz LONG TEXTi�aLMANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES 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_valuerr$rN)r>r�SENTINEL_VALUE�writer@rr<� call_count�assert_called_with�lenrL)rr<r� orig_writer?rrr�test_write_no_tags�s zSqueezerTest.test_write_no_tagscCs�xzdddddgD]f}|��}t|j_|j}|�|�}|�|j�|d�t�|�|jd�|�|d�|�t |j �d�qWd S) z@Test Squeezer's overriding of the EditorWindow's write() method.r rNz LONG TEXTi�aLMANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES 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$rN)r>rPrQrOr@rr<rRrSrTrL)rrr<rUr?rrr�test_write_not_stdout�s z"SqueezerTest.test_write_not_stdoutcCs�|��}xndD]f}ttd�|_}|�|�}d|_|�|j�|d�t�|�|jd�|� |d�|�t |j�d�qWxjddd gD]Z}ttd�|_}|�|�}d|_|�|j�|d�d �|�|jd�|�t |j�d�q�Wd S)z@Test Squeezer's overriding of the EditorWindow's write() method.)r rN)rO�2�stdoutr$rz LONG TEXTi�aLMANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES MANY_LINES 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>rrPrQr@�auto_squeeze_min_linesrr<rRrSrTrL)rr<rrUr?rrr�test_write_stdout�s" zSqueezerTest.test_write_stdoutcCsd|jdd�}|j}|�|�}d|_tdd�|_|�dd�|�|�dd �d �|�t |j �d�dS) zATest that the auto-squeezing creates an ExpandingButton properly.T)r;r-r.)rOzTEXT TEXT TEXT TEXT TEXT TEXT rZz1.0�endr(r$N)r>rr@r[rrJrQr�getrTrL)rr<r=r?rrr�test_auto_squeeze�s zSqueezerTest.test_auto_squeezecCs �xdD]�}|jdd�}|j}|�|�}tdd�|_|�dd|�|�dd�|�|�dd �d �|�t |j �d�|jt�d�}|�|d �|�|�dd �d�|�t |j �d�|�|j djd�|j dj t�d�|�|�dd �d �|�t |j �d�qWdS)z$Test the squeeze_current_text event.)rZrWT)r;r.)rOz1.0z SOME TEXT �insertr]zSOME TEXT r)�event�breakz r$z SOME TEXTN)r>rr@rrJr`rArr^rTrL�squeeze_current_text_event�s�expand)rZtag_namer<r=r?�retvalrrr�test_squeeze_current_text_event�s" z,SqueezerTest.test_squeeze_current_text_eventcCs�|jdd�}|j}|�|�}tdd�|_|�ddd�|�dd�|�|�dd �d �|�t |j �d�|jt�d�}|�|d �|�|�dd �d �|�t |j �d�dS)z@Test that the event doesn't squeeze text without a relevant tag.T)r;r.)rOz1.0z SOME TEXT ZTAGr`r]zSOME TEXT r)rarbN)r>rr@rrJr`rArr^rTrLrc)rr<r=r?rfrrr�/test_squeeze_current_text_event_no_allowed_tags�s zrr@rrJr`rArcrrTrLr^� assertTrueZcompare)rr<r=r?rfrrr�/test_squeeze_text_before_existing_squeezed_texts$ zr@r[rD�strr�reloadr)rr<r?Zorig_auto_squeeze_min_linesZnew_auto_squeeze_min_linesrrr�test_reloads zSqueezerTest.test_reloadcCst��dS)z@Test that Squeezer.reload() runs without any instances existing.N)rrn)rrrr�!test_reload_no_squeezer_instances,sz.SqueezerTest.test_reload_no_squeezer_instances)F)N)N)r4r5r6r7r>r@r:rDrKrMrVrXr\r_rgrhrkrorprrrrr8Rs r8c@sTeZdZdZdd�Zeded�dd��Zdd �Zd d�Z dd �Z dd�Zdd�ZdS)�ExpandingButtonTestz$Tests for the ExpandingButton class.cCs$t|�}t�}t|�|j_d|_|S)z0Helper for tests: Create a mock Squeezer object.rY)rrrr<rr[)rrr?rrr�make_mock_squeezer5s z&ExpandingButtonTest.make_mock_squeezerzidlelib.squeezer.Hovertip)�autospeccCs�|��}|jj}tddd|�}|�|jd�|�|j|�|�d|�d�k�|�|� dd�d�|� d |��d t� �r�dnd}|� ||��|�|jd �|j|ttd�|jdd }|� d|��dS)z1Test the simplest creation of an ExpandingButton.rN�TAGSrYz50 linesrz1.0r]r(zz�2�3r$)Zhover_delayrzright-clickN)rrr<rr rrdZmasterrjZcgetr^ZassertInZbindrZisAquaTkrRrSr � call_args�lower)rZMockHovertipr?r=�expandingbuttonZright_button_codeZtooltip_textrrrrM?szExpandingButtonTest.test_initcCs�|��}tddd|�}|j}|jd|d�|j|_|jt�d�}|�|d�|�|�dd�d �|� d �}|�|�d|�d�|�|� dd�d|f�|�|jjj d�|jj�|�dS)zTest the expand event.rNrtrYz1.0)�window)raNr]zTEXT zend-1cr$)rrr r� window_create� base_textrerrr^�indexZ tag_nextrangerL�removerRrS)rr?ryr=rfZtext_end_indexrrr�test_expand\s zExpandingButtonTest.test_expandc Cs�|��}dd}t|dd|�}|��|�|j�|j}|jd|d�|j|_td��$}d|j _ d|j_ |jt �d �}Wd QRX|�|d�|�|j�dd�d �td��$}d|j _ d|j_ |jt �d �}Wd QRX|�|d �|�|j�dd�|�d S)z@Test that expanding very long output asks user for confirmation.�ai��rtrYz1.0)rzzidlelib.squeezer.tkMessageBoxF)raNrbzend-1cr T)rrr Zset_is_dangerousrjZis_dangerousrr{r|rZaskokcancelrOZaskyesnorerrr^)rr?rryr=Zmock_msgboxrfrrr�test_expand_dangerous_oupput|s( z0ExpandingButtonTest.test_expand_dangerous_oupputcCsp|��}tddd|�}t�|_t�|_|jt�d�}|�|d�|�|jjd�|�|jjd�|j�d�dS)zTest the copy event.rNrtrY)raNr$) rrr rZclipboard_clearZclipboard_append�copyrrRrS)rr?ryrfrrr� test_copy�szExpandingButtonTest.test_copyc Csl|��}tddd|�}t�|_tdtd��8}|jt�d�|�|jd�|�|j dd d�Wd QRXd S)zTest the view event.rNrtrYzidlelib.squeezer.view_text)rs)rar$rr"N) rrr rZ selection_ownrrZviewrrRrw)rr?ryZmock_view_textrrr� test_view�szExpandingButtonTest.test_viewc Cs�|��}tddd|�}td��j}t�}||_t�}d|_d|_|j|d�|�|j j t|j��x"|jD]^}}|j j |td�qhWWdQRXdS) zTest the context menu.rNrtrYztkinter.Menur*)ra)�labelZcommandN)rrr rrrO�x�yZcontext_menu_eventrZadd_commandrRrTZrmenu_specsZassert_any_callr )rr?ryZ mock_MenuZ mock_menuZ mock_eventr��datarrr� test_rmenu�s zExpandingButtonTest.test_rmenuN) r4r5r6r7rrrrrMrr�r�r�r�rrrrrq1s (rq�__main__r")� verbosity)%r7�textwraprZtkinterrrZunittestZ unittest.mockrrrrr Ztest.supportr Zidlelib.configrZidlelib.squeezerrr rZidlelibrZidlelib.textviewrZidlelib.tooltiprZidlelib.pyshellrrPrZTestCaserr8rqr4rlrrrr�s(/`*