Edit File: pytestplugin.cpython-37.pyc
B ��4]� � @ s y d dl ZW n ek r, ddlmZ Y nX d dlZd dlZd dlZd dlZd dlZyd dl Z dZ W n ek r~ dZ Y nX dd� Z dd � Z d d � Z d d � Ze r�d dlZdd� Zdd� Zdd� Zdd� Zdadd� Zdd� Zdd� Zd d � Zd d � Zd d!� ZdS )"� N� )� plugin_baseTFc s, | � d�� � fdd�}t�|� t�� d S )N� sqlalchemyc sp |� dd ���r.G �fdd�dtj�� � |d< |� dd ���r^G � �fdd�dtj�� � |d< �j| f|� d S )N�callbackc s e Zd Zd� fdd� ZdS )z=pytest_addoption.<locals>.make_option.<locals>.CallableActionNc s � |||� d S )N� )�self�parser� namespace�values� option_string)� callback_r �Y/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/testing/plugin/pytestplugin.py�__call__ s zFpytest_addoption.<locals>.make_option.<locals>.CallableAction.__call__)N)�__name__� __module__� __qualname__r r )r r r �CallableAction s r �action�zeroarg_callbackc s. e Zd Zd�� fdd� Zd�fdd� Z� ZS ) z=pytest_addoption.<locals>.make_option.<locals>.CallableActionFNc s" t � | �j||dd|||d� d S )Nr T)�option_strings�dest�nargs�const�default�required�help)�super�__init__)r r r r r r )r � __class__r r r + s zFpytest_addoption.<locals>.make_option.<locals>.CallableAction.__init__c s � |||� d S )Nr )r r r r r )r r r r = s zFpytest_addoption.<locals>.make_option.<locals>.CallableAction.__call__)FFN)N)r r r r r � __classcell__r )r r )r r r * s )�pop�argparse�ActionZ addoption)�name�kw)�group)r r r r � make_option s z%pytest_addoption.<locals>.make_option)Zgetgroupr Z setup_optionsZ read_config)r r& r )r% r �pytest_addoption s + r' c C s� t �d� t| d�r2t�| j� t�| jd � n&| jjrXt j � | jj�rXt � | jj� t� | j� t�tt| jdd��� t�t jj� d S )Nz sqlalchemy.testing.assertions� slaveinput�follower_identZ cov_sourceF)�pytestZregister_assert_rewrite�hasattrr Z!restore_important_follower_configr( Zconfigure_follower�optionZ write_idents�os�path�exists�removeZ pre_beginZset_coverage_flag�bool�getattrZ set_skip_test�skip� Exception)�configr r r �pytest_configureJ s r6 c C s t �� d S )N)r Z post_begin)�sessionr r r �pytest_sessionstart_ s r8 c C s t �� d S )N)r Zfinal_process_cleanup)r7 r r r �pytest_sessionfinishc s r9 c C sH t �| j� dt�� jdd� | jd< ddlm} |�| jd � d S )Nztest_%sr � r) )� provision) r Z!memoize_important_follower_configr( �uuidZuuid4�hex�sqlalchemy.testingr; Zcreate_follower_db)�noder; r r r �pytest_configure_nodej s r@ c C s ddl m} |�| jd � d S )Nr )r; r) )r> r; Zdrop_follower_dbr( )r? �errorr; r r r �pytest_testnodedownu s rB c C s t �dd� �}dd� |D �|d d �< tdd� |D ��}x�|D ]x}xrt�|j|jj�D ]\}||jk rT||j }xBtj |j |jjd�� � D ]&}x |� � D ]} || j � | � q�W q�W qTW q<W g } x<|D ]4} | jj|kr�| �|| jj | j � q�| � | � q�W t| dd� d �|d d �< d S ) Nc S s t �t�S )N)� collections� defaultdict�listr r r r �<lambda>� � z/pytest_collection_modifyitems.<locals>.<lambda>c S s. g | ]&}t |jtj�r|jjj�d �s|�qS )�_)� isinstance�parentr* �Instancer# � startswith)�.0�itemr r r � <listcomp>� s z1pytest_collection_modifyitems.<locals>.<listcomp>c s s | ] }|j V qd S )N)rJ )rM rN r r r � <genexpr>� s z0pytest_collection_modifyitems.<locals>.<genexpr>)rJ c S s | j j j j| j j j| jfS )N)rJ r# )rN r r r rF � s )�key)rC rD �setr Zgenerate_sub_tests�clsrJ �moduler* �Classr Zcollectr# �append�extend�sorted) r7 r5 �itemsZ rebuilt_itemsZ test_classesZ test_classZsub_clsZ per_cls_dict�inst�tZnewitemsrN r r r � pytest_collection_modifyitems{ s, r\ c C s\ t �|�r"t�|�r"tj|| d�S t �|�rTt| tj�rTt� | j |�rTtj || d�S g S d S )N)rJ ) �inspect�isclassr Z want_classr* rU � isfunctionrI rK Z want_methodrS �Function)Z collectorr# �objr r r �pytest_pycollect_makeitem� s rb c sR t � tj�sd S td krFt� jj� � jja� fdd�}� jj�|� t� � d S )Nc s t � jj� d ad S )N)�class_teardownrJ �_current_classr )rN r r �finalize� s z&pytest_runtest_setup.<locals>.finalize)rI r* r` rd � class_setuprJ Z addfinalizer� test_setup)rN re r )rN r �pytest_runtest_setup� s rh c C s t | � d S )N)� test_teardown)rN r r r �pytest_runtest_teardown� s rj c C s t �| | jjj| jj| j� d S )N)r Z before_testrJ rT r rS r# )rN r r r rg � s rg c C s t �| � d S )N)r Z after_test)rN r r r ri � s ri c C s t �| j� d S )N)r Zstart_test_classrS )rN r r r rf � s rf c C s t �| j� d S )N)r Zstop_test_classrS )rN r r r rc � s rc )Zsqla_plugin_baser � ImportError� r! rC r] r- r* ZxdistZ has_xdistr' r6 r8 r9 r< r@ rB r\ rb rd rh rj rg ri rf rc r r r r �<module> s< 2 6