Edit File: test_reflection.cpython-37.pyc
B ��4]\� � @ sf d dl Z d dlZd dlZddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z dd lm Z dd l m Z dd l mZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dl m Z d dl m Z d dl m Z d\Z Z!G dd � d e j"�Z#G d d � d e j"�Z$G d d � d e j"�Z%d!Z&dS )"� N� )�assert_raises_message)�config)�engines)�eq_)�expect_warnings)�fixtures)�is_)�Column)�Table� )�event)�exc)� ForeignKey)�inspect)�Integer)�MetaData)�String)�testing)�types)� Inspector)�DDL)�Index)� quoted_name)NNc @ s$ e Zd ZdZedd� �Zdd� ZdS )� HasTableTestTc C s& t d|tdtdd�tdtd��� d S )N� test_table�idT)� primary_key�data�2 )r r r r )�cls�metadata� r" �[/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/suite/test_reflection.py� define_tables# s zHasTableTest.define_tablesc C sB t j�� �.}t jj�|d�s t�t jj�|d�r4t�W d Q R X d S )Nr Znonexistent_table)r �db�begin�dialectZ has_table�AssertionError)�selfZconnr" r" r# �test_has_table, s zHasTableTest.test_has_tableN)�__name__� __module__� __qualname__� __backend__� classmethodr$ r* r" r" r" r# r s r c @ s� e Zd Zd ZZdZedd� �Zedd� �Zedd� �Z ed d � �Z ed d � �Z ed d� �Z e jjdd� �Ze jjdd� �Ze jjdd� �Ze jd�dd��Ze jjdd� �Ze jje jje jjdd� ���Ze jjd d � �Ze jje jj d d � ��Z e jj d d!� �Z e jj e jj d"d#� ��Z!d�d$d%�Z"e jje jj d&d'� ��Z#e jj$d(d)� �Z%e jj$e jj d*d+� ��Z&e jje jj$d,d-� ��Z'd�d.d/�Z(e jjd0d1� �Z)e jd2d3� �Z*e jjd4d5� �Z+e jjd6d7� �Z,e jje jd8d9� ��Z-e jje jj d:d;� ��Z.e jj/d<d=� �Z0e jj/e jj$e jjd>d?� ���Z1e jj$d@dA� �Z2e jj$e jj dBdC� ��Z3e jd�dDdE��Z4e jj5dFdG� �Z6e jje jj5e jj dHdI� ���Z7e jje jdJdK� ��Z8e jd�dLdM��Z9e jj dNdO� �Z:e jj e jj dPdQ� ��Z;e jj<e jj dRdS� ��Z=e jj>dTdU� �Z?e jj@dVdW� �ZAe jdXdY� �ZBdZd[� ZCe jd�d\d]��ZDe jjEd^d_� �ZFe jjEe jj d`da� ��ZGe jdbdc� �ZHe jjEddde� �ZIe jjEdfdg� �ZJe jjKe jdhdi� ��ZLe jjMdjdk� �ZNe jj/e jjMdldm� ��ZOe jj/dndo� �ZPe jjMe jj dpdq� ��ZQe jd�drds��ZRe jjSdtdu� �ZTe jjSe jj dvdw� ��ZUe jd�dxdy��ZVe jd�dzd{��ZWe jjd|d}� �ZXe jje jj d~d� ��ZYe �Zd�d��e jd�d�d����Z[d�d�� Z\e jj d�d�� �Z]e jje jd�d�� ��Z^dS )��ComponentReflectionTestNTc C s4 t jjjr*ddlm} tjt|j d�d�S t j S d S )Nr )�pool)Z poolclass)�options) r � requirementsZindependent_connections�enabled� sqlalchemyr1 r �testing_engine�dictZ StaticPoolr% )r r1 r" r" r# � setup_bind7 s z"ComponentReflectionTest.setup_bindc C s* | � |d � tjjjr&| � |tjj� d S )N)�define_reflected_tablesr �requires�schemasr4 r � test_schema)r r! r" r" r# r$ B s z%ComponentReflectionTest.define_tablesc C s |r|d }nd}t jjjrztd|tdtjdd�tdt�d�d d �td t� d�d d �td tj tj d | dd��|dd�}nBtd|tdtjdd�tdt�d�d d �td t� d�d d �|dd�}td|tdtj dd�tdtj t� d| ��tdt� d��|dd� td|tdtj �tdtj t� |j j��tdt� d��tjddd�|dd� td |td tj dd d �tdt� d�d d!�td"t� d�d#d!�|d$d%� t jjj�rX|d k�r�td&|td tj dd�tdt� d��td't d(t jj ��dtjjjd)� nntd*|td tj dd�td+t d,tjjj ��tdt� d��|dd� td-|td tj dd�tdt� d��|dd� t jjj�r�| �||� |�s�td.|td/t� d��dd0�}td1|td tj dd�td/t� d��dd0�}td2|j j�� � td3|j j�� � t jjj�r�| � ||� |�st jj j�r| � |� d S )4N�.� �users�user_idT)r �test1� F)�nullable�test2�parent_user_idz%susers.user_id� user_id_fk)�name)�schema� test_needs_fk� dingalings� dingaling_id� address_idz %semail_addresses.address_idr � �email_addresses�remote_user_idZ email_address� � email_ad_pk� comment_testr z id comment)r �commentzdata % comment)rS �d2z)Comment types type speedily ' " \ '' Fun!z the test % ' " \ table comment)rH rS � local_table� remote_idz%s.remote_table_2.id)rI rH � remote_table�local_idz%s.local_table.id�remote_table_2�noncol_idx_test_nopk�q)rI �noncol_idx_test_pk�noncol_idx_nopk� noncol_idx_pk) r r: � self_referential_foreign_keysr4 r r �sa�INTZCHARZFloatr r r �cr@ ZPrimaryKeyConstraint�cross_schema_fk_reflectionr r< r% r' �default_schema_name�index_reflection� define_indexr r[ �desc�view_column_reflection� define_views�temp_table_reflection�define_temp_tables)r r! rH Z schema_prefixr? rZ r\ r" r" r# r9 H s� z/ComponentReflectionTest.define_reflected_tablesc C s� t �d�rdgdd�}n ddgi}td|tdtjd d �td t�d ��td tj�tjd dd�t�dd �f|�}t j j j r�t j j j r�t �|dtd�� t �|dtd�� d S )N�oraclezGLOBAL TEMPORARYz PRESERVE ROWS)�prefixesZoracle_on_commitrm Z TEMPORARY�user_tmpr T)r rG r �foo� user_tmp_uq)rG � user_tmp_ix� after_createz:create temporary view user_tmp_v as select * from user_tmp� before_dropzdrop view user_tmp_v)r �againstr r r` ra ZVARCHAR�UniqueConstraintr r: �view_reflectionr4 �temporary_viewsr �listenr )r r! �kwrn r" r" r# rk � s* z*ComponentReflectionTest.define_temp_tablesc C s2 t d|jj|jj� t d|jj|jj|jj� d S )N� users_t_idx� users_all_idx)r rb rA rD r@ )r r! r? r" r" r# rf � s z$ComponentReflectionTest.define_indexc C sb x\dD ]T}|}|r d||f }|d }d||f }t �|dt|�� t �|dtd| �� qW d S )N)r? rN z%s.%sZ_vz"CREATE VIEW %s AS SELECT * FROM %srr rs z DROP VIEW %s)r rx r )r r! rH � table_name�fullnameZ view_name�queryr" r" r# ri � s z$ComponentReflectionTest.define_viewsc C s$ t tj�}| �tjj|�� k� d S )N)r r r% �assert_r r< Zget_schema_names)r) �inspr" r" r# �test_get_schema_names s z-ComponentReflectionTest.test_get_schema_namesc C s4 t �� }t|jd�rt�t|� t|jd�s0t�d S )Nrd )r r6 �hasattrr' r( r )r) Zenginer" r" r# �test_dialect_initialize s z/ComponentReflectionTest.test_dialect_initializec C s t tj�}t|jtjjj� d S )N)r r r% r rd r' )r) r� r" r" r# � test_get_default_schema_name s z4ComponentReflectionTest.test_get_default_schema_name�tablec s� ddddddg� | j }t|j�}|dkrR|�|�}|�� dd g}tt|�|� nl|rld d � |�|�D �}n |�|�}� fd d �|D �}|d kr�dddg}t||� ndddg}tt|�|� d S )NrR r\ rZ rU rW rY �view�email_addresses_v�users_vc S s g | ]}|d r|d �qS )r r" )�.0�recr" r" r# � <listcomp>: s zAComponentReflectionTest._test_get_table_names.<locals>.<listcomp>c s g | ]}|� kr|�qS r" r" )r� �t)�_ignore_tablesr" r# r� @ s � foreign_keyr? rN rJ ) r! r �bindZget_view_names�sortr �sortedZ get_sorted_table_and_fkc_names�get_table_names) r) rH � table_type�order_by�metar� � table_names�answer�tablesr" )r� r# �_test_get_table_names" s. z-ComponentReflectionTest._test_get_table_namesc C s& t | j�}|�� }tt|�dg� d S )Nrn )r r� Zget_temp_table_namesr r� )r) r� �temp_table_namesr" r" r# �test_get_temp_table_namesI s z1ComponentReflectionTest.test_get_temp_table_namesc C s& t | j�}|�� }tt|�dg� d S )N� user_tmp_v)r r� Zget_temp_view_namesr r� )r) r� r� r" r" r# �test_get_temp_view_namesO s z0ComponentReflectionTest.test_get_temp_view_namesc C s | � � d S )N)r� )r) r" r" r# �test_get_table_namesW s z,ComponentReflectionTest.test_get_table_namesc C s | j dd� d S )Nr� )r� )r� )r) r" r" r# �test_get_table_names_fks[ s z0ComponentReflectionTest.test_get_table_names_fksc C s | � � d S )N)�_test_get_comments)r) r" r" r# �test_get_comments` s z)ComponentReflectionTest.test_get_commentsc C s | � tjj� d S )N)r� r r r< )r) r" r" r# � test_get_comments_with_schemad s z5ComponentReflectionTest.test_get_comments_with_schemac C st t tj�}t|jd|d�ddi� t|jd|d�dd i� tdd� |jd|d�D �dd d �d d d �d dd �g� d S )NrR )rH �textz the test % ' " \ table commentr? c S s g | ]}|d |d d��qS )rG rS )rG rS r" )r� r� r" r" r# r� u s z>ComponentReflectionTest._test_get_comments.<locals>.<listcomp>z id commentr )rS rG zdata % commentr z)Comment types type speedily ' " \ '' Fun!rT )r r r% r Zget_table_comment� get_columns)r) rH r� r" r" r# r� i s z*ComponentReflectionTest._test_get_commentsc C s | � tjj� d S )N)r� r r r< )r) r" r" r# � test_get_table_names_with_schema� s z8ComponentReflectionTest.test_get_table_names_with_schemac C s | j dd� d S )Nr� )r� )r� )r) r" r" r# �test_get_view_names� s z+ComponentReflectionTest.test_get_view_namesc C s | j tjjdd� d S )Nr� )r� )r� r r r< )r) r" r" r# � test_get_view_names_with_schema� s z7ComponentReflectionTest.test_get_view_names_with_schemac C s | � � | j dd� d S )Nr� )r� )r� )r) r" r" r# �test_get_tables_and_views� s z1ComponentReflectionTest.test_get_tables_and_viewsc C sv t tj�}| jj| jj }}ddg}|dkr4ddg}t|j�}�x0t|||f�D �] \}} |} |j || d�} | � t | �dkt | �� x�t | j �D ]�\} } t| j| | d � | | d j}| j}t|tjj�r�|j}t�d �r�|tjtjfkr�tj}| � t t|j�� |j�� tj tj tjtjtj tj tj!g��dkd | j| j| | d |f � | j"s�| | d d ks�t#�q�W qPW d S ) Nr? rN r� r� r� )rH r rG �typerl z%s(%s), %s(%s)�default)$r r r% r� r? rN r r� �zipr� r �len� enumerate�columnsr rG � __class__r� � isinstancer` r Z TypeEnginert � sql_typesZDateZDateTime�set�__mro__� intersectionr �NumericZTimer Z_Binaryr r( )r) rH r� r� r? � addressesr� r� r| r� Z schema_name�cols�i�col�ctypeZ ctype_defr" r" r# �_test_get_columns� sH z)ComponentReflectionTest._test_get_columnsc C s | � � d S )N)r� )r) r" r" r# �test_get_columns� s z(ComponentReflectionTest.test_get_columnsc G sD t d| jfdd� t|�D ��� }|�� dd� t| jj��d�D �S )Nr� c S s g | ]\}}t d | |��qS )zt%d)r )r� r� �type_r" r" r# r� � s z<ComponentReflectionTest._type_round_trip.<locals>.<listcomp>c S s g | ] }|d �qS )r� r" )r� rb r" r" r# r� � s )r r! r� �creater r� r� )r) r r� r" r" r# �_type_round_trip� s z(ComponentReflectionTest._type_round_tripc C sH xB| � t�dd��D ],}t|tj�s(t�t|jd� t|jd� qW d S )N� rB )r� r� r� r� r( r Z precisionZscale)r) �typr" r" r# �test_numeric_reflection� s z/ComponentReflectionTest.test_numeric_reflectionc C s4 | � t�d��d }t|tj�s$t�t|jd� d S )N�4 r )r� r� r r� r( r �length)r) r� r" r" r# �test_varchar_reflection� s z/ComponentReflectionTest.test_varchar_reflectionc C s\ t d| jtdtdd�tdtdd��}|�� ttdd� t| jj�� d�D ��ddd �� d S ) Nr� �aT)rC �bFc s s | ]}|d |d fV qdS )rG rC Nr" )r� r� r" r" r# � <genexpr>� s zCComponentReflectionTest.test_nullable_reflection.<locals>.<genexpr>)r� r� ) r r! r r r� r r7 r r� r� )r) r� r" r" r# �test_nullable_reflection� s z0ComponentReflectionTest.test_nullable_reflectionc C s | j tjjd� d S )N)rH )r� r r r< )r) r" r" r# � test_get_columns_with_schema s z4ComponentReflectionTest.test_get_columns_with_schemac C sn t | j�}| jj}t|j�}|�d�}| �t|�dkt|�� x*t|j �D ] \}}t |j || d � qJW d S )Nrn r rG ) r r� r� rn r r� r r� r� r� r rG )r) r� rn r� r� r� r� r" r" r# �test_get_temp_table_columns s z3ComponentReflectionTest.test_get_temp_table_columnsc C s2 t | j�}|�d�}tdd� |D �dddg� d S )Nr� c S s g | ] }|d �qS )rG r" )r� r� r" r" r# r� s zFComponentReflectionTest.test_get_temp_view_columns.<locals>.<listcomp>r rG ro )r r� r� r )r) r� r� r" r" r# �test_get_temp_view_columns s z2ComponentReflectionTest.test_get_temp_view_columnsc C s | j dd� d S )Nr� )r� )r� )r) r" r" r# �test_get_view_columns s z-ComponentReflectionTest.test_get_view_columnsc C s | j tjjdd� d S )Nr� )rH r� )r� r r r< )r) r" r" r# �!test_get_view_columns_with_schema s z9ComponentReflectionTest.test_get_view_columns_with_schemac C s� | j }| jj| jj }}t|j�}|j|j|d�}|d }t|dg� |j|j|d�}|d } t| dg� t j j � � � t|d d� W d Q R X d S )N)rH �constrained_columnsr@ rL rG rQ ) r! r� r? rN r r� Zget_pk_constraintrG r r r: Zreflects_pk_names�fail_if) r) rH r� r? r� r� Z users_consZ users_pkeysZ addr_consZ addr_pkeysr" r" r# �_test_get_pk_constraint% s z/ComponentReflectionTest._test_get_pk_constraintc C s | � � d S )N)r� )r) r" r" r# �test_get_pk_constraint6 s z.ComponentReflectionTest.test_get_pk_constraintc C s | j tjjd� d S )N)rH )r� r r r< )r) r" r" r# �"test_get_pk_constraint_with_schema: s z:ComponentReflectionTest.test_get_pk_constraint_with_schemac C s0 | j }| jj}t|j�}ttjd|j|j � d S )Nz+.*get_primary_keys\(\) method is deprecated) r! r� r? r r� r �sa_excZSADeprecationWarningZget_primary_keysrG )r) r� r? r� r" r" r# � test_deprecated_get_primary_keys@ s z8ComponentReflectionTest.test_deprecated_get_primary_keysc C s: | j }| jj| jj }}t|j�}|}tjjj r�|j |j |d�}|d }tjj � � � t|d d� W d Q R X t|d |� t|d |j � t|d dg� tjjj r�t|d d g� |j |j |d�} | d }tjj� � � | �|d d k � W d Q R X t|d |� t|d |j � t|d dg� t|d d g� d S ) N)rH r rG rF �referred_schema�referred_table�referred_columnsr@ r� rE rO )r! r� r? rN r r� r r: r_ r4 �get_foreign_keysrG Znamed_constraintsr� r Z implicitly_named_constraintsr ) r) rH r� r? r� r� Zexpected_schemaZ users_fkeys�fkey1Z addr_fkeysr" r" r# �_test_get_foreign_keysM s, z.ComponentReflectionTest._test_get_foreign_keysc C s | � � d S )N)r� )r) r" r" r# �test_get_foreign_keysn s z-ComponentReflectionTest.test_get_foreign_keysc C s | j tjjd� d S )N)rH )r� r r r< )r) r" r" r# �!test_get_foreign_keys_with_schemar s z9ComponentReflectionTest.test_get_foreign_keys_with_schemac C s | � dtjjj dtjj dtjj �\}}}ttj�}|�|j �}t t |�d� |d }t |d tjj� t |d |j � t |d d g� t |d d g� |j|j tjjd �}t t |�d� |d }|d d tjjjfks�t �t |d |j � t |d d g� t |d d g� d S )Nz%s.local_tablez%s.remote_tablez%s.remote_table_2� r r� r� r� r r� rV )rH rX ) r� r r% r' rd r r< r r� rG r r� r( ) r) rU rW rY r� Z local_fkeysr� Z remote_fkeysZfkey2r" r" r# �"test_get_inter_schema_foreign_keysw s, z:ComponentReflectionTest.test_get_inter_schema_foreign_keysc C s | j dd� d S )NZCASCADE)Zondelete)� _test_get_foreign_key_options)r) r" r" r# �%test_get_foreign_key_options_ondelete� s z=ComponentReflectionTest.test_get_foreign_key_options_ondeletec C s | j dd� d S )NzSET NULL)Zonupdate)r� )r) r" r" r# �%test_get_foreign_key_options_onupdate� s z=ComponentReflectionTest.test_get_foreign_key_options_onupdatec s | j }td|tdtdd�dd� td|tdtdd�tdttjdd d ��td td ��dd� td |tdtdd�tdtd�dd�tdt�tjdgdgfddi|��dd� |�� t |j �}|� d�d d � t t � fdd�� D ��i � |� d �d d � t t � fdd�� D ��|� d S )N�xr T)r )rI r� Zx_idzx.id�xid)rG �test� �userrG r F)rC �tidztable.idZmyfkr r2 c 3 s" | ]}� | r|� | fV qd S )Nr" )r� �k)�optsr" r# r� � s zHComponentReflectionTest._test_get_foreign_key_options.<locals>.<genexpr>c 3 s" | ]}� | r|� | fV qd S )Nr" )r� r� )r� r" r# r� � s )r! r r r r` r r ZForeignKeyConstraint� create_allr r� r� r r7 )r) r2 r� r� r" )r� r# r� � s8 z5ComponentReflectionTest._test_get_foreign_key_optionsc C sb dd� |D �}xN|D ]F}|d |ks(t �||�|d � }x |D ]}t|| || � q@W qW d S )Nc S s g | ] }|d �qS )rG r" )r� �dr" r" r# r� � s z@ComponentReflectionTest._assert_insp_indexes.<locals>.<listcomp>rG )r( �indexr )r) �indexes�expected_indexesZ index_namesZe_indexr� �keyr" r" r# �_assert_insp_indexes� s z,ComponentReflectionTest._assert_insp_indexesc C sP | j }t|j�}|jd|d�}dddgdd�ddddgd d�g}| �||� d S ) Nr? )rH FrA rD rz )�unique� column_namesrG r@ r{ )r! r r� � get_indexesr� )r) rH r� r� r� r� r" r" r# �_test_get_indexes� s z)ComponentReflectionTest._test_get_indexesc C s | � � d S )N)r� )r) r" r" r# �test_get_indexes� s z(ComponentReflectionTest.test_get_indexesc C s | j tjjd� d S )N)rH )r� r r r< )r) r" r" r# � test_get_indexes_with_schema� s z4ComponentReflectionTest.test_get_indexes_with_schemac C s� | j }t|j�}|�|�}d|d�g}| �||� t|||jd�}tt|j�d� t t |j�d j |� tt |j�d j |� d S )NF)r� rG )� autoload_withr� r ) r! r r� r� r� r r r� r� r �listr� rG )r) �tnameZixnamer� r� r� r� r� r" r" r# �_test_get_noncol_index� s z.ComponentReflectionTest._test_get_noncol_indexc C s | � dd� d S )NrZ r] )r� )r) r" r" r# �test_get_noncol_index_no_pk s z3ComponentReflectionTest.test_get_noncol_index_no_pkc C s | � dd� d S )Nr\ r^ )r� )r) r" r" r# �test_get_noncol_index_pk s z0ComponentReflectionTest.test_get_noncol_index_pkc C s� t d| jtdtd��tdtd��� t�| jdtd�� t�| jdtd�� | j�� t| jj �}t d��" t |� d�d dgd d �g� W d Q R X d S ) Nr� r� rM �yrr z+CREATE INDEX t_idx ON t(lower(x), lower(y))z CREATE INDEX t_idx_2 ON t(x)z>Skipped unsupported reflection of expression-based index t_idxZt_idx_2r )rG r� r� ) r r! r r r rx r r� r r� r r r� )r) r� r" r" r# �%test_reflect_expression_based_indexes s$ z=ComponentReflectionTest.test_reflect_expression_based_indexesc C s | � � d S )N)� _test_get_unique_constraints)r) r" r" r# �test_get_unique_constraints, s z3ComponentReflectionTest.test_get_unique_constraintsc C sF t | j�}|�d�}x|D ]}|�dd � qW t|dgdd�g� d S )Nrn �duplicates_indexrG rp )r� rG )r r� �get_unique_constraints�popr )r) r� � reflected�reflr" r" r# �&test_get_temp_table_unique_constraints0 s z>ComponentReflectionTest.test_get_temp_table_unique_constraintsc C sR t | j�}|�d�}x|D ]}|�dd � qW tdd� |D �ddgdd�g� d S ) Nrn Zdialect_optionsc S s g | ]}|d dkr|�qS )rG rq r" )r� �idxr" r" r# r� D s zGComponentReflectionTest.test_get_temp_table_indexes.<locals>.<listcomp>Fro rq )r� r� rG )r r� r� r r )r) r� r� �indr" r" r# �test_get_temp_table_indexes; s z3ComponentReflectionTest.test_get_temp_table_indexesc C s | j tjjd� d S )N)rH )r r r r< )r) r" r" r# �'test_get_unique_constraints_with_schemaN s z?ComponentReflectionTest.test_get_unique_constraints_with_schemac C s� t ddgd�ddddgd�ddddgd�dd d gd�d dgd�d dgd�gt�d �d�}| j}td|tdt�d��tdt�d��tdtj�td t�d��td t�d��|d�}x*|D ]"}|� tj |d d |d i�� q�W |� � t |j �}t |jd|d�t�d �d�}t� }x>t||�D ]0\} } | �dd �} | �r>|�| � t| | � �qW t� } td| |j |d�}tdd� |jD ��} tdd� |jD ���dg�}| �|��r�t�|�r�t|| � t|t� � d S )NZunique_ar� )rG r� Z unique_a_b_cr� rb Z unique_c_a_bZunique_asc_keyZascr� z i.have.dotsz i have spacesrG )r� ZtesttblrP rM )rH r� r )r� rH c S s g | ] }|j �qS r" )rG )r� r r" r" r# r� � s zHComponentReflectionTest._test_get_unique_constraints.<locals>.<listcomp>c S s g | ]}t |tj�r|j�qS r" )r� r` ru rG )r� Zuqr" r" r# r� � s )r� �operator� itemgetterr! r r r` r r Zappend_constraintru r� r r� r r� r� r �addr r r� � constraints� differencer� r( )r) rH �uniques� orig_metar� Zuc� inspectorr Znames_that_duplicate_index�origr ZdupeZreflected_metadataZ idx_namesZuq_namesr" r" r# r S s^ z4ComponentReflectionTest._test_get_unique_constraintsc C s | � � d S )N)�_test_get_check_constraints)r) r" r" r# �test_get_check_constraints� s z2ComponentReflectionTest.test_get_check_constraintsc C s | j tjjd� d S )N)rH )r r r r< )r) r" r" r# �!test_get_check_constraints_schema� s z9ComponentReflectionTest.test_get_check_constraints_schemac s� | j }td|tdt� �tjddd�tjddd�|d� |�� t|j�}t |j d|d�t � d �d �}d d � � � fd d�|D �}t |ddd�ddd�g� d S )NZsa_ccr� za > 1 AND a < 5Zcc1)rG za = 1 OR (a > 2 AND a < 5)Zcc2)rH rG )r� c S s d� t�d| �� tj��S )N� zand|\d|=|a|or|<|>)�join�re�findall�lower�I)�sqltextr" r" r# � normalize� s zFComponentReflectionTest._test_get_check_constraints.<locals>.normalizec s" g | ]}|d � |d �d��qS )rG r )rG r r" )r� �item)r r" r# r� � s zGComponentReflectionTest._test_get_check_constraints.<locals>.<listcomp>za > 1 and a < 5)rG r za = 1 or a > 2 and a < 5)r! r r r r` ZCheckConstraintr� r r� r� Zget_check_constraintsr r r )r) rH r r r r" )r r# r � s( z3ComponentReflectionTest._test_get_check_constraintsc C sL | j }d}d}t|j�}|j||d�}| �|� |j||d�}| �|� d S )Nr� r� )rH )r! r r� Zget_view_definitionr )r) rH r� Z view_name1Z view_name2r� Zv1Zv2r" r" r# �_test_get_view_definition� s z1ComponentReflectionTest._test_get_view_definitionc C s | � � d S )N)r! )r) r" r" r# �test_get_view_definition� s z0ComponentReflectionTest.test_get_view_definitionc C s | j tjjd� d S )N)rH )r! r r r< )r) r" r" r# �$test_get_view_definition_with_schema� s z<ComponentReflectionTest.test_get_view_definition_with_schema� postgresqlzPG specific featurec C s0 | j }t|j�}|�||�}| �t|t�� d S )N)r! r r� Z get_table_oidr r� �int)r) r| rH r� r� Zoidr" r" r# �_test_get_table_oid� s z+ComponentReflectionTest._test_get_table_oidc C s | � d� d S )Nr? )r& )r) r" r" r# �test_get_table_oid� s z*ComponentReflectionTest.test_get_table_oidc C s | j dtjjd� d S )Nr? )rH )r& r r r< )r) r" r" r# � test_get_table_oid_with_schema� s z6ComponentReflectionTest.test_get_table_oid_with_schemac C sR | j }t|j�}x<dD ]4\}}|�|�}dd� |D �| }|�dd�st�qW dS )a� test that 'autoincrement' is reflected according to sqla's policy. Don't mark this test as unsupported for any backend ! (technically it fails with MySQL InnoDB since "id" comes before "id2") A backend is better off not returning "autoincrement" at all, instead of potentially returning "False" for an auto-incrementing primary key column. ))r? r@ )rN rL )rJ rK c S s i | ]}||d �qS )rG r" )r� rb r" r" r# � <dictcomp> s zBComponentReflectionTest.test_autoincrement_col.<locals>.<dictcomp>Z autoincrementTN)r! r r� r� �getr( )r) r� r� r� Zcnamer� Zid_r" r" r# �test_autoincrement_col� s z.ComponentReflectionTest.test_autoincrement_col)Nr� N)N)Nr� )N)N)N)N)N)N)N)_r+ r, r- Z run_insertsZ run_deletesr. r/ r8 r$ r9 rk rf ri r r: Zschema_reflectionr� r� r� Zprovide_metadatar� r� r� rv rw r� Ztable_reflectionr� Z!foreign_key_constraint_reflectionr� Zcomment_reflectionr� r; r� r� r� rh r� r� r� r� r� r� r� r� r� r� rj r� r� r� r� r� Z!primary_key_constraint_reflectionr� r� r� r� r� r� rc r� Z1foreign_key_constraint_option_reflection_ondeleter� Z1foreign_key_constraint_option_reflection_onupdater� r� r� r� re r� r� r� r� r� Zindexes_with_expressionsr� Z unique_constraint_reflectionr r r r r Zcheck_constraint_reflectionr r r r! r" r# Zonly_onr& r' r( r+ r" r" r" r# r0 2 s� $ % 8 "- N ' r0 c @ s0 e Zd ZdZdZedd� �Zdd� Zdd� Zd S ) �NormalizedNameTest)Zdenormalized_namesTc C sL t tddd�|tdtdd�� t tddd�|tdtdd�tdtd��� d S ) N�t1T)�quoter )r �t2�t1idzt1.id)r r r r r )r r! r" r" r# r$ s z NormalizedNameTest.define_tablesc C s� t tj�}ttddd�|dd�}|jd }|jj�|jj �s@t �t tj�}|j dd� d� |jd jj�|jd jj �s|t �d S ) Nr/ T)r. )Zautoloadr- c S s | � � dkS )N)r- r/ )r )rG �mr" r" r# �<lambda>, � zINormalizedNameTest.test_reflect_lowercase_forced_tables.<locals>.<lambda>)�only) r r r% r r r� rb r0 Z referencesr r( Zreflect)r) �m2Zt2_refZt1_refZm3r" r" r# �$test_reflect_lowercase_forced_tables$ s z7NormalizedNameTest.test_reflect_lowercase_forced_tablesc C sP dd� t tj��� D �}t|d �� |d �� � t|d �� |d �� � d S )Nc S s g | ]}|� � d kr|�qS ))r- r/ )r )r� r� r" r" r# r� 1 s z;NormalizedNameTest.test_get_table_names.<locals>.<listcomp>r r� )r r r% r� r �upperr )r) Z tablenamesr" r" r# r� / s z'NormalizedNameTest.test_get_table_namesN) r+ r, r- � __requires__r. r/ r$ r6 r� r" r" r" r# r, s r, )r0 r r, )'r r r5 r` r> r r r r r r r rH r r r r r� r r r r r r r r� Zengine.reflectionr r r Z sql.elementsr r! r? Z TablesTestr r0 r, �__all__r" r" r" r# �<module> sH g(