관리-도구
편집 파일: pytestplugin.cpython-38.pyc
U -?�fl � @ s� d dl mZ d dlZd dlZd dlmZ d dlZd dlZd dlZd dl Z d dl Z d dl Z d dl m Z d dlZd dlZze s�d dlZW n ek r� ddlmZ Y nX dd� Zd d �d d �Zd aejdd�dd� �Zdd� Zdd� Zdd� Zdd� Z dd� Z G d d � d �Z d d � Z d d!� Z d"d#� Z!d<d$d%�Z"d&d'� Z#da$d(d)� Z%ej&dd*�d+d,� �Z'd-d.� Z(da)d/d0� Z*ejd1d2�d3d4� �Z+ejd5d2�d6d7� �Z,d8d9� Z-G d:d;� d;ej.�Z/dS )=� )� annotationsN)�update_wrapper)� TYPE_CHECKING� )� plugin_basec s$ | � d�� � fdd�}t�|� d S )N� sqlalchemyc sn |� dd �� � r.G � fdd�dtj�}||d< |� dd ���r\G �fdd�dtj�}||d< �j| f|� d S )N�callbackc s e Zd Zd� fdd� ZdS )�=pytest_addoption.<locals>.make_option.<locals>.CallableActionNc s � |||� d S �N� ��self�parser� namespace�values� option_string�� callback_r �V/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/plugin/pytestplugin.py�__call__* s �Fpytest_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 ) r FNc 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# �� __class__r r r% 5 s �zFpytest_addoption.<locals>.make_option.<locals>.CallableAction.__init__c s � |||� d S r r r �r r r r G s r )FFN)N)r r r r% r � __classcell__r r( r& r r 4 s � �)�pop�argparse�ActionZ addoption)�name�kwr ��group)r r r � make_option% s z%pytest_addoption.<locals>.make_option)Zgetgroupr Z setup_options)r r1 r r/ r �pytest_addoption" s +r2 z pytest.Config��configc C s t �| j� t jst jr`d�tt j�dd� t jD � �}| jjrX| j jd|� �7 _n|| j_| j � d�rz| j � t � � t | d�r�t �| j� t �| jd � n&| jjr�tj�| jj�r�t�| jj� t �| j� t �tt| jdd��� t �t� | jj �rd a d S ) Nz and c S s g | ]}d |� ��qS )znot r )�.0�tagr r r � <listcomp>X s z$pytest_configure.<locals>.<listcomp>Zxdist� workerinput�follower_identZ cov_sourceFT) r Z read_configZrootpathZ exclude_tagsZ include_tags�join�list�optionZmarkexpr� pluginmanagerZ hasplugin�register� XDistHooks�hasattrZ!restore_important_follower_configr8 Zconfigure_followerZ write_idents�os�path�exists�removeZ pre_beginZset_coverage_flag�bool�getattrZset_fixture_functions�PytestFixtureFunctions�dump_pyannotate�DUMP_PYANNOTATE)r4 Znew_exprr r r �pytest_configureS s4 �� � � rJ FT)Zautousec c s. t rddlm} | �� d V t r*| �� d S )Nr �� collect_types)rI �pyannotate_runtimerL �start�stoprK r r r �collect_types_fixture| s rP c C s� dd l }ddl m} ddlm} ddlm} d}tjjr<dnd}d|� d |� d �d |j � �g}|� r�dd l m } |� d |j � d�� n|� d|� �� | j j�d�} | r�| �d|� |D ]} | �| � q�n td|d� |D ] } t| � q�d S )Nr )� __version__)�has_compiled_ext)�_CYEXTENSION_MSGzsqlalchemy installationz no user sitezuser site loadedz SQLAlchemy z (�)zPath: )�utilz!compiled extension enabled, e.g. � z compiled extension not enabled; Zterminalreporter�=z=========================)r rQ Zsqlalchemy.utilrR Zsqlalchemy.util._has_cyrS �sys�flags� no_user_site�__file__Zsqlalchemy.cyextensionrU �appendr4 r= Z get_pluginZ write_sepZ write_line�print) �sessionr rQ rR rS Zgreet�siteZmsgsrU �pm�mr r r �_log_sqlalchemy_info� s* � rb c C s$ ddl m} t| � |�tj� d S �Nr ��asyncio)�sqlalchemy.testingre rb Z _assume_asyncr Z post_begin)r^ re r r r �pytest_sessionstart� s rg c C sB ddl m} |�tj� | jjjr>ddlm } |� | jjj� d S )Nr rd rK ) rf re �_maybe_async_provisioningr Zfinal_process_cleanupr4 r< rH rM rL Z dump_stats)r^ re rL r r r �pytest_sessionfinish� s ri c C s ddl m} |�� d S rc )rf re � _shutdown)r4 re r r r �pytest_unconfigure� s rk c s> | j jjr:ddlm} tj�d�� � fdd�}|j|d� d S )Nr rK �lib/sqlalchemyc s< t j�t j�| ��} dt j�| � g�kr,d S d| kr8d S | S )Nrl Ztesting)rA rB �normpath�abspath� commonpath)�filename�Zlib_sqlalchemyr r �_filter� s �z)pytest_collection_finish.<locals>._filter)Zfilter_filename) r4 r< rH rM rL rA rB rn Zinit_types_collection)r^ rL rr r rq r �pytest_collection_finish� s rs c @ s e Zd Zdd� Zdd� ZdS )r? c C sX ddl m} ddl m} t�|j� dt�� jdd� |jd< |� |j |jd � d S )Nr �� provisionrd ztest_%s� r9 ) rf ru re r Z!memoize_important_follower_configr8 �uuid�uuid4�hexrh Zcreate_follower_db)r �noderu re r r r �pytest_configure_node� s �z XDistHooks.pytest_configure_nodec C s0 ddl m} ddl m} |�|j|jd � d S )Nr rt rd r9 )rf ru re rh Zdrop_follower_dbr8 )r rz �errorru re r r r �pytest_testnodedown� s �z XDistHooks.pytest_testnodedownN)r r r r{ r} r r r r r? � s r? c s� ddl m} t�dd� ��dd� |D �|d d �< dd� |D ��� fd d �� � ��fd d �}|�|� g }|D ]2}|j}|�kr�|��| |j � ql|�|� qlt |d d� d�|d d �< d S )Nr rd c S s t �t�S r )� collections� defaultdictr; r r r r �<lambda>� � z/pytest_collection_modifyitems.<locals>.<lambda>c S s4 g | ],}|� tj�d k r|� tj�j�d�s|�qS )N�_)� getparent�pytest�Classr- � startswith�r5 �itemr r r r7 � s �z1pytest_collection_modifyitems.<locals>.<listcomp>c S s h | ]}|� tj��qS r )r� r� r� r� r r r � <setcomp> s z0pytest_collection_modifyitems.<locals>.<setcomp>c 3 s4 | � � D ]&}t|tj�r(� |�E d H q|V qd S r )�collect� isinstancer� Z Collector)�elementZ inst_or_fn)r� r r r� s z.pytest_collection_modifyitems.<locals>.collectc s �D � ]�} t � }t| jdd�s,t| jdd�r4dh}n t| jdd�rJdh}nt� }dd� | �� D �}|| }|�|�}|D ]}| �|� qxt�| j| j |�D ]h}|| jk r��| j }| � t j �}t j j|j|d �}|D ]}|�|� qԈ |�D ]} || j �| � q�q�qd S ) NZ __backend__FZ __only_on__�backendZ__sparse_backend__Zsparse_backendc S s h | ] }|j �qS r )r- )r5 �markr r r r� s zLpytest_collection_modifyitems.<locals>.setup_test_classes.<locals>.<setcomp>�r- �parent)�setrF �cls� frozensetZ iter_markers�unionZ add_markerr Zgenerate_sub_tests�moduler� r� �Moduler� � from_parentr r- r\ ) Z test_classZ add_markersZexisting_markersZ all_markers�markerZsub_clsZ per_cls_dictr� Znew_cls�fn�r� Z rebuilt_itemsZ test_classesr r �setup_test_classes sD �� � � z9pytest_collection_modifyitems.<locals>.setup_test_classesc S s | � tj�j| � tj�j| jfS r )r� r� r� r- r� )r� r r r r� E s �)�key) rf re r~ r rh r� �extendr- r\ �sorted)r^ r4 �itemsre r� Znewitemsr� Zcls_r r� r � pytest_collection_modifyitems� s* �� * �r� c sx t �|�rJt�||�rJddlm} |jr0t|�}� fdd�t� j |�D �S t � |�rp� j d k rpt� � j |�rpd S g S d S )Nr r3 c s g | ]}t jj|j� d ��qS )r� )r� r� r� r )r5 Zparametrize_cls�� collectorr r r7 T s � �z-pytest_pycollect_makeitem.<locals>.<listcomp>) �inspect�isclassr Z want_classrf r4 � any_async�_apply_maybe_async�_parametrize_clsr� � isfunctionr� Z want_method)r� r- �objr4 r r� r �pytest_pycollect_makeitemM s ��� �r� c C s t | d�r| j} q t�| �S )N� __wrapped__)r@ r� r� �iscoroutinefunction�r� r r r � _is_wrapped_coroutine_functiong s r� c s� ddl m� t| ��� D ]�\}}t|�s2t|t�rt|dd�s|�d�rt |�sd}t|t�rh|j }d}t � fdd��}||�}|r�t|�}d|_ t | ||� q|r�| �� d d � D ]}|tkr�t|d� q�| S ) Nr rd �_maybe_async_appliedFZtest_Tc s � j | f|�|�S r )� _maybe_async�r� �args�kwargsrd r r � make_async} s z&_apply_maybe_async.<locals>.make_asyncr )rf re �varsr� �callabler� � classmethodrF r� r� �__func__�_pytest_fn_decoratorr� �setattr�mro�objectr� )r� Zrecurser- �valueZis_classmethodr� Zdo_asyncr� r rd r r� n s8 �� ��� r� c C s� d|j kr|gS |j}g }tjdd� |D �� D ]�}i }tdd� |D �|�D ]>\}}|s^td��t�d|�}t||j�D ]\} } | || < qvqJd� dd � |D ��} d |j | f } t � t | |f|�} t | | | � |�| � q.|S ) z6implement a class-based version of pytest parametrize.�_sa_parametrizec S s g | ] \}}|�qS r r )r5 �argname�paramsr r r r7 � s z$_parametrize_cls.<locals>.<listcomp>c S s g | ] }|d �qS )r r )r5 Z _sa_paramr r r r7 � s z*need argnames for class-based combinationsz,\s*r� c s s. | ]&}|j �d �D ]}t�dd|�V qqdS )�-z\W� N)�id�split�re�sub)r5 �param�tokenr r r � <genexpr>� s �z#_parametrize_cls.<locals>.<genexpr>z%s_%s)�__dict__r� � itertools�product�zip� TypeErrorr� r� r r: r �type�__new__r� r\ )r� r� r� �classesZfull_param_setZ cls_variablesr� r� Z argname_split�arg�valZparametrized_namer- Znewclsr r r r� � s2 � � � r� c C s@ ddl m} t| tj�r<td kr<|�tj| j � | � tj �ad S rc ) rf re r� r� �Function�_current_classrh r Z!start_test_class_outside_fixturesr� r� r� �r� re r r r �pytest_runtest_setup� s �r� )Z hookwrapperc c s ddl m} |�tj| � d V td k �r|d ksD|�tj�tk �rd az�z|� tj | j � W n� t k r� } z|t jr�|js�dt j f|_nZ|jd r�t|jd t�r�t|j�}|d dt j 7 < t|�|_n| jdt jf7 _� W 5 d }~X Y nX W 5 d a X d S )Nr rd z__Original test failure__: ���z __Original test failure__: z__Original test failure__)rf re r� r Z after_testr� r� r� r� �_current_reportrh Z stop_test_class_outside_fixturesr� � ExceptionZfailedr� Z longreprtextr� �strr; �tuple)r� Znextitemre �er� r r r �pytest_runtest_teardown� sF �� ��� �� �r� c C s, ddl m} |�tj| | jj| j| j� d S rc ) rf re r� r Z before_testr� r r� r- r� r r r �pytest_runtest_call� s �r� c C s | j dkr| ad S )N�call)�whenr� )�reportr r r �pytest_runtest_logreport s r� �class)�scopec c sV ddl m} | j}t|d�r(|�|j� d V t|d�rD|�|j� |�tj|� d S )Nr rd �setup_test_class�teardown_test_class) rf re r� r@ r� r� r� r Zstop_test_class)�requestre r� r r r �setup_class_methods s r� �functionc c s� ddl m} | j}t|d�r(|�|j� t|d�r>|�|j� d V |�tj|� t|d�rh|�|j � t|d�r~|�|j � d S )Nr rd � setup_test�setUp�tearDown� teardown_test) rf re �instancer@ r� r� r� r Zafter_test_fixturesr� r� )r� re r r r r �setup_test_methods$ s r� c s8 ddl m� ddlm� dd� � d � ���fdd� }|S ) z:Port of langhelpers.decorator with pytest-specific tricks.r )�format_argspec_plus)�inspect_getfullargspecc S s t | |� || S r )�exec)�code�env�fn_namer r r �_exec_code_in_envi s z/_pytest_fn_decorator.<locals>._exec_code_in_envr c s� �| �}|r|j �|� tdd| jd�}|��|dd�� d| }� |�| d�| j�}|szt| d| �j|_| |_t|| �S | j |_ | j|_t | d �r�| j |_ |S d S ) N� __target_fn� __orig_fn)r� r� r- F)�groupedzVdef %(name)s%(grouped_args)s: return %(__target_fn)s(%(__orig_fn)s, %(apply_kw)s) )r� r� r� � pytestmark) r� r� �dictr �updaterF � __defaults__r� r r r@ r� )r� �add_positional_parameters�spec�metadatar� Z decorated�r� r� r� �targetr r �decorateq s6 ��� � z&_pytest_fn_decorator.<locals>.decorate)r )Zsqlalchemy.util.langhelpersr� Zsqlalchemy.util.compatr� )r r r r r r� c s !r� c @ sf e Zd Zdd� Zedd� �Zdd� Zdd� eed d� d �Z d d � Z d d� Z dd� Z dd� Z dd� ZdS )rG c O s t jj||�S r )r� �skipr� )r r� r. r r r �skip_test_exception� s z*PytestFixtureFunctions.skip_test_exceptionc C s t jS r )r� r� �r r r r � add_to_marker� s z$PytestFixtureFunctions.add_to_markerc C s t j�dd�S )Nr� r� )r� r� Z usefixturesr r r r �mark_base_test_class� s �z+PytestFixtureFunctions.mark_base_test_classc C s | S r r �r� r r r r� � r� z PytestFixtureFunctions.<lambda>c C s t | d�r| jS t| �jS )Nr )r@ r r� r r r r r� � s )�i�r�s�nc s� ddl m� t|�dkr2t|d d�r2t|d �}|�dd���fdd�}|�d d�}g }d �|�r | j� tjdd d � t |�D ��� }� fd d �t |�D �}|D ]\�t �t �s��f�|��\}} ||�dd� } | r�d�|� | | d� �fdd�|D ��f� q�nD|D ]>�t �t ��s�f�|��\}} | �r0d�|� || df� �qg �|D ]6\} } }��rf| | f7 } tj| d|i�} �� | � �qL����fdd�} | S )a Facade for pytest.mark.parametrize. Automatically derives argument names from the callable which in our case is always a method on a class with positional arguments. ids for parameter sets are derived using an optional template. r �� exclusionsr �__next__�argnamesNc s< g }g }| D ]&}t |� j�r(|�|� q |�|� q ||fS r )r� �compoundr\ )r� �resultZgathered_exclusions�ar r r �_filter_exclusions� s z?PytestFixtureFunctions.combinations.<locals>._filter_exclusions�id_Fc S s g | ]\}}|d kr|�qS ))r r r r r �r5 �idx�charr r r r7 � s �z7PytestFixtureFunctions.combinations.<locals>.<listcomp>c s* g | ]"\}}|� krt �|�� | f�qS r )�operator� itemgetterr )�_combination_id_fnsr r r7 � s �Tr� c 3 s | ]\}}||� ��V qd S r r )r5 �getterZcomb_fn)r� r r r� � s z6PytestFixtureFunctions.combinations.<locals>.<genexpr>r� c s� t �| �r:�rtd��d| jkr&g | _| j�� �f� | S t �| �jdd � }� d kr\|}n t� d� �}�r�t dd� |D ��}d|� �}||g7 }t �fdd ��}|| |fd �} t j �|��| �S d S ) Nz5exclusions not supported for class level combinationsr� r z, *c s s | ]}|� d �rdV qdS )� _exclusionsr N)r� )r5 r r r r r� s zHPytestFixtureFunctions.combinations.<locals>.decorate.<locals>.<genexpr>Z _exclusions_c s4 |d }|r"� � � j|� }|| �} | |d d� |�S )Nr� )r �add)r� r� r. r Zexlur r r �check_exclusions! s zOPytestFixtureFunctions.combinations.<locals>.decorate.<locals>.check_exclusions)r� )r� r� �NotImplementedErrorr� r� r\ �getfullargspecr� r� r� �sumr� r� r� Z parametrize)r� Z _fn_argnamesZ _argnamesZ existing_exlZcurrent_exclusion_namer! )r r �has_exclusions� pytest_paramsr r r s4 � � �z5PytestFixtureFunctions.combinations.<locals>.decorate)r )rf r �lenr@ r; r* r r r � enumerater� r� r\ r: r� r� ) r Zarg_setsr. r r Ztobuild_pytest_paramsZ _arg_getterZfnsZ fn_paramsZparam_exclusions� parametersr� r r )r r� r r r% r& r � combinations� sh ��� � ��� � &z#PytestFixtureFunctions.combinationsc G s |d }t j|dd � d|i�S )Nr r r� )r� r� )r r) �identr r r � param_ident1 s z"PytestFixtureFunctions.param_identc sz ddl m� ddl m� t|�dkrFt|d �rF|d }|dd � }nd }tj||��� ��fdd�}|rr||�S |S d S )Nr r3 rd r c s �j r� �| �} �| �} | S r )r� Z_maybe_async_wrapperr� �re r4 �fixturer r �wrapK s z,PytestFixtureFunctions.fixture.<locals>.wrap)rf r4 re r' r� r� r. )r r� r. r� r/ r r- r r. 5 s z PytestFixtureFunctions.fixturec C s t j�d�S )NZPYTEST_CURRENT_TEST)rA �environ�getr r r r �get_current_test_nameZ s z,PytestFixtureFunctions.get_current_test_namec s$ ddl m� t� fdd��}||�S )Nr rd c s � j | f|�|� d S r )Z_run_coroutine_functionr� rd r r r ` s z3PytestFixtureFunctions.async_test.<locals>.decorate)rf re r� )r r� r r rd r � async_test] s z!PytestFixtureFunctions.async_testN)r r r r �propertyr r �reprr� r r* r, r. r2 r3 r r r r rG � s � %rG )T)0� __future__r r+ r~ � functoolsr r� r� r rA r� rX �typingr rw r� Zsqla_plugin_baser � ImportErrorr� r2 rJ rI r. rP rb rg ri rk rs r? r� r� r� r� r� r� r� Zhookimplr� r� r� r� r� r� r� ZFixtureFunctionsrG r r r r �<module> s\ 1& a ! 5 >2