Current File : //opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyc
�
�M!Vc @ sX d Z y d d l Z Wn! e k
r9 d d l m Z n Xd d l Z e j d k Z d d l Z d d l Z d d l
Z
d d l Z d d l Z y d d l
Z
e Z Wn e k
r� e Z n Xd � Z d � Z d � Z e re j d � Z d
� Z d � Z n d � Z d
� Z d a d � Z d � Z d � Z d � Z d � Z d � Z! d S( s� NOTE: copied/adapted from SQLAlchemy master for backwards compatibility;
this should be removable when Alembic targets SQLAlchemy 1.0.0.
i����Ni ( t plugin_basei i c s9 | j d � � � f d � } t j | � t j � d S( Nt
sqlalchemyc sX | j d d � � � rD d t j f � f d � � Y} | | d <n � j | | � d S( Nt callbackt CallableActionc s e Z d � f d � Z RS( c s � | | | � d S( N( ( t selft parsert namespacet valuest
option_string( t callback_( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt __call__$ s N( t __name__t
__module__t NoneR
( ( R ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyR # s t action( t popR
t argparset Actiont addoption( t namet kwR ( t group( R sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt make_option s
( t getgroupR t
setup_optionst read_config( R R ( ( R sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt pytest_addoption s
c C sr t | d � r6 t j | j � t j | j d � n t j | j � t t | j d t � � } t j
| � d S( Nt
slaveinputt follower_identt
cov_source( t hasattrR t! restore_important_follower_configR t configure_followert pre_begint optiont boolt getattrt Falset set_coverage_flag( t configt coverage( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt pytest_configure/ s c C s t j � d S( N( R t
post_begin( t session( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt pytest_sessionstart< s c C sO t j | j � d t t � | j d <d d l m } | j | j d � d S( Ns test_%sR i����( t provision( R t! memoize_important_follower_configR t nextt _follower_countt alembic.testingR- t create_follower_db( t nodeR- ( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt pytest_configure_nodeB s c C s( d d l m } | j | j d � d S( Ni����( R- R ( R1 R- t drop_follower_dbR ( R3 t errorR- ( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt pytest_testnodedownL s c C sg t j t � } g | D]! } t | j t j � r | ^ q | (t d � | D� � } x� | D]� } x� t j | j
| j j � D]g } | | j
k r} | | j
} xB t j | j
d | j j �j � D] } | j | j � � q� Wq} q} Wq[ Wg }
xX | D]P } | j j
| k r<|
j | | j j
� g | | j j
(q� |
j | � q� Wt |
d d � �| (d S( Nc s s | ] } | j Vq d S( N( t parent( t .0t item( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pys <genexpr>` s R8 t keyc S s% | j j j j | j j j | j f S( N( R8 R ( R: ( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt <lambda>v s ( t collectionst defaultdictt listt
isinstanceR8 t pytestt Instancet setR t generate_sub_testst clst modulet ClassR t collectt extendt appendt sorted( R+ R' t itemst
rebuilt_itemsR: t test_classest
test_classt sub_clst list_t instt newitems( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt pytest_collection_modifyitemsQ s( ,
"
c C s� t j | � r1 t j | � r1 t j | d | �St j | � rz t | t j � rz t j | j
| � rz t j | d | �Sg Sd S( NR8 ( t inspectt isclassR t
want_classRA RG t
isfunctionR@ RB t want_methodRE t Function( t collectorR t obj( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt pytest_pycollect_makeitem} s c sq t � t j � s d St d k rc t � j j � � j j a � f d � } � j j j | � n t � � d S( Nc s t � j j � d a d S( N( t class_teardownR8 R
t _current_class( ( R: ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt finalize� s ( R@ RA RZ R_ R
t class_setupR8 t addfinalizert
test_setup( R: R` ( ( R: sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt pytest_runtest_setup� s c C s t | � d S( N( t
test_teardown( R: ( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt pytest_runtest_teardown� s c C s, t j | | j j j | j j | j � d S( N( R t before_testR8 RF R RE R ( R: ( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyRc � s c C s t j | � d S( N( R t
after_test( R: ( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyRe � s c C s t j | j � d S( N( R t start_test_classRE ( R: ( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyRa � s c C s t j | j � d S( N( R t stop_test_classRE ( R: ( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyR^ � s ( i i (" t __doc__t alembic_plugin_baseR t ImportErrort t syst version_infot py3kRA R RU R= t itertoolst xdistt Truet has_xdistR% R R) R, t countR0 R4 R7 RT R] R
R_ Rd Rf Rc Re Ra R^ ( ( ( sT /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/pytestplugin.pyt <module> s@
,
Mini Shell