o [h@sddlZddlZddlmZddlmZddlmZddlm Z e e Z dZ Gdd d eZd d Zd d ZddZddZejddZejddZejddZejddZejddZejddZejddZd d!Zejd"d#ZdS)$N)config)engines)url)compatc@s0eZdZddZeddZddZddZd S) registercCs i|_dSNfns)selfr /home/ubuntu/experiments/live_experiments/Pythonexperiments/Otree/venv/lib/python3.10/site-packages/sqlalchemy/testing/provision.py__init__s zregister.__init__cCstd|SN*)rfor_db)clsfnr r rinitsz register.initcsfdd}|S)Ncs|j<Sr r )rdbnamer r rdecorates z!register.for_db..decorater )r rrr rrrszregister.for_dbcGspt|tjr t|}n t|tjr|}n|jj}|}||j vr-|j ||g|RS|j d|g|RSr) isinstancer string_typessa_urlmake_urlURLdbrget_backend_namer )r cfgargrbackendr r r__call__s    zregister.__call__N)__name__ __module__ __qualname__r classmethodrrr#r r r rrs   rcC0tD]}td||jjt||j|qdS)NzCREATE database %s, URI %r)_configs_for_db_operationloginforr create_dbfollower_identr r r rcreate_follower_db+ r/cCszt|}||rt||}i}t||t||}t|||| t j ||||}|r;t|||Sr )rr get_dialectload_provisioningfollower_url_from_mainupdate_db_optsrZtesting_enginepost_configure_engineconnectcloserConfigrconfigure_follower)db_urloptionsZ file_configr.dialectdb_optsengr r r r setup_config1s      r?cCr()NzDROP database %s, URI %r)r)r*r+rrdrop_dbr-r r rdrop_follower_dbFr0rAccst}tjD]}|jq tjD]}|jj}|}||j|j |j f}||vr5|V| |qtjD]}|jq;dSr ) setrr8Z all_configsrZdisposerrusernamehostdatabaseadd)hostsr rr"Z host_confr r rr)Ls   r)cCtd|j)zDynamically create a database for testing. Used when a test run will employ multiple processes, e.g., when run via `tox` or `pytest -n4`. z"no DB creation routine for cfg: %sNotImplementedErrorrr r>identr r rr,_sr,cCrH)z8Drop a database that we dynamically created for testing.zno DB drop routine for cfg: %srIrKr r rr@isr@cCdS)zCSet database options (db_opts) for a test database that we created.Nr )r:r=r r rr4or4cCrM)zPerform extra steps after configuring an engine for testing. (For the internal dialects, currently only used by sqlite.) Nr )renginer.r r rr5usr5cCst|}||_|S)zCreate a connection URL for a dynamically-created test database. :param url: the connection URL specified when the test run was invoked :param ident: the pytest-xdist "worker identifier" to be used as the database name )rrrErrLr r rr3~s r3cCrM)z@Create dialect-specific config settings for a follower database.Nr )r rLr r rr9rNr9cCrM)a[Remove databases that were created during the test process, after the process has ended. This is an optional step that is invoked for certain backends that do not reliably release locks on the database as long as a process is still in use. For the internal dialects, this is currently only necessary for mssql and oracle. Nr rPr r r run_reap_dbss rQc Cstdtt}tt}i}t|B}|D]7}|}|d\}}t |}||vr:| ||<|| | |j f} || ||| |qWdn1sZwY|D]} t|| d} || } t| | qadS)NzReaping databases... r)r*r+ collections defaultdictrBopenstripsplitrrr1r2rrDrFlistrQ) Z idents_fileurlsZidentsZdialectsfile_lineZdb_namer:Zurl_objZurl_keyrrLr r rreap_dbss,         r\cCrH)aOSpecify keyword arguments for creating a temporary Table. Dialect-specific implementations of this method will return the kwargs that are passed to the Table method when creating a temporary table for testing, e.g., in the define_temp_tables method of the ComponentReflectionTest class in suite/test_reflection.py z.no temp table keyword args routine for cfg: %srI)r r>r r rtemp_table_keyword_argss r])rSloggingrrrOrrutilr getLoggerr$r*ZFOLLOWER_IDENTobjectrr/r?rAr)rr,r@r4r5r3r9rQr\r]r r r rs<