
    -Sh                         d Z ddlmZ ddlmZ ddlmZ ddlZddlZddlmZ ddl	m
Z
 ddl	mZ  G d	 d
ej                  Zd Zedk    r ej                     dS dS )zTests for augment.rename.    )absolute_import)division)print_functionN)rename)scope)
test_utilsc                   J    e Zd Zd Zd Zd Zd Zd Zd Zd Z	d Z
d	 Zd
 ZdS )
RenameTestc                 h   d}t          j        |          }|                     t          j        |dd                     |                     |t          j        d                     t          j        |          }|                     t          j        |dd                     |                     |t          j        d                     t          j        |          }|                     t          j        |dd                     |                     |t          j        |                     d S )Nz$import aaa.bbb.ccc
aaa.bbb.ccc.foo()aaa.bbbxxx.yyyz$import xxx.yyy.ccc
xxx.yyy.ccc.foo()aaa.bbb.ccczimport xxx.yyy
xxx.yyy.foo()bbbastparse
assertTruer   rename_externalcheckAstsEqualassertFalseselfsrcts      _/var/www/html/movieo_spanner_bot/venv/lib/python3.11/site-packages/pasta/augment/rename_test.pytest_rename_external_in_importz)RenameTest.test_rename_external_in_import   s    
1C	#AOOF*1iCCDDD39%LMMNNN	#AOOF*1mYGGHHH39%DEEFFF	#AV+Aui@@AAA39S>>*****    c                     d}t          j        |          }|                     t          j        |dd                     |                     |t          j        d                     d S )Nz#import aaa.bbb.ccc as ddd
ddd.foo()r   r   z#import xxx.yyy.ccc as ddd
ddd.foo()r   r   r   r   r   r   r   s      r   *test_rename_external_in_import_with_asnamez5RenameTest.test_rename_external_in_import_with_asname-   s\    
0C	#AOOF*1iCCDDD39%KLLMMMMMr   c                     d}t          j        |          }|                     t          j        |dd                     |                     |t          j        d                     d S )Nz import aaa, aaa.bbb, aaa.bbb.cccr   r   z import aaa, xxx.yyy, xxx.yyy.cccr   r   s      r   /test_rename_external_in_import_multiple_aliasesz:RenameTest.test_rename_external_in_import_multiple_aliases3   s\    
,C	#AOOF*1iCCDDD39%GHHIIIIIr   c                 h   d}t          j        |          }|                     t          j        |dd                     |                     |t          j        d                     t          j        |          }|                     t          j        |dd                     |                     |t          j        d                     t          j        |          }|                     t          j        |dd                     |                     |t          j        |                     d S )Nz%from aaa.bbb.ccc import ddd
ddd.foo()r   r   z%from xxx.yyy.ccc import ddd
ddd.foo()r   z!from xxx.yyy import ddd
ddd.foo()r   r   r   s      r   "test_rename_external_in_importfromz-RenameTest.test_rename_external_in_importfrom9   s    
2C	#AOOF*1iCCDDD39%MNNOOO	#AOOF*1mYGGHHH39%IJJKKK	#AV+Aui@@AAA39S>>*****r   c                     d}t          j        |          }|                     t          j        |dd                     |                     |t          j        d                     d S )Nz!from aaa.bbb import ccc
ccc.foo()r   r   zfrom xxx import yyy
yyy.foo()r   r   s      r   (test_rename_external_in_importfrom_aliasz3RenameTest.test_rename_external_in_importfrom_aliasG   s\    
.C	#AOOF*1mYGGHHH39%EFFGGGGGr   c                     d}t          j        |          }|                     t          j        |dd                     |                     |t          j        d                     d S )Nz(from aaa.bbb import ccc as abc
abc.foo()r   r   z$from xxx import yyy as abc
abc.foo()r   r   s      r   4test_rename_external_in_importfrom_alias_with_asnamez?RenameTest.test_rename_external_in_importfrom_alias_with_asnameM   s\    
5C	#AOOF*1mYGGHHH39%LMMNNNNNr   c                     d}t          j        |          }t          j        |          }|                     t          j        ||dd                     |                     |t          j        d                     d S )N	aaa.bbb()aaaxxxz	xxx.bbb())r   r   r   analyzer   r   _rename_readsr   r   r   r   scs       r   test_rename_reads_namez!RenameTest.test_rename_reads_nameS   sl    
C	#A	q		BOOF(Qu==>>>39[1122222r   c                     d}t          j        |          }t          j        |          }t	          j        ||dd           |                     |t          j        d                     d S )Nr*   r+   r   zxxx.yyy.bbb()r   r   r   r-   r   r.   r   r/   s       r   #test_rename_reads_name_as_attributez.RenameTest.test_rename_reads_name_as_attributeZ   s`    
C	#A	q		B
Qy11139_5566666r   c                     d}t          j        |          }t          j        |          }t	          j        ||dd           |                     |t          j        d                     d S )Naaa.bbb.ccc()r   r   zxxx.yyy.ccc()r3   r/   s       r   test_rename_reads_attributez&RenameTest.test_rename_reads_attributea   s`    
C	#A	q		B
Q	955539_5566666r   c                    d}t          j        |          }t          j        |          }t	          j        ||dd           t	          j        ||dd           |                     |t          j        |                     d S )Nr6   zaaa.bbb.ccc.dddr   zbbb.aaar3   r/   s       r   test_rename_reads_noopz!RenameTest.test_rename_reads_nooph   sv    
C	#A	q		B
Q 19===
Q	955539S>>*****r   N)__name__
__module____qualname__r   r    r"   r$   r&   r(   r1   r4   r7   r9    r   r   r
   r
      s        + + +N N NJ J J+ + +H H HO O O3 3 37 7 77 7 7+ + + + +r   r
   c                      t          j                    } |                     t          j        t                               | S )N)unittest	TestSuiteaddTests	makeSuiter
   )results    r   suiterD   q   s2    &//($Z00111	-r   __main__)__doc__
__future__r   r   r   r   r?   pasta.augmentr   
pasta.baser   r   TestCaser
   rD   r:   mainr=   r   r   <module>rL      s      ' & & & & &       % % % % % % 



                    ! ! ! ! ! !Q+ Q+ Q+ Q+ Q+$ Q+ Q+ Q+h  
 z(-///// r   