U
    ^g%i                     @   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Zprimary_keythis is some tablecomment)r   r   r   r   _fixturer
   selfm1m2Zdiffs r   j/var/www/html/api-medvista/venv/lib/python3.8/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   Z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Zexisting_nullableZexisting_typeZ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'   r*   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)   r*   r   r   r   r   test_alter_column_comment   s>    z2AutogenerateCommentsTest.test_alter_column_commentN)__name__
__module____qualname__Z__backend____requires__r   r!   r"   r#   r&   r,   r-   r.   r   r   r   r   r      s   ((r   N)Z
sqlalchemyr   r   r   r   r   Z_autogen_fixturesr   Ztestingr
   r   r   r   r   r   r   r   <module>   s   