관리-도구
편집 파일: pytestplugin.cpython-37.opt-1.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�valuesZ 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�argparseZActionZ 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_beginZoption�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, )r/ r0 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 )r0 r r r �<lambda>v s z/pytest_collection_modifyitems.<locals>.<lambda>)�key)� collections� defaultdict�list�setr Zgenerate_sub_tests�clsr, �moduler- �Classr Zcollect�extend�append�sorted) r! r �itemsZ rebuilt_itemsZ test_classesZ test_classZsub_clsZlist_�instZnewitemsr0 r r r � pytest_collection_modifyitemsQ s( rA c C s\ t �|�r"t�|�r"tj|| d�S t �|�rTt| tj�rTt� | j |�rTtj || d�S g S d S )N)r, ) �inspectZisclassr Z want_classr- r; Z isfunctionr+ r. Z want_methodr9 �Function)Z collectorr �objr r r �pytest_pycollect_makeitem} s rE 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 )r0 r r �finalize� s z&pytest_runtest_setup.<locals>.finalize)r+ r- rC rG � class_setupr, Z addfinalizer� test_setup)r0 rH r )r0 r �pytest_runtest_setup� s rK c C s t | � d S )N)� test_teardown)r0 r r r �pytest_runtest_teardown� s rM c C s t �| | jjj| jj| j� d S )N)r Z before_testr, r: r r9 r )r0 r r r rJ � s rJ c C s t �| � d S )N)r Z after_test)r0 r r r rL � s rL c C s t �| j� d S )N)r Zstart_test_classr9 )r0 r r r rI � s rI c C s t �| j� d S )N)r Zstop_test_classr9 )r0 r r r rF � s rF ) �__doc__Zalembic_plugin_baser � ImportError� �sys� version_infoZpy3kr- r rB r5 � itertoolsZxdistZ has_xdistr r r" �countr% r( r* rA rE rG rK rM rJ rL rI rF r r r r �<module> s@ ,