Edit File: pytestplugin.cpython-37.pyc
B �M!V � @ s d Z y ddlZW n ek r0 ddlmZ Y nX ddlZejdkZddlZddl Z ddl Z ddl Z ddl Z yddl Z dZW n ek r� dZY nX dd � Zd d � Zd d � Zer�e �d�Zdd� Zdd� Zdd� Zdd� Zdadd� Zdd� Zdd� Zd d � Z d d � Z d d!� Z dS )"z�NOTE: copied/adapted from SQLAlchemy master for backwards compatibility; this should be removable when Alembic targets SQLAlchemy 1.0.0. � N� )� plugin_base)� r TFc s, | � d�� � fdd�}t�|� t�� d S )NZ sqlalchemyc s@ |� 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 �T/opt/alt/python37/lib/python3.7/site-packages/alembic/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)�pop�argparse�ActionZ addoption)�name�kwr )�group)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 sR t | d�r&t�| j� t�| jd � t�| j� tt| jdd��}t� |� d S )N� slaveinput�follower_identZ cov_sourceF) �hasattrr Z!restore_important_follower_configr Zconfigure_followerZ pre_begin�option�bool�getattrZset_coverage_flag)�configZcoverager r r �pytest_configure/ s r# c C s t �� d S )N)r Z post_begin)�sessionr r r �pytest_sessionstart< s r% c C s> t �| j� dtt� | jd< ddlm} |�| jd � d S )Nztest_%sr r )� provision)r Z!memoize_important_follower_configr �next�_follower_count�alembic.testingr& Zcreate_follower_db)�noder& r r r �pytest_configure_nodeB s r+ c C s ddl m} |�| jd � d S )Nr )r& r )r) r& Zdrop_follower_dbr )r* �errorr& r r r �pytest_testnodedownL s r- c C s t �t�}dd� |D �|d d �< tdd� |D ��}xl|D ]d}x^t�|j|jj�D ]H}||jk rP||j }x.t j |j |jjd�� � D ]}|� |� � � q�W qPW q8W g } xJ|D ]B} | jj|kr�| � || jj � g || jj d d �< q�| �| � q�W t| dd� d�|d d �< d S ) Nc S s g | ]}t |jtj�r|�qS r )� isinstance�parent�pytest�Instance)�.0�itemr r r � <listcomp>^ s z1pytest_collection_modifyitems.<locals>.<listcomp>c s s | ] }|j V qd S )N)r/ )r2 r3 r r r � <genexpr>` s z0pytest_collection_modifyitems.<locals>.<genexpr>)r/ c S s | j j j j| j j j| jfS )N)r/ r )r3 r r r �<lambda>v s z/pytest_collection_modifyitems.<locals>.<lambda>)�key)� collections� defaultdict�list�setr Zgenerate_sub_tests�clsr/ �moduler0 �Classr Zcollect�extend�append�sorted) r$ r" �itemsZ rebuilt_itemsZ test_classesZ test_classZsub_clsZlist_�instZnewitemsr3 r r r � pytest_collection_modifyitemsQ s( rD c C s\ t �|�r"t�|�r"tj|| d�S t �|�rTt| tj�rTt� | j |�rTtj || d�S g S d S )N)r/ ) �inspect�isclassr Z want_classr0 r> � isfunctionr. r1 Z want_methodr< �Function)Z collectorr �objr r r �pytest_pycollect_makeitem} s rJ c sR t � tj�sd S td krFt� jj� � jja� fdd�}� jj�|� t� � d S )Nc s t � jj� d ad S )N)�class_teardownr/ �_current_classr )r3 r r �finalize� s z&pytest_runtest_setup.<locals>.finalize)r. r0 rH rL � class_setupr/ Z addfinalizer� test_setup)r3 rM r )r3 r �pytest_runtest_setup� s rP c C s t | � d S )N)� test_teardown)r3 r r r �pytest_runtest_teardown� s rR c C s t �| | jjj| jj| j� d S )N)r Z before_testr/ r= r r< r )r3 r r r rO � s rO c C s t �| � d S )N)r Z after_test)r3 r r r rQ � s rQ c C s t �| j� d S )N)r Zstart_test_classr< )r3 r r r rN � s rN c C s t �| j� d S )N)r Zstop_test_classr< )r3 r r r rK � s rK ) �__doc__Zalembic_plugin_baser � ImportError� �sys� version_infoZpy3kr0 r rE r8 � itertoolsZxdistZ has_xdistr r# r% �countr( r+ r- rD rJ rL rP rR rO rQ rN rK r r r r �<module> s@ ,