관리-도구
편집 파일: plugin_base.cpython-37.pyc
B ��4]�L � @ s� d Z ddlmZ ddlZddlZejdkZer8ddlZnddlZda da da da da dadZdadadadadae� ae� adadd� Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z da dd� Z!dd� Z"dd� Z#dd� Z$dd � Z%d d � Z&d d � Z'g Z(g Z)d!d"� Z*d#d$� Z+e*d%d&� �Z,e*d'd(� �Z-e*d)d*� �Z.e+d+d,� �Z/e+d-d.� �Z0e+d/d0� �Z1d1d2� Z2e+d3d4� �Z3e+d5d6� �Z4e+d7d8� �Z5e+d9d:� �Z6d;d<� Z7d=d>� Z8d?d@� Z9dAdB� Z:dCdD� Z;dEdF� Z<dGdH� Z=dIdJ� Z>dKdL� Z?dMdN� Z@dVdPdQ�ZAdRdS� ZBdTdU� ZCdS )Wz�Testing extensions. this module is designed to work as a testing-framework-agnostic library, created so that multiple test frameworks can be supported at once (mostly so that we can migrate to new ones). The current target is py.test. � )�absolute_importN)� r c C s8 | dddt dd� | dddt dd� | dd dd d d � | d dtdd� | dd dddd � | ddddd� | ddddd� | ddd d d� | d dd d � | d!dd"d#d� | d$dd%d&d'� | d(dd)d*d+d,� | d-ddtd.d� | d/dd0d*d1d,� | d2dtdd3d4� | d5dtdd6d4� | d7dd8d*d9d,� | d:dd;d*d<d,� d S )=Nz --log-info�callback�stringz,turn on info logging for <LOG> (multiple OK))�action�typer �helpz --log-debugz-turn on debug logging for <LOG> (multiple OK)z--db�append�dbzBUse prefab database uri. Multiple OK, first one is run by default.)r r �destr z--dbszList available prefab dbs)r Zzeroarg_callbackr z--dburi�dburiz8Database uri. Multiple OK, first one is run by default.z --dropfirst� store_true� dropfirstz,Drop all tables in the target database first)r r r z--backend-only� backend_onlyz<Run only tests marked with __backend__ or __sparse_backend__z --nomemory�nomemoryz Don't run memory profiling testsz--postgresql-templatedbz^name of template database to use for PostgreSQL CREATE DATABASE (defaults to current database))r r z--low-connections�low_connectionsz>Use a low number of distinct connections - i.e. for Oracle TNSz--write-idents� write_identszCwrite out generated follower idents to <file>, when -n<num> is used)r r r z --reversetop� reversetopFzTUse a random-ordering set implementation in the ORM (helps reveal dependency issues))r r �defaultr z--requirementsz3requirements class for testing, overrides setup.cfgz--with-cdecimal�cdecimalzDMonkeypatch the cdecimal library into Python 'decimal' for all testsz --include-tagz Include tests with tag <tag>)r r r r z --exclude-tagz Exclude tests with tag <tag>z--write-profilesZwrite_profilesz$Write/update failing profiling data.z--force-write-profilesZforce_write_profilesz,Unconditionally write/update profiling data.)�_log� _list_dbs�_requirements_opt� _include_tag� _exclude_tag)Z make_option� r �X/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/plugin/plugin_base.py� setup_options2 s� r c C s ddl m} | |_dS )z�Configure required state for a follower. This invokes in the parent process and typically includes database creation. r )� provisionN)�sqlalchemy.testingr �FOLLOWER_IDENT)Zfollower_identr r r r �configure_follower� s r! c C s t td�| d< dS )aT Store important configuration we will need to send to a follower. This invokes in the parent process after normal config is set up. This is necessary as py.test seems to not be using forking, so we start with nothing in memory, *but* it isn't running our argparse callables, so we have to just copy all of that over. )� include_tags� exclude_tags�memoized_configN)r" r# )�dict_r r r �!memoize_important_follower_config� s r& c C s( t �| d d � t�| d d � dS )zfRestore important configuration needed by a follower. This invokes in the follower process. r$ r" r# N)r" �updater# )r% r r r �!restore_important_follower_config� s r( c C s t �� at�ddg� d S )Nz setup.cfgztest.cfg)� configparser� ConfigParser� file_config�readr r r r � read_config� s r- c C s | a xtD ]}|t t� q W dS )z7things to set up early, before coverage might be setup.N)�options� pre_configurer+ )�opt�fnr r r � pre_begin� s r2 c C s | t _d S )N)r. Z has_coverage)�valuer r r �set_coverage_flag� s r4 c C s | a d S )N)�_skip_test_exception)�excr r r � set_skip_test� s r7 c C sp xt D ]} | tt� qW ddlma ddlmamama ddlm a m a m a ddlm a ddlm a t �� dS )z9things to set up later, once we know coverage is running.r )�testing)�fixtures�engines� exclusions)� assertions�warnings� profiling)�config)�utilN)�post_configurer. r+ � sqlalchemyr8 r r9 r: r; r< r= r> r? r@ Z setup_filters)r1 r r r � post_begin� s rC c C sR t sdd l a t �� | �d�r2t �|��t j� n | �d�rNt �|��t j� d S )Nr z-infoz-debug)�logging� basicConfig�endswith� getLogger�setLevel�INFO�DEBUG)�opt_strr3 �parserr r r r s r c G sF t d� x.tt�d��D ] }t d|t�d|�f � qW t�d� d S )Nz0Available --db options (use --dburi to override)r z%20s %sr )�print�sortedr+ r. �get�sys�exit)�args�macror r r r s r c C s t |� d S )N)�_setup_requirements)rK r3 rL r r r r s r c C s t �|�dd�� d S )N�-�_)r# �add�replace)rK r3 rL r r r r s r c C s t �|�dd�� d S )NrU rV )r" rW rX )rK r3 rL r r r r s r c C s t �| � | S )N)r/ r )r1 r r r �pre( s rY c C s t �| � | S )N)rA r )r1 r r r �post- s rZ c C s | a d S )N)r. )r0 r+ r r r �_setup_options2 s r[ c C s | j rt�d� d S )NZmemory_intensive)r r# rW )r0 r+ r r r � _set_nomemory8 s r\ c C s | j rdd l }|tjd< d S )Nr Zdecimal)r rP �modules)r. r+ r r r r �_monkeypatch_cdecimal> s r^ c C s ddl m} t|_d S )Nr )r? )r r? r5 )r. r+ r? r r r �_init_skiptestF s r_ c C s" ddl m} ddlm} ddl m} | jr6t| j�}ng }| jr�xR| jD ]H}xBt� d|�D ]2}||� d�krzt d| ��qZ|� |� d|�� qZW qHW |s�|� |� dd�� d |_xl|D ]d}| jr�|jr�t| jd �� } | �|jd | d � W d Q R X |�|| ||j�} |js�| �| |� q�W d S ) Nr )r? )r8 )r z[,\s]+r z:Unknown URI specifier '%s'. Specify --dbs for known uris.r �a� � )r r? rB r8 r r �listr �re�splitr. � RuntimeErrorr rO �_currentr r �open�writeZ setup_configZset_as_current) r. r+ r? r8 r Zdb_urlsZdb_tokenr Zdb_urlZfile_�cfgr r r � _engine_uriM s2 "rk c C s |� dd�}t|� d S )N� sqla_testing�requirement_cls)rO rT )r. r+ rm r r r � _requirementsu s rn c C s~ ddl m} ddlm} |jd k r&d S | �d�\}}t|�}x$|�d�dd � D ]}t||�}qPW t||�}|� |_|_d S )Nr )r? )r8 �:�.� ) r r? rB r8 � requirementsre � __import__�getattr�requires)�argumentr? r8 �modnameZclsname�modZ componentZreq_clsr r r rT | s rT c C s� ddl m}m} ddlm} ddlm}m} | j�r��xN|j � � D �]>}|j }||�} y | � � } W n t k rv Y n,X x(| D ] } |�|�|�| |�� ��� q~W |jj�|��ry| j dd�} W n t k r� Y n0X x,| D ]$} |�|�|j| |�� dd��� q�W |�|| � |jj�|��r4|j|| |jd� ||d�rBddlm} x6| �d �D ](} |�| �| j| d | d d ��� �qVW qBW d S ) Nr )r? r@ )�against)�schema�inspect� test_schema)rz � postgresql)r} �*�namerz )r rz ) r r? r@ Z sqlalchemy.testing.exclusionsry rB rz r{ r �Config� all_configsr Zget_view_names�NotImplementedError�executeZ _DropViewZTableZMetaDatarr Zschemas�enabled_for_configZdrop_all_tablesr| Zsqlalchemy.dialectsr} Z get_enumsZ DropEnumTypeZENUM)r. r+ r? r@ ry rz r{ rj �eZ inspectorZ view_namesZvnamer} �enumr r r �_prep_testing_database� sJ r� c C s | j rddlm} |� d S )Nr )�randomize_unitofwork)r Zsqlalchemy.orm.utilr� )r. r+ r� r r r �_reverse_topological� s r� c C s ddl m} t|_||_d S )Nr )r? )r r? r. r+ )r0 r+ r? r r r �_post_setup_options� s r� c C s$ ddl m} |�|�dd��|_d S )Nr )r> rl Z profile_file)r r> ZProfileStatsFilerO Z_profile_stats)r. r+ r> r r r �_setup_profiling� s r� c C sL t | tj�sdS | j�d�r dS tjjrDt| dd�sDt| dd�sDdS dS d S )NFrV � __backend__�__sparse_backend__T) � issubclassr9 ZTestBase�__name__� startswithr? r. r rt )�clsr r r � want_class� s r� c C s� |j �d�sdS |jd kr dS trXt| d�r@t�| j��tt �pVt|d�oV|j �tt �S t rzt| d�rzt�| j��tt �S t r�t|d�r�|j �tt �S dS d S )NZtest_F�__tags__�_sa_exclusion_extendT) r� r� � __module__r" �hasattrr; Ztagsr� Z include_testr# r� )r� r1 r r r � want_method� s" r� c c s� t | dd�st | dd�r�t | dd�}xtt| |d�D ]\}| j}t�dd|j�}t�dd|�}d | j|f }t|| f||d ��}t|||� |V q2W n| V d S ) Nr� Fr� )�sparsez [_\[\]\.]+rV z_+$� z%s_%s)�_sa_orig_cls_name�__only_on_config__)rt �_possible_configs_for_clsr� rd �subr r �setattr)r� �moduler� rj � orig_nameZ alpha_namer �subclsr r r �generate_sub_tests s r� c C s t | � t| � d S )N)� _do_skips� _setup_engine)r� r r r �start_test_class% s r� c C s* t j�� ztjst�� W d t� X d S )N)r: �testing_reaperZ_stop_test_ctxr. r r< �global_cleanup_assertions�_restore_engine)r� r r r �stop_test_class* s r� c C s t j�t� d S )N)r? rg �resetr8 r r r r r� 5 s r� c C s t j�� t�� t� d S )N)r: r� Z_stop_test_ctx_aggressiver< r� r� r r r r �final_process_cleanup9 s r� c C s, t | dd �r(tj| jd�}tj�|t� d S )N�__engine_options__)r. )rt r: Ztesting_enginer� r? rg Z push_enginer8 )r� Zengr r r r� ? s r� c C s& t |d|j�}d|||f }|t_d S )Nr� z%s.%s.%s)rt r� r> Z _current_test)�testZtest_module_nameZ test_classZ test_namer Zid_r r r � before_testE s r� c C s t j�� d S )N)r: r� Z_after_test_ctx)r� r r r � after_testQ s r� Fc C s� t tj�� �}| jrDtj| j� }x"t|�D ]}||�r*|�|� q*W t | dd �r�tjt � | j �� }x"t|�D ]}||�sl|�|� qlW t | dd �r�|� | jg� t| d��r tj}xXt|�D ]L}xF| jD ]<}t ||�}|�|�} | r�|�|� |d k �r|�| � P q�W q�W t| d��r�t � } tj}xDt|�D ]8}x0| jD ]&}t ||�}|�|��s:| �|� �q:W �q.W |�| ��r�|�| � |�r�i } x8tt|dd� d��D ] } | jj } | | k�r�| | | < �q�W | � � S |S )N� __only_on__r� � __requires__�__prefer_requires__c S s | j j| j jjfS )N)r r ZdialectZserver_version_info)rj r r r �<lambda>� s z+_possible_configs_for_cls.<locals>.<lambda>)�key) �setr? r� r� Z__unsupported_on__r; �db_specrc �removert r@ �to_listr� �intersection_updater� r� rr r� Zmatching_config_reasons�extendr� r� rW � difference�difference_update�reversedrN r r �values)r� �reasonsr� r� �spec� config_objrr � requirement�checkZ skip_reasons� non_preferredZ per_dialectrj r r r r r� U sZ r� c C s� g }t | |�}t| dd�rJx.t| d�D ] }|� r&t�d| j|jf � q&W |s�d| jd�dd� tj�� D ��d�|�f }t�|� nVt| d�r�t � }t j t � | j�� }x |D ]}||�s�|�|� q�W |�|�r�|�|� tj|kr�t|�� | � d S ) NZ __skip_if__Fz'%s' skipped by %sz.'%s' unsupported on any DB implementation %s%sz, c s s< | ]4}d |j jd�dd� t�|j �D ��|j jf V qdS )z '%s(%s)+%s'rp c s s | ]}t |�V qd S )N)�str)�.0�digr r r � <genexpr>� s z&_do_skips.<locals>.<genexpr>.<genexpr>N)r r �joinr; Z_server_versionZdriver)r� r� r r r r� � s z _do_skips.<locals>.<genexpr>�__prefer_backends__)r� rt r? Z skip_testr� r� r� r� r� r� r; r� r@ r� r� rW r� r� rg � _setup_config�pop)r� r� r� �c�msgr� r� r� r r r r� � s2 r� c C s t j�| t� d S )N)r? rg �pushr8 )r� �ctxr r r r� � s r� )NF)D�__doc__� __future__r rd rP � version_infoZpy3kr) r* r9 r: r; r= r> r< rr r? r8 r@ r+ rD r� r" r# r. r r! r& r( r- r2 r4 r5 r7 rC r r r r r r/ rA rY rZ r[ r\ r^ r_ rk rn rT r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> s� ( 9 A*