o
    Df                     @   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   d/home/ubuntu/webapp/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    