XylotrechusZ
3
��2aeU � @ 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 )NZguic s � j � � j� d S )N)Zupdate_idletasksZdestroy� )�rootr �2/opt/alt/python36/lib64/python3.6/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 Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�CountLinesTestc C s | j |t|||��S )N)� assertEqualr )�self�expected�text� linewidth�tabwidthr r r �check# s zCountLinesTest.checkc C s | j td�d� d S )N� r )r r )r r r r �test_count_empty) s z CountLinesTest.test_count_emptyc C s | j td�d� d S )Nz
text� )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� d S )Nztext
� )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� d S )Nz1
2
3
� )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 ddd dd� | j dd ddd� d S ) Nr% � � � )r r r r r# r' � �
� � � � �
r) 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__r r" r$ r& r( r3 r r r r r ! s r c @ s� e Zd Zdd� Zd$dd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zed d �Zeejejejejfd d ��Zed d!� �Zd"d#� ZdS )%� SqueezerTestc C s t � }t|_|S )N)r r � __class__)r �editwinr r r �make_mock_editor_windowS s z$SqueezerTest.make_mock_editor_windowNc C s |d kr| j � }t|�S )N)r: r )r Z
editor_windowr r r �make_squeezer_instance[ s z#SqueezerTest.make_squeezer_instancec C s$ x d
D ]\}}| j ||d� qW d S )Nr+ �P r% �O r* r' �x )r r �r+ r< �r% r= �r* r< �r' r> )r? r@ rA rB )�_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_||d d���fdd�|j_t� }� fdd�}||j_td |�}|j� | j |j
� d S )
Nr- r# r% )ZborderZpadxZfontc s � | S )Nr )�opt)� 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 r9 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 d Q R X d S )N)rM �TEXTz*idlelib.squeezer.count_lines_with_wrapping) r: rQ r; r �SENTINEL_VALUEr �assertIs� count_lines�assert_called_with)r r r r9 �squeezerZmock_count_linesr r r r rC � s
z%SqueezerTest._test_count_lines_helperc C s2 | j � }| j|�}| j|j|� | j|jg � d S )N)r: r; rT r9 r �expandingbuttons)r r9 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 )
Nr! rR z LONG TEXTi� z MANY_LINES
�d )rM r% r ) r: r rS �writer; r r9 �
call_countrV �lenrX )r r9 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 )
Nr! rR z LONG TEXTi� z MANY_LINES
rZ �stderrr% r ) r: rS r[ rM r; r r9 r\ rV r] rX )r r r9 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 d S )Nr< r* )r r r! rR )rM �2 �stdoutr% r z LONG TEXTi� z MANY_LINES
rZ )r! rR )
r: rQ r rS r[ r; �auto_squeeze_min_linesr r9 r\ rV r] rX )r r9 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 ) N�iomarkz1.0r, r1 )rM zTEXT
rc �end�
r% )r r �mark_setr: r r; rd r rU r[ r �getr] rX )r r � text_widgetr9 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 d S )Nrc r` rf z1.0r1 )rM z
SOME
TEXT
�insertrg z SOME
TEXT
r )�event�breakz
r% z SOME
TEXT)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 r9 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 )
Nrf z1.0r1 )rM z
SOME
TEXT
ZTAGrm rg z SOME
TEXT
r )rn ro )r r ri r: r rp rq r; r rU rm r rj r] rX rr )r r rk r9 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 �� d S )Nrf z1.0r1 )rM z
SOME
TEXT
rc rm )rn r% z MORE
STUFF
ro rg z
r# r �<)r r ri r: r rp rq r; r rU rm rr r r] rX rj �
assertTrueZcompare)r r rk r9 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
configType�section�option�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� d S )N�mainr zauto-squeeze-min-linesZ42�* )ZassertIsInstancer rd �intr Z SetOption�reloadr )r r r r � test_reloadK s zSqueezerTest.test_reload)N)r4 r5 r6 r: r; rD rQ rC rY r_ ra re rl rv rw rz r r{ � classmethodr Z NOT_GIVENr� r� r� r r r r r7 Q s,
" r7 c @ sP e Zd Zdd� Zeded�dd� �Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dS )�ExpandingButtonTestc C s$ t | �}t� }t|�|j_d|_|S )Nrb )r r r r9 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� � d S )NrR �TAGSrb z50 linesr z1.0rg rh z<Double-Button-1>z <Button-%s>�2�3r% )Z hover_delayr z right-click)r� r9 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|� d S ) NrR r� rb z1.0)�window)rn rg 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 )N�ar- r, r� rb z1.0)r� z idlelib.squeezer.tkMessageBoxF)rn ro 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� d S )NrR r� rb )rn r% ) 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 )
NrR r� rb zidlelib.squeezer.view_text)r� )rn r% r r# )
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 d Q R X d S )NrR r� rb z tkinter.Menur- )rn )�labelZcommand)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 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 r7 r� r4 r� r r r r �<module> s( 0 *