Current File : /opt/hc_python/lib/python3.12/site-packages/alembic/testing/__pycache__//assertions.cpython-312.pyc
�
V��gL � �p � U d dl mZ d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z
d dl m Z d dl m
Z
d dlmZ d d lmZ d d
lmZ d d lmZ d d lmZ d d
lmZ d dlmZ d dlmZ d� Zd� Zd� Zd� Zd� Z d#d�Z G d� d� Z ej> d$d�� Z d%d�Z! d&d�Z"d'd�Z#i Z$d e%d < d � Z&d � Z'd � Z(d!� Z)d"� Z*y)(� )� annotationsN)�Any)�Dict)�exc)�default)�URL)�_expect_warnings)�eq_)�is_)�is_false)�is_not_)�is_true)�ne_�� decoratorc �~ � | j | j ur%| j sJ d| �d| j �d�� �yy)aH assert that any exception we're catching does not have a __context__
without a __cause__, and that __suppress_context__ is never set.
Python 3 will report nested as exceptions as "during the handling of
error X, error Y occurred". That's not what we want to do. we want
these exceptions in a cause chain.
z
Exception z> was correctly raised but did not set a cause, within context z as its cause.N)� __context__� __cause__�__suppress_context__)� exceptions �I/opt/hc_python/lib/python3.12/site-packages/alembic/testing/assertions.py� _assert_proper_exception_contextr sI � � � � �Y�%8�%8�8��.�.�
� �)�/�/�
1�
�u� /� 9� c �" � t | |||d�� S )NT��
check_context��_assert_raises��
except_cls� callable_�args�kws r �
assert_raisesr$ + s � � �*�i��r�� N�Nr c � � t | |||� S �Nr r s r �assert_raises_context_okr' / s � � �*�i��r� :�:r c �$ � t | ||||d�� S )NT)�msgr r �r r) r! r" �kwargss r �assert_raises_messager, 3 s � � ��I�t�V��D� � r c �" � t | ||||�� S )N)r) r r* s r � assert_raises_message_context_okr. 9 s � � �*�i��v�3� G�Gr c � � t | ||� 5 } ||i |�� d d d � |j S # 1 sw Y j S xY wr& )�_expect_raises�error)r r! r" r+ r) r �ecs r r r ? s< � �
�
�C� � 7�2��4�"�6�"�
8�
�8�8�O�
8�
�8�8�O�s � +�?c � � e Zd ZU dZded< y)�_ErrorContainerNr r1 )�__name__�
__module__� __qualname__r1 �__annotations__� r r r4 r4 G s � ��E�3�r r4 c # � K � t � }|rt j � d } |�� d}|sJ d� �y # | $ r�}||_ d}|�V|rt |� |k( sFJ |� d|� �� �t j |t |� t
j � s J |� d|� �� �|r
s t |� t t |� j d� � Y d }~��d }~ww xY w�w)Nr FT� != z !~ zutf-8z#Callable did not raise an exception) r4 �sys�exc_infor1 �str�re�search�UNICODEr �print�encode)r r) r �
text_exactr2 � are_we_already_in_a_traceback�success�errs r r0 r0 K s� � �� �
� �B��(+� � ��q�(9�%�
(����� �9�9�9�7�� �
(������ �?���3�x�3��9�3�%�t�C�5�(9�9���y�y��c�#�h��
�
�;�N��u�D���=N�N�;� �!>� ,�S� 1�
�c�#�h�o�o�g�&�'�'��
(�s+ �$C �7 �
C �C�BC�C �C�C c � � t | |�� S )Nr �r0 )r r s r �
expect_raisesrJ e s � � �*�M� B�Br c � � t | |||�� S )N)r) r rD rI )r r) r rD s r �expect_raises_messagerL i s � � ���=�Z� � r c �� � t j dd| � } t j dd| � } t j dd|� }t j dd|� }| |k( sJ |xs | �d|��� �y )Nz^\s+?|\n� z {2,}� r; )r? �sub)�a�br) s r �eq_ignore_whitespacerS q se � �
���{�B��"�A�
���x��a� �A�
���{�B��"�A�
���x��a� �A� ��6�-�3�-��1�-�-�6r zDict[Any, Any]�
_dialect_modsc � � | �| dk( rt j � S t j | � j � � }| dk( r d|_ |S | dk( rd|_ |S )Nr �
postgresqlT�mssqlF)r �DefaultDialectr �create� get_dialect�implicit_returning�legacy_schema_aliasing)�name�ds r � _get_dialectr_ } sf � � �|�t�y�(��%�%�'�'� *�C�J�J�t� � (� (� *� ,�� �<� �#'�A� � �� �W�_�',�A� $��r c �$ � t t | fi |��S )a� Context manager which expects one or more warnings.
With no arguments, squelches all SAWarnings emitted via
sqlalchemy.util.warn and sqlalchemy.util.warn_limited. Otherwise
pass string expressions that will match selected warnings via regex;
all non-matching warnings are sent through.
The expect version **asserts** that the warnings were in fact seen.
Note that the test suite sets SAWarning warnings to raise exceptions.
)r �Warning��messagesr# s r �expect_warningsrd � s � � �G�X� 4�� 4�4r c �$ � � t � fd�� }|S )z�Decorator form of expect_warnings().
Note that emits_warning does **not** assert that the warnings
were in fact seen.
c �f �� t t g���ddi�5 | |i |��cd d d � S # 1 sw Y y xY w)N�assert_F)r �DeprecationWarning)�fnr" r# rc s �r �decoratez2emits_python_deprecation_warning.<locals>.decorate� s. �� �
� 0�
K�(�
K�%�
K��t�?�r�?� L�
K�
K�s �'�0r )rc rj s ` r � emits_python_deprecation_warningrk � s �� � �#� �#� �Or c �8 � t t j | fi |��S r& )r �sa_exc�SADeprecationWarningrb s r � expect_sqlalchemy_deprecatedro � s � � �F� 7� 7�� H�R� H�Hr c �8 � t t j | fi |��S r& )r rm �RemovedIn20Warningrb s r � expect_sqlalchemy_deprecated_20rr � s � � �F� 5� 5�x� F�2� F�Fr )NF)NFF)T)TFr& )+�
__future__r �
contextlibr? r< �typingr r �
sqlalchemyr rm �sqlalchemy.enginer r � sqlalchemy.testing.assertionsr r
r r r
r r �sqlalchemy.utilr r r$ r' r, r. r r4 �contextmanagerr0 rJ rL rS rT r8 r_ rd rk ro rr r9 r r �<module>r{ s� �� "� � �
� � � $� %� !� :� -� -� 2� 1� 1� -� %�
�*O�;�� H� BG��� � ���:?�:� �:�2C�
5:��.� !#�
�~� "�
�
5�
� I�Gr 