OwlCyberSecurity - MANAGER
Edit File: test_autogen_computed.cpython-312.pyc
� i��g � � � d dl Zd dl mZ d dl mZ d dl mZ d dl mZ ddlmZ dd lm Z dd l m Z dd l m Z dd l m Z dd l m Z ddl mZ G d� dee� Zy)� N)�Column)�Integer)�MetaData)�Table� )�AutogenFixtureTest� )�testing)�eq_)�is_)�is_true)�mock)�TestBasec � � e Zd ZdZdZd� Zd� Z ej d� d� d� d� � d � � Z ej d � d � d � d � � d� � Z y)�AutogenerateComputedTest)�computed_columnsTc �^ � t � }t � }t d|t dt d�� � t d|t dt d�� t dt t j 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 t |j � d� y ) N�user�idT�� primary_key�foo�5r � add_column� r )r r r r �sa�Computed�_fixturer �name�computedr � isinstancer � persisted�str�sqltext��self�m1�m2�diffs�cs �Z/opt/hc_python/lib/python3.12/site-packages/alembic/testing/suite/test_autogen_computed.py�test_add_computed_columnz1AutogenerateComputedTest.test_add_computed_column s� � � �Z�� �Z�� �f�b�&��w�D�A�B� � � � �4��d� 3� �5�'�2�;�;�s�#3� 4� � � � �b�"�%�� �E�!�H�Q�K� �&� �E�!�H�Q�K�� � �E�!�H�Q�K� � �e�$� �!�H�Q�K� � ��� �1�b�k�k�*�+� �A�K�K�� � �C�� � �N�C� � c �B � t � }t � }t d|t dt d�� t dt t j d� � � t d|t dt d�� � | j ||� }t |d d d� t |d d d� |d d }t |j d� t t |j t j � � t t |j t j � � y ) Nr r Tr r r r � remove_columnr r ) r r r r r r r r r r r! r �server_defaultr% s r+ �test_remove_computed_columnz4AutogenerateComputedTest.test_remove_computed_column, s� � � �Z�� �Z�� � � � �4��d� 3� �5�'�2�;�;�s�#3� 4� � �f�b�&��w�D�A�B�� � �b�"�%�� �E�!�H�Q�K� �)� �E�!�H�Q�K�� � �!�H�Q�K�� �A�F�F�E��� �1�:�:�r�{�{�3�4�� �1�+�+�R�[�[�9�:r- c �0 � d t j d� fS �N�bar*5�r r � r- r+ �<lambda>z!AutogenerateComputedTest.<lambda>D s � ��r�{�{�7�+�,r- c �0 � t j d� d fS r3 r5 r6 r- r+ r7 z!AutogenerateComputedTest.<lambda>E s � �"�+�+�g�&��-r- c �Z � t j d� t j dd�� fS )Nr4 �bar * 42T)r" r5 r6 r- r+ r7 z!AutogenerateComputedTest.<lambda>F s! � � �K�K�� � �K�K� �d� 3� r- c �V � t j d� t j d� fS )Nr4 r: r5 r6 r- r+ r7 z!AutogenerateComputedTest.<lambda>J s � ����W�%�r�{�{�:�'>�?r- c �p � t j |fi t � ��\ }}t � }t � }|�g n|g}|�g n|g}t d|t dt d�� t dt � t dt g|��� � t d|t dt d�� t dt � t dt g|��� � t j d� 5 }| j ||� }d d d � t j t j d� g� t t � g � y # 1 sw Y �IxY w) Nr r Tr �barr �alembic.util.warnz/Computed default on user.foo cannot be modified)r �resolve_lambda�localsr r r r r �patchr r � mock_calls�call�list�r& � test_case� arg_before� arg_afterr' r( � mock_warnr) s r+ �!test_cant_change_computed_warningz:AutogenerateComputedTest.test_cant_change_computed_warningC s � � !(� 6� 6�y� M�F�H� M� � �I� �Z�� �Z��%�-�R�J�<� �#�+�B�)�� � � � � �4��d� 3� �5�'� "� �5�'� /�J� /� � � � � �4��d� 3� �5�'� "� �5�'� .�I� .� � �Z�Z�+� ,� ��M�M�"�b�)�E� -� � � � � �Y�Y�H� I� J� � �D��K�� � -� ,�s �D,�,D5c � � y)N)NNr6 r6 r- r+ r7 z!AutogenerateComputedTest.<lambda>o s � � r- c �V � t j d� t j d� fS )Nr r5 r6 r- r+ r7 z!AutogenerateComputedTest.<lambda>p s � ����S�!�2�;�;�s�#3�4r- c �V � t j d� t j d� fS r3 r5 r6 r- r+ r7 z!AutogenerateComputedTest.<lambda>q s � ����W�%�r�{�{�7�';�<r- c �V � t j d� t j d� fS )Nr4 z bar * 5r5 r6 r- r+ r7 z!AutogenerateComputedTest.<lambda>r s � ����W�%�r�{�{�?�'C�Dr- c �H � t j |fi t � ��\ }}t � }t � }|�g n|g}|�g n|g}t d|t dt d�� t dt � t dt g|��� � t d|t dt d�� t dt � t dt g|��� � t j d� 5 }| j ||� }d d d � t j g � t t � g � y # 1 sw Y �5xY w)Nr r Tr r= r r> ) r r? r@ r r r r r rA r r rB rD rE s r+ �test_computed_unchangedz0AutogenerateComputedTest.test_computed_unchangedn s� � � !(� 6� 6�y� M�F�H� M� � �I� �Z�� �Z��%�-�R�J�<� �#�+�B�)�� � � � � �4��d� 3� �5�'� "� �5�'� /�J� /� � � � � �4��d� 3� �5�'� "� �5�'� .�I� .� � �Z�Z�+� ,� ��M�M�"�b�)�E� -� �I� � �"�%� �D��K�� � -� ,�s �D�D!N) �__name__� __module__� __qualname__� __requires__� __backend__r, r1 r � combinationsrJ rP r6 r- r+ r r sv � �(�L��K�!�0;�. �W���,� -� � @�� �� �D �W��� �4�<�D� � � � r- r )� sqlalchemyr r r r r �_autogen_fixturesr � r r r r r r r r6 r- r+ �<module>rZ s: �� � � � � � 1� � � � � � �@ � 1�8� @ r-