B � fxN�@s�dZddlmZddlmZmZddlZddlmZm Z m Z m Z m Z ddl mZddlmZddlmZmZmZdd lmZdd lmZdd lmZdd lmZe jZd 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 �Zd d �Z dd�Z dd�Z dS)�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 |d d�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 �Zd d �Zdd�Z dd�Z dd�Z dd�Z dd�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:gs  zSqueezerTest.make_text_widgetcCs6t�|||�}t�||||�|�tj||||�dS)N)r Z GetOptionZ SetOptionr)rZ configTypeZsectionZoption�value�prev_valrrr� set_idleconf_option_with_cleanuposz-SqueezerTest.set_idleconf_option_with_cleanupc Csz|��}|�|�}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 �d S) 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 �|j t�d �}|�|d �|�|�dd �d�|�t |j �d�|�|j d j d�|j d j t�d �|�|�dd �d �|�t |j �d �qWdS)z$Test the squeeze_current_text event.)rZrWT)r;r.)rOz1.0z SOME TEXT �insertr]z SOME 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 �|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.)rOz1.0z SOME TEXT ZTAGr`r]z SOME 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_reloadcCs t��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 d d �Z dd�Z dd�Z dS)�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�d nd }|� ||� ��|�|jd �|j|ttd�|jdd }|� d|���dS)z1Test the simplest creation of an ExpandingButton.rN�TAGSrYz50 linesrz1.0r]r(zz �2�3r$)Z hover_delayrz right-clickN)rrr<rr rrdZmasterrjZcgetr^ZassertInZbindrZisAquaTkrRrSr � call_args�lower)rZ MockHovertipr?r=�expandingbuttonZright_button_codeZ tooltip_textrrrrM?szExpandingButtonTest.test_initcCs�|��}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.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 _ |j t �d �}Wd QRX|�|d �|�|j�dd �d �td��$}d|j _ d|j _ |j t �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_dangerousrjZ is_dangerousrr{r|rZ askokcancelrOZaskyesnorerrr^)rr?rryr=Z mock_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�s zExpandingButtonTest.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�s zExpandingButtonTest.test_viewc Cs�|��}tddd|�}td��j}t�}||_t�}d|_d|_|j|d�|�|j j t |j ��x"|j D]^}}|j j |td�qhWWdQRXdS) zTest the context menu.rNrtrYz tkinter.Menur*)ra)�labelZcommandN)rrr rrrO�x�yZcontext_menu_eventrZ add_commandrRrTZ rmenu_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 Z test.supportr Zidlelib.configr Zidlelib.squeezerr r rZidlelibrZidlelib.textviewrZidlelib.tooltiprZidlelib.pyshellrrPrZTestCaserr8rqr4rlrrrr�s(       /`*