
    ZeIi                         d dl 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 d	dlmZ d	dlmZ d	dlmZ  G d de	e          ZdS )    N)Column)Integer)MetaData)Table)sqla_compat   )AutogenFixtureTest   )testing)config)eq_)is_true)TestBasec                      e Zd ZdZdZd Zd Zd Zd Z e	j
        d ed	          f ed	          df ed	           edd
          f ed           ed          f edddd           edddd          f eddd           eddd          f          ej        j        d                         Zd Zd ZdS )AutogenerateIdentityTest)identity_columnsTc                    t                      }t                      }t          d|t          dt          j                             t          d|t          dt          j                  t          dt
          t          j        dd          d                     |                     ||          }t          |d	         d	         d
           t          |d	         d         d           t          |d	         d         j	        d           |d	         d         j
        }t          t          |t          j                             t          |j        d           t          |j        d           d S )Nuserotherid      start	incrementTprimary_keyr   
add_column   r
   )r   r   r   saTextr   Identity_fixturer   nameidentityr   
isinstancer   r   )selfm1m2diffsis        p/var/www/html/MCyber-Diagnostic/venv/lib/python3.11/site-packages/alembic/testing/suite/test_autogen_identity.pytest_add_identity_columnz1AutogenerateIdentityTest.test_add_identity_column   s7   ZZZZfb&"'223337BG$$!q111 	  	
	
 
	
 
	
 b"%%E!HQK&&&E!HQK   E!HQKd###!HQK 
1bk**+++AGQAK    c                 v   t                      }t                      }t          d|t          dt          t	          j        dd          d                     t          d|           |                     ||          }t          |d         d         d	           t          |d         d         d           |d         d         }t          |j        d           t          t          |j        t          j                             t          |j        j        d           t          |j        j        d           d S )
Nr   r   r   r
   r   Tr   r   remove_column)r   r   r   r   r    r"   r#   r   r$   r   r&   r%   r   r   )r'   r(   r)   r*   cs        r,   test_remove_identity_columnz4AutogenerateIdentityTest.test_remove_identity_column1   s   ZZZZ!q111 	  		
 		
 		
 	fbb"%%E!HQK)))E!HQK   !HQKAFD
1:r{33444AJa   AJ !$$$$$r.   c           
         t                      }t                      }||fD ]<}t          j        d          }t          d|t	          dt
          |                     =|                     ||          }t          |g            d S )Nr   r   r   r   )r   r    r"   r   r   r   r#   r   )r'   r(   r)   mid_r*   s         r,   test_no_change_identity_columnz7AutogenerateIdentityTest.test_no_change_identity_columnM   s~    ZZZZb 	9 	9A+A&&&C&!VD'3778888b"%%E2r.   c                 D   t                      }t                      }t          j        rddd}nddd}t          d|t	          dt
          t          j        d                               t          j        dddi|}t          d|t	          dt
          |                     |                     ||          }t          j
        j        d	k    rAt          t          |          d
           t          |d         d         d         d           d S t          |g            d S )NT)oracle_on_nulloracle_order)on_nullorderr   r   r   r4   r   oracler   r   modify_default )r   r   identity_has_dialect_kwargsr   r   r   r    r"   r#   r   dbr$   r   lenr   )r'   r(   r)   argsr6   r*   s         r,   test_dialect_kwargs_changesz4AutogenerateIdentityTest.test_dialect_kwargs_changesY   s
   ZZZZ2 	4&*DAADD#d33Dfb&w!0D0D0DEEFFFk***T**fb&w44555b"%%9>X%%CJJ"""aA 011111rNNNNNr.   Nr   r4   r   r   F)alwaysr   r   d   )r   minvaluemaxvaluecycle
   r
   i'  )r   r   rH   i  c           	         |rt          j        di |fnd}|rt          j        di |fnd}t                      }t                      }t          d|t	          dt
          g|R  t	          dt           j                             t          d|t	          dt
          g|R  t	          dt           j                             |                     ||          }t          t          |d                   d           |d         d         }t          |d         d           t          |d         d           t          |d	         d           |d
         }|d         }	d }
 |
||            |
||	           d S )Nr?   r   r   r   r   r   r>   r   r
   r      c                     | rat          t          |t          j                             |                                 D ]#\  }}t          t          ||          |           $d S t          |dv            d S )N)NF)r   r&   r    r"   itemsr   getattr)kwidtkvs       r,   checkz<AutogenerateIdentityTest.test_change_identity.<locals>.check   s     .
344555HHJJ , ,DAqQ++++, , },-----r.   )
r    r"   r   r   r   r   r!   r#   r   rB   )r'   beforeafter
arg_before	arg_afterr(   r)   r*   oldnewrT   s              r,   test_change_identityz-AutogenerateIdentityTest.test_change_identitym   s     28?bk++F++--R
/4<R[))5))++"	ZZZZ4.:...7BG$$		
 	
 	
 	4-9---7BG$$		
 	
 	
 b"%%CaMM1aE!H&'''E!HfE!HdAhAh	. 	. 	. 	fceSr.   c                 L   t                      }t                      }t          d|t          dt                    t          dt          j                             t          d|t          dt          t	          j        dd                    t          dt          j                             |                     ||          }t          t          |d                   d           |d         d         }t          |d         d	           t          |d         d           t          |d
         d           t          |d         d            |d         }t          t          |t          j                             t          |j        d           t          |j        d           d S )Nr   r   r   r     r   rH   r   r   r>   r
   r   rL   )r   r   r   r   r    r!   r"   r#   r   rB   r   r&   r   rH   )r'   r(   r)   r*   addeds        r,   test_add_identity_to_columnz4AutogenerateIdentityTest.test_add_identity_to_column   si   ZZZZ4!!7BG$$		
 	
 	
 	4"+A"E"E"EFF7BG$$		
 	
 	
 b"%%CaMM1aE!H&'''E!HfE!HdE!Hda
5"+..///EKEND!!!!!r.   c                    t                      }t                      }t          d|t          dt          t	          j        dd                    t          dt          j                             t          d|t          dt                    t          dt          j                             |                     ||          }t          t          |d                   d           |d         d         }t          |d         d	           t          |d         d           t          |d
         d           t          |d         d            |d         }t          t          |t          j                             d S )Nr   r   r   r]   r^   r   r   r   r>   r
   rL   r   )r   r   r   r   r    r"   r!   r#   r   rB   r   r&   )r'   r(   r)   r*   removeds        r,    test_remove_identity_from_columnz9AutogenerateIdentityTest.test_remove_identity_from_column   sE   ZZZZ4"+A"E"E"EFF7BG$$		
 	
 	
 	4!!7BG$$		
 	
 	
 b"%%CaMM1aE!H&'''E!HfE!HdE!Hd(
7BK0011111r.   )__name__
__module____qualname____requires____backend__r-   r2   r7   rD   r   combinationsdictr   requirementsidentity_columns_alterr[   r`   rc   r?   r.   r,   r   r      s       (LK  :% % %8
 
 
  ( W	tt!}}}	A	A12223	U			TT.../Dq1s$???Dq1s%@@@	

 DrQ666DrQ666	
  /( ( 0/ (T" " "@2 2 2 2 2r.   r   )
sqlalchemyr    r   r   r   r   alembic.utilr   _autogen_fixturesr	    r   r   r   r   r   r   r?   r.   r,   <module>rq      s                               $ $ $ $ $ $ 1 1 1 1 1 1                              R2 R2 R2 R2 R218 R2 R2 R2 R2 R2r.   