Edit File: noseplugin.cpython-37.pyc
B �M!V� � @ sp d Z y ddlZW n ek r0 ddlmZ Y nX ddlZddlZddlmZ da ej dkZ G dd� de�Z dS ) z� Enhance nose with extra options and behaviors for running SQLAlchemy tests. NOTE: copied/adapted from SQLAlchemy master for backwards compatibility; this should be removable when Alembic targets SQLAlchemy 1.0.0. � N� )� plugin_base)�Plugin)� r c s~ e Zd ZdZdZdZejfdd�Z� fdd�Z dd � Z d d � Z d d � Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Z� ZS )�NoseSQLAlchemyTZ sqla_testing�d c s6 t �| ||� |j� � fdd�}t�|� t�� d S )Nc s4 |� dd �� � r$� fdd�}||d< �| f|� d S )N�callbackc s � |||� d S )N� )�optionZopt_str�value�parser)� callback_r �R/opt/alt/python37/lib/python3.7/site-packages/alembic/testing/plugin/noseplugin.py�wrap_. s z:NoseSQLAlchemy.options.<locals>.make_option.<locals>.wrap_)�pop)�name�kwr )�opt)r r � make_option+ s z+NoseSQLAlchemy.options.<locals>.make_option)r �options� add_optionr Z setup_optionsZ read_config)�selfr �envr r )r r r ' s zNoseSQLAlchemy.optionsc s, t t| ��||� t�|� t�|j� d S )N)�superr � configurer Z pre_beginZset_coverage_flagZenable_plugin_coverage)r r Zconf)� __class__r r r 6 s zNoseSQLAlchemy.configurec C s ddl ma t�� d S )Nr )�fixtures)Zalembic.testingr r Z post_begin)r r r r �begin< s zNoseSQLAlchemy.beginc C s dS )N� r )r �testr r r � describeTestB s zNoseSQLAlchemy.describeTestc C s dS )NFr )r �fnr r r � wantFunctionE s zNoseSQLAlchemy.wantFunctionc C s0 t r t|jd�sdS |jj}n|j}t�||�S )N�clsF)�py3k�hasattr�__self__r# Zim_classr Z want_method)r r! r# r r r � wantMethodH s zNoseSQLAlchemy.wantMethodc C s t �|�S )N)r Z want_class)r r# r r r � wantClassQ s zNoseSQLAlchemy.wantClassc C s$ t �||jjj|jj|jjj� d S )N)r Z before_testr r# � __module__�method�__name__)r r r r r � beforeTestT s zNoseSQLAlchemy.beforeTestc C s t �|� d S )N)r Z after_test)r r r r r � afterTestZ s zNoseSQLAlchemy.afterTestc C s( t |t�rt|tj�sd S t�|� d S )N)� isinstance�type� issubclassr �TestBaser Zstart_test_class)r �ctxr r r � startContext] s zNoseSQLAlchemy.startContextc C s( t |t�rt|tj�sd S t�|� d S )N)r. r/ r0 r r1 r Zstop_test_class)r r2 r r r � stopContextc s zNoseSQLAlchemy.stopContext)r+ r) � __qualname__�enabledr �score�os�environr r r r r" r' r( r, r- r3 r4 � __classcell__r r )r r r ! s r ) �__doc__Zalembic_plugin_baser � ImportErrorr r8 �sysZ nose.pluginsr r � version_infor$ r r r r r �<module> s