3 ��2aeU�@s�ddlmZddlmZmZddlZddlmZmZm Z m Z mZddlm Z ddlmZddlmZmZmZddlmZdd lmZdd lmZddlmZe jZdd �ZGdd�dej�ZGdd�dej�Z Gdd�dej�Z!e"dkr�ej#dd�dS)�)� namedtuple)�Text�TkN)�Mock�NonCallableMagicMock�patch�sentinel�ANY)�requires)�idleConf)�count_lines_with_wrapping�ExpandingButton�Squeezer)�macosx)� view_text)�Hovertip)�PyShellcs0td�t��j��fdd�}|j|��S)NZguics�j��j�dS)N)Zupdate_idletasksZdestroy�)�rootr�2/opt/alt/python36/lib64/python3.6/test_squeezer.py�cleanup_rootsz&get_test_tk_root..cleanup_root)r rZwithdraw� addCleanup)Z test_instancerr)rr�get_test_tk_roots rc@s<eZdZdd�Zdd�Zdd�Zdd�Zd d �Zdd�Zd S)�CountLinesTestcCs|j|t|||��S)N)�assertEqualr)�self�expected�text� linewidth�tabwidthrrr�check#szCountLinesTest.checkcCs|jtd�d�dS)N�r)rr)rrrr�test_count_empty)szCountLinesTest.test_count_emptycCs|jtd�d�dS)Nz text�)rr)rrrr�!test_count_begins_with_empty_line-sz0CountLinesTest.test_count_begins_with_empty_linecCs|jtd�d�dS)Nztext �)rr)rrrr�test_count_ends_with_empty_line1sz.CountLinesTest.test_count_ends_with_empty_linecCs|jtd�d�dS)Nz1 2 3 �)rr)rrrr�test_count_several_lines5sz'CountLinesTest.test_count_several_linescCs|jddddd�|jddddd�|jddddd�|jddddd�|jddddd�|jddd dd�|jd dd dd�|jddd dd�|jddddd�|jddddd�|jddddd�|jddd dd�|jddddd�|jddddd�dS)Nr%� ��)rrrrr#r'�� r)z z z z z z z z z z z z z )r )rrrr�test_tab_width9szCountLinesTest.test_tab_widthN) �__name__� __module__�__qualname__r r"r$r&r(r3rrrrr!src@s�eZdZdd�Zd$dd�Zdd�Zdd �Zd d�Zdd �Zdd�Z dd�Z dd�Zdd�Zdd�Z dd�Zdd�Zedd�Zeejejejejfdd��Zed d!��Zd"d#�ZdS)%�SqueezerTestcCst�}t|_|S)N)rr� __class__)r�editwinrrr�make_mock_editor_windowSsz$SqueezerTest.make_mock_editor_windowNcCs|dkr|j�}t|�S)N)r:r)rZ editor_windowrrr�make_squeezer_instance[sz#SqueezerTest.make_squeezer_instancecCs$xd D]\}}|j||d�qWdS)Nr+�Pr%�Or*r'�x)rr�r+r<�r%r=�r*r<�r'r>)r?r@rArB)�_test_count_lines_helper)rrrrrr�test_count_linesaszSqueezerTest.test_count_linesc s�d�d}d}||j_|�d|||jj_||dd��fdd�|j_t�}�fdd�}||j_td |�}|j�|j |j �dS) Nr-r#r%)ZborderZpadxZfontcs�|S)Nr)�opt)� text_optsrr�|szDSqueezerTest._prepare_mock_editwin_for_count_lines..cs|dkr�Std��dS)N�0z%measure should only be called on '0'!)� ValueError)�char)� CHAR_WIDTHrr�measure�szCSqueezerTest._prepare_mock_editwin_for_count_lines..measurezidlelib.squeezer.Font)Zget_tk_tabwidth�return_valuerZwinfo_width�cgetrrLr�startr�stop) rr9rrZBORDER_WIDTHZ PADDING_WIDTHZ mock_fontrLZpatcherr)rKrFr�%_prepare_mock_editwin_for_count_linesks z2SqueezerTest._prepare_mock_editwin_for_count_linescCsh|j�}|j|||�|j|�}ttd�}d}td|��&|j|j|�t�|j|||�WdQRXdS)N)rM�TEXTz*idlelib.squeezer.count_lines_with_wrapping) r:rQr;r�SENTINEL_VALUEr�assertIs�count_lines�assert_called_with)rrrr9�squeezerZmock_count_linesrrrrrC�s z%SqueezerTest._test_count_lines_helpercCs2|j�}|j|�}|j|j|�|j|jg�dS)N)r:r;rTr9r�expandingbuttons)rr9rWrrr� test_init�s zSqueezerTest.test_initcCs�|j�}xxddddddgD]`}ttd�|_}|j|�}|j|jj|f�t�|j|jd�|j|f�|jt |j �d �qWdS) Nr!rRz LONG TEXTi�zMANY_LINES �d)rMr%r)r:rrS�writer;rr9� call_countrV�lenrX)rr9r� orig_writerWrrr�test_write_no_tags�s zSqueezerTest.test_write_no_tagscCs�x~ddddddgD]f}|j�}t|j_|j}|j|�}|j|jj|d�t�|j|jd�|j|d�|jt |j �d �qWdS) Nr!rRz LONG TEXTi�zMANY_LINES rZ�stderrr%r)r:rSr[rMr;rr9r\rVr]rX)rrr9r^rWrrr�test_write_not_stdout�s z"SqueezerTest.test_write_not_stdoutcCs�|j�}|j|ddd�xndD]f}ttd�|_}|j|�}d|_|j|jj|d�t�|j|j d �|j |d�|jt|j�d �qWxnddd dgD]Z}ttd�|_}|j|�}d|_|j|jj|d�d�|j|j d �|jt|j�d �q�WdS)Nr<r*)rrr!rR)rM�2�stdoutr%rz LONG TEXTi�zMANY_LINES rZ)r!rR) r:rQrrSr[r;�auto_squeeze_min_linesrr9r\rVr]rX)rr9rr^rWrrr�test_write_stdout�s& zSqueezerTest.test_write_stdoutcCs�t|�}t|�}|jdd�|j�}||_|j|�}d|_tdd�|_|j ddd�|j |jdd�d �|j t|j �d �dS)N�iomarkz1.0r,r1)rMzTEXT rc�end� r%)rr�mark_setr:rr;rdrrUr[r�getr]rX)rr�text_widgetr9rWrrr�test_auto_squeeze�s zSqueezerTest.test_auto_squeezecCs,t|�}�xdD�]}t|�}|jdd�|j�}||_|j_|j|�}tdd�|_ |j dd|�|jdd�|j|jdd �d �|jt |j�d�|jt�d�}|j|d �|j|jdd �d�|jt |j�d�|j|jdjd�|jdjt�d�|j|jdd �d �|jt |j�d�qWdS)Nrcr`rfz1.0r1)rMz SOME TEXT �insertrgzSOME TEXT r)�event�breakz r%z SOME TEXT)rcr`)rrrir:r�per�bottomr;rrUrmrrjr]rX�squeeze_current_text_event�s�expand)rrZtag_namerkr9rW�retvalrrr�test_squeeze_current_text_event�s( z,SqueezerTest.test_squeeze_current_text_eventcCs�t|�}t|�}|jdd�|j�}||_|j_|j|�}tdd�|_ |j ddd�|jdd�|j|jdd�d �|jt |j�d �|jt�d�}|j|d�|j|jdd�d �|jt |j�d �dS) Nrfz1.0r1)rMz SOME TEXT ZTAGrmrgzSOME TEXT r)rnro)rrrir:rrprqr;rrUrmrrjr]rXrr)rrrkr9rWrurrr�/test_squeeze_current_text_event_no_allowed_tagss zszSqueezerTest._make_sigcCs|dd�\}}|j||�S)Nr#��)r�)r|Z mock_call_obj�args�kwargsrrr�get_GetOption_signatureFsz$SqueezerTest.get_GetOption_signaturecCs8|jtjt�tjdddd�tj�|jtjd�dS)N�mainrzauto-squeeze-min-linesZ42�*)ZassertIsInstancerrd�intrZ SetOption�reloadr)rrrr�test_reloadKszSqueezerTest.test_reload)N)r4r5r6r:r;rDrQrCrYr_rarerlrvrwrzrr{�classmethodrZ NOT_GIVENr�r�r�rrrrr7Qs, "r7c@sPeZdZdd�Zeded�dd��Zdd�Zd d �Zdd�Z d d�Z dd�ZdS)�ExpandingButtonTestcCs$t|�}t�}t|�|j_d|_|S)Nrb)rrrr9rrd)rrrWrrr�make_mock_squeezerWs z&ExpandingButtonTest.make_mock_squeezerzidlelib.squeezer.Hovertip)�autospeccCs�|j�}|jj}tddd|�}|j|jd�|j|j|�|jd|jd�k�|j|j dd�d�|j d |j��d tj �r�dnd}|j ||j��|j|jd �|j|ttd�|jdd }|j d|j��dS)NrR�TAGSrbz50 linesrz1.0rgrhzz�2�3r%)Zhover_delayrzright-click)r�r9rr rrsZmasterryrNrjZassertInZbindrZisAquaTkr\rVr � call_args�lower)rZMockHovertiprWrk�expandingbuttonZright_button_codeZtooltip_textrrrrYaszExpandingButtonTest.test_initcCs�|j�}tddd|�}|j}|jd|d�|j|_|jt�d�}|j|d�|j|jdd�d�|j d �}|j|jd|�d�|j|j dd�d|f�|j|jjj d �|jjj|�dS)NrRr�rbz1.0)�window)rnrgzTEXT zend-1cr%)r�r r� window_create� base_textrtrrrj�indexZ tag_nextrangerX�remover\rV)rrWr�rkruZtext_end_indexrrr�test_expand~s zExpandingButtonTest.test_expandcCs�|j�}dd}t|dd|�}|j�|j|j�|j}|jd|d�|j|_td��$}d |j _ d |j_ |jt �d �}WdQRX|j|d�|j|jjdd�d �td��$}d|j _ d|j_ |jt �d �}WdQRX|j|d�|j|jjdd�|�dS)N�ar-r,r�rbz1.0)r�zidlelib.squeezer.tkMessageBoxF)rnrozend-1cr!Ti��)r�r Zset_is_dangerousryZis_dangerousrr�r�rZaskokcancelrMZaskyesnortrrrj)rrWrr�rkZmock_msgboxrurrr�test_expand_dangerous_oupput�s( z0ExpandingButtonTest.test_expand_dangerous_oupputcCsp|j�}tddd|�}t�|_t�|_|jt�d�}|j|d�|j|jjd�|j|jjd�|jjd�dS)NrRr�rb)rnr%) r�r rZclipboard_clearZclipboard_append�copyrr\rV)rrWr�rurrr� test_copy�szExpandingButtonTest.test_copyc Csl|j�}tddd|�}t�|_tdtd��8}|jt�d�|j|jd�|j|j dd d�WdQRXdS) NrRr�rbzidlelib.squeezer.view_text)r�)rnr%rr#) r�r rZ selection_ownrrZviewrr\r�)rrWr�Zmock_view_textrrr� test_view�szExpandingButtonTest.test_viewcCs�|j�}tddd|�}td��j}t�}||_t�}d|_d|_|j|d�|j|j j t|j��x"|jD]^}}|j j |td�qhWWdQRXdS)NrRr�rbztkinter.Menur-)rn)�labelZcommand)r�r rrrM�x�yZcontext_menu_eventrZadd_commandr\r]Zrmenu_specsZassert_any_callr )rrWr�Z mock_MenuZ mock_menuZ mock_eventr��datarrr� test_rmenu�s zExpandingButtonTest.test_rmenuN)r4r5r6r�rrrYr�r�r�r�r�rrrrr�Ss )r��__main__r#)� verbosity)$�collectionsrZtkinterrrZunittestZ unittest.mockrrrrr Ztest.supportr Zidlelib.configrZidlelib.squeezerrr rZidlelibrZidlelib.textviewrZidlelib.tooltiprZidlelib.pyshellrrSrZTestCaserr7r�r4r�rrrr�s(0*