B CVvg39@sddlZddlZddlmZddlmZmZddlmZddl m Z ddl m Z ddl mZddlmZdd lmZd ZGd d d ZdS) N)StringIO)expectedFailureskip)apps)settings) serializers)router)atomic) import_stringZtest_c@seZdZdZddZddZddZd,d d Zd dZddZ ddZ ddZ ddZ d-ddZ d.ddZd/ddZddZd0dd Zd1d"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd!S)2BaseDatabaseCreationzs Encapsulate backend-specific differences pertaining to creation and destruction of the test database. cCs ||_dS)N) connection)selfr rD/tmp/pip-install-o3oxmrkh/Django/django/db/backends/base/creation.py__init__szBaseDatabaseCreation.__init__cCs |jS)N)r _nodb_cursor)r rrrrsz!BaseDatabaseCreation._nodb_cursorcCstj|tjdS)N)sysstderrwriteoslinesep)r msgrrrlogszBaseDatabaseCreation.logFTc Cs0ddlm}|}|dkrBd}|r(d}|d||||f|||||j|tj |jj d<||jj d<zR|jj dd d krtj }d d t Dt_ |d t|ddd |jj ddWd|jj dd d kr|t_ X|r||j_|d|jj d|jtjddkr,||S)z Create a test database, prompting the user for confirmation if the database already exists. Return the name of the test database created. r) call_commandrZCreatingzUsing existingz %s test database for alias %s...NAMETESTZMIGRATEFcSsi|] }d|jqS)N)label).0Zapprrr Csz7BaseDatabaseCreation.create_test_db..ZmigrateT) verbosity interactivedatabaseZ run_syncdbNZcreatecachetable)r"ZRUNNING_DJANGOS_TEST_SUITEtrue)Zdjango.core.managementr_get_test_db_namer_get_database_display_str_create_test_dbr closer DATABASESalias settings_dictZMIGRATION_MODULESrget_app_configsmaxserialize_db_to_stringZ_test_serialized_contentsZensure_connectionrenvironget mark_expected_failures_and_skips) r r autoclobber serializekeepdbrtest_database_nameactionZold_migration_modulesrrrcreate_test_db sB       z#BaseDatabaseCreation.create_test_dbcCs|d|jjd<dS)z Set this database up to be used in testing as a mirror of a primary database whose settings are given. rN)r r*)r Zprimary_settings_dictrrrset_as_test_mirrorfsz'BaseDatabaseCreation.set_as_test_mirrorcs.fdd}t}tjd|d|d|S)z Serialize all data in the database into a JSON string. Designed only for test runner usage; will not handle large amounts of data. c3sddlm}|j}xtD]}|jdk r |j|jkr |jt j kr xX| D]L}|j jrPtjj|rP|jjj|j jj}|EdHqPWq WdS)Nr)MigrationLoader)Zdjango.db.migrations.loaderr8r rr+Z models_modulerZ migrated_appsnamerZTEST_NON_SERIALIZED_APPSZ get_models_metaZ can_migraterZallow_migrate_modelr)Z _base_managerusingZorder_bypkiterator)r8loaderZ app_configmodelZqueryset)r rr get_objectsts       z@BaseDatabaseCreation.serialize_db_to_string..get_objectsjsonN)indentstream)rrr2getvalue)r r@outr)r rr-ms z+BaseDatabaseCreation.serialize_db_to_stringc Cst|}t}t|jjdb|j>x6tjd||jjdD]}|| |j j j j q@WWdQRX|jj|dWdQRXdS)zw Reload the database with data from a string generated by the serialize_db_to_string() method. )r;rAN) table_names)rsetr r r)Zconstraint_checks_disabledrZ deserializesaveaddobject __class__r:Zdb_tableZcheck_constraints)r datarFobjrrrdeserialize_db_from_strings  z/BaseDatabaseCreation.deserialize_db_from_stringcCs d|jj|dkrd|ndfS)zR Return display string for a database for use in various actions. z'%s'%sz ('%s'))r r))r r Z database_namerrrr%sz.BaseDatabaseCreation._get_database_display_strcCs0|jjddr |jjddSt|jjdS)z Internal implementation - return the name of the test DB that will be created. Only useful when called from create_test_db() and _create_test_db() and when no external munging is done with the 'NAME' settings. rr)r r*TEST_DATABASE_PREFIX)r rrrr$sz&BaseDatabaseCreation._get_test_db_namecCs|d|dS)Nz%CREATE DATABASE %(dbname)s %(suffix)s)execute)r cursor parametersr3rrr_execute_create_test_dbsz,BaseDatabaseCreation._execute_create_test_dbc Cs:|}|jj||d}|}y||||Wntk r*}z|rZ|S|d||sxt d|}|s|dkry@|dkr|d| ||f| d|||||Wn<tk r}z|d|t d Wd d }~XYnXn|d t dWd d }~XYnXWd QRX|S) zF Internal implementation - create the test db tables. )Zdbnamesuffixz+Got an error creating the test database: %szXType 'yes' if you would like to try deleting the test database '%s', or 'no' to cancel: yesrz,Destroying old test database for alias %s...zDROP DATABASE %(dbname)sz-Got an error recreating the test database: %srONzTests cancelled.)r$r ops quote_namesql_table_creation_suffixrrU Exceptionrinputr%rRrexit) r r r1r3r4Ztest_db_paramsrSeZconfirmrrrr&s6    &z$BaseDatabaseCreation._create_test_dbcCsL|jjd}|dkr:d}|r d}|d||||f||||dS)z( Clone a test database. rrzCloning test databasezUsing existing clonez%s for alias %s...N)r r*rr%_clone_test_db)r rVr r1r3Zsource_database_namer5rrr clone_test_dbs z"BaseDatabaseCreation.clone_test_dbcCs |jj}|dd|d|iS)zX Return a modified connection settings dict for the n-th clone of a DB. rz{}_{})r r*format)r rVZorig_settings_dictrrrget_test_db_clone_settingssz/BaseDatabaseCreation.get_test_db_clone_settingscCs tddS)zI Internal implementation - duplicate the test db tables. znThe database backend doesn't support cloning databases. Disable the option to run tests in parallel processes.N)NotImplementedError)r rVr r3rrrr_sz#BaseDatabaseCreation._clone_test_dbNcCs|j|dkr |jjd}n||d}|dkr\d}|rBd}|d||||f|sl||||dk r|tj|jj d<||jjd<dS)zv Destroy a test database, prompting the user for confirmation if the database already exists. NrrZ DestroyingZ Preservingz %s test database for alias %s...) r r'r*rbrr%_destroy_test_dbrr(r))r Zold_database_namer r3rVr4r5rrrdestroy_test_dbs   z$BaseDatabaseCreation.destroy_test_dbc Cs0|}|d|jj|WdQRXdS)zF Internal implementation - remove the test db tables. zDROP DATABASE %sN)rrRr rXrY)r r4r rSrrrrd!s z%BaseDatabaseCreation._destroy_test_dbc Csx\|jjjD]N}|d\}}}|dd}|tjkr t|}t||}t ||t |q Wxv|jjj D]d\}} xZ| D]R}|d\}}}|dd}|tjkr|t|}t||}t ||t ||q|WqnWdS)z Mark tests in Django's test suite which are expected failures on this database and test which should be skipped on this database. .rN)r featuresZdjango_test_expected_failures rpartitionsplitrZINSTALLED_APPSr getattrsetattrrZdjango_test_skipsitemsr) r Z test_nameZtest_case_name_Ztest_method_nameZtest_appZ test_caseZ test_methodreasontestsrrrr0-s     z5BaseDatabaseCreation.mark_expected_failures_and_skipscCsdS)zQ SQL to append to the end of the test table creation statements. rPr)r rrrrZDsz.BaseDatabaseCreation.sql_table_creation_suffixcCs$|jj}|d|d|d|fS)z Return a tuple with elements of self.connection.settings_dict (a DATABASES setting value) that uniquely identify a database accordingly to the RDBMS particularities. ZHOSTZPORTZENGINE)r r*r$)r r*rrrtest_db_signatureJs z&BaseDatabaseCreation.test_db_signature)rFTF)F)F)rFF)F)NrFN)__name__ __module__ __qualname____doc__rrrr6r7r-rNr%r$rUr&r`rbr_rerdr0rZrprrrrr s( F   )    r )rriorZunittestrrZ django.appsrZ django.confrZ django.corerZ django.dbrZdjango.db.transactionr Zdjango.utils.module_loadingr rQr rrrrs