OwlCyberSecurity - MANAGER
Edit File: test_autogen_identity.cpython-312.pyc
� i��g� � � � d dl Zd dl mZ d dl mZ d dl mZ d dl mZ d dlmZ ddlm Z d d l m Z d d l m Z d d l m Z d d l mZ d dl mZ G d� de e� Zy)� N)�Column)�Integer)�MetaData)�Table)� sqla_compat� )�AutogenFixtureTest� )�testing)�config)�eq_)�is_true)�TestBasec �^ � e Zd ZdZdZd� Zd� Zd� Zd� Z e j d e d� � f e d� � df e d� � e dd � � f e d � � e d� � f e dddd�� e dddd �� f e ddd�� e ddd�� f� e j j d� � � Zd� Zd� Zy)�AutogenerateIdentityTest)�identity_columnsTc �t � t � }t � }t d|t dt j � � t d|t dt j � t dt t j dd�� d�� � | j ||� }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� y ) 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 �Z/opt/hc_python/lib/python3.12/site-packages/alembic/testing/suite/test_autogen_identity.py�test_add_identity_columnz1AutogenerateIdentityTest.test_add_identity_column s� � � �Z�� �Z�� �f�b�&��"�'�'�2�3� � � � �7�B�G�G� $� ���� � �!�q�1� � � � � � �b�"�%�� �E�!�H�Q�K� �&� �E�!�H�Q�K�� � �E�!�H�Q�K� � �d�#� �!�H�Q�K� � ��� �1�b�k�k�*�+� �A�G�G�Q�� �A�K�K��� c �, � t � }t � }t d|t dt t j dd�� d�� � t d|� | j ||� }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� y ) 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� � � �Z�� �Z�� � � � ���� � �!�q�1� � � � �f�b��� � �b�"�%�� �E�!�H�Q�K� �)� �E�!�H�Q�K�� � �!�H�Q�K�� �A�F�F�D��� �1�:�:�r�{�{�3�4� �A�J�J� � �a� � �A�J�J� � �!�$r. c �� � t � }t � }||fD ]4 }t j d�� }t d|t dt |� � �6 | j ||� }t |g � y )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\ � � �Z�� �Z���b��A��+�+�A�&�C� �&�!�V�D�'�3� 7� 8� � � � �b�"�%�� �E�2�r. c �� � t � }t � }t j rddd�}nddd�}t d|t dt t j d�� � � t j d ddi|��}t d|t dt |� � | j ||� }t j j d k( r+t t |� d � t |d d d d � y t |g � y )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� � � �Z�� �Z�� � 2� 2�&*�D�A�D� #�d�3�D� �f�b�&��w�� � �!�0D�E�F��k�k�*��*�T�*�� �f�b�&��w��4�5�� � �b�"�%�� �9�9�>�>�X� %� �C��J�� "� ��a��� �A�� 0� 1� ��r�Nr. Nr r4 r r F)�alwaysr r �d )r �minvalue�maxvalue�cycle� r i' )r r rH i c �d � |rt j di |��fnd}|rt j di |��fnd}t � }t � }t d|t dt g|��� t dt j � � t d|t dt g|��� t dt j � � | j ||� }t t |d � d� |d d }t |d d� t |d d� t |d d� |d }|d } d � } | ||� | || � y ) Nr? r r r r r r> r r r � c �� � | rRt t |t j � � | j � D ] \ }}t t ||� |� � y t |dv � y )N)NF)r r& r r" �itemsr �getattr)�kw�idt�k�vs r, �checkz<AutogenerateIdentityTest.test_change_identity.<locals>.check� sK � ��� �3�� � �4�5� �H�H�J�D�A�q����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�b�k�k�+�F�+�-�R� �/4�R�[�[�)�5�)�+�"� � �Z�� �Z�� � � � �4�� .�:� .� �7�B�G�G� $� � � � � �4�� -�9� -� �7�B�G�G� $� � � � �b�"�%�� �C��a��M�1� ��a��� �� �E�!�H�&�'� �E�!�H�f� � �E�!�H�d���A�h���A�h�� .� �f�c�� �e�S�r. c � � t � }t � }t d|t dt � t dt j � � t d|t dt t j dd�� � t dt j � � | 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� y ) 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� s � � �Z�� �Z�� � � � �4�� !� �7�B�G�G� $� � � � � �4��"�+�+�A��"E� F� �7�B�G�G� $� � � � �b�"�%�� �C��a��M�1� ��a��� �� �E�!�H�&�'� �E�!�H�f� � �E�!�H�d�� �E�!�H�d���a���� �5�"�+�+�.�/� �E�K�K��� �E�N�N�D�!r. c �T � t � }t � }t d|t dt t j dd�� � t dt j � � t d|t dt � t dt j � � | j ||� }t t |d � d� |d d }t |d d � t |d d� t |d d� t |d d � |d }t t |t j � � y ) 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� s� � � �Z�� �Z�� � � � �4��"�+�+�A��"E� F� �7�B�G�G� $� � � � � �4�� !� �7�B�G�G� $� � � � �b�"�%�� �C��a��M�1� ��a��� �� �E�!�H�&�'� �E�!�H�f� � �E�!�H�d�� �E�!�H�d����(��� �7�B�K�K�0�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��� �t�!�}� � �A� �� � �A� ��1��2�3� �U� �T�� .�/� �q�1�s�$� ?� �q�1�s�%� @� � �r�Q�� 6� �r�Q�� 6� � � ���/�/�(� 0� � (�T "�@ 2r. r )� sqlalchemyr r r r r � alembic.utilr �_autogen_fixturesr � r r r r r r r? r. r, �<module>rq s: �� � � � � � $� 1� � � � � �R2� 1�8� R2r.