o
    ‹åg‹  ã                   @   s‚   d dl mZ d dl mZ d dl mZ d dl mZ d dl mZ ddlmZ dd	lm	Z	 dd
lm
Z
 ddlmZ G dd„ deeƒZdS )é    )ÚColumn)ÚFloat)ÚMetaData)ÚString)ÚTableé   )ÚAutogenFixtureTesté   )Úeq_)Úmock)ÚTestBasec                   @   sT   e Zd ZdZdZdd„ Zdd„ Zdd„ Zd	d
„ Zdd„ Z	dd„ Z
dd„ Zdd„ ZdS )ÚAutogenerateCommentsTestT)Úcommentsc                 C   sb   t ƒ }t ƒ }td|tdtdƒdd�dd� td|tdtdƒdd�dd� |  ||¡}t|g ƒ d S )NÚ
some_tableÚtesté
   T©Úprimary_keyúthis is some table©Úcomment)r   r   r   r   Ú_fixturer
   ©ÚselfÚm1Úm2Údiffs© r   úg/var/www/html/flaskapp/venv/lib/python3.10/site-packages/alembic/testing/suite/test_autogen_comments.pyÚ%test_existing_table_comment_no_change   s    üüz>AutogenerateCommentsTest.test_existing_table_comment_no_changec                 C   sŒ   t ƒ }t ƒ }td|tdtdƒdd�ƒ td|tdtdƒdd�dd� |  ||¡}t|d d d	ƒ t|d d
 jdƒ t|d d d ƒ d S )Nr   r   r   Tr   r   r   r   Úadd_table_commentr   é   ©r   r   r   r   r   r
   r   r   r   r   r   Útest_add_table_comment(   s   üz/AutogenerateCommentsTest.test_add_table_commentc                 C   sz   t ƒ }t ƒ }td|tdtdƒdd�dd� td|tdtdƒdd�ƒ |  ||¡}t|d d d	ƒ t|d d
 jd ƒ d S )Nr   r   r   Tr   r   r   r   Úremove_table_commentr   r"   r   r   r   r   Útest_remove_table_comment;   s   üz2AutogenerateCommentsTest.test_remove_table_commentc                 C   s�   t ƒ }t ƒ }td|tdtdƒdd�dd� td|tdtdƒdd�dd� |  ||¡}t|d	 d	 d
ƒ t|d	 d jdƒ t|d	 d dƒ d S )Nr   r   r   Tr   r   r   zthis is also some tabler   r    r   r!   r"   r   r   r   r   Útest_alter_table_commentM   s$   üüz1AutogenerateCommentsTest.test_alter_table_commentc              	   C   sr   t ƒ }t ƒ }td|tdtdƒdd�tdtdd�ƒ td|tdtdƒdd�tdtdd�ƒ |  ||¡}t|g ƒ d S )	Nr   r   r   Tr   Úamountú
the amountr   )r   r   r   r   r   r   r
   r   r   r   r   Ú&test_existing_column_comment_no_changee   s    üüz?AutogenerateCommentsTest.test_existing_column_comment_no_changec              
   C   sŠ   t ƒ }t ƒ }td|tdtdƒdd�tdtƒƒ td|tdtdƒdd�tdtdd�ƒ |  ||¡}t|d	d dddtjd
dœd dfggƒ d S ©Nr   r   r   Tr   r'   r(   r   Úmodify_commentF©Úexisting_nullableÚexisting_typeÚexisting_server_default©	r   r   r   r   r   r   r
   r   ÚANYr   r   r   r   Útest_add_column_comment{   s>   üüýõÿÿþz0AutogenerateCommentsTest.test_add_column_commentc              
   C   sŠ   t ƒ }t ƒ }td|tdtdƒdd�tdtdd�ƒ td|tdtdƒdd�tdtƒƒ |  ||¡}t|d	d dddtjd
dœdd fggƒ d S r*   r0   r   r   r   r   Útest_remove_column_comment£   s>   üüýõÿÿþz3AutogenerateCommentsTest.test_remove_column_commentc              
   C   sŽ   t ƒ }t ƒ }td|tdtdƒdd�tdtdd�ƒ td|tdtdƒdd�tdtd	d�ƒ |  ||¡}t|d
d dddtjddœdd	fggƒ d S )Nr   r   r   Tr   r'   r(   r   zthe adjusted amountr+   Fr,   r0   r   r   r   r   Útest_alter_column_commentË   s>   üüýõÿÿþz2AutogenerateCommentsTest.test_alter_column_commentN)Ú__name__Ú
__module__Ú__qualname__Ú__backend__Ú__requires__r   r#   r%   r&   r)   r2   r3   r4   r   r   r   r   r      s    ((r   N)Ú
sqlalchemyr   r   r   r   r   Ú_autogen_fixturesr   Útestingr
   r   r   r   r   r   r   r   Ú<module>   s    