NEw/QwddlZddlZddlZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z dd l mZd d lmZd d lm Z d d lmZd dlmZd dlmZGddeZGddeZGddeZGddeZGddeeZGddeeejZGddeZGddeZdS) N) assertions)configschema)drop_all_tables) BasicEntity)ComparableEntity)adict)event)util)declarative_base)DeclarativeMeta)sort_tables_and_constraintscbeZdZdZdZdZdZddZdZdZ e j dZ dS)TestBaseNc|s J|dSNr)selfvalmsgs /srv/buildsys-work-dir/castor/build_node/builder-2/WGSG1/unpkd_srcs/cloudlinux-venv-1.0.6/venv/lib/python3.11/site-packages/sqlalchemy/testing/fixtures.pyassert_zTestBase.assert_-sCscRt|dr|dSdS)NsetUp)hasattrrrs rsetupzTestBase.setup1s/ 4 ! !  JJLLLLL  rcRt|dr|dSdS)NtearDown)rr#r s rteardownzTestBase.teardown5s/ 4 $ $  MMOOOOO  rc#"Ktj}|} |V||dS#||wxYwr)rdbconnectbeginrollbackclose)rconntranss r connectionzTestBase.connection9szy  ""  JJJ NN    JJLLLLL NN    JJLLLLs A$$*Br) __name__ __module__ __qualname__ __requires____unsupported_on__ __only_on__ __skip_if__rr!r$rfixturer-rrrrrsLKKV^rrcheZdZdZdZdZdZdZdZdZ dZ dZ dZ e dZe dZe dZe dZdZd Zd Zd Zd Ze d Ze dZe dZe dZe dZe dZe dZdZdZe dZ dS) TablesTestonceeachNc~|||dSr) _init_class_setup_once_tables_setup_once_insertsclss r setup_classzTablesTest.setup_classms<      !!!!!rc&|jdkr|jdkrd|_|jdvsJt|_t|_||_tj |_ |j|j _dS)Nr9r8)r9N) run_define_tablesrun_create_tables run_insertsr othertables setup_bindbindsaMetaDatametadatar>s rr;zTablesTest._init_classus  F * *$..(.%?n4444GG WW >>##{}} H rc|jdkr]||j5}||ddddS#1swxYwYdSdSNr8rD_load_fixturesrHr( insert_data)r?r+s rr=zTablesTest._setup_once_insertss ?f $ $    !! &T%%% & & & & & & & & & & & & & & & & & & % $AA #A c|jdkrj||j|jdkr|j|j|j|jjdSdSrMrB define_tablesrKrC create_allrHrFupdater>s rr<zTablesTest._setup_once_tablessu  F * *   cl + + +$.. ''111 J  cl1 2 2 2 2 2 + *rcH|jdkrj||j|jdkr|j|j|j|jjdS|jdkr!|j|jdSdSNr9rSr s r_setup_each_tableszTablesTest._setup_each_tabless  !V + +   t} - - -%// ((333 K  t}3 4 4 4 4 4  #v - - M $ $TY / / / / /. -rc|jdkr]||j5}||ddddS#1swxYwYdSdSrXrN)rr+s r_setup_each_insertszTablesTest._setup_each_insertss  v % %    ! ! !"" 'd  &&& ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' ' & %rQc |jdkrX|j|jdkrt |j|j|jn%|jdkrt |j|j|jdkr|jdkr|j5}tdt|jj DD]q} | | +#tjj$r0}t#jd|d|t&jYd}~jd}~wwxYw ddddS#1swxYwYdSdSdS)Nr9cg|] \}}|| Srr).0tfkss r z4TablesTest._teardown_each_tables..s-$Q= )==rzError emptying table z: )file)rBrFclearrCrrKrH run_deletesr'reversedrvaluesexecutedeleterIexc DBAPIErrorrprint_sysstderr)rr+tableexs r_teardown_each_tablesz TablesTest._teardown_each_tabless  !V + + K     %// ty999 M   ! ! ! !  #v - - DM49 5 5 5  !V + +0@F0J0J""$$ %(C M07799))E U\\^^44446, >CeeRRH!$                   , +0J0Js=8AE;>'D&%E;&E*:&E% E;%E**E;;E?E?cV||dSr)rYr[r s rr!zTablesTest.setups, !!!   """""rc.|dSr)rpr s rr$zTablesTest.teardowns ""$$$$$rc|jrt|j|j|jdkr||jd|j_|j d|_dSdSrM)rCrrKrHrun_dispose_bind dispose_bindrun_setup_bindr>s r_teardown_once_metadata_bindz'TablesTest._teardown_once_metadata_bindsk   4 CL#( 3 3 3  6 ) )   SX & & &     )CHHH * )rc.|dSr)rwr>s rteardown_classzTablesTest.teardown_classs ((*****rctjSr)rr&r>s rrGzTablesTest.setup_binds yrct|dr|dSt|dr|dSdS)Ndisposer*)rr|r*)r?rHs rruzTablesTest.dispose_bindsR 4 # #  LLNNNNN T7 # #  JJLLLLL  rcdSrr)r?rKs rrTzTablesTest.define_tables rciSrrr>s rfixtureszTablesTest.fixturess rcdSrr)r?r-s rrPzTablesTest.insert_datar~rc>||j||dSr)assert_sql_countrH)rcountfns r sql_count_zTablesTest.sql_count_s" diU33333rc>||j||dSr) assert_sqlrH)r callable_ statementss rsql_eq_zTablesTest.sql_eq_s   9j99999rciic}|D]X\}t|dkrttjr |j|d<|dd|<Yt|jj D]M\}vr |j  fd|DNdS)z4Insert rows as represented by the fixtures() method.r rrNcVg|]%}tt|&Sr)dictzip)r^ column_valuesheadersrns rraz-TablesTest._load_fixtures..s?%WU^];;<<r) ritemslen isinstancer string_typesrFrrKrfrHrginsert)r?rowsdatar`rrns @@rrOzTablesTest._load_fixturess7B <<>>//11 # #KE44yy1}}%!233 * 5)!!WGENqrr(DKK5 L  & & ( (   JE3}G## H   )-e      r)!r.r/r0rvrBrCrDrdrtrHrKrFrE classmethodr@r;r=r<rYr[rpr!r$rwryrGrurTrrPrrrOrrrr7r7TsNKK DH F E""[" % %[ %&&[& 33[3000''' :###%%%  [ ++[+[[   [ [  [ 444:::[rr7cDeZdZejdZdZfdZxZS) RemovesEventsctSr)setr s r _event_fnszRemovesEvents._event_fnss uu rc h|j|||ftj|||fi|dSr)raddr listen)rtargetnamerkws r event_listenzRemovesEvents.event_listensA VT2./// VT2,,,,,,,rc|jD]}tj|tt|}t |dr|dSdS)Nr$)rr removesuperrrr$)rkeysuper_ __class__s rr$zRemovesEvents.teardownse?  C L#   }d++ 6: & &  OO       r) r.r/r0rmemoized_propertyrrr$ __classcell__rs@rrrsd ---rrc$eZdZedZdS)_ORMTestctjjtjdSr)rIormsessionclose_all_sessions clear_mappersr>s rryz_ORMTest.teardown_class s2 ))+++ rN)r.r/r0rryrrrrrs-[rrceZdZdS)ORMTestNr.r/r0rrrrr&sDrrceZdZdZdZdZedZedZdZ dZ edZ ed Z ed Z d Zd Zed ZdZdZedZedZdS) MappedTestr8r9Nc||jt|_||||dSr)r;classesr r<_setup_once_classes_setup_once_mappersr=r>s rr@zMappedTest.setup_class3ss  ; ''CK     !!! !!! !!!!!rcV||dSr)_teardown_once_classrwr>s rryzMappedTest.teardown_class?s,   """ ((*****rc||||dSr)rY_setup_each_classes_setup_each_mappersr[r s rr!zMappedTest.setupDsP !!!   """   """   """""rctjj|||dSr)rIrrr_teardown_each_mappers_teardown_each_classesrpr s rr$zMappedTest.teardownJsT ))+++ ##%%% ##%%% ""$$$$$rcj|jtdSr)rrcrryr>s rrzMappedTest._teardown_once_classPs. !!!!!rcT|jdkr||jdSdSrMrun_setup_classes_with_register_classes setup_classesr>s rrzMappedTest._setup_once_classesU5  F * *  & &s'8 9 9 9 9 9 + *rcT|jdkr||jdSdSrMrun_setup_mappersr setup_mappersr>s rrzMappedTest._setup_once_mappersZrrcT|jdkr||jdSdSrXrr s rrzMappedTest._setup_each_mappers_5  !V + +  ' '(: ; ; ; ; ; , +rcT|jdkr||jdSdSrXrr s rrzMappedTest._setup_each_classescrrc |jGfddt}Gddtj|t}Gddt |}Gddt |}||_||_|d S) zRun a setup method, framing the operation with a Base class that will catch new subclasses to be established within the "classes" registry. ceZdZfdZdS)6MappedTest._with_register_classes..FindFixturecL||<t||||dSr)type__init__r? classnamebasesdict_ cls_registrys rrz?MappedTest._with_register_classes..FindFixture.__init__qs+*- Y' c9eU;;;;;rNr.r/r0rrsr FindFixturerps. < < < < < < ._BaseNrrrr_Baseru DrrceZdZdS)0MappedTest._with_register_classes..BasicNrrrrBasicrxrrrceZdZdS)5MappedTest._with_register_classes..ComparableNrrrr Comparabler{rrrN) rrrwith_metaclassobjectr r rr)r?rrrrrrs @rrz!MappedTest._with_register_classesgs{  < < < < < < <$ < < <      D' V<<        K        )5    # rc\|jdkr tjdSdSrM)rrIrrr s rrz!MappedTest._teardown_each_mapperss4  !V + + F " " " " " , +rcR|jdkr|jdSdSrM)rrrcr s rrz!MappedTest._teardown_each_classess2  !V + + L    , +rcdSrrr>s rrzMappedTest.setup_classesr~rcdSrrr>s rrzMappedTest.setup_mappersr~r)r.r/r0rrrrr@ryr!r$rrrrrrrrrrrrrrr*sWG " "[ "++[+### %%% ""["::[:::[:<<<<<<  [ 4###!!!  [   [   rrcLeZdZdZdZedZefdZxZS)DeclarativeMappedTestr8cdSrrr>s rr<z(DeclarativeMappedTest._setup_once_tablesr~rcj|jGfddt}Gddt}t|j||}||_t t|||jj r-|j r(|j tj dSdSdS)NceZdZfdZdS)LDeclarativeMappedTest._with_register_classes..FindFixtureDeclarativec@||<tj||||dSr)rrrs rrzUDeclarativeMappedTest._with_register_classes..FindFixtureDeclarative.__init__s**- Y'(iFFFFFrNrrsrFindFixtureDeclarativers5 G G G G G G GrrceZdZejZdS)FDeclarativeMappedTest._with_register_classes..DeclarativeBasicN)r.r/r0rTable __table_cls__rrrDeclarativeBasicrs"LMMMrr)rK metaclassr?)rrrrrKrrrrrFrCrUrr&)r?rrr _DeclBaserrs @rrz,DeclarativeMappedTest._with_register_classess{  G G G G G G G_ G G G  ) ) ) ) )v ) ) )%\,    ) #S))@@DDD <  /3#8 / L # #FI . . . . . / / / /r) r.r/r0rrrr<rrrs@rrrsi  [ ////[/////rrcZeZdZdxZZdZdZejdZ dZ e dZ dS)ComputedReflectionFixtureTestNT)computed_columnstable_reflectionz[\[\]\(\)\s`'\"]*c\|jd|S)N)regexpsublower)rtexts r normalizez'ComputedReflectionFixtureTest.normalizes${r4((..000rc ddlm}ddlm}ddlm}ddlm}ddlm}|d||d|d |d ||d ||d |d|d|d||d|d |d ||d||d }|jjj rE|d||d|d |d ||d||dtj }|jj j rg| |d||dd|jjj r+| |d||dd|jjj ri| |d||dd |jjj r/| |d||dd dSdSdS)Nr )Integer)testing)Column)Computed)rcomputed_default_tableidT) primary_keynormal computed_colz normal + 42 with_default42)server_defaultcomputed_column_tablecomputed_no_flagz normal / 42rcomputed_virtualz normal + 2F) persistedz normal / 2computed_storedz normal - 42z normal * 42)rrrrrrrrequiresschemasenabledr test_schemacomputed_columns_virtual append_columncomputed_columns_stored) r?rKrrrrrr_t2s rrTz+ComputedReflectionFixtureTest.define_tabless######%%%%%%""""""  $  F4d 3 3 3 F8W % % F>7HH],C,C D D F>74 @ @ @     E #  F4d 3 3 3 F8W % % F%w0G0G H H      # + 'tW$777x)))7HH]4K4KLL) B   4 <  OO&H\U;;;   '/   F* ???   3 ;  OO%H]d;;;   '/   F) $???    r) r.r/r0rDrd __backend__r1recompilerr rrTrrrrrsi $$K+K;L RZ, - -F111AA[AAArr)r(rl sqlalchemyrIrrrrenginesrentitiesr r rr r ext.declarativerrrrrr7rrrAssertsExecutionResultsrrrrrrr/s~ $$$$$$!!!!!!&&&&&&......------000000#####v###lwwwwwwwwtF"v     h   i i i i i :z'Ii i i X / / / / /J / / /FMMMMMJMMMMMr