JezK
Edit File: plugin_base.pyc
� 1��fc @@ s" d Z d d l m Z d d l Z d d l a d d l Z d d l Z e Z t j d � Z e j d6 k Z e r� d d l Z e j Z n. d d l Z d d l Z d e f d � � YZ d a d a d a d a d a d a d a d Z d a d a d a d a d 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 � Z0 g Z1 g Z2 d � Z3 d � Z4 e3 d � � Z5 e3 d � � Z6 e3 d � � Z7 e3 d � � Z8 e4 d � � Z9 e3 d � � Z: e4 d � � Z; e4 d � � Z< d � Z= e4 d � � Z> e4 d! � � Z? e4 d" � � Z@ e4 d# � � ZA d$ � ZB d% � ZC d& � ZD d' � ZE d( � ZF d) � ZG d* � ZH d+ � ZI d, � ZJ d- � ZK d. � ZL d/ � ZM d e d0 � ZN d1 � ZO d2 � ZP d3 e f d4 � � YZQ d aR d5 � ZS d S(7 s� Testing extensions. this module is designed to work as a testing-framework-agnostic library, created so that multiple test frameworks can be supported at once (mostly so that we can migrate to new ones). The current target is pytest. i ( t absolute_importNs sqlalchemy.testing.plugin_basei t ABCc B@ s e Z e j Z RS( ( t __name__t __module__t abct ABCMetat __metaclass__( ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR ) s c C@ s | d d d d t d t d d �| d d d d t d t 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 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 t d d/ �| d0 d d d d1 d d2 �| d3 d t d d4 d d5 �| d6 d d d d7 d' t d d8 �| d9 d d d t d t d d: �| d; d d d d< d' t d d= �| d> d d d t d t d d? �| d@ d d d t d t d dA �| dB d d d dC d' t d dD �| dE d d d dF d' t d dG �| dH d t d dI d dJ �| dK d t d d d' g d dL d dM �d S(N Ns --log-infot actiont callbackt typet 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 --dbst zeroarg_callbacks List available prefab dbss --dburit dburis8 Database uri. Multiple OK, first one is run by default.s --dbdrivert dbdrivers| Additional database drivers to include in tests. These are linked to the existing database URLs by the provisioning system.s --dropfirstt store_truet dropfirsts, Drop all tables in the target database firsts --disable-asyncios8 disable test / fixtures / provisoning running in asyncios --backend-onlyt backend_onlys< Run only tests marked with __backend__ or __sparse_backend__s --nomemoryt nomemorys Don't run memory profiling testss --notimingintensivet notimingintensives Don't run timing intensive testss --profile-sortt defaultt cumulativet profilesorts* Type of sort for profiling standard outputs --profile-dumpt profiledumps2 Filename where a single profile run will be dumpeds --postgresql-templatedbs^ name of template database to use for PostgreSQL CREATE DATABASE (defaults to current database)s --low-connectionst low_connectionss> Use a low number of distinct connections - i.e. for Oracle TNSs --write-identst write_identssC write out generated follower idents to <file>, when -n<num> is useds --reversetopt reversetopsT 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 --write-profilest write_profiless$ Write/update failing profiling data.s --force-write-profilest force_write_profiless, Unconditionally write/update profiling data.s --dump-pyannotatet dump_pyannotates/ Run pyannotate and dump json info to given files --mypy-extra-test-patht mypy_extra_test_pathssl Additional test directories to add to the mypy tests. This is used only when running mypy tests. Multiple OK( t strt _logt _list_dbst Falset _requirements_optt _include_tagt _exclude_tag( t make_option( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt setup_optionsA s 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 sqlalchemy.testingR+ t FOLLOWER_IDENT( t follower_identR+ ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt configure_follower� s c C@ s i t d 6t d 6| d <d S( sS 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 pytest 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 include_tagst exclude_tagst memoized_configN( R0 R1 ( t dict_( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt! memoize_important_follower_config� s c C@ s. t j | d d � t j | d d � d S( sf Restore important configuration needed by a follower. This invokes in the follower process. R2 R0 R1 N( R0 t updateR1 ( R3 ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/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( ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/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_configureR9 ( t optt fn( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt pre_begin s c C@ s | t _ d S( N( R< t has_coverage( t value( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt set_coverage_flag% s c C@ s� x t D] } | t t � q Wd d l m a d d l m a m a m a d 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 testing( t fixturest enginest exclusions( t assertionst warningst profiling( t configR+ ( t utilN( t post_configureR< R9 t sqlalchemyRD R, RE RF RG RH RI RJ RK R+ RL t setup_filters( R? ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/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_strRB t parser( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/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 sortedR9 R< t gett syst exit( t argst macro( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR$ H s c C@ s t | � d S( N( t _setup_requirements( RX RB RY ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR& O s c C@ s t j | j d d � � d S( Nt -t _( R1 t addt replace( RX RB RY ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR( S s c C@ s t j | j d d � � d S( NRa Rb ( R0 Rc Rd ( RX RB RY ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR' W s c C@ s t j | � | S( N( R= R ( R? ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt pre_ s c C@ s t j | � | S( N( RM R ( R? ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt postd s c C@ s | a d S( N( R< ( R> R9 ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt _setup_optionsi s c C@ s | j r t j d � n d S( Nt memory_intensive( R R1 Rc ( R> R9 ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt _set_nomemoryo s c C@ s | j r t j d � n d S( Nt timing_intensive( R R1 Rc ( R> R9 ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt _set_notimingintensiveu s c C@ s) | j r% d d l } | t j d <n d S( Ni t decimal( R R\ t modules( R< R9 R ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt _monkeypatch_cdecimal{ s c C@ s d d l m } t � | _ d S( Ni ( RK ( R, RK t _fixture_fn_classt _fixture_functions( R< R9 RK ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt _init_symbols� s c C@ s0 | j s t r, d d l m } t | _ n d S( Ni ( t asyncio( t disable_asynciot py3kR, Rr R% t ENABLE_ASYNCIO( R> R9 Rr ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt _set_disable_asyncio� s c C@ s� d d l m } d d l m } d d l m } | j rK t | j � } n g } | j p] g } | j r� xr | j D]d } x[ t j d | � D]G } | | j d � k r� t d | � � q� | j | j d | � � q� Wqs Wn | s | j | j d d � � n d | _ t | j | | � � } x� | D]� } t j d | � | j r�| j r�t | j d � �# } | j | j d | d � Wd QXn | j | | | | j � } | j s(| j | | � q(q(Wd S( Ni ( RD ( RK ( R+ s [,\s]+R s: Unknown URI specifier '%s'. Specify --dbs for known uris.R s Adding database URL: %st at s ( RN RD R, RK R+ R t listR R t ret splitR< t RuntimeErrorR R[ t Nonet _currentt generate_db_urlst logt infoR R- t opent writet setup_configt set_as_current( R< R9 RD RK R+ t db_urlst extra_driverst db_tokenR t expanded_urlst db_urlt file_t cfg( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt _engine_uri� s8 $ % c C@ s | j d d � } t | � d S( Nt sqla_testingt requirement_cls( R[ R` ( R< R9 R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt _requirements� s c C@ s� d d l m } d d l m } | j d k r3 d S| j d � \ } } t | � } x* | j d � d D] } t | | � } qh Wt | | � } | � | _ | _ t | _ d S( Ni ( RK ( RD t :t .i ( R, RK RN RD t requirementsR} R{ t __import__t getattrt requirest bootstrapped_as_sqlalchemy( t argumentRK RD t modnamet clsnamet modt componentt req_cls( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR` � s c C@ s] d d l m } | j rY d d l m } x- | j j � D] } | j | | j � q9 Wn d S( Ni ( RK ( R+ ( R, RK R R+ t Configt all_configst drop_all_schema_objectsR ( R< R9 RK R+ R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt _prep_testing_database� s c C@ s' | j r# d d l m } | � n d S( Ni ( t randomize_unitofwork( R t sqlalchemy.orm.utilR� ( R< R9 R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt _reverse_topological� s c C@ s& d d l m } t | _ | | _ d S( Ni ( RK ( R, RK R< R9 ( R> R9 RK ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt _post_setup_options� s c C@ sD d d l m } | j | j d d � d | j d | j �| _ d S( Ni ( RJ R� t profile_filet sortt dump( R, RJ t ProfileStatsFileR[ R R t _profile_stats( R< R9 RJ ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt _setup_profiling� s c C@ sz t | t j � s t S| j d � r) t St j j rr t | d t � rr t | d t � rr t | d t � rr t St Sd S( NRb t __backend__t __sparse_backend__t __only_on__( t issubclassRE t TestBaseR% t startswithRK R< R R� t True( t namet cls( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/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% R R} R0 t hasattrRG t tagsR� t include_testR1 R� R� ( R� R? ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt want_method s" c c@ s� t | d t � s$ t | d t � r� t | d t � } x� t | d | �D]� } | j } t j d d | j � } t j d d | � } d | j | f } t | | f i | d 6| d 6� } t | | | � | VqI Wn | Vd S( NR� R� t sparses [_\[\]\.]+Rb s _+$t s %s_%st _sa_orig_cls_namet __only_on_config__( R� R% t _possible_configs_for_clsR Rz t subR� R t setattr( R� t moduleR� R� t orig_namet alpha_nameR� t subcls( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt generate_sub_tests- s c C@ s t | � t | � d S( N( t _do_skipst _setup_engine( R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt! start_test_class_outside_fixturesG s c C@ s t j | � t j j � d S( N( RE t stop_test_class_inside_fixturesRF t testing_reaper( R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt stop_test_classL s c C@ sL t j j � t j t t j | � z t j s< t j � n Wd t � Xd S( N( RF R� t stop_test_class_outside_fixturesR+ RK R R< R RH t global_cleanup_assertionst _restore_engine( R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR� U s c C@ s t j r t j j t � n d S( N( RK R~ t resetRD ( ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR� _ s c C@ s" t j j � t j � t � d S( N( RF R� t final_cleanupRH R� R� ( ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt final_process_cleanupd s c C@ sW t | d d � rS t | j � } d | d <t j d | � } t j j | t � n d S( Nt __engine_options__t classt scopeR< ( R� R} t dictR� RF t testing_engineRK R~ t push_engineRD ( R� t optst eng( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR� j s c C@ s9 t | d | j � } d | | | f } t j | � d S( NR� s %s.%s.%s( R� R RJ t _start_current_test( t testt test_module_namet test_classt test_nameR� t id_( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt before_testr s c C@ s t j � t j j � d S( N( RE t after_testRF R� ( R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR� ~ s c C@ s t j j | � d S( N( RF R� t after_test_outside_fixtures( R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt after_test_fixtures� 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 d � r� | j | j g � n t | d � r�t j } x| t | � D]k } xb | j D]W } t | | � } | j | � } | r&| j | � | d k ry| j | � n Pq&q&WqWn 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 | r�i } xH t t | d d � �� D]+ } | j j } | | k rL| | | <qLqLW| j � S| S( NR� R� t __requires__t __prefer_requires__t keyc S@ s" | j j | j j | j j j f S( N( R R� t drivert dialectt server_version_info( R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt <lambda>� s ( t setRK R� R� t __unsupported_on__RG t db_specRy t removeR� R} RL t to_listR� t intersection_updateR� R� R� R� t matching_config_reasonst extendR� t enabled_for_configRc t differencet difference_updatet reversedRZ R R� t values( R� t reasonsR� R� t spect config_objR� t requirementt checkt skip_reasonst non_preferredt per_dialectR� R ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR� � sZ c C@ si g } t | | � } t | d t � rm xC t | d � D]/ } | � r7 t j d | j | j f � q7 q7 Wn | s� d | j d j d � t j j � D� � d j | � f } t j | � n� t | d � r@t � } t j t j | j � � } x* | D]" } | | � s� | j | � q� q� W| j | � r@| j | � q@n t j | k ret | j � | � n d S( Nt __skip_if__s '%s' skipped by %ss. '%s' unsupported on any DB implementation %s%ss , c s@ sM | ]C } d | j j d j d � t j | j � D� � | j j f Vq d S( s '%s(%s)+%s'R� c s@ s | ] } t | � Vq d S( N( R" ( t .0t dig( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pys <genexpr>� s N( R R� t joinRG t _server_versionR� ( R R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pys <genexpr>� s t __prefer_backends__( R� R� R% RK t skip_testR R R� R� R� R� RG R� RL R� R Rc R� R� R~ t _setup_configt pop( R� R� R� t ct msgR R� R� ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR� � s2 ! c C@ s t j j | t � d S( N( RK R~ t pushRD ( R� t ctx( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR � s t FixtureFunctionsc B@ sk e Z e j d � � Z e j d � � Z e j d � � Z e j d � � Z d � Z e j d � � Z RS( c O@ s t � � d S( N( t NotImplementedError( t selft argt kw( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt skip_test_exception� s c O@ s t � � d S( N( R ( R R^ R ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt combinations� s c O@ s t � � d S( N( R ( R R^ R ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt param_ident s c O@ s t � � d S( N( R ( R R R ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt fixture s c C@ s t � � d S( N( R ( R ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt get_current_test_name s c C@ s t � � d S( N( R ( R ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt mark_base_test_class s ( R R R t abstractmethodR R R R R R ( ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyR � s c C@ s | a d S( N( Ro ( t fixture_fn_class( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt set_fixture_functions s ( i i (T t __doc__t __future__R R RQ Rz R\ R% R� RT R� t version_infoRt R7 R R8 t collectionst collections_abct objectR} RE RF RG RI RJ R+ RH R� RK RD RL R9 R� R0 R1 R< R* R/ R4 R6 R; R@ RC RP R# R$ R& R( R' R= RM Re Rf Rg Ri Rk Rn Rq Rv R� R� R` R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R� R R Ro R ( ( ( sX /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/testing/plugin/plugin_base.pyt <module> s� � . B *