B
� f�: � @ s d Z ddlmZ ddlmZ ddlZddlmZ ddlmZ ddl m
Z
m Z ddlm Z dd l
mZ ejjZe �d
�e �d
�e �d
�e �d
�d �Zd Zd
d� Zdd� ZG dd� dej�ZG dd� dej�ZG dd� dej�ZG dd� dej�Ze dk�rej dd� dS ) z Test colorizer, coverage 93%.� )� colorizer)�requiresN)�mock)�partial)�Tk�Text)�config)�
Percolator� )�main� highlight�keys�
extensionsa+ if True: int ('1') # keyword, builtin, string, comment
elif False: print(0) # 'string' in comment
else: float(None) # if in comment
if iF + If + IF: 'keyword matching must respect case'
if'': x or'' # valid string-keyword no-space combinations
async def f(): await g()
'x', '''x''', "x", """x"""
c C s t tj_d S )N)�testcfgr �idleConf�userCfg� r r �E/opt/alt/python37/lib64/python3.7/idlelib/idle_test/test_colorizer.py� setUpModule! s r c C s t tj_d S )N)�usercfgr r r r r r r �tearDownModule% s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� FunctionTestc C s | � t�dd�d� d S )NZtest)�a�bZcdz(?Pa|b|cd))� assertEqualr �any)�selfr r r �test_any+ s zFunctionTest.test_anyc C s | � t�� � d S )N)�
assertTruer Zmake_pat)r r r r �
test_make_pat/ s zFunctionTest.test_make_patc C s� t j}| j}d}|�|�}||�� d d� |�||�� �}||�� d d� |�||�� �}||�� d d� |�||�� �}||�� d d � |�||�� �}||�� d d� d S )
Nz def f():
print("hello")
�KEYWORD�def�SYNC�
�BUILTIN�print�STRINGz"hello")r �progr �search� groupdict�end)r r' �eq�line�mr r r � test_prog3 s
zFunctionTest.test_progc C s: t j}|�d�}| �|� |�d�}| �|�d�d� d S )NZnospacez spacer )r �idprog�match� assertIsNoner �group)r r/ r- r r r � test_idprogB s
zFunctionTest.test_idprogN)�__name__�
__module__� __qualname__r r r. r3 r r r r r ) s r c @ s, e Zd Zedd� �Zedd� �Zdd� ZdS )�ColorConfigTestc C s* t d� t� }| _|�� t|�| _d S )N�gui)r r �root�withdrawr �text)�clsr9 r r r �
setUpClassL s zColorConfigTest.setUpClassc C s | ` | j�� | j�� | `d S )N)r; r9 �update_idletasks�destroy)r< r r r �
tearDownClassS s
z ColorConfigTest.tearDownClassc C sn | j }| j}t�|� ||d d� ||d d� ||d d� ||d d� ||d d � ||d
d� d S ) N�
backgroundz#ffffff�
foregroundz#000000ZselectbackgroundZgrayZselectforegroundZinsertbackgroundZblackZinactiveselectbackground)r; r r Z color_config)r r; r+ r r r �test_color_configZ s
z!ColorConfigTest.test_color_configN)r4 r5 r6 � classmethodr= r@ rC r r r r r7 J s r7 c @ sD e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d
� Zd d � Z d
S )� ColorDelegatorInstantiationTestc C s. t d� t� }| _|�� t|� }| _d S )Nr8 )r r r9 r: r r; )r< r9 r; r r r r= i s z*ColorDelegatorInstantiationTest.setUpClassc C s | ` | j�� | j�� | `d S )N)r; r9 r> r? )r< r r r r@ p s
z-ColorDelegatorInstantiationTest.tearDownClassc C s t �� | _d S )N)r �ColorDelegator�color)r r r r �setUpw s z%ColorDelegatorInstantiationTest.setUpc C s* | j �� | j�dd� | j �� | ` d S )Nz1.0r* )rG �closer; �delete�
resetcache)r r r r �tearDownz s
z(ColorDelegatorInstantiationTest.tearDownc C s | j }| �|tj� d S )N)rG �assertIsInstancer rF )r rG r r r � test_init� s z)ColorDelegatorInstantiationTest.test_initc C s: | j }| �|j� | �|j� | �|j� | �|j� d S )N)rG r1 �after_idr �allow_colorizing� assertFalse�
colorizing�stop_colorizing)r rG r r r �test_init_state� s
z/ColorDelegatorInstantiationTest.test_init_stateN)
r4 r5 r6 rD r= r@ rH rL rN rT r r r r rE g s rE c @ s� e Zd Zedd� �Zedd� �Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
e j �
ejd�dd� �Ze j �
ejd�dd� �Zdd� Zdd� Ze j �
ejd�dd� �Ze j �
ejd�dd � �Ze j �
ejd �e j �
ejd�d d � ��Zd S )!�ColorDelegatorTestc C s8 t d� t� }| _|�� t|� }| _t|�| _d S )Nr8 )r r r9 r: r r; r �
percolator)r< r9 r; r r r r= � s
z ColorDelegatorTest.setUpClassc C s0 | j j�� | ` | `| j�� | j�� | `d S )N)rV ZredirrI r; r9 r> r? )r< r r r r@ � s
z ColorDelegatorTest.tearDownClassc C s t �� | _| j�| j� d S )N)r rF rG rV Z insertfilter)r r r r rH � s
zColorDelegatorTest.setUpc C s8 | j �� | j�| j � | j�dd� | j �� | ` d S )Nz1.0r* )rG rI rV Z removefilterr; rJ rK )r r r r rL � s
zColorDelegatorTest.tearDownc C s: | j }| �|jtj� | �| jj�dd|j �d d� d S )N�after�info� �timer)
rG rM �delegater Z Delegatorr r9 �tk�callrO )r rG r r r �test_setdelegate� s z#ColorDelegatorTest.test_setdelegatec
C s| t tjjdd�}xf| jj�� D ]V\}}| j|d��< | �d|� | �d|� |dkrj| � |||�
� d�� W d Q R X q W d S )Nz IDLE Classic)�theme)�tagrA rB )r" �TODO)�element) r r r � GetHighlightrG �tagdefs�items�subTestZassertInr �lower)r r r` Zcolorsr r r �test_LoadTagDefs� s z#ColorDelegatorTest.test_LoadTagDefsc C s� | j }ttjjdd�}xx| jjD ]l}xfdD ]^}| j||d��F |dkr\| �|� ||�d� n"| �|� ||�||�
� d�| � W d Q R X q*W q W | �|� � d d � d S )
Nz IDLE Classic)r_ )rA rB )r` �plane)r" ra r
)rb �����Zsel) r; r r r rc rG rd rf r Ztag_cgetrg � tag_names)r r; r r` ri r r r �test_config_colors� s
&z%ColorDelegatorTest.test_config_colors� notify_rangec C sb | j }|�dd� | �|�dd�d� |�dd� |�dd� | �|�dd�d� |�d d
� d S ) N�insertZfooz1.0r* zfoo
z1.0+3cZbarbazz
foobarbaz
z1.3z1.3+6c)r; rn r �get�assert_called_with)r � mock_notifyr; r r r � test_insert� s z ColorDelegatorTest.test_insertc C s| | j }|�dd� | �|�dd�d� |�d� | �|�dd�d� |�d� |�dd � | �|�dd�d
� |�d� d S ) Nrn Z abcdefghiz1.0r* z
abcdefghi
z1.7z abcdefgi
z1.3z1.6zabcgi
)r; rn r ro rJ rp )r rq r; r r r � test_delete� s
z ColorDelegatorTest.test_deletec C s | j }| j}| j}|j}|| jj�dd|�d d� | �|j� | �|j � | �
|j � d|_|� dd� | �|j � ||j|� |�
|� d |_|� dd� | �
|j � | �|j� || jj�dd|j�d d� | �|j|� |�
|j� d |_d |_ |� d
d � | �|j� d S ) NrW rX rY rZ Tz1.0r* z1.0+3cFz1.4z1.4+10c)r; rG r rO r9 r\ r] rQ rR rS r rP rm � after_cancelZassertIsNotNone�assertNotEqualr1 )r r; rG r+ Zsave_idr r r �test_notify_range� s0
z$ColorDelegatorTest.test_notify_rangec C sP | j }| j}| �|j� | �|j� | �|j� || jj� dd|j
�d d� |� � | � |j
� | �|j� | �|j� | �|j� d|_|� � | � |j
� | �|j� | �|j� | �|j� |� � | � |j
� | �|j� | �|j� | �|j� d|_|� � || jj� dd|j
�d d� | �|j� | �|j� | �|j� d S )NrW rX rY rZ TF)
rG r rQ rR rS r rP r9 r\ r] rO Ztoggle_colorize_eventr1 )r rG r+ r r r �test_toggle_colorize_event s8 z-ColorDelegatorTest.test_toggle_colorize_event�recolorize_mainc C s | j }| j}| j}|�|j� |j}d |_|�� |�� ||_d|_|�� |�� d|_d|_ |�� |�� d|_ |�� | �
|j � | �
|j � |� � ||j
d� || jj�dd|j�d d� |�ddd � |�� | �
|j � | �
|j � |� � ||j
d
� | �|j� d S ) NFTrY rW rX rZ ra z1.0r* � )r; rG r rt rO r[ �
recolorizeZassert_not_calledrP rR rQ rS Z
assert_calledZ
call_countr9 r\ r] �
tag_remover1 )r Z mock_recmainr; rG r+ Z
save_delegater r r �test_recolorize3 s< z"ColorDelegatorTest.test_recolorizec
C sz | j }| j}| j}|�dt� d}|�ddd� |�� x8|�� D ],}| j|d�� ||� |�d� W d Q R X qBW |�
ddd� |�� x8|D ]0\}}| j|d�� ||�|�|� W d Q R X q�W ||� dd�d� ||� d d�d
� ||� d d �d
� ||� dd �d� ||� dd �d� ||� dd�d� ||� dd�d� ||� dd�d� ||� dd�d� ||� dd�d� d S )Nrn ) )z1.0)r )z1.2r )z1.3)r )z1.7r )z1.9)r$ )z1.14)r& )z1.19)�COMMENT)z2.1)r )z2.18r )z2.25)r} )z3.6)r$ )z3.12)r )z3.21)r} )z4.0)r )z4.3r )z4.6r )z5.2)r& )z5.8)r )z5.10)r& )z6.0)r )z6.10)Z
DEFINITION)z6.11r )z7.0)r& )z7.4r )z7.5)r& )z7.12r )z7.14)r& )z1.55)r" )z2.50)r" )z3.34)r" ra z1.0r* )r` r )�indexr )z1.0z1.2r} z2.0)z2.22z2.43r" )z2.43z3.0r& )z4.17z4.53z7.0)z7.0z7.3z7.3)z7.5z7.12z7.12)z7.14z7.17z7.17)z7.19z7.26)z7.26z9.0) r; rG r rn �sourcer{ rx rk rf �
tag_ranges�tag_addZ
tag_nextrange) r rq r; rG r+ Zexpectedr` r~ Z
expected_tagsr r r �test_recolorize_main` s2 z'ColorDelegatorTest.test_recolorize_mainrz c
C s� | j }| j}|�dt� |�� |�dd� |�dd� |�dd� x8|jD ].}| j|d�� | �|� |�d� W d Q R X qLW |�
� x8|jD ].}| j|d�� | � |� |�d� W d Q R X q�W d S )Nrn ZERRORz1.0ra Zhit)r` r ) r; rG rn r rx r� rd rf ru r� Z removecolorsr )r rq Zmock_recolorizer; rG r` r r r �test_removecolors� s z$ColorDelegatorTest.test_removecolorsN)r4 r5 r6 rD r= r@ rH rL r^ rh rl r Zpatch�objectr rF rr rs rv rw r| r� r� r r r r rU � s $(-0rU �__main__ry )� verbosity) �__doc__Zidlelibr Z test.supportr Zunittestr � functoolsr Ztkinterr r r Zidlelib.percolatorr r r r ZIdleUserConfParserr r r r ZTestCaser r7 rE rU r4 r r r r r � s0
! '