OwlCyberSecurity - MANAGER
Edit File: test_autogen_comments.cpython-312.pyc
� i��g� � � � d dl mZ d dl mZ d dl mZ d dl mZ d dl mZ ddlmZ dd lm Z dd lm Z dd lm Z G d � d ee � Z y)� )�Column)�Float)�MetaData)�String)�Table� )�AutogenFixtureTest� )�eq_)�mock)�TestBasec �D � e Zd ZdZdZd� Zd� Zd� Zd� Zd� Z d� Z d � Z d � Z y ) �AutogenerateCommentsTestT)�commentsc � � t � }t � }t d|t dt d� d�� d�� t d|t dt 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 �Z/opt/hc_python/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�� � � � �6�6�"�:�4� 8�(� � � � � �6�6�"�:�4� 8�(� � � � �b�"�%�� �E�2�� c �\ � t � }t � }t d|t dt d� d�� � t d|t dt 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�� �l�B��v�v�b�z�t� L�M� � � � �6�6�"�:�4� 8�(� � � � �b�"�%�� �E�!�H�Q�K�,�-� �E�!�H�Q�K� � �!5�6� �E�!�H�Q�K�� r" c �8 � t � }t � }t d|t dt d� d�� d�� t d|t dt 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�� � � � �6�6�"�:�4� 8�(� � �l�B��v�v�b�z�t� L�M�� � �b�"�%�� �E�!�H�Q�K�/�0� �E�!�H�Q�K� � ��&r" c �` � t � }t � }t d|t dt d� d�� d�� t d|t dt d� d�� d�� | j ||� }t |d d d � t |d d j d� t |d d d� y ) Nr r r Tr r r zthis is also some tabler r$ r r% r& r s r �test_alter_table_commentz1AutogenerateCommentsTest.test_alter_table_commentM s� � � �Z�� �Z�� � � � �6�6�"�:�4� 8�(� � � � � �6�6�"�:�4� 8�-� � � � �b�"�%�� �E�!�H�Q�K�,�-� �E�!�H�Q�K� � �!:�;� �E�!�H�Q�K�-�.r" c �4 � t � }t � }t d|t dt d� d�� t dt d�� � t d|t dt d� d�� t dt 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�� � � � �6�6�"�:�4� 8� �8�U�L� 9� � � � � �6�6�"�:�4� 8� �8�U�L� 9� � � � �b�"�%�� �E�2�r" c �f � t � }t � }t d|t dt d� d�� t dt � � t d|t dt d� d�� t dt d�� � | j ||� }t |d d dddt j d d �d dfgg� 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�� � � � �6�6�"�:�4� 8� �8�U� #� � � � � �6�6�"�:�4� 8� �8�U�L� 9� � � � �b�"�%�� � � )� �$� �15�-1�X�X�7<�� �$� �� � r" c �f � t � }t � }t d|t dt d� d�� t dt d�� � t d|t dt d� d�� t dt � � | j ||� }t |d d dddt j d d �dd fgg� y r2 r8 r s r �test_remove_column_commentz3AutogenerateCommentsTest.test_remove_column_comment� s� � � �Z�� �Z�� � � � �6�6�"�:�4� 8� �8�U�L� 9� � � � � �6�6�"�:�4� 8� �8�U� #� � � � �b�"�%�� � � )� �$� �15�-1�X�X�7<�� %� � �� � r" c �j � t � }t � }t d|t dt d� d�� t dt d�� � t d|t dt d� d�� t dt d �� � | j ||� }t |d d dddt j d d �dd fgg� y ) Nr r r Tr r. r/ r zthe adjusted amountr3 Fr4 r8 r s r �test_alter_column_commentz2AutogenerateCommentsTest.test_alter_column_comment� s� � � �Z�� �Z�� � � � �6�6�"�:�4� 8� �8�U�L� 9� � � � � �6�6�"�:�4� 8� �8�U�,A� B� � � � �b�"�%�� � � )� �$� �15�-1�X�X�7<�� %�-� �� � 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 � 1�8� e r"