Edit File: plugin_base.cpython-37.opt-1.pyc
B �M!V�C � @ s d Z ddlmZ yddlmZ W n ek r@ ddlmZ Y nX ddlZddl Z ej dkZ e rjddl Z nddl Z dadadZdadadadZdadadadai ae� ae� a da dd� Z d d � Z 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*g Z+g Z,d!d"� Z-d#d$� Z.e-d%d&� �Z/e-d'd(� �Z0e-d)d*� �Z1e.d+d,� �Z2e.d-d.� �Z3e.d/d0� �Z4d1d2� Z5e.d3d4� �Z6e.d5d6� �Z7e.d7d8� �Z8d9d:� Z9d;d<� Z:d=d>� Z;d?d@� Z<dAdB� Z=dCdD� Z>dEdF� Z?dGdH� Z@dIdJ� ZAdQdKdL�ZBdMdN� ZCdOdP� ZDdS )RaG Testing extensions. this module is designed to work as a testing-framework-agnostic library, so that we can continue to support nose and also begin adding new functionality via py.test. NOTE: copied/adapted from SQLAlchemy master for backwards compatibility; this should be removable when Alembic targets SQLAlchemy 1.0.0 � )�absolute_import)�SkipTest)�SkippedN)� r c C s | dddt dd� | dddt dd� | dd dd d d � | d dtdd� | dd dddd � | ddddd� | ddddd� | ddd d d� | d dd d d� | d!dd"d#d$d%� | d&ddtd'd� | d(dd)d#d*d%� | d+dtdd,d-� | d.dtdd/d-� | d0dd1d2� | d3d4d5d d6d%� d S )7Nz --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 r r 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__z --mockpool�mockpoolz0Use mock pool (asserts only one connection used)z--low-connectionsZlow_connectionsz>Use a low number of distinct connections - i.e. for Oracle TNSz --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 --serversidez"Turn on server side cursors for PG)r r z--mysql-engineZstoreZ mysql_enginez\Use the specified MySQL storage engine for all tables, default is a db-default/InnoDB combo.)�_log� _list_dbs�_requirements_opt� _include_tag� _exclude_tag)Z make_option� r �S/opt/alt/python37/lib/python3.7/site-packages/alembic/testing/plugin/plugin_base.py� setup_options9 sP 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)�alembic.testingr �FOLLOWER_IDENT)Zfollower_identr r r r �configure_followerj s r! c C s t ttd�| 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. )�db_opts� include_tags� exclude_tags�memoized_configN)r" r# r$ )�dict_r r r �!memoize_important_follower_configu s r' c C s: t �| d d � t�| d d � t�| d d � dS )zfRestore important configuration needed by a follower. This invokes in the follower process. r% r" r# r$ N)r" �updater# r$ )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 r3 c C s | t _d S )N)r/ Z has_coverage)�valuer r r �set_coverage_flag� s r5 c C sT xt D ]} | tt� qW ddlmamama ddlmama ddl m a t� � dS )z9things to set up later, once we know coverage is running.r )�config�warnings� exclusions)�engines�fixtures)�utilN) �post_configurer/ r, r r6 r7 r8 r9 r: � sqlalchemyr; Z setup_filters)r2 r r r � post_begin� s r> c C sR t sdd l a t �� | �d�r2t �|��t j� n | �d�rNt �|��t j� d S )Nr z-infoz-debug)�loggingZ basicConfig�endswithZ getLoggerZsetLevel�INFO�DEBUG)�opt_strr4 �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)�argsZmacror r r r � s r c C s t |� d S )N)�_setup_requirements)rC r4 rD r r r r � s r c C s t �|�dd�� d S )N�-�_)r$ �add�replace)rC r4 rD r r r r � s r c C s t �|�dd�� d S )NrL rM )r# rN rO )rC r4 rD r r r r � s r c C s t �| � | S )N)r0 r )r2 r r r �pre� s rP c C s t �| � | S )N)r< r )r2 r r r �post� s rQ c C s | a d S )N)r/ )r1 r, r r r �_setup_options� s rR c C s | j rdtd< d S )NTZserver_side_cursors)Z serversider" )r/ r, r r r �_server_side_cursors� s rS c C s | j rdd l }|tjd< d S )Nr Zdecimal)r rH �modules)r/ r, r r r r �_monkeypatch_cdecimal� s rU c C s� ddl m} ddl m} | jr*t| j�}ng }| jr�xR| jD ]H}xBt�d|�D ]2}||�d�krnt d| ��qN|� |� d|�� qNW q<W |s�|� |� dd�� x0|D ](}|� |t | ||j�}|js�|�|� q�W d S )Nr )r6 )r z[,\s]+r z:Unknown URI specifier '%s'. Specify --dbs for known uris.r )r r6 r r �listr �re�splitr/ � RuntimeErrorr rG Z setup_configr" r �_currentZset_as_current) r/ r, r6 r Zdb_urlsZdb_tokenr Zdb_url�cfgr r r � _engine_uri� s( r\ c C s | j r ddlm} |jtd< d S )Nr )�poolZ poolclass)r r= r] Z AssertionPoolr" )r/ r, r] r r r � _engine_pool s r^ c C s |� dd�}t|� d S )NZ sqla_testing�requirement_cls)rG rK )r/ r, r_ r r r � _requirements s r` c C sl ddl m} |jd k rd S | �d�\}}t|�}x$|�d�dd � D ]}t||�}qDW t||�}|� |_d S )Nr )r6 �:�.� )r r6 � requirementsrX � __import__�getattr)Zargumentr6 �modnameZclsname�modZ componentZreq_clsr r r rK # s rK c C s ddl m} ddlm} ddlm} ddlm} |jrDddlm } nddl m } |j }| j �r�x�|j�� D �]�}|j} || �} y | �� } W n tk r� Y n,X x(| D ] } | �|�|�| |�� ��� q�W |jj�|��r4y| jdd�} W n tk �r Y n2X x.| D ]&} | �|�|j| |�� dd��� �q W x6t| jd d ��D ]"} | �|� |�| |�� ��� �qFW |jj�|��r�x<t| jd dd ��D ]&} | �|� |j| |�� dd��� �q�W ||d �rl|j rldd l m } x6| � d�D ](}| �|�!|j"|d |d d��� �q�W qlW d S )Nr )r6 )�against)�schema)r; )�inspect)� InspectorZ test_schemaZ foreign_key)�order_by)rm rj � postgresql)rn �*�namerj )rp rj )#r r6 Zalembic.testing.exclusionsri r= rj Zalembicr; Zsqla_08rk Z sqlalchemy.engine.reflectionrl Z from_enginer �Config� all_configsr Zget_view_names�NotImplementedError�executeZ _DropViewZTableZMetaDatard Zschemas�enabled_for_config�reversedZget_table_namesZ DropTableZsqla_100Zsqlalchemy.dialectsrn Z get_enumsZ DropEnumTypeZENUM)r/ r, r6 ri rj r; rk rl r[ �eZ inspectorZ view_namesZvnameZtnamern �enumr r r �_prep_testing_database5 s\ ry c C s | j rddlm} |� d S )Nr )�randomize_unitofwork)r Zsqlalchemy.orm.utilrz )r/ r, rz r r r �_reverse_topologicalt s r{ c C s ddl m} t|_||_d S )Nr )r6 )r r6 r/ r, )r1 r, r6 r r r �_post_setup_options{ s r| c C s@ t | tj�sdS | j�d�r dS tjjr8t| dd�s8dS dS d S )NFrM � __backend__T) � issubclassr: ZTestBase�__name__� startswithr6 r/ r rf )�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# �hasattrr8 Ztagsr� Z include_testr$ r� )r� r2 r r r � want_method� s" r� c c sv t | dd�rlxdt| �D ]P}d| j|jj|jjf }t|| fdd|jj|jjf i�}t|||� |V qW n| V d S )Nr} Fz%s_%s_%s� __only_on__z%s+%s)rf �_possible_configs_for_clsr r rp �driverr �setattr)r� �moduler[ rp �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 � d S )N)�_restore_engine)r� r r r �stop_test_class� s r� c C s t j�� d S )N)r6 rZ �resetr r r r r� � s r� c C s* t | dd �r&tj| jd�}tj�|� d S )N�__engine_options__)r/ )rf r9 Ztesting_enginer� r6 rZ Z push_engine)r� Zengr r r r� � s r� c C s d S )Nr )�testZtest_module_nameZ test_classZ test_namer r r � before_test� s r� c C s d S )Nr )r� r r r � after_test� s r� c C sf t tj�� �}| jrDtj| j� }x"t|�D ]}||�r*|�|� q*W t | dd �r�tjt � | j �� }x"t|�D ]}||�sl|�|� qlW t | d�r�tj}xVt|�D ]J}xD| jD ]:}t ||�}|�|�}|r�|�|� |d k r�|�|� P q�W q�W t | d��rbt � } tj}xDt|�D ]8}x0| jD ]&}t ||�}|�|��s | �|� �q W �qW |�| ��rb|�| � |S )Nr� � __requires__�__prefer_requires__)�setr6 rq rr Z__unsupported_on__r8 �db_specrV �removerf r; �to_listr� r� rd r� Zmatching_config_reasons�extendr� ru rN � difference�difference_update) r� �reasonsrr �spec� config_objrd Z requirement�checkZ skip_reasons� non_preferredr r r r� � sB r� c C s g }t | |�}t| dd�rHx,t| d�D ] }|� r&td| j|jf ��q&W |s�t| dd�rjd| j| jf }n,d| jd�dd � tj�� D ��d�|�f }t|��nVt | d �r�t � }t j t �| j�� }x |D ]}||�s�|�|� q�W |�|�r�|�|� tj|k�rt|�� | � d S ) NZ __skip_if__Fz'%s' skipped by %sr} z('%s' unsupported for implementation '%s'z.'%s' unsupported on any DB implementation %s%sz, c s s: | ]2}d |j jd�dd� |j jjD ��|j jf V qdS )z '%s(%s)+%s'rb c s s | ]}t |�V qd S )N)�str)�.0�digr r r � <genexpr> s z&_do_skips.<locals>.<genexpr>.<genexpr>N)r rp �joinZdialectZserver_version_infor� )r� r� r r r r� s z _do_skips.<locals>.<genexpr>�__prefer_backends__)r� rf r r r� r� r6 rq rr r� r� r8 r� r; r� r� rN r� r� rZ � _setup_config�pop)r� r� rr �c�msgr� r� r� r r r r� � s8 r� c C s t j�| � d S )N)r6 rZ �push)r� Zctxr r r r� + s r� )N)E�__doc__Z __future__r Znoser � ImportErrorZ_pytest.runnerr rH rW � version_infoZpy3kr* r+ r: r9 r r8 r7 Z assertionsrd r6 r; r, r? r" r� r# r$ r/ r r! r' r) r. r3 r5 r> r r r r r r0 r< rP rQ rR rS rU r\ r^ r` rK ry r{ r| r� r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> s� 1 ? +,