Current File : //opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyc
�
�M!Vc @@ s� d Z d d l m Z y d d l m Z Wn! e k
rM d d l m Z n Xd d l Z d d l Z e j
d, k Z e r� d d l Z n d d l
Z d a d a d Z d a d a d a d Z d a d a d a d a i a e � a e � a d a d � Z d � Z d � Z! d
� Z" d � Z# d � Z$ d
� Z% d � Z& d � Z' d � Z( d � Z) d � Z* d � Z+ g Z, g Z- d � Z. d � Z/ e. d � � Z0 e. d � � Z1 e. d � � Z2 e/ d � � Z3 e/ d � � Z4 e/ d � � Z5 d � Z6 e/ d � � Z7 e/ d � � Z8 e/ d � � Z9 d � Z: d! � Z; d" � Z< d# � Z= d$ � Z> d% � Z? d& � Z@ d' � ZA d( � ZB d d) � ZC d* � ZD d+ � ZE d S(- sG Testing extensions.
this module is designed to work as a testing-framework-agnostic library,
so that we can continue to support nose and also begin adding new
functionality via py.test.
NOTE: copied/adapted from SQLAlchemy master for backwards compatibility;
this should be removable when Alembic targets SQLAlchemy 1.0.0
i ( t absolute_import( t SkipTest( t SkippedNi c
C@ s� | d d d d d d t d d �| d d d d d d t d d �| d
d d d d d d
d d �| d d d d t d d �| d d d d d d d d d �| d d d d d d d �| d d d d d d d �| d d d d d d d �| d d d d d d d �| d! d d d d" d# t d d$ �| d% d d d d d t d d& �| d' d d d d( d# t d d) �| d* d d d t d d d d+ �| d, d d d t d d d d- �| d. d d d d/ �| d0 d d1 d d2 d# d d d3 �d S(4 Ns
--log-infot actiont callbackt typet stringt helps, turn on info logging for <LOG> (multiple OK)s --log-debugs- turn on debug logging for <LOG> (multiple OK)s --dbt appendt destt dbsB Use prefab database uri. Multiple OK, first one is run by default.s --dbss List available prefab dbss --dburit dburis8 Database uri. Multiple OK, first one is run by default.s --dropfirstt
store_truet dropfirsts, Drop all tables in the target database firsts --backend-onlyt backend_onlys& Run only tests marked with __backend__s
--mockpoolt mockpools0 Use mock pool (asserts only one connection used)s --low-connectionst low_connectionss> Use a low number of distinct connections - i.e. for Oracle TNSs --reversetopt
reversetopt defaultsT Use a random-ordering set implementation in the ORM (helps reveal dependency issues)s --requirementss3 requirements class for testing, overrides setup.cfgs --with-cdecimalt cdecimalsD Monkeypatch the cdecimal library into Python 'decimal' for all testss
--include-tags Include tests with tag <tag>s
--exclude-tags Exclude tests with tag <tag>s --serversides" Turn on server side cursors for PGs --mysql-enginet storet mysql_engines\ Use the specified MySQL storage engine for all tables, default is a db-default/InnoDB combo.( t _logt _list_dbst Falset _requirements_optt _include_tagt _exclude_tagt None( t make_option( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt
setup_options9 sP c C@ s d d l m } | | _ d S( s� Configure required state for a follower.
This invokes in the parent process and typically includes
database creation.
i ( t provisionN( t alembic.testingR t FOLLOWER_IDENT( t follower_identR ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt configure_followerj s c C@ s# i t d 6t d 6t d 6| d <d S( sT Store important configuration we will need to send to a follower.
This invokes in the parent process after normal config is set up.
This is necessary as py.test seems to not be using forking, so we
start with nothing in memory, *but* it isn't running our argparse
callables, so we have to just copy all of that over.
t db_optst include_tagst exclude_tagst memoized_configN( R$ R% R& ( t dict_( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt! memoize_important_follower_configu s
c C@ sC t j | d d � t j | d d � t j | d d � d S( sf Restore important configuration needed by a follower.
This invokes in the follower process.
R' R$ R% R& N( R$ t updateR% R& ( R( ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt! restore_important_follower_config� s c C@ s# t j � a t j d d g � d S( Ns setup.cfgs test.cfg( t configparsert ConfigParsert file_configt read( ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt read_config� s c C@ s( | a x t D] } | t t � q
Wd S( s7 things to set up early, before coverage might be setup.N( t optionst
pre_configureR. ( t optt fn( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt pre_begin� s
c C@ s
| t _ d S( N( R1 t has_coverage( t value( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt set_coverage_flag� s c C@ sn x t D] } | t t � q Wd d l m a m a m a d d l m a m a d d l m
a
t j � d S( s9 things to set up later, once we know coverage is running.i ( t configt warningst
exclusions( t enginest fixtures( t utilN( t post_configureR1 R. R R9 R: R; R< R= t
sqlalchemyR> t
setup_filters( R4 ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt
post_begin� s
c C@ sy t s d d l a t j � n | j d � rJ t j | � j t j � n+ | j d � ru t j | � j t j � n d S( Ni s -infos -debug( t loggingt basicConfigt endswitht getLoggert setLevelt INFOt DEBUG( t opt_strR7 t parser( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyR � s
c G@ sQ d GHx8 t t j d � � D]! } d | t j d | � f GHq Wt j d � d S( Ns0 Available --db options (use --dburi to override)R
s %20s %si ( t sortedR. R1 t gett syst exit( t argst macro( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyR � s c C@ s t | � d S( N( t _setup_requirements( RJ R7 RK ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyR � s c C@ s t j | j d d � � d S( Nt -t _( R& t addt replace( RJ R7 RK ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyR � s c C@ s t j | j d d � � d S( NRS RT ( R% RU RV ( RJ R7 RK ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyR � s c C@ s t j | � | S( N( R2 R ( R4 ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt pre� s
c C@ s t j | � | S( N( R? R ( R4 ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt post� s
c C@ s
| a d S( N( R1 ( R3 R. ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt _setup_options� s c C@ s | j r t t d <n d S( Nt server_side_cursors( t
serversidet TrueR$ ( R1 R. ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt _server_side_cursors� s c C@ s) | j r% d d l } | t j d <n d S( Ni t decimal( R RN t modules( R1 R. R ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt _monkeypatch_cdecimal� s c C@ s- d d l m } d d l m } | j r; t | j � } n g } | j r� xr | j D]d } x[ t j d | � D]G } | | j d � k r� t d | � � qm | j
| j d | � � qm WqT Wn | s� | j
| j d d � � n xE | D]= } | j | t
| | | j � } | j s� | j | � q� q� Wd S( Ni ( R9 ( R s [,\s]+R
s: Unknown URI specifier '%s'. Specify --dbs for known uris.R ( R R9 R R t listR
t ret splitR1 t RuntimeErrorR RM t setup_configR$ R! t _currentt set_as_current( R1 R. R9 R t db_urlst db_tokenR
t db_urlt cfg( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt _engine_uri� s(
$
c C@ s- | j r) d d l m } | j t d <n d S( Ni ( t poolt poolclass( R R@ Rm t
AssertionPoolR$ ( R1 R. Rm ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt _engine_pool s c C@ s | j d d � } t | � d S( Nt sqla_testingt requirement_cls( RM RR ( R1 R. Rr ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt
_requirements s c C@ s� d d l m } | j d k r# d S| j d � \ } } t | � } x* | j d � d D] } t | | � } qX Wt | | � } | � | _ d S( Ni ( R9 t :t .i ( R R9 t requirementsR Rc t
__import__t getattr( t argumentR9 t modnamet clsnamet modt componentt req_cls( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyRR # s c C@ s� d d l m } d d l m } d d l m } d d l m } | j r\ d d l m } n d d l
m } | j } | j
r�x<| j j � D](} | j } | | � }
y |
j � } Wn t k
r� n: Xx6 | D]. } | j | j | j | | j � � � � q� W| j j j | � r�y |
j d d � } Wn t k
rDq�Xx? | D]4 } | j | j | j | | j � d d �� � qLWn xH t |
j d
d � � D]. }
| j | j | j |
| j � � � � q�W| j j j | � rAxW t |
j d
d d d � � D]4 }
| j | j | j |
| j � d d �� � qWn | | d � r� | j r� d d
l m } xJ |
j d � D]6 } | j | j! | j" d | d d | d � � � qyWq� q� Wn d S( Ni ( R9 ( t against( t schema( R> ( t inspect( t InspectorR� t test_schemat order_byt foreign_keyt
postgresql( R� t *t name(# R R9 t alembic.testing.exclusionsR R@ R� t alembicR> t sqla_08R� t sqlalchemy.engine.reflectionR� t from_engineR
t Configt all_configsR
t get_view_namest NotImplementedErrort executet _DropViewt Tablet MetaDataRv t schemast enabled_for_configt reversedt get_table_namest DropTablet sqla_100t sqlalchemy.dialectsR� t get_enumst DropEnumTypet ENUM( R1 R. R9 R R� R> R� R� Rk t et inspectort
view_namest vnamet tnameR� t enum( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt _prep_testing_database5 s\
c C@ s' | j r# d d l m } | � n d S( Ni ( t randomize_unitofwork( R t sqlalchemy.orm.utilR� ( R1 R. R� ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt _reverse_topologicalt s c C@ s& d d l m } t | _ | | _ d S( Ni ( R9 ( R R9 R1 R. ( R3 R. R9 ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt _post_setup_options{ s c C@ sW t | t j � s t S| j j d � r, t St j j rO t | d t � rO t St
Sd S( NRT t __backend__( t
issubclassR= t TestBaseR t __name__t
startswithR9 R1 R Rx R\ ( t cls( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt
want_class� s
c C@ s� | j j d � s t S| j d k r) t St r~ t | d � r\ t j | j � j
t t � p} t | d � o} | j j
t t � St r� t | d � r� t j | j � j
t t � St r� t | d � r� | j j
t t � St
Sd S( Nt test_t __tags__t _sa_exclusion_extend( R� R� R t
__module__R R% t hasattrR; t tagsR� t include_testR& R� R\ ( R� R4 ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt want_method� s"
c c@ s� t | d t � r� x� t | � D]o } d | j | j j | j j f } t | | f i d | j j | j j f d 6� } t | | | � | Vq Wn | Vd S( NR� s %s_%s_%ss %s+%st __only_on__( Rx R t _possible_configs_for_clsR� R
R� t driverR t setattr( R� t moduleRk R� t subcls( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt generate_sub_tests� s "# c C@ s t | � t | � d S( N( t _do_skipst
_setup_engine( R� ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt start_test_class� s
c C@ s t � d S( N( t _restore_engine( R� ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt stop_test_class� s c C@ s t j j � d S( N( R9 Rf t reset( ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyR� � s c C@ s> t | d d � r: t j d | j � } t j j | � n d S( Nt __engine_options__R1 ( Rx R R< t testing_engineR� R9 Rf t push_engine( R� t eng( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyR� � s c C@ s d S( N( ( t testt test_module_namet
test_classt test_name( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt before_test� s c C@ s d S( N( ( R� ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt
after_test� s c
C@ s t t j j � � } | j rf t j | j � } x3 t | � D]" } | | � r= | j | � q= q= Wn t | d d � r� t j t j | j
� � } x3 t | � D]" } | | � s� | j | � q� q� Wn t | d � r`t j } x| t | � D]k } xb | j D]W } t | | � } | j | � } | r� | j | � | d k rQ| j | � n Pq� q� Wq� Wn t | d � r�t � } t j } xV t | � D]H } x? | j D]4 } t | | � } | j | � s�| j | � q�q�Wq�W| j | � r�| j | � q�n | S( NR� t __requires__t __prefer_requires__( t setR9 R� R� t __unsupported_on__R; t db_specRa t removeRx R R> t to_listR� R� Rv R� t matching_config_reasonst extendR� R� RU t
differencet difference_update(
R� t reasonsR� t spect
config_objRv t requirementt checkt skip_reasonst
non_preferred( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyR� � sB
c C@ s� g } t | | � } t | d t � rl xB t | d � D]. } | � r7 t d | j | j f � � q7 q7 Wn | s� t | d t � r� d | j | j f } n; d | j d j d � t j j � D� � d j | � f } t | � � n� t
| d � rit � } t j
t j | j � � } x* | D]" } | | � s!| j | � q!q!W| j | � ri| j | � qin t j | k r�t | j � | � n d S( Nt __skip_if__s '%s' skipped by %sR� s( '%s' unsupported for implementation '%s's. '%s' unsupported on any DB implementation %s%ss , c s@ sJ | ]@ } d | j j d j d � | j j j D� � | j j f Vq d S( s '%s(%s)+%s'Ru c s@ s | ] } t | � Vq d S( N( t str( t .0t dig( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pys <genexpr> s N( R
R� t joint dialectt server_version_infoR� ( R� R� ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pys <genexpr> s t __prefer_backends__( R� Rx R R R� R� R� R9 R� R� R� R� R; R� R> R� R� RU R� R� Rf t
_setup_configt pop( R� R� R� t ct msgR� R� R� ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyR� � s8
c C@ s t j j | � d S( N( R9 Rf t push( R� t ctx( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyR� + s ( i i (F t __doc__t
__future__R t noseR t ImportErrort _pytest.runnerR RN Rb t version_infot py3kR, R- R R= R< R R; R: t
assertionsRv R9 R> R. RC R$ R� R% R& R1 R R# R) R+ R0 R5 R8 RB R R R R R R2 R? RW RX RY R] R` Rl Rp Rs RR R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� ( ( ( sS /opt/alt/python27/lib/python2.7/site-packages/alembic/testing/plugin/plugin_base.pyt <module> s�
1 ? + ,
Mini Shell