관리-도구
편집 파일: plugin_base.cpython-38.pyc
U -?�fJT � @ st U d dl mZ d dlZd dlmZ d dlZd dlad dlZd dlm Z d dl Z d dl Z d dl m Z d dlmZ dZt�d�ZdadadadadadadadZdada da da dae � Z e � Z!da"d e#d < d d � Z$d d� Z%dd� Z&dd� Z'dd� Z(dd� Z)dd� Z*dd� Z+dd � Z,d d � Z-d d � Z.d!d"� Z/d#d$� Z0d%d&� Z1d'd(� Z2g Z3g Z4d)d*� Z5d+d,� Z6e5d-d.� �Z7e5d/d0� �Z8e6d1d2� �Z9e6d3d4� �Z:e5d5d6� �Z;e6d7d8� �Z<e6d9d:� �Z=d;d<� Z>e6d=d>� �Z?e6d?d@� �Z@e6dAdB� �ZAdCdD� ZBdEdF� ZCdGdH� ZDdIdJ� ZEdKdL� ZFdMdN� ZGdOdP� ZHdQdR� ZIdSdT� ZJdUdV� ZKdWdX� ZLdYdZ� ZMded[d\�ZNd]d^� ZOd_d`� ZPG dadb� dbejQ�ZRdaSdcdd� ZTdS )f� )� annotationsN)� Namespace)�Path)�Any)�asyncioFz sqlalchemy.testing.plugin_baser �optionsc C s� | ddt tdd� | ddt tdd� | ddt d d d � | d dtd d� | ddt ddd � | ddt ddd � | ddddd� | dddd � | d dtd �d d� | d dtd!�d"d� | d#dtd$�d%d� | d&dtd'�d(d� | d)t d*d+d,d-� | d.t d/d0d1� | d2dd3d4d� | d5t d6d7d1� | d8dt td9d� | d:dtt d;d<� | d=dtt d>d<� | d?dd@dAdBdC� | dDddEdAdFdC� | dGt dHdId1� | dJt dg dKdLdM� | dNt dOdP� | dQddRd � d S )SNz --log-info�callbackz,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 --dbdriver�dbdriverz|Additional database drivers to include in tests. These are linked to the existing database URLs by the provisioning system.z --dropfirst� store_true� dropfirstz,Drop all tables in the target database first)r r r z--disable-asyncioz8disable test / fixtures / provisoning running in asyncio)r r z--backend-only�backendz}Run only tests marked with __backend__ or __sparse_backend__; this is now equivalent to the pytest -m backend mark expressionz --nomemoryZmemory_intensivezpDon't run memory profiling tests; this is now equivalent to the pytest -m 'not memory_intensive' mark expressionz--notimingintensiveZtiming_intensivezpDon't run timing intensive tests; this is now equivalent to the pytest -m 'not timing_intensive' mark expressionz--nomypyZmypyz_Don't run mypy typing tests; this is now equivalent to the pytest -m 'not mypy' mark expressionz--profile-sortZ cumulative� profilesortz*Type of sort for profiling standard output)r �defaultr r z--profile-dump� profiledumpz2Filename where a single profile run will be dumped)r 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 usedz--requirementsz3requirements class for testing, overrides setup.cfgz --include-tagzAInclude tests with tag <tag>; legacy, use pytest -m 'tag' instead)r r r r z --exclude-tagzEExclude tests with tag <tag>; legacy, use pytest -m 'not tag' insteadz--write-profilesZwrite_profilesFz$Write/update failing profiling data.)r r r r z--force-write-profilesZforce_write_profilesz,Unconditionally write/update profiling data.z--dump-pyannotateZdump_pyannotatez/Run pyannotate and dump json info to given filez--mypy-extra-test-pathZmypy_extra_test_pathszlAdditional test directories to add to the mypy tests. This is used only when running mypy tests. Multiple OK)r r r r r z--postgresql-templatedbz^name of template database to use for PostgreSQL CREATE DATABASE (defaults to current database))r r z--oracledb-thick-modez:enables the 'thick mode' when testing with oracle+oracledb)�str�_log� _list_dbs�_set_tag_include�_set_tag_exclude�_requirements_opt� _include_tag� _exclude_tag)� make_option� r" �U/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/plugin/plugin_base.py� setup_options; 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 dS )z�Store important configuration we will need to send to a follower. This invokes in the parent process after normal config is set up. Hook is currently not used. Nr" �Zdict_r" r" r# �!memoize_important_follower_config� s r+ c C s dS )z�Restore important configuration needed by a follower. This invokes in the follower process. Hook is currently not used. Nr" r* r" r" r# �!restore_important_follower_config s r, c C s* t �� at�t| d �t| d �g� d S )Nz setup.cfgztest.cfg)� configparser� ConfigParser� file_config�readr )� root_pathr" r" r# � read_config s �r2 c C s | a tD ]}|t t� qdS )z7things to set up early, before coverage might be setup.N)r � pre_configurer/ )�opt�fnr" r" r# � pre_begin s r6 c C s | t _d S �N)r Z has_coverage)�valuer" r" r# �set_coverage_flag s r9 c C sp t D ]} | tt� qddlma ddlmamama ddlm a m a m a ddlm a m a ddlma t �� dS )z9things to set up later, once we know coverage is running.r ��testing)�fixtures�engines� exclusions)� assertions�warnings� profiling)�configr& )�utilN)�post_configurer r/ � sqlalchemyr; r' r< r= r> r? r@ rA rB r& rC Z setup_filters�r5 r" r" r# � post_begin$ s rG 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_strr8 �parserr" r" r# r 6 s r c G sV t d krtt�� � td� tt �d��D ] }td|t �d|�f � q*t� d� d S )Nz0Available --db options (use --dburi to override)r z%20s %sr ) r/ r2 r �cwd�print�sortedr �get�sys�exit)�args�macror" r" r# r C s r c C s t |� d S r7 )�_setup_requirementsrO r" r" r# r N s r c s � fdd�}|S )Nc s t | � |� d S r7 )r rO ��tagr" r# �_do_include_tagS s z)_set_tag_include.<locals>._do_include_tagr" )r\ r] r" r[ r# r R s r c s � fdd�}|S )Nc s t | � |� d S r7 )r rO r[ r" r# �_do_exclude_tagZ s z)_set_tag_exclude.<locals>._do_exclude_tagr" )r\ r^ r" r[ r# r Y s r c C s t �|�dd�� d S �N�-�_)� exclude_tags�add�replacerO r" r" r# r ` s r c C s t �|�dd�� d S r_ )� include_tagsrc rd rO r" r" r# r d s r c C s t �| � | S r7 )r3 r rF r" r" r# �prel s rf c C s t �| � | S r7 )rD r rF r" r" r# �postq s rg c C s | a d S r7 �r �r4 r/ r" r" r# �_setup_optionsv s rj c C s, ddl m} |�ddd� |�ddd� d S )Nr )�registryzsqlite.pysqlite_numericz#sqlalchemy.dialects.sqlite.pysqliteZ _SQLiteDialect_pysqlite_numericzsqlite.pysqlite_dollarZ _SQLiteDialect_pysqlite_dollar)Zsqlalchemy.dialectsrk �register)r4 r/ rk r" r" r# � _register_sqlite_numeric_dialect| s ��rm c C s^ t j�dd�dkrZddlm} z|dd� W n, tk rX } ztd�|�W 5 d }~X Y nX d S ) NZREQUIRE_SQLALCHEMY_CEXT�0�1r )�has_compiled_extT)Zraise_zEREQUIRE_SQLALCHEMY_CEXT is set but can't import the cython extensions)�os�environrU Zsqlalchemy.utilrp � ImportError�AssertionError)r4 r/ rp �errr" r" r# � __ensure_cext� s ��rv c C s ddl m} t� |_d S �Nr �rB )r' rB �_fixture_fn_classZ_fixture_functions)r r/ rB r" r" r# � _init_symbols� s rz c C s | j r dt_d S )NF)Zdisable_asyncior ZENABLE_ASYNCIOri r" r" r# �_set_disable_asyncio� s r{ c sp ddl m} ddlm} ddlm} ddlm� | jrBt| j�}ng }| j pNg }| j r�| j D ]D}t � d|�D ]2}||� d�kr�td| ��ql|�|�d|�� qlq\|s�|�|�dd �� d |_| j�r |j�r � fd d �|D �D ]<} t| jd ��&} | �|j� d | jdd�� d�� W 5 Q R X q�t|�||��} | D ]8} t�d| � |�| | ||j�} |j�s2| �| |� �q2d S )Nr r: rx r% )�urlz[,\s]+r z:Unknown URI specifier '%s'. Specify --dbs for known uris.r c s g | ]}� � |��qS r" )Zmake_url)�.0�db_url�Zsa_urlr" r# � <listcomp>� s z _engine_uri.<locals>.<listcomp>�a� F)Z hide_password� zAdding database URL: %s) rE r; r' rB r& Zsqlalchemy.enginer| r �listr r �re�splitr � RuntimeErrorr rU �_currentr r( �open�writeZrender_as_stringZgenerate_db_urls�log�infoZ setup_configZset_as_current) r r/ r; rB r& Zdb_urlsZ extra_driversZdb_tokenr r~ �file_Z expanded_urls�cfgr" r r# � _engine_uri� sL ��� �r� c C s |� dd�}t|� d S )N� sqla_testing�requirement_cls)rU rZ )r r/ r� r" r" r# � _requirements� s r� c C sr ddl m} ddlm} | �d�\}}t|�}|�d�dd � D ]}t||�}q@t||�}|� |_|_t |_ d S )Nr rx r: �:�.� ) r' rB rE r; r� � __import__�getattr� requirements�requires�bootstrapped_as_sqlalchemy)�argumentrB r; �modnameZclsname�mod� componentZreq_clsr" r" r# rZ � s rZ c C s@ ddl m} | jr<ddl m} |j�� D ]}|�||j� q(d S )Nr rx r% )r' rB r r& �Config� all_configsZdrop_all_schema_objectsr )r r/ rB r& r� r" r" r# �_prep_testing_database� s r� c C s ddl m} t|_||_d S rw )r' rB r r/ )r4 r/ rB r" r" r# �_post_setup_options� s r� c C s. ddl m} |j|�dd�| j| jd�|_d S )Nr )rA r� Z profile_file)�sort�dump)r' rA ZProfileStatsFilerU r r Z_profile_stats)r r/ rA r" r" r# �_setup_profiling s �r� c C s& t |tj�sdS | �d�r dS dS d S )NFra T)� issubclassr< ZTestBase� startswith)�name�clsr" r" r# � want_class s r� c C s& |j �d�sdS |jd kr dS dS d S )NZtest_FT)�__name__r� � __module__)r� r5 r" r" r# � want_method s r� c c s� d|ksd|kr�d|k}t | |d�D ]\}| j}t�dd|j�}t�dd|�}d| j|f }t|| f||d ��}t|||� |V q$n| V d S ) Nr Zsparse_backend)�sparsez [_\[\]\.]+ra z_+$� z%s_%s)�_sa_orig_cls_name�__only_on_config__)�_possible_configs_for_clsr� r� �subr� r �setattr) r� �module�markersr� r� � orig_nameZ alpha_namer� �subclsr" r" r# �generate_sub_tests! s � r� c C s t | � t| � d S r7 )� _do_skips� _setup_engine�r� r" r" r# �!start_test_class_outside_fixtures9 s r� c C s t �| � tj�� d S r7 )r< Z stop_test_class_inside_fixturesr= �testing_reaperr� r" r" r# �stop_test_class> s r� c C s: t j�� t�ttj| � ztjs*t � � W 5 t� X d S r7 ) r= r� � stop_test_class_outside_fixturesr& rB r �_restore_enginer r r? �global_cleanup_assertionsr� r" r" r# r� G s r� c C s t jrt j�t� d S r7 )rB r� �resetr; r" r" r" r# r� Q s r� c C s t j�� t�� t� d S r7 )r= r� Z final_cleanupr? r� r� r" r" r" r# �final_process_cleanupV s r� c C s< t | dd �r8t| j�}d|d< tj|d�}tj�|t� d S )N�__engine_options__�class�scoperh ) r� �dictr� r= Ztesting_enginerB r� Z push_enginer; )r� �optsZengr" r" r# r� \ s r� c C s* t |d|j�}d|||f }t�|� d S )Nr� z%s.%s.%s)r� r� rA Z_start_current_test)�testZtest_module_nameZ test_classZ test_namer� Zid_r" r" r# � before_testd s r� c C s t �� tj�� d S r7 )r< � after_testr= r� �r� r" r" r# r� o s r� c C s t j�| � d S r7 )r= r� Zafter_test_outside_fixturesr� r" r" r# �after_test_fixturest s r� c C s� t tj�� �}| jr@tj| j� }t|�D ]}||�r(|�|� q(t | dd �r~tjt � | j �� }t|�D ]}||�sf|�|� qft | dd �r�|� | jg� t| d�r�tj}t|�D ]H}| jD ]<}t ||�}|�|�} | r�|�|� |d k r�|�| � q�q�q�t| d��rft � } tj}t|�D ]4}| jD ]&}t ||�}|�|��s$| �|� �q$�q|�| ��rf|�| � |�r�i } tt|dd� d��D ] } | jj } | | k�r�| | | < �q�| � � S |S )N� __only_on__r� � __requires__�__prefer_requires__c S s | j j| j j| j jjfS r7 )r r� �driver�dialectZserver_version_info)r� r" r" r# �<lambda>� s �z+_possible_configs_for_cls.<locals>.<lambda>)�key) �setrB r� r� Z__unsupported_on__r> �db_specr� �remover� rC �to_listr� �intersection_updater� �hasattrr� r� Zmatching_config_reasons�extendr� Zenabled_for_configrc � difference�difference_update�reversedrT r r� �values)r� �reasonsr� r� �spec� config_objr� � requirement�checkZ skip_reasons� non_preferredZ per_dialectr� r r" r" r# r� x s^ �� r� c C s� g }t | |�}t| dd�rFt| d�D ] }|� r$t�d| j|jf � q$|s�d| j| jd�dd� tj�� D ��d�|�f }t�|� nRt | d�r�t � }t j t �| j�� }|D ]}||�s�|�|� q�|�|�r�|�|� tj|kr�t|�� | � d S ) NZ __skip_if__Fz'%s' skipped by %sz1'%s.%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'r� c s s | ]}t |�V qd S r7 )r )r} �digr" r" r# � <genexpr>� s �z&_do_skips.<locals>.<genexpr>.<genexpr>N)r r� �joinr> Z_server_versionr� )r} r� r" r" r# r� � s � ���z _do_skips.<locals>.<genexpr>�__prefer_backends__)r� r� rB Z skip_testr� r� r� r� r� r� r� r> r� rC r� r� rc r� r� r� � _setup_config�pop)r� r� r� �c�msgr� r� r� r" r" r# r� � s8 � � � r� c C s t j�| t� d S r7 )rB r� �pushr; )r� �ctxr" r" r# r� � s r� c @ sn e Zd Zejdd� �Zejdd� �Zejdd� �Zejdd� �Zd d � Z ejd d �d d��Z ej dd� �Z dS )�FixtureFunctionsc O s t � �d S r7 ��NotImplementedError��self�arg�kwr" r" r# �skip_test_exception� s z$FixtureFunctions.skip_test_exceptionc O s t � �d S r7 r� �r� rX r� r" r" r# � combinations� s z FixtureFunctions.combinationsc O s t � �d S r7 r� r� r" r" r# � param_ident� s z FixtureFunctions.param_identc O s t � �d S r7 r� r� r" r" r# �fixture� s zFixtureFunctions.fixturec C s t � �d S r7 r� �r� r" r" r# �get_current_test_name� s z&FixtureFunctions.get_current_test_namer )�returnc C s t � �d S r7 r� r� r" r" r# �mark_base_test_class� s z%FixtureFunctions.mark_base_test_classc C s t � �d S r7 r� r� r" r" r# � add_to_marker s z FixtureFunctions.add_to_markerN) r� r� � __qualname__�abc�abstractmethodr� r� r� r� r� r �abstractpropertyr r" r" r" r# r� � s r� c C s | a d S r7 )ry )Zfixture_fn_classr" r" r# �set_fixture_functions s r )NF)U� __future__r r �argparser r- rH rq �pathlibr r� rV �typingr r' r r� rK r� r< r= r> r@ rA r& r? r� rB r; rC r/ r� re rb r �__annotations__r$ r) r+ r, r2 r6 r9 rG r r r r r r r r3 rD rf rg rj rm rv rz r{ r� r� rZ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �ABCr� ry r r" r" r" r# �<module> s� 6 1 B+