Edit File: plugin_base.cpython-37.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-engine�storeZ 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 r4 c C s | t _d S )N)r0 Z has_coverage)�valuer r r �set_coverage_flag� s r6 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_configurer0 r- r r7 r8 r9 r: r; � sqlalchemyr< Z setup_filters)r3 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)�logging� basicConfig�endswith� getLogger�setLevel�INFO�DEBUG)�opt_strr5 �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- r0 �get�sys�exit)�args�macror r r r � s r c C s t |� d S )N)�_setup_requirements)rG r5 rH r r r r � s r c C s t �|�dd�� d S )N�-�_)r% �add�replace)rG r5 rH r r r r � s r c C s t �|�dd�� d S )NrQ rR )r$ rS rT )rG r5 rH r r r r � s r c C s t �| � | S )N)r1 r )r3 r r r �pre� s rU c C s t �| � | S )N)r= r )r3 r r r �post� s rV c C s | a d S )N)r0 )r2 r- r r r �_setup_options� s rW c C s | j rdtd< d S )NTZserver_side_cursors)Z serversider# )r0 r- r r r �_server_side_cursors� s rX c C s | j rdd l }|tjd< d S )Nr Zdecimal)r rL �modules)r0 r- r r r r �_monkeypatch_cdecimal� s rZ 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 )r7 )r z[,\s]+r z:Unknown URI specifier '%s'. Specify --dbs for known uris.r )r r7 r r �listr �re�splitr0 � RuntimeErrorr rK Z setup_configr# r! �_currentZset_as_current) r0 r- r7 r Zdb_urlsZdb_tokenr Zdb_url�cfgr r r � _engine_uri� s( ra c C s | j r ddlm} |jtd< d S )Nr )�poolZ poolclass)r r> rb Z AssertionPoolr# )r0 r- rb r r r � _engine_pool s rc c C s |� dd�}t|� d S )NZ sqla_testing�requirement_cls)rK rP )r0 r- rd r r r � _requirements s re c C sl ddl m} |jd k rd S | �d�\}}t|�}x$|�d�dd � D ]}t||�}qDW t||�}|� |_d S )Nr )r7 �:�.� )r r7 � requirementsr] � __import__�getattr)�argumentr7 �modnameZclsname�modZ componentZreq_clsr r r rP # s rP 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 )r7 )�against)�schema)r< )�inspect)� InspectorZ test_schemaZ foreign_key)�order_by)rs rp � postgresql)rt �*�namerp )rv rp )#r r7 Zalembic.testing.exclusionsro r> rp �alembicr< Zsqla_08rq Z sqlalchemy.engine.reflectionrr Z from_enginer �Config� all_configsr Zget_view_names�NotImplementedError�executeZ _DropViewZTableZMetaDatari Zschemas�enabled_for_config�reversedZget_table_namesZ DropTableZsqla_100Zsqlalchemy.dialectsrt Z get_enumsZ DropEnumTypeZENUM)r0 r- r7 ro rp r< rq rr r` �eZ inspectorZ view_namesZvnameZtnamert �enumr r r �_prep_testing_database5 s\ r� c C s | j rddlm} |� d S )Nr )�randomize_unitofwork)r Zsqlalchemy.orm.utilr� )r0 r- r� r r r �_reverse_topologicalt s r� c C s ddl m} t|_||_d S )Nr )r7 )r r7 r0 r- )r2 r- r7 r r r �_post_setup_options{ s r� c C s@ t | tj�sdS | j�d�r dS tjjr8t| dd�s8dS dS d S )NFrR � __backend__T) � issubclassr; ZTestBase�__name__� startswithr7 r0 r rk )�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$ �hasattrr9 Ztagsr� Z include_testr% r� )r� r3 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)rk �_possible_configs_for_clsr� r rv �driverr �setattr)r� �moduler` rv �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)r7 r_ �resetr r r r r� � s r� c C s* t | dd �r&tj| jd�}tj�|� d S )N�__engine_options__)r0 )rk r: Ztesting_enginer� r7 r_ 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__)�setr7 rx ry Z__unsupported_on__r9 �db_specr[ �removerk r< �to_listr� r� ri r� Zmatching_config_reasons�extendr� r| rS � difference�difference_update) r� �reasonsry �spec� config_objri � 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'rg c s s | ]}t |�V qd S )N)�str)�.0�digr r r � <genexpr> s z&_do_skips.<locals>.<genexpr>.<genexpr>N)r rv �joinZdialectZserver_version_infor� )r� r� r r r r� s z _do_skips.<locals>.<genexpr>�__prefer_backends__)r� rk r r� r� r� r7 rx ry r� r� r9 r� r< r� r� rS r� r� r_ � _setup_config�pop)r� r� ry �c�msgr� r� r� r r r r� � s8 r� c C s t j�| � d S )N)r7 r_ �push)r� �ctxr r r r� + s r� )N)E�__doc__� __future__r Znoser � ImportErrorZ_pytest.runnerr rL r\ � version_infoZpy3kr+ r, r; r: r r9 r8 Z assertionsri r7 r< r- r@ r# r� r$ r% r0 r r" r( r* r/ r4 r6 r? r r r r r r1 r= rU rV rW rX rZ ra rc re rP r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> s� 1 ? +,