관리-도구
편집 파일: test_squeezer.cpython-36.pyc
3 � feU � @ 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 )� )� namedtuple)�Text�TkN)�Mock�NonCallableMagicMock�patch�sentinel�ANY)�requires)�idleConf)�count_lines_with_wrapping�ExpandingButton�Squeezer)�macosx)� view_text)�Hovertip)�PyShellc s0 t d� t� � � j� � fdd�}| j|� � S )z*Helper for tests: Create a root Tk object.Zguic s � j � � j� d S )N)Zupdate_idletasksZdestroy� )�rootr �D/opt/alt/python36/lib64/python3.6/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 @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )�CountLinesTestz1Tests for the count_lines_with_wrapping function.c C s | j |t|||��S )N)� assertEqualr )�self�expected�text� linewidth�tabwidthr r r �check# s zCountLinesTest.checkc C s | j 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 | j 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 | j 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_line1 s z.CountLinesTest.test_count_ends_with_empty_linec C s | j td�d� dS )z Test with several lines of text.z1 2 3 � N)r r )r r r r �test_count_several_lines5 s z'CountLinesTest.test_count_several_linesc C s | j ddddd� | j ddddd� | j ddddd� | j ddddd� | j ddddd� | j ddd dd� | j d dd dd� | j ddd dd� | j ddddd� | j ddddd� | j ddddd� | j ddd dd� | j dd d dd� | j dd ddd� dS ) z-Test with various tab widths and line widths.r% � � � )r r r r r# r' � � � � � � � Nr) z z z z z z z z z z z z z )r )r r r r �test_tab_width9 s z CountLinesTest.test_tab_widthN) �__name__� __module__� __qualname__�__doc__r r" r$ r& r( r3 r r r r r ! s r c @ s� e Zd ZdZdd� Zd%dd�Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zed d �Zeejejejejfd d ��Zed!d"� �Zd#d$� ZdS )&� SqueezerTestz Tests for the Squeezer class.c C s t � }t|_|S )z$Create a mock EditorWindow instance.)r r � __class__)r �editwinr r r �make_mock_editor_windowS s z$SqueezerTest.make_mock_editor_windowNc C s |dkr| j � }t|�S )z<Create an actual Squeezer instance with a mock EditorWindow.N)r; r )r Z editor_windowr r r �make_squeezer_instance[ s z#SqueezerTest.make_squeezer_instancec C s$ x dD ]\}}| j ||d� qW d S )z�Test Squeezer.count_lines() with various inputs. This checks that Squeezer.count_lines() calls the count_lines_with_wrapping() function with the appropriate parameters. r+ �P r% �O r* r' �x )r r N�r+ r= �r% r> �r* r= �r' r? )r@ rA rB rC )�_test_count_lines_helper)r r r r r r �test_count_linesa s z SqueezerTest.test_count_linesc s� d� d}d}||j _|� d|| |jj_||dd���fdd�|j_t� }� fdd �}||j_td |�}|j� | j |j � dS ) z<Prepare a mock EditorWindow object for Squeezer.count_lines.r- r# r% N)ZborderZpadxZfontc s � | S )Nr )Zopt)� text_optsr r �<lambda>| s zDSqueezerTest._prepare_mock_editwin_for_count_lines.<locals>.<lambda>c s | dkr � S t d��d S )N�0z%measure should only be called on '0'!)� ValueError)�char)� CHAR_WIDTHr r �measure� s zCSqueezerTest._prepare_mock_editwin_for_count_lines.<locals>.measurezidlelib.squeezer.Font) Zget_tk_tabwidth� return_valuer Z winfo_width�cgetr rL r �startr �stop) r r: r r Z BORDER_WIDTHZ PADDING_WIDTHZ mock_fontrL Zpatcherr )rK rF r �%_prepare_mock_editwin_for_count_linesk s z2SqueezerTest._prepare_mock_editwin_for_count_linesc C sh | j � }| j|||� | j|�}ttd�}d}td|��& | j|j|�t� |j|||� W dQ R X dS )z Helper for test_count_lines.)rM �TEXTz*idlelib.squeezer.count_lines_with_wrappingN) r; rQ r< r �SENTINEL_VALUEr �assertIs� count_lines�assert_called_with)r r r r: �squeezerZmock_count_linesr r r r rD � s z%SqueezerTest._test_count_lines_helperc C s2 | j � }| j|�}| j|j|� | j|jg � dS )z(Test the creation of Squeezer instances.N)r; r< rT r: r �expandingbuttons)r r: rW r r r � test_init� s zSqueezerTest.test_initc C s� | j � }xxdddd dd gD ]`}ttd� |_}| j|�}| j|jj|f �t� | j|jd� |j|f � | jt |j �d � q W d S ) z@Test Squeezer's overriding of the EditorWindow's write() method.r! rR z LONG TEXTi� z MANY_LINES �d )rM r% r N) r; r rS �writer< r r: � call_countrV �lenrX )r r: r � orig_writerW r r r �test_write_no_tags� s z SqueezerTest.test_write_no_tagsc C s� x~dddd dd gD ]f}| j � }t|j_|j}| j|�}| j|jj|d�t� | j|jd� |j|d� | jt |j �d � qW d S ) z@Test Squeezer's overriding of the EditorWindow's write() method.r! rR z LONG TEXTi� z MANY_LINES rZ �stderrr% r N) r; rS r[ rM r< r r: r\ rV r] rX )r r r: r^ rW r r r �test_write_not_stdout� s z"SqueezerTest.test_write_not_stdoutc C s� | j � }| j|ddd� xndD ]f}ttd� |_}| j|�}d|_| j|jj|d�t� | j|j d � |j |d� | jt |j �d � q W xnd d d d gD ]Z}ttd� |_}| j|�}d|_| j|jj|d�d� | j|j d � | jt |j �d � q�W dS )z@Test Squeezer's overriding of the EditorWindow's write() method.r= r* )r r r! rR )rM �2 �stdoutr% r z LONG TEXTi� z MANY_LINES rZ N)r! rR ) r; rQ r rS r[ r< �auto_squeeze_min_linesr r: r\ rV r] rX )r r: r r^ rW r r r �test_write_stdout� s& z SqueezerTest.test_write_stdoutc C s� t | �}t|�}|jdd� | j� }||_| j|�}d|_tdd�|_|j dd d� | j |j dd�d � | j t |j �d � d S ) zATest that the auto-squeezing creates an ExpandingButton properly.�iomarkz1.0r, r1 )rM zTEXT rc �end� r% N)r r �mark_setr; r r< rd r rU r[ r �getr] rX )r r � text_widgetr: rW r r r �test_auto_squeeze� s z SqueezerTest.test_auto_squeezec C s, t | �}�x dD �]}t|�}|jdd� | j� }| |_|j_| j|�}tdd�|_ |j dd|� |jdd� | j |j dd �d � | j t |j�d � |jt� d �}| j |d � | j |j dd �d� | j t |j�d� | j |jd jd� |jd jt� d � | j |j dd �d � | j t |j�d � qW dS )z$Test the squeeze_current_text event.rc r` rf z1.0r1 )rM z SOME TEXT �insertrg z SOME TEXT r )�event�breakz r% z SOME TEXTN)rc r` )r r ri r; r �per�bottomr< r rU rm r rj r] rX �squeeze_current_text_event�s�expand)r r Ztag_namerk r: rW �retvalr r r � test_squeeze_current_text_event� s( z,SqueezerTest.test_squeeze_current_text_eventc C s� t | �}t|�}|jdd� | j� }| |_|j_| j|�}tdd�|_ |j ddd� |jdd� | j |j dd�d � | j t |j�d � |jt� d �}| j |d � | j |j dd�d � | j t |j�d � d S )z@Test that the event doesn't squeeze text without a relevant tag.rf z1.0r1 )rM z SOME TEXT ZTAGrm rg z SOME TEXT r )rn ro N)r r ri r; r rp rq r< r rU rm r rj r] rX rr )r r rk r: rW ru r r r �/test_squeeze_current_text_event_no_allowed_tags s z<SqueezerTest.test_squeeze_current_text_event_no_allowed_tagsc C s t | �}t|�}|jdd� | j� }| |_|j_| j|�}tdd�|_ |j ddd� |jdd� |j t� d� | j t |j�d � |j dd d� |jdd� |j t� d�}| j |d � | j |jdd �d � | j t |j�d� | j|j|jd d|jd �� dS )z2Test squeezing text before existing squeezed text.rf z1.0r1 )rM z SOME TEXT rc rm )rn r% z MORE STUFF ro rg z r# r �<N)r r ri r; r rp rq r< r rU rm rr r r] rX rj � assertTrueZcompare)r r rk r: rW ru r r r �/test_squeeze_text_before_existing_squeezed_text s* z<SqueezerTest.test_squeeze_text_before_existing_squeezed_text�GetOptionSignaturez:configType section option default type warn_on_default rawc C s | j |||||||�S )N)r{ )�clsZ configTypeZsectionZoption�default�typeZwarn_on_default�rawr r r � _make_sig> s zSqueezerTest._make_sigc C s |dd � \}}| j ||�S )Nr# ���)r� )r| Z mock_call_obj�args�kwargsr r r �get_GetOption_signatureF s z$SqueezerTest.get_GetOption_signaturec C s8 | j tjt� tjdddd� tj� | jtjd� dS )z Test the reload() class-method.�mainr zauto-squeeze-min-linesZ42�* N)ZassertIsInstancer rd �intr Z SetOption�reloadr )r r r r � test_reloadK s zSqueezerTest.test_reload)N)r4 r5 r6 r7 r; r< rE rQ rD rY r_ ra re rl rv rw rz r r{ � classmethodr Z NOT_GIVENr� r� r� r r r r r8 Q 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.rb )r r r r: r rd )r r rW r r r �make_mock_squeezerW s z&ExpandingButtonTest.make_mock_squeezerzidlelib.squeezer.Hovertip)�autospecc C s� | j � }|jj}tddd|�}| j|jd� | j|j|� | jd|jd�k� | j|j dd�d� | j d |j � � d t j � r�d nd }| j ||j � � | j|jd � |j|ttd� |jd d }| j d|j� � dS )z1Test the simplest creation of an ExpandingButton.rR �TAGSrb z50 linesr z1.0rg rh z<Double-Button-1>z <Button-%s>�2�3r% )Z hover_delayr z right-clickN)r� r: r r r rs Zmasterry rN rj ZassertInZbindr ZisAquaTkr\ rV r � call_args�lower)r Z MockHovertiprW rk �expandingbuttonZright_button_codeZ tooltip_textr r r rY a s z ExpandingButtonTest.test_initc C s� | j � }tddd|�}|j}|jd|d� |j|_|jt� d�}| j|d� | j|jdd�d � |j d �}| j|jd|�d� | j|j dd�d|f� | j|j j j d � |j j j|� dS ) zTest the expand event.rR r� rb z1.0)�window)rn Nrg zTEXT zend-1cr% )r� r r � window_create� base_textrt r r rj �indexZ tag_nextrangerX �remover\ rV )r rW r� rk ru Ztext_end_indexr r r � test_expand~ s z ExpandingButtonTest.test_expandc C s� | j � }dd }t|dd|�}|j� | j|j� |j}|jd|d� |j|_td��$}d |j _ d |j _ |j t � d �}W d Q R X | j|d � | j|jjdd �d� td��$}d|j _ d|j _ |j t � d �}W d Q R X | j|d � | j|jjdd �|� d S )z@Test that expanding very long output asks user for confirmation.�ar- r, r� rb z1.0)r� z idlelib.squeezer.tkMessageBoxF)rn Nro zend-1cr! Ti�� )r� r Zset_is_dangerousry Z is_dangerousr r� r� r Z askokcancelrM Zaskyesnort r r rj )r rW r r� rk Z mock_msgboxru r r r � test_expand_dangerous_oupput� s( z0ExpandingButtonTest.test_expand_dangerous_oupputc C sp | j � }tddd|�}t� |_t� |_|jt� d�}| j|d� | j|jjd� | j|jjd� |jjd� dS )zTest the copy event.rR r� rb )rn Nr% ) r� r r Zclipboard_clearZclipboard_append�copyr r\ rV )r rW r� ru r r r � test_copy� s z ExpandingButtonTest.test_copyc C sl | j � }tddd|�}t� |_tdtd��8}|jt� d� | j|jd� | j|j d d d� W d Q R X d S ) zTest the view event.rR r� rb zidlelib.squeezer.view_text)r� )rn r% r r# N) r� r r Z selection_ownr r Zviewr r\ r� )r rW r� Zmock_view_textr r r � test_view� s z ExpandingButtonTest.test_viewc C s� | j � }tddd|�}td��j}t� }||_t� }d|_d|_|j|d� | j|j j t |j �� x"|j D ]^}}|j j |td� qhW W dQ R X dS ) zTest the context menu.rR r� rb z tkinter.Menur- )rn )�labelZcommandN)r� r r r rM �x�yZcontext_menu_eventr Z add_commandr\ r] Z rmenu_specsZassert_any_callr )r rW r� Z mock_MenuZ mock_menuZ mock_eventr� �datar r r � test_rmenu� s z ExpandingButtonTest.test_rmenuN) r4 r5 r6 r7 r� r r rY r� r� r� r� r� r r r r r� S s )r� �__main__r# )� verbosity)$� collectionsr 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 rS r ZTestCaser r8 r� r4 r� r r r r �<module> s( 0 *