
    +h                         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ee      Zy)    )Column)Float)MetaData)String)Table   )AutogenFixtureTest   )eq_)mock)TestBasec                   D    e Zd ZdZdZd Zd Zd Zd Zd Z	d Z
d	 Zd
 Zy)AutogenerateCommentsTestT)commentsc           
          t               }t               }t        d|t        dt        d      d      d       t        d|t        dt        d      d      d       | j	                  ||      }t        |g        y )N
some_tabletest
   Tprimary_keythis is some tablecomment)r   r   r   r   _fixturer   selfm1m2diffss       ^/var/www/html/venv/lib/python3.12/site-packages/alembic/testing/suite/test_autogen_comments.py%test_existing_table_comment_no_changez>AutogenerateCommentsTest.test_existing_table_comment_no_change   so    ZZ66":48(		
 	66":48(		
 b"%E2    c           
      \   t               }t               }t        d|t        dt        d      d             t        d|t        dt        d      d      d       | j	                  ||      }t        |d   d   d	       t        |d   d
   j                  d       t        |d   d   d        y )Nr   r   r   Tr   r   r   r   add_table_commentr      r   r   r   r   r   r   r   r   s       r    test_add_table_commentz/AutogenerateCommentsTest.test_add_table_comment(   s    ZZlBvvbzt LM66":48(		
 b"%E!HQK,-E!HQK!56E!HQKr"   c           
      8   t               }t               }t        d|t        dt        d      d      d       t        d|t        dt        d      d             | j	                  ||      }t        |d   d   d	       t        |d   d
   j                  d        y )Nr   r   r   Tr   r   r   r   remove_table_commentr   r&   r   s       r    test_remove_table_commentz2AutogenerateCommentsTest.test_remove_table_comment;   s    ZZ66":48(		
 	lBvvbzt LMb"%E!HQK/0E!HQK&r"   c           
      `   t               }t               }t        d|t        dt        d      d      d       t        d|t        dt        d      d      d       | j	                  ||      }t        |d	   d	   d
       t        |d	   d   j                  d       t        |d	   d   d       y )Nr   r   r   Tr   r   r   zthis is also some tabler   r$   r   r%   r&   r   s       r    test_alter_table_commentz1AutogenerateCommentsTest.test_alter_table_commentM   s    ZZ66":48(		
 	66":48-		
 b"%E!HQK,-E!HQK!:;E!HQK-.r"   c           
      4   t               }t               }t        d|t        dt        d      d      t        dt        d             t        d|t        dt        d      d      t        dt        d             | j                  ||      }t        |g        y )	Nr   r   r   Tr   amount
the amountr   )r   r   r   r   r   r   r   r   s       r    &test_existing_column_comment_no_changez?AutogenerateCommentsTest.test_existing_column_comment_no_changee   s    ZZ66":488UL9		
 	66":488UL9		
 b"%E2r"   c                 f   t               }t               }t        d|t        dt        d      d      t        dt                     t        d|t        dt        d      d      t        dt        d             | j                  ||      }t        |d	d dddt        j                  d
dd dfgg       y 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   s       r    test_add_column_commentz0AutogenerateCommentsTest.test_add_column_comment{   s    ZZ66":488U#		
 	66":488UL9		
 b"% )$ 15-1XX7<
 $	
r"   c                 f   t               }t               }t        d|t        dt        d      d      t        dt        d             t        d|t        dt        d      d      t        dt                     | j                  ||      }t        |d	d dddt        j                  d
ddd fgg       y r2   r8   r   s       r    test_remove_column_commentz3AutogenerateCommentsTest.test_remove_column_comment   s    ZZ66":488UL9		
 	66":488U#		
 b"% )$ 15-1XX7<
 %	
r"   c                 j   t               }t               }t        d|t        dt        d      d      t        dt        d             t        d|t        dt        d      d      t        dt        d	             | j                  ||      }t        |d
d dddt        j                  dddd	fgg       y )Nr   r   r   Tr   r.   r/   r   zthe adjusted amountr3   Fr4   r8   r   s       r    test_alter_column_commentz2AutogenerateCommentsTest.test_alter_column_comment   s    ZZ66":488UL9		
 	66":488U,AB		
 b"% )$ 15-1XX7<
 %-	
r"   N)__name__
__module____qualname____backend____requires__r!   r'   r*   r,   r0   r:   r<   r>    r"   r    r   r      s6    K L,&'$/0,&
P&
P'
r"   r   N)
sqlalchemyr   r   r   r   r   _autogen_fixturesr	   testingr   r   r   r   rD   r"   r    <module>rH      s1         1   e
18 e
r"   