← Back
Editing: test_squeezer.cpython-37.pyc
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( / ` *
Save File
Cancel