관리-도구
편집 파일: test_dialect.cpython-38.pyc
U -?�f�Y � @ s� d dl Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lm Z dd lm Z dd lm Z ddl m Z ddl mZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lm Z dd l m Z dd l m Z G d d � d ej!�Z"G d!d"� d"ej!�Z#G d#d$� d$ej$�Z%G d%d&� d&ej!�Z&G d'd(� d(ej$�Z'G d)d*� d*ej!�Z(G d+d,� d,ej!�Z)G d-d.� d.ej*e)�Z+G d/d0� d0ej!�Z,G d1d2� d2ej$�Z-dS )3� N� )�testing� )� assert_raises)�config)�engines)�eq_)�fixtures)� is_not_none)�is_true)�ne_)�provide_metadata)� expect_raises)�expect_raises_message)� requirements)� set_default_schema_on_connection)�Column)�Table� )� bindparam)�dialects)�event)�exc)�Integer)�literal_column)�select)�String)�Compiled)�inspect_getfullargspecc @ s e Zd ZdZdd� ZdS )�PingTestTc C s0 t j�� � }tt jj�|jj�� W 5 Q R X d S �N)r �db�connectr �dialectZdo_ping� connection�dbapi_connection)�self�conn� r( �U/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/suite/test_dialect.py� test_do_ping+ s �zPingTest.test_do_pingN)�__name__� __module__� __qualname__� __backend__r* r( r( r( r) r ( s r c @ s: e Zd ZdZdd� Zejee� �d�dd� �Zdd� Z d S ) �ArgSignatureTesta� test that all visit_XYZ() in :class:`_sql.Compiler` subclasses have ``**kw``, for #8988. This test uses runtime code inspection. Does not need to be a ``__backend__`` test as it only needs to run once provided all target dialects have been imported. For third party dialects, the suite would be run with that third party as a "--dburi", which means its compiler classes will have been imported by the time this test runs. c c sR t jD ] } | �d�st�d| � qtg}|rN|�d�}|�|�� � |V q*d S )N�_zsqlalchemy.dialects.%sr ) r �__all__� startswith� importlib� import_moduler �pop�extend�__subclasses__)�d�stack�clsr( r( r) �_all_subclasses@ s z ArgSignatureTest._all_subclasses)�paramsc c s |j V d S r )�param)r& �requestr( r( r) �all_subclassesL s z ArgSignatureTest.all_subclassesc C sL |}|j D ]<}|�d�r t||�}t|�}t|jd|j� d|� d�� q d S )N�visit_zCompiler visit method �.z:() does not accommodate for **kw in its argument signature)�__dict__r2 �getattrr r �varkwr+ )r& r? r: �k�methZinspr( r( r) � test_all_visit_methods_accept_kwP s �z1ArgSignatureTest.test_all_visit_methods_accept_kwN) r+ r, r- �__doc__r; r �fixture�listr? rG r( r( r( r) r/ 2 s r/ c @ s: e Zd ZdZdZdZedd� �Zej dd� �Z dd � Z d S ) � ExceptionTestz�Test basic exception wrapping. DBAPIs vary a lot in exception behavior so to actually anticipate specific exceptions from real round trips, we need to be conservative. �eachTc C s( t d|tdtddd�tdtd��� d S )N� manual_pk�idTF�� primary_keyZ autoincrement�data�2 �r r r r �r: �metadatar( r( r) � define_tablesk s �zExceptionTest.define_tablesc C sd t j�� �P}|�� }|�| jj�� ddd�� tt j |j| jj�� ddd�� |� � W 5 Q R X d S )Nr �d1�rN rQ ) r r! r" �begin�execute�tablesrM �insertr r ZIntegrityError�rollback)r& r' �transr( r( r) �test_integrity_errort s � �z"ExceptionTest.test_integrity_errorc C s� t j�� �v}z |�ttd��� ds(t�W nB tjk rl } z"t |�}t |j �t |�ks\t�W 5 d }~X Y nX t |t �s|t�W 5 Q R X d S )Nu méilF) r r! r" rZ r r �AssertionErrorr � DBAPIError�str�orig� isinstance)r& r' �errZerr_strr( r( r) � test_exception_with_non_ascii� s (z+ExceptionTest.test_exception_with_non_asciiN) r+ r, r- rH � run_deletesr. � classmethodrV r Z$duplicate_key_raises_integrity_errorr_ rf r( r( r( r) rK _ s rK c @ sl e Zd ZdZdZdd� Zdd� Zdd� Zd d � Ze j j d d � �Z e j j e j j d d� ��Ze j jdd� �ZdS )�IsolationLevelTestT��isolation_levelc C sF t �t�}|d }|d }t|��d|g�}|r8|�� S t�d� d S )N�default� supported� AUTOCOMMITz(no non-default isolation level available)r �get_isolation_levelsr �set� differencer5 Z skip_test)r& �levelsrl rm �sr( r( r) � _get_non_default_isolation_level� s z3IsolationLevelTest._get_non_default_isolation_levelc C s t tjjjt�t�d � d S )Nrl )r r r! r# Zdefault_isolation_levelr ro )r& r( r( r) � test_default_isolation_level� s �z/IsolationLevelTest.test_default_isolation_levelc C sl | � � }tj�� �P}|�� }t||� |j|d� t|�� |� |j� |j j � t|�� |� W 5 Q R X d S )Nrj ) rt r r! r" �get_isolation_levelr �execution_optionsr r# �reset_isolation_levelr$ r% )r& Z non_defaultr' �existingr( r( r) � test_non_default_isolation_level� s �z3IsolationLevelTest.test_non_default_isolation_levelc C s� t �t�}|d }t|��dg�D ]z}tj�� �>}|j|d� t|� � |� |� � }|� � t|� � |� W 5 Q R X tj�� �}t|� � |d � W 5 Q R X q"d S )Nrm rn rj rl ) r ro r rp rq r! r" rw r rv rY r] )r& rr Z all_levels�levelr' r^ r( r( r) �test_all_levels� s �z"IsolationLevelTest.test_all_levelsc C sJ |}t tjdd|jjd�t�t�d �f �� |j dd� W 5 Q R X dS )z4test for the new get_isolation_level_values() method�NInvalid value '%s' for isolation_level. Valid isolation levels for '%s' are %s�FOO�, rm rj N) r r � ArgumentErrorr# �name�joinr ro r rw )r& �connection_no_transr$ r( r( r) �#test_invalid_level_execution_option� s ���� z6IsolationLevelTest.test_invalid_level_execution_optionc C sR |t dd�d�}ttjdd|jjd�t�t �d �f �� |� � W 5 Q R X dS )z�test for the new get_isolation_level_values() method and support for the dialect-level 'isolation_level' parameter. r~ rj ��optionsr} r rm N) �dictr r r� r# r� r� r ro r r" )r& �testing_engine�engr( r( r) � test_invalid_level_engine_param� s ���� z2IsolationLevelTest.test_invalid_level_engine_paramc C s� t �t�}|d }tt|d ��|dg��d }|d|id�}|�� �}t|�� |� W 5 Q R X |�� � }|j |d� t|�� |� W 5 Q R X |�� �}t|�� |� W 5 Q R X d S )Nrl rm rn r rk r� rj ) r ro r �sortedrp rq r" r rv rw )r& r� rr rl rm �er' r( r( r) �%test_dialect_user_setting_is_restored s �� z8IsolationLevelTest.test_dialect_user_setting_is_restoredN)r+ r, r- r. � __requires__rt ru rz r| r �requiresZget_isolation_level_valuesr� Z#dialect_level_isolation_level_paramr� � independent_readonly_connectionsr� r( r( r( r) ri � s ri c @ sd e Zd ZdZdZdZedd� �Zdd� Zdd � Z d d � Z d d � Z e j je �dddg�dd� ��ZdS )�AutocommitIsolationTestrL )� autocommitTc C s, t d|tdtddd�tdtd��dd� d S ) N� some_tablerN TFrO rQ rR )Ztest_needs_acidrS rT r( r( r) rV s �z%AutocommitIsolationTest.define_tablesc C s� |� � }|�| jj�� ddd�� |�� t|�t| jjj j ��|rHdnd � |�� |� � � |�| jj� � � W 5 Q R X d S )Nr z some datarX ) rY rZ r[ r� r\ r] r �scalarr �crN �delete)r& r' r� r^ r( r( r) �_test_conn_autocommits( s � � z.AutocommitIsolationTest._test_conn_autocommitsc C s< |}|j dd�}| �|d� |j�|jj� | �|d� d S )Nrn rj TF)rw r� r# rx r$ r% )r& r� r' �c2r( r( r) �test_autocommit_on8 s z*AutocommitIsolationTest.test_autocommit_onc C s |}| � |d� d S )NF)r� �r& r� r' r( r( r) �test_autocommit_offA s z+AutocommitIsolationTest.test_autocommit_offc C sB |}|j dd�}| �|d� |j t�t�d d� | �|d� d S )Nrn rj Trl F)rw r� r ro r r� r( r( r) �.test_turn_autocommit_off_via_default_iso_levelE s ��zFAutocommitIsolationTest.test_turn_autocommit_off_via_default_iso_level�use_dialect_settingFc C s� |r|ddid�}n|� j dd�}t�t�}|d }|�� �}| �|d� W 5 Q R X |�� � }|j |d� | �|d� W 5 Q R X |�� �}| �|d� W 5 Q R X dS ) z test #10147rk rn r� rj rl TFN)rw r ro r r" r� )r& r� r� r� rr rl r' r( r( r) �#test_dialect_autocommit_is_restoredS s � z;AutocommitIsolationTest.test_dialect_autocommit_is_restoredN)r+ r, r- rg r� r. rh rV r� r� r� r� r r� r� Z variationr� r( r( r( r) r� s r� c @ s e Zd Zedd� �ZdS )� EscapingTestc C s� | j }td|tdtd���}|�tj� tj�� ��}|�|� � t dd�� |�|� � t dd�� t |� t |jj��|jjtd�k��d� t |� t |jj��|jjtd�k��d� W 5 Q R X d S ) z|test that the DBAPI accommodates for escaped / nonescaped percent signs in a way that matches the compiler �trQ rR z some % value)rQ zsome %% other valuez'some % value'z'some %% other value'N)rU r r r �creater r! rY rZ r\ r� r r� r r� rQ �wherer )r& �mr� r' r( r( r) � test_percent_sign_round_tripq s, ��� ���z)EscapingTest.test_percent_sign_round_tripN)r+ r, r- r r� r( r( r( r) r� p s r� c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d � Zd S ) � WeCanSetDefaultSchemaWEventsTestT)Zdefault_schema_name_switchc C s8 t jjj}t�� }|�� � W 5 Q R X t|jj|� d S r )r r! r# �default_schema_namer r� r" r )r& r� r� r( r( r) �test_control_case� s z2WeCanSetDefaultSchemaWEventsTest.test_control_casec C sd t jjj}t�� }t�|d�dd� �}|�� � }|j� |�}t |t j � W 5 Q R X t |jj|� d S )Nr" c S s t t| tj� d S r �r r � test_schema�r% Zconnection_recordr( r( r) � on_connect� s �zMWeCanSetDefaultSchemaWEventsTest.test_wont_work_wo_insert.<locals>.on_connect) r r! r# r� r r� r � listens_forr" �_get_default_schema_namer r� )r& r� r� r� r' �what_it_should_ber( r( r) �test_wont_work_wo_insert� s z9WeCanSetDefaultSchemaWEventsTest.test_wont_work_wo_insertc C s` t �� }tj|ddd�dd� �}|�� � }|j�|�}t|tj � W 5 Q R X t|jj tj � d S )Nr" T�r\ c S s t t| tj� d S r r� r� r( r( r) r� � s �zRWeCanSetDefaultSchemaWEventsTest.test_schema_change_on_connect.<locals>.on_connect) r r� r r� r" r# r� r r r� r� )r& r� r� r' r� r( r( r) � test_schema_change_on_connect� s z>WeCanSetDefaultSchemaWEventsTest.test_schema_change_on_connectc C s� t �� }tj|ddd�dd� �}|�� �F}|�� }|j�|�}t|t j � |� � |j�|�}t|t j � W 5 Q R X t|jj t j � d S )Nr" Tr� c W s t t| tj� d S r r� )r% �argr( r( r) r� � s �z\WeCanSetDefaultSchemaWEventsTest.test_schema_change_works_w_transactions.<locals>.on_connect) r r� r r� r" rY r# r� r r r� r] r� )r& r� r� r' r^ r� r( r( r) �'test_schema_change_works_w_transactions� s zHWeCanSetDefaultSchemaWEventsTest.test_schema_change_works_w_transactionsN) r+ r, r- r. r� r� r� r� r� r( r( r( r) r� � s r� c @ s e Zd ZdS )�&FutureWeCanSetDefaultSchemaWEventsTestN)r+ r, r- r( r( r( r) r� � s r� c @ s� e Zd ZdZejdddddddd d d d d ddddddddddd�Zeejj dd� ��Z ej dd � �Z ed d � �Z ed d � �Zd!S )"�DifficultParametersTestT)Zboring)zper cent)z per % cent)z%percent)zpar(ens))zpercent%(ens)yah)zcol:ons)Z_starts_with_underscore)zdot.s)zmore :: %colons%)�_name)Z___name)z[BracketsAndCase])Z 42numbers)z percent%signs)z has spaces)z /slashes/)z more/slashes)zq?marks)Z1param)z1col:on� paramname)�argnamesc C s� |}t d|tdtdd�t|td�dd��}|�|� |�|�� �dd|d i�� t|j | �� |j | d k�}t |� |�d � t|j | �� |j | t |�k�}|�||d i��� }t |j| d � t|j | �� |j | �d d g��}|�|��� }d S ) Nr� rN T)rP rR F)Znullabler z some namezsome other_name)r r r r r� rZ r\ �valuesr r� r� r r� r �first�_mapping�in_)r& r� r$ rU r� r� �stmt�rowr( r( r) �!test_round_trip_same_named_column� s$ � "�z9DifficultParametersTest.test_round_trip_same_named_columnc c sv t d|tdt�tdtd��tdtd���}|�|� |�|�� dddd �d d d d �d ddd �dddd �g� |V d S )N�mytable�myidr� rR �descr �aZa_desc)r� r� r� r �bZb_descr r� Zc_desc� r8 Zd_desc)r r r r r� rZ r\ )r& rU r$ r� r( r( r) �multirow_fixture s" � �� z(DifficultParametersTest.multirow_fixturec C sD |}t |jj��|jjt|dd�k�}|�||di�}t|d� d S )N�x��valuer� r )r r� r� r� r� r r� r �r& r� r$ r� Ztbl1r� �resr( r( r) � test_standalone_bindparam_escape5 s �z8DifficultParametersTest.test_standalone_bindparam_escapec C s` |}t |jj��|jj�t|ddgd����|jj�}|�||ddgi�� � }t |ddg� d S )Nr� r� r� r8 r r� ) r r� r� r� r� r� r Zorder_byZscalars�allr r� r( r( r) �*test_standalone_bindparam_escape_expanding@ s ���zBDifficultParametersTest.test_standalone_bindparam_escape_expandingN)r+ r, r- r. r � combinationsZtough_parametersr r Z unusual_column_name_charactersr� rI r� r� r� r( r( r( r) r� � sD � & r� c @ s^ e Zd ZdZdZedd� �Zejdd� �Z dd� Z d d � Z d d � Z d d� Z dd� Zdd� ZdS )�ReturningGuardsTestz=test that the various 'returning' flags are set appropriatelyTc C s( t d|tdtddd�tdtd��� d S )Nr� rN TFrO rQ rR rS rT r( r( r) rV T s �z!ReturningGuardsTest.define_tablesc s | j j�� �fdd�}|S )Nc s | � �jj�} |r�|spttjd� jj� d� jj� d���0 � � | |dddi|dddi|d dd ig�}W 5 Q R X n>� � | |dddi|dddi|d dd ig�}t |� � d d d g� nT|s�t tj �� � � | |dddi� W 5 Q R X n$� � | |dddi�}t |� � d g� d S )NzDialect z\+zW with current server capabilities does not support .*RETURNING when executemany is usedr rQ rW r �d2r �d3)r )r )r )Z returningr� rN r r ZStatementErrorr# r� ZdriverrZ r r� r ra )r� Z executemanyZ id_param_nameZexpect_success�result�r$ r� r( r) �goa sB � �� �� � �z(ReturningGuardsTest.run_stmt.<locals>.go)r[ r� )r& r$ r� r( r� r) �run_stmt] s /z ReturningGuardsTest.run_stmtc C s& | j j}|�� }||dd|jj� d S )NFrN )r[ r� r\ r# Zinsert_returning�r& r$ r� r� r� r( r( r) �test_insert_single� s z&ReturningGuardsTest.test_insert_singlec C s& | j j}|�� }||dd|jj� d S )NTrN )r[ r� r\ r# Z insert_executemany_returningr� r( r( r) �test_insert_many� s �z$ReturningGuardsTest.test_insert_manyc C s` | j j}|�|�� ddd�ddd�ddd�g� |�� �|jjtd�k�}||d d|j j � d S � Nr rW rX r r� r r� �b_idF) r[ r� rZ r\ �updater� r� rN r r# Zupdate_returningr� r( r( r) �test_update_single� s �� z&ReturningGuardsTest.test_update_singlec C s` | j j}|�|�� ddd�ddd�ddd�g� |�� �|jjtd�k�}||d d|j j � d S � Nr rW rX r r� r r� r� T) r[ r� rZ r\ r� r� r� rN r r# Z update_executemany_returningr� r( r( r) �test_update_many� s �� �z$ReturningGuardsTest.test_update_manyc C s` | j j}|�|�� ddd�ddd�ddd�g� |�� �|jjtd�k�}||d d|j j � d S r� ) r[ r� rZ r\ r� r� r� rN r r# Zdelete_returningr� r( r( r) �test_delete_single� s �� z&ReturningGuardsTest.test_delete_singlec C s` | j j}|�|�� ddd�ddd�ddd�g� |�� �|jjtd�k�}||d d|j j � d S r� ) r[ r� rZ r\ r� r� r� rN r r# Z delete_executemany_returningr� r( r( r) �test_delete_many� s �� �z$ReturningGuardsTest.test_delete_manyN)r+ r, r- rH r. rh rV r rI r� r� r� r� r� r� r� r( r( r( r) r� O s 4 r� ).r3 � r r r r r r r r r r Z assertionsr r r Z provisionr Zschemar r r r r r r r r r Z sql.compilerr �utilr ZTestBaser r/ Z TablesTestrK ri r� r� r� ZFutureEngineMixinr� r� r� r( r( r( r) �<module> sP -8 Y!C �u