`fbXdZddlZddlmZddlmZmZmZmZm Z ddl Z er ddl m Z ddl mZnexZ ZejdZd Ze dd Ze jd dZe jd dZe jd dZe jd dZe jd dZe jd dZe jd dZe jd dZe jd dZe jdZe jd dZ e jdZ!e jdZ"dS)z;Fixtures and testing utilities for :pypi:`pytest `.N)contextmanager) TYPE_CHECKINGAnyMappingSequenceUnion)Celery)WorkController NO_WORKERc2|dddS)z)Register additional pytest configuration.markerszBcelery(**overrides): override celery configuration for a test caseN)addinivalue_line)configs G/var/www/html/env/lib/python3.11/site-packages/celery/contrib/pytest.pypytest_configurers,  WFc+Kddlm}m}|sin|}|dd||d|}|||5|VddddS#1swxYwYdS)z=Utility context used to setup Celery app for pytest fixtures.)TestAppsetup_default_appF)set_as_currentenable_loggingr)use_trapN) testing.apprr)rr parametersrrrtest_apps r _create_appr s87777777%5:Jw%  H  8h 7 7 7s<AAsession)scopecdS)zYou can override this fixture to enable the app trap. The app trap raises an exception whenever something attempts to use the current or default apps. Frrrruse_celery_app_trapr#5s  5rc#K|jd}t|fi|r|jni}t d|||d|5}|s(|||VddddS#1swxYwYdS)z1Session Fixture: Return app for session fixtures.celeryrrrNr)nodeget_closest_markerdictkwargsr set_default set_currentrequest celery_configcelery_parameterscelery_enable_loggingr#markrapps rcelery_session_appr4@s < * *8 4 4D - ? ?$#>4;;B ? ?F  $91 1    #&"  OO    OO    s/A??BBc#Kddlm}tsh|D]}|j||D]}|||j|fd|i|5} | VddddS#1swxYwYdSdS)z?Session Fixture: Start worker that lives throughout test suite.rworkerpoolN)testingr7r loaderimport_task_module register_task start_worker) r.r4celery_includescelery_class_taskscelery_worker_poolcelery_worker_parametersr7module class_taskws rcelery_session_workerrETs  % A AF  % 8 8 @ @ @ @, 9 9J  , ,Z 8 8 8 8 V !3==&8=#;== @AGGG                   sA++A/2A/cdS)z0You can override this fixture to enable logging.Frrrrr1r1ls  5rcdS)zYou can override this include modules when a worker start. You can have this return a list of module names to import, these can be task modules, modules registering signals, and so on. rrrrrr>r>ss  2rcdS)zYou can override this fixture to set the worker pool. The "solo" pool is used by default, but you can set this to return e.g. "prefork". solorrrrr@r@~s  6rciS)zRedefine this fixture to configure the test Celery app. The config returned by your fixture will then be used to configure the :func:`celery_app` fixture. rrrrr/r/  IrciS)zRedefine this fixture to change the init parameters of test Celery app. The dict returned by your fixture will then be used as parameters when instantiating :class:`~celery.Celery`. rrrrr0r0rKrciS)a"Redefine this fixture to change the init parameters of Celery workers. This can be used e. g. to define queues the worker will consume tasks from. The dict returned by your fixture will then be used as parameters when instantiating :class:`~celery.worker.WorkController`. rrrrrArAs  Irc#K|jd}t|fi|r|jni}t d|||d|5}|VddddS#1swxYwYdS)z/Fixture creating a Celery application instance.r%r&Nr)r'r(r)r*rr-s r celery_apprOs < * *8 4 4D - ? ?$#>4;;B ? ?F  $91 1    #& sAAAcgS)zARedefine this fixture to register tasks with the test Celery app.rrrrr?r?s  Irc#Kddlm}tsN|D]}|j||j|fd|i|5}|VddddS#1swxYwYdSdS)zAFixture: Start worker in a thread, stop it when the test returns.rr6r8N)r9r7r r:r;r=)r.rOr>r@rAr7rBrDs r celery_workerrRs  % 9 9F   0 0 8 8 8 8 V ==&8=#;== @AGGG                  sAAAc.|dS)z!Fixture that sets app as current.N)r,)rOs rdepends_on_current_apprTsr)FFN)#__doc__os contextlibrtypingrrrrrpytestr%r r7r objectenvirongetr rrfixturer#r4rEr1r>r@r/r0rArOr?rRrTrrrr^s AA %%%%%%?????????????? %'''''''$$F^ JNN; ' ' $(i   ! i   ! &i   ! .i   !  i   ! i   ! i   ! i   ! i     !     i   !  $r