8 f5P*dZddlmZddlZddlmZddlmZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZddlZddlmZdd lmZdd lmZe rddlZddlZdd lmZeeed ee fZ!eee Z"ee#e#e!e#e"fZ$gd Z%ej&dd^dZ'ej&dd_dZ(ej&dd`dZ)ej&ddadZ*ej&ddbdZ+ej&ddbdZ,ej&ddbdZ-ej&ddcd'Z.ej&ddd)Z/ded+Z0d^d,Z1dfd/Z2ej&dgd1Z3ej&dgd2Z4ej&dhd4Z5ej&did6Z6ej&djd8Z7ej&dkd:Z8ej&dld;Z9ej&dmd<Z:ej&dnd>Z;ej&dod?Z<ej&dpdAZ=ej&dqdCZ>GdDdEZ?ej&dFZ@ej&ddrdGZAej&dHIdsdJZBGdKdLeZCe dtdudWZDej&dvdYZEej&dvdZZFGd[d\eZGej&dwd]ZHdS)xzAll pytest-django fixtures) annotationsN)contextmanager)partial) TYPE_CHECKINGAnyCallableClassVarContextManager GeneratorIterableListLiteralOptionalProtocolTupleUnion)live_server_helper)is_django_unittestskip_if_no_django)DjangoDbBlocker__all__)django_db_setupdbtransactional_dbdjango_db_reset_sequencesdjango_db_serialized_rollback admin_userdjango_user_modeldjango_username_fieldclient async_client admin_clientrfasync_rfsettings live_server_live_server_helperdjango_assert_num_queriesdjango_assert_max_num_queries"django_capture_on_commit_callbackssession)scopereturnNonecrttjd}|rt|dSdS)NTOX_PARALLEL_ENVsuffix)rosgetenv_set_suffix_to_test_databases)tox_environments L/var/www/equiseq/venv/lib/python3.11/site-packages/pytest_django/fixtures.py'django_db_modify_db_settings_tox_suffixr:BsGi 233O>%_======>>requestpytest.FixtureRequestctt|jdid}|rt |dSdS)N workerinputworkeridr3)rgetattrconfiggetr7)r< xdist_suffixs r9)django_db_modify_db_settings_xdist_suffixrELsX7>="==AA*MML;%\::::::;;r;r:rEc"tdSNr)r:rEs r9,django_db_modify_db_settings_parallel_suffixrHVs r;rHc"tdSrGr)rHs r9django_db_modify_db_settingsrJ^sr;boolc8|jd S)N nomigrationsrBgetvalue)r<s r9django_db_use_migrationsrPes~&&~66 66r;c:|jd}|S)Nreuse_dbrN)r<rRs r9django_db_keepdbrSjs^,,Z88H Or;c:|jd}|S)N create_dbrN)r<rUs r9django_db_createdbrVpsn--k::I r;django_test_environmentdjango_db_blockerrrPrSrVrJGenerator[None, None, None]c#Kddlm}m}i} |st|r|sd| d<|5|d |jjjdd| } dddn #1swxYwYdV|s|5 || |jjjnF#t$r9} |j tj d | Yd} ~ nd} ~ wwxYwddddS#1swxYwYdSdS) z8Top level fixture to ensure test databases are availabler)setup_databasesteardown_databasesTkeepdbF) verbosity interactiveN)r^z.Error when trying to teardown test databases: ) django.test.utilsr[r\_disable_migrationsunblockrBoptionverbose Exceptionnodewarnpytest PytestWarning) r<rWrXrPrSrVrJr[r\setup_databases_argsdb_cfgexcs r9rrvsFEEEEEEE #. 2.)-X&  " " $ $    n+3  #                  EEE   & & ( (   ""6W^5J5RSSSSS    !!()aZ])a)abb                   sMA!!A%(A%C8B%$C8% C(//C#C8#C((C88C<?C<rc # Kddlm t|rdVdS|jd}|rt |\}}}}nd\}}}}p|pd|jvpd|jv|pd|jv}|pd|jv}|ddlddl r j j }n j j }||| | G fd d |} | | d } | dV| |  d kr| |dS)Nr)VERSION django_dbFFNFNrr(rrczeZdZZZZZs$edfd Zedfd Z xZ SxZ S)/_django_db_helper..PytestDjangoTestCaseNr/r0ctjj|dkrdjjj_dSdS)NrF)supertestTestCase setUpClassr transactionAtomic_ensure_durabilityclsro __class__djangos r9rzz:_django_db_helper..PytestDjangoTestCase.setUpClasssMfk*C00;;===V##FKFI)0CCC$#r;cdkrdjjj_t jj|dS)NruT)rr{r|r}rwrxry tearDownClassr~s r9rz=_django_db_helper..PytestDjangoTestCase.tearDownClasssEV##FJFI)0Cfk*C00>>@@@@@r;r/r0) __name__ __module__ __qualname__reset_sequencesserialized_rollback databasesavailable_apps classmethodrzr __classcell__)rro_available_apps _databases_reset_sequences_serialized_rollbackr transactionals@r9PytestDjangoTestCaserss*2  !"I  &,N A  L L L L L L L[ L  A A A A A A A[ A A A A A A A A Ar;r__init__) methodName)rvr)rrorrgget_closest_markervalidate_django_db fixturenamesrc django.db django.testrxTransactionTestCaseryrz _pre_setup_post_teardownrdoClassCleanupsrestore)r<rrXmarkerrrrrtest_case_classr test_caserorrrrrrs @@@@@@@r9_django_db_helperrsT '""  \ , ,[ 9 9F , v & &      N ,        a  a '"6 6 _-7K_:_ &^*EI]*]O-'7+??/ +9 +.&.J$O!A!A!A!A!A!A!A!A!A!A!A!A!A!A!A!AF##%%%$$ ;;;I  EEE &&(((&,,...r; _DjangoDbc: ddd }||ji|jS)aoValidate the django_db marker. It checks the signature and creates the ``transaction``, ``reset_sequences``, ``databases``, ``serialized_rollback`` and ``available_apps`` attributes on the marker which will have the correct values. Sequence reset, serialized_rollback, and available_apps are only allowed when combined with transaction. FNr{rKrr_DjangoDbDatabasesrr_DjangoDbAvailableAppsr/rc|||||fSrGr`)r{rrrrs r9apifunz"validate_django_db..apifunsOY8K^[[r;rq) r{rKrrKrrrrKrrr/r)argskwargs)rrs r9rr sL" %(,$)15 \\\\\ 66; 0&- 0 00r;cddlm}ddlm}Gdd}||_Gdd|j}||_dS)Nrr')migrateceZdZd dZd dZdS) ._disable_migrations..DisableMigrationsitemstrr/rKcdS)NTr`selfrs r9 __contains__z;_disable_migrations..DisableMigrations.__contains__&4r;r0cdSrGr`rs r9 __getitem__z:_disable_migrations..DisableMigrations.__getitem__)rr;N)rrr/rK)rrr/r0)rrrrrr`r;r9DisableMigrationsr%s<          r;rceZdZfdZxZS)1_disable_migrations..MigrateSilentCommandc@d|d<tj|i|S)Nrr^)rwhandle)rrrrs r9rz8_disable_migrations..MigrateSilentCommand.handle/s("#F; !577>42622 2r;)rrrrr)rs@r9MigrateSilentCommandr.s8 3 3 3 3 3 3 3 3 3r;r) django.confr'django.core.management.commandsrMIGRATION_MODULESCommand)r'rrrs r9rbrb!s$$$$$$777777"3!2!4!4H33333w333 +GOOOr;r4rc*ddlm}|jD]r}|did}|s|ddkr:d|d}|dkrL|di|d ||dd<sdS) NrrTESTNAMEENGINEzdjango.db.backends.sqlite3test_z:memory:_)rr' DATABASESvaluesrC setdefault)r4r' db_settings test_names r9r7r76s$$$$$$)0022 > > OOFB//33F;;  68$(DDD5 F 355I  " " vr***)2&=&=V&=&= FF## > >r;rcdS)aRequire a django test database. This database will be setup with the default fixtures and will have the transaction management disabled. At the end of the test the outer transaction that wraps the test itself will be rolled back to undo any changes to the database (in case the backend supports transactions). This is more limited than the ``transactional_db`` fixture but faster. If both ``db`` and ``transactional_db`` are requested, ``transactional_db`` takes precedence. Nr`rs r9rrKr;cdS)auRequire a django test database with transaction support. This will re-initialise the django database for each test and is thus slower than the normal ``db`` fixture. If you want to use the database with transactions you must request this resource. If both ``db`` and ``transactional_db`` are requested, ``transactional_db`` takes precedence. Nr`rs r9rr\rr;rcdS)aaRequire a transactional test database with sequence reset support. This requests the ``transactional_db`` fixture, and additionally enforces a reset of all auto increment sequences. If the enquiring test relies on such values (e.g. ids as primary keys), you should request this resource to ensure they are consistent across tests. Nr`)rrs r9rrlrr;rcdS)aRequire a test database with serialized rollbacks. This requests the ``db`` fixture, and additionally performs rollback emulation - serializes the database contents during setup and restores it during teardown. This fixture may be useful for transactional tests, so is usually combined with ``transactional_db``, but can also be useful on databases which do not support transactions. Note that this will slow down that test suite by approximately 3x. Nr`)rrs r9rr|rr;django.test.Clientc>tddlm}|S)zA Django test client instance.rClient)rrrrs r9r"r"s-"""""" 688Or;django.test.AsyncClientc>tddlm}|S)z$A Django test async client instance.r AsyncClient)rrrrs r9r#r#s.'''''' ;==r;c"ddlm}|S)z!The class of Django's user model.r)get_user_model)django.contrib.authr)rrs r9r r s$322222 >  r;c|j}|S)z=The fieldname for the username used with Django's user model.)USERNAME_FIELD)r fields r9r!r!s#1E Lr;r!c|}|}|dkrdnd} |j|}n<#|j$r/i}d|jvrd|d<d|d<|||<|jjdi|}YnwxYw|S)zA Django admin user. This uses an existing user with username "admin", or creates a new one with password "password". emailzadmin@example.comadminpasswordr`)_default_managerget_by_natural_key DoesNotExistREQUIRED_FIELDScreate_superuser)rr r! UserModelusername_fieldusernameuser user_datas r9rrs"I*N&4&?&?""WH H )<tddlm}|S)zRequestFactory instancerRequestFactory)rrrrs r9r%r%s0****** >  r;django.test.AsyncRequestFactoryc>tddlm}|S)zAsyncRequestFactory instancerAsyncRequestFactory)rrrrs r9r&r&s1//////    r;c>eZdZUgZded<d dZd dZd d Zdd Zd S)SettingsWrapperzClassVar[list[Any]] _to_restoreattrrr/r0cddlm}|}|ddlm}t |||j|dS)Nroverride_settingsr)rrenablerr'delattrrappend)rrroverrider's r9 __delattr__zSettingsWrapper.__delattr__sq111111$$&&(((((($ )))))r;cddlm}|di||i}||j|dS)Nrrr`)rrrrr)rrvaluerrs r9 __setattr__zSettingsWrapper.__setattr__s[111111$$55e}55 )))))r;c.ddlm}t||S)Nrr)rr'rA)rrr's r9 __getattr__zSettingsWrapper.__getattr__ s$((((((x&&&r;cpt|jD]}||jdd=dSrG)reversedrdisable)rrs r9finalizezSettingsWrapper.finalizesF !122  H         QQQ   r;N)rrr/r0)rrr) rrrr__annotations__rr r rr`r;r9rrsw')K)))) * * * *****''''       r;rc#rKtt}|V|dS)zAA Django settings object which restores changes after the testrunN)rrr)wrappers r9r'r's@G MMM r;c#Kt|jdptjdpd}t j|}|V|dS)atRun a live Django server in the background during tests The address the server is started from is taken from the --liveserver command line option or if this is not provided from the DJANGO_LIVE_TEST_SERVER_ADDRESS environment variable. If neither is provided ``localhost`` is used. See the Django documentation for its full syntax. NOTE: If the live server needs database access to handle a request your test will have to request database access. Furthermore when the tests want to see data added by the live-server (or the other way around) transactional database access will be needed as data inside a transaction is not shared between the live server and test code. Static assets will be automatically served when ``django.contrib.staticfiles`` is available in INSTALLED_APPS. liveserverDJANGO_LIVE_TEST_SERVER_ADDRESS localhostN)rrBrOr5r6r LiveServerstop)r<addrservers r9r(r(su(  --  96 7 7    *4 0 0F LLL KKMMMMMr;T)autousec#Kd|jvrdVdS|d|d}|jdV|jdS)aZHelper to make live_server work, internal to pytest-django. This helper will dynamically request the transactional_db fixture for a test which uses the live_server fixture. This allows the server and test to access the database without having to mark this explicitly which is handy since it is usually required and matches the Django behaviour. The separate helper is required since live_server can not request transactional_db directly since it is session scoped instead of function-scoped. It will also override settings only for the duration of the test. r(Nr)rgetfixturevalue_live_server_modified_settingsrr)r<r(s r9r)r)@s G000  .///))-88K.55777 EEE.6688888r;c eZdZdZ d dd Zd S)DjangoAssertNumQueriesz]The type of the `django_assert_num_queries` and `django_assert_max_num_queries` fixtures..numint connection Any | Noneinfo str | Noner/'django.test.utils.CaptureQueriesContextcdSrGr`)rr!r#r%s r9__call__zDjangoAssertNumQueries.__call__` r;N)..)r!r"r#r$r%r&r/r'rrr__doc__r)r`r;r9r r \s?11 "%        r;r rB pytest.Configr!r"exactr#r$r%r&>Generator[django.test.utils.CaptureQueriesContext, None, None]c#Kddlm}|ddlm}n|}|ddk}||5}|Vt |} |r|| k} n| |k} | rud|d} |s| dz } | dkrd nd } | d | d | d z } |r| d|z } |r-d|jD} | dd| zz } n| dz } tj | ddddS#1swxYwYdS)Nr)CaptureQueriesContext)r#rezExpected to perform z queries zor less rwaswerezbut  z done c3&K|] }|dV dS)sqlNr`).0qs r9 z&_assert_num_queries..s&CCQ%CCCCCCr;z Queries: ======== z z (add -v option to show queries)) rar1rr# getoptionlencaptured_queriesjoinrifail)rBr!r.r#r%r1connrecontext num_performedfailedmsgverbsqlss r9_assert_num_queriesrGis8777770000000y))A-G  t $ $ G  )M)FF"S(F  7777C "z!)Q..55FD 5-55$555 5C #{D{{" :CC'*BCCC3fkk$6G6GGG99 K   )sBCC#&C# pytestconfigc,tt|SrGrrGrHs r9r*r*s & 5 55r;c0tt|dS)NF)r.rJrKs r9r+r+s & E B B BBr;c"eZdZdZdddd d Zd S) DjangoCaptureOnCommitCallbacksz=The type of the `django_capture_on_commit_callbacks` fixture..)usingexecuterOrrPrKr/'ContextManager[list[Callable[[], Any]]]cdSrGr`)rrOrPs r9r)z'DjangoCaptureOnCommitCallbacks.__call__r*r;N)rOrrPrKr/rQr+r`r;r9rNrNs@GG          r;rNcddlm}|jS)Nrry)rrycaptureOnCommitCallbacksrTs r9r,r,s$$$$$$  ,,r;r)r<r=r/r0)r:r0rEr0r/r0)rHr0r/r0)r<r=r/rK)r<r=rWr0rXrrPrKrSrKrVrKrJr0r/rY)r<r=rr0rXrr/rY)r/r)r4rr/r0)rr0r/r0)rr0rr0r/r0)rr0rr0r/r0)r/r)r/r)rr0)r/r)rr0r!r)rr0r/r)r/r)r/r)r<r=)r<r=r/rY)TNN) rBr-r!r"r.rKr#r$r%r&r/r/)rHr-r/r )r/rN)Ir, __future__rr5 contextlibr functoolsrtypingrrrr r r r r rrrrrrir django_compatr lazy_djangorrrrrrrrKrrfixturer:rErHrJrPrSrVrrrrbr7rrrrr"r#r r!rr$r%r&rr'r(r)r rGr*r+rNr,r`r;r9r^s^ """""" %%%%%%  ------******"MMM!!!!!!eGI$6 $EFG!$s), $0$8NN O    .i   >>>! >i   ;;;! ;i   ! i   !  i   777! 7i   !  i   !  i   $$$! $Ng g g g T11110++++*>>>>*             ( <    !!!!        Bi   ! @99996      X    ! #####L6666CCCC      X    ------r;