o y\<*@sddlZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z d d lmZd d lm Z d d lmZd dlmZGdddeZGdddeZGdddeZGdddeZGdddeeZGdddeeejZGdddeZdS)N) assertions)config)schema)drop_all_tables) BasicEntity)ComparableEntity)adict)event)util)declarative_base)DeclarativeMetac@s:eZdZdZdZdZdZdZd ddZddZ ddZ dS) TestBaseNcCs|sJ|dSNr)selfvalmsgrrM/var/www/html/env/lib/python3.10/site-packages/sqlalchemy/testing/fixtures.pyassert_/szTestBase.assert_cCt|dr |dSdS)NsetUp)hasattrrrrrrsetup3  zTestBase.setupcCr)NtearDown)rrrrrrteardown7rzTestBase.teardownr) __name__ __module__ __qualname__ __whitelist__ __requires____unsupported_on__ __only_on__ __skip_if__rrrrrrrrs  rc@seZdZdZdZdZdZdZdZdZ dZ dZ dZ e ddZe ddZe dd Ze d d Zd d ZddZddZddZddZe ddZe ddZe ddZe ddZe ddZe d d!Ze d"d#Zd$d%Zd&d'Ze d(d)Z dS)* TablesTestonceeachNcCs|||dSr) _init_class_setup_once_tables_setup_once_insertsclsrrr setup_classUs zTablesTest.setup_classcCsZ|jdkr|jdkr d|_|jdvsJt|_t|_||_t |_ |j|j _dS)Nr)r()r)N) run_define_tablesrun_create_tables run_insertsr othertables setup_bindbindsaMetaDatametadatar-rrrr*]s    zTablesTest._init_classcC"|jdkr||dSdSNr(r2_load_fixtures insert_datar-rrrr,ks  zTablesTest._setup_once_insertscCsF|jdkr!||j|jdkr|j|j|j|jjdSdSr;)r0 define_tablesr9r1 create_allr6r4updater-rrrr+qs   zTablesTest._setup_once_tablescCs|jdkr7|j|jdkrt|j|j|j||j|jdkr-|j|j|j |jjdS|jdkrLt|j|j|j|jdSdSNr)) r0r4clearr1rr9r6r?r@rArrrr_setup_each_tablesys       zTablesTest._setup_each_tablescCr:rBr<rrrr_setup_each_insertss  zTablesTest._setup_each_insertsc Cs|jdkrT|jdkrV|j;}t|jjD]+}z ||Wqt j j yA}zt j d||ftjdWYd}~qd}~wwWddS1sMwYdSdSdS)Nr)zError emptying table %s: %r)file)r0 run_deletesr6connectreversedr9 sorted_tablesexecutedeleter7exc DBAPIErrorr print_sysstderr)rconntableexrrr_teardown_each_tabless  "z TablesTest._teardown_each_tablescC||dSr)rDrErrrrrs zTablesTest.setupcC |dSr)rUrrrrrs zTablesTest.teardowncCsJ|jr t|j|j|jdkr||jd|j_|jdur#d|_dSdSr;)r1rr9r6run_dispose_bind dispose_bindrun_setup_bindr-rrr_teardown_once_metadata_binds    z'TablesTest._teardown_once_metadata_bindcCrWr)r[r-rrrteardown_classs zTablesTest.teardown_classcCstjSr)rdbr-rrrr5zTablesTest.setup_bindcCs0t|dr |dSt|dr|dSdS)Ndisposeclose)rr_r`)r.r6rrrrYs    zTablesTest.dispose_bindcCdSrr)r.r9rrrr?zTablesTest.define_tablescCsiSrrr-rrrfixturesrbzTablesTest.fixturescCrarrr-rrrr>rbzTablesTest.insert_datacCs||j||dSr)assert_sql_countr6)rcountfnrrr sql_count_zTablesTest.sql_count_cCs||j||dSr) assert_sqlr6)r callable_ statementsrrrsql_eq_rhzTablesTest.sql_eq_csii}|D]$\}t|dkrq ttjr!|j|d<|dd|<q |jjD]vr;q4|j  fdd|Dq4dS)z4Insert rows as represented by the fixtures() method.r rrNcsg|] }tt|qSr)dictzip).0 column_valuesheadersrSrr sz-TablesTest._load_fixtures..) rcitemslen isinstancer string_typesr4r9rJr6rKinsert)r.rowsdatarrqrr=s$       zTablesTest._load_fixtures)!rr r!rZr0r1r2rGrXr6r9r4r3 classmethodr/r*r,r+rDrErUrrr[r\r5rYr?rcr>rgrlr=rrrrr'<sT            r'cs2eZdZejddZddZfddZZS) RemovesEventscCstSr)setrrrr _event_fnsr^zRemovesEvents._event_fnscKs,|j|||ftj|||fi|dSr)r~addr listen)rtargetnamerfkwrrr event_listenszRemovesEvents.event_listencs:|jD]}tj|qtt|}t|dr|dSdS)Nr)r~r removesuperr|rr)rkeysuper_ __class__rrrs     zRemovesEvents.teardown) rr r!r memoized_propertyr~rr __classcell__rrrrr|s  r|c@seZdZeddZdS)_ORMTestcCstjjjtjdSr)r7ormsessionSession close_all clear_mappersr-rrrr\sz_ORMTest.teardown_classN)rr r!r{r\rrrrrsrc@ eZdZdS)ORMTestNrr r!rrrrrrc@seZdZdZdZdZeddZeddZdd Z d d Z ed d Z eddZ eddZ ddZddZeddZddZddZeddZeddZdS) MappedTestr(r)NcCs>||jdur t|_||||dSr)r*classesr r+_setup_once_classes_setup_once_mappersr,r-rrrr/ s  zMappedTest.setup_classcCrVr)_teardown_once_classr[r-rrrr\s zMappedTest.teardown_classcCs$||||dSr)rD_setup_each_classes_setup_each_mappersrErrrrrs zMappedTest.setupcCs*tjjj|||dSr)r7rrrr_teardown_each_mappers_teardown_each_classesrUrrrrr!s zMappedTest.teardowncCs|jtdSr)rrCrr\r-rrrr's  zMappedTest._teardown_once_classcC|jdkr ||jdSdSr;run_setup_classes_with_register_classes setup_classesr-rrrr, zMappedTest._setup_once_classescCrr;run_setup_mappersr setup_mappersr-rrrr1rzMappedTest._setup_once_mapperscCrrBrrrrrr6 zMappedTest._setup_each_mapperscCrrBrrrrrr:rzMappedTest._setup_each_classescsl|jGfdddt}Gdddt|t}Gdddt|}Gdddt|}||_||_|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. ceZdZfddZdS)z6MappedTest._with_register_classes..FindFixturec||<t||||Sr)type__init__r. classnamebasesdict_ cls_registryrrrHz?MappedTest._with_register_classes..FindFixture.__init__Nrr r!rrrrr FindFixtureGrc@r)z0MappedTest._with_register_classes.._BaseNrrrrr_BaseLrrc@r)z0MappedTest._with_register_classes..BasicNrrrrrBasicOrrc@r)z5MappedTest._with_register_classes..ComparableNrrrrr ComparableRrrN) rrr with_metaclassobjectrrrr)r.rfrrrrrrrr>s z!MappedTest._with_register_classescCs|jdkr tjdSdSr;)rr7rrrrrrrYs z!MappedTest._teardown_each_mapperscCs|jdkr |jdSdSr;)rrrCrrrrr`s z!MappedTest._teardown_each_classescCrarrr-rrrrdrbzMappedTest.setup_classescCrarrr-rrrrhrbzMappedTest.setup_mappers)rr r!rrrr{r/r\rrrrrrrrrrrrrrrrrs4       rc@s,eZdZdZdZeddZeddZdS)DeclarativeMappedTestr(cCrarrr-rrrr+qrbz(DeclarativeMappedTest._setup_once_tablescsn|jGfdddt}Gdddt}t|j||d}||_||jjr3|jr5|jt j dSdSdS)Ncr)zLDeclarativeMappedTest._with_register_classes..FindFixtureDeclarativecrr)rrrrrrrzrzUDeclarativeMappedTest._with_register_classes..FindFixtureDeclarative.__init__NrrrrrFindFixtureDeclarativeyrrc@seZdZejZdS)zFDeclarativeMappedTest._with_register_classes..DeclarativeBasicN)rr r!rTable __table_cls__rrrrDeclarativeBasic~s r)r9 metaclassr.) rrrr r9rr4r1r@rr])r.rfrr _DeclBaserrrrusz,DeclarativeMappedTest._with_register_classesN)rr r!rrr{r+rrrrrrms r)rP sqlalchemyr7rrrenginesrentitiesrrr r r ext.declarativer rrrr'r|rrAssertsExecutionResultsrrrrrrs*            *l