o
    MhúRW
  ã                   @   s�   d dl mZmZ d dlmZ d dlmZ d dlmZ	 d dl
mZ d dlmZ d dlmZ e	ƒ dd	„ ƒZe	ƒ d
d„ ƒZde_G dd„ dejƒZdS )é    )Úabsolute_importÚunicode_literals)Úcache)Úgen_unique_id)Útask)Újail)Úapp)Úunittestc                 C   s   | |  S )N© )Úir
   r
   úP/var/www/html/env/lib/python3.10/site-packages/djcelery/tests/test_worker_job.pyÚmytask   s   r   c                 C   s   ddl m} t|ƒS )Nr   ©Ú
connection)Ú	django.dbr   Úid)r   r   r
   r
   r   Úget_db_connection   s   r   Tc                   @   s$   e Zd Zdd„ Zdd„ Zdd„ ZdS )ÚTestJailc                    sd   ddl m‰  dˆ _ˆ j‰‡ ‡fdd„}|ˆ _ztttƒ tjdgi ƒ |  	ˆ j¡ W ˆˆ _d S ˆˆ _w )Nr   r   Fc                     s   dˆ _ ˆ| i |¤ŽS ©NT)Ú_was_closed©ÚargsÚkwargs©r   Úold_connection_closer
   r   Úmonkeypatched_connection_close"   s   zTTestJail.test_django_db_connection_is_closed.<locals>.monkeypatched_connection_closeé   )
r   r   r   Úcloser   r   r   r   ÚnameÚ
assertTrue)Úselfr   r
   r   r   Ú#test_django_db_connection_is_closed   s   z,TestJail.test_django_db_connection_is_closedc                 C   sz   t tjdd ƒ}dt_t tdd ƒ}|rttdƒ dd„ }|tj_tttƒ tj	dgi ƒ |  
tj¡ |tj_|r;|t_d S d S )Nr   FÚparse_backend_uric                  _   ó
   dt _d S r   ©r   r   r   r
   r
   r   Úmonkeypatched_cache_close4   ó   
zRTestJail.test_django_cache_connection_is_closed.<locals>.monkeypatched_cache_closeé   )Úgetattrr   r   Údelattrr   r   r   r   r   r   r   r"   ©r    Úold_cache_closeÚold_cache_parse_backendr%   r
   r
   r   Ú&test_django_cache_connection_is_closed-   s   

ÿz/TestJail.test_django_cache_connection_is_closedc                 C   sz   t tjdd ƒ}dt_t tdd ƒ}dd„ t_dd„ }|tj_tttƒ tj	dgi ƒ |  
tj¡ |tj_|r9|t_d S t`d S )	Nr   Fr"   c                 S   s   g d¢S )N)ÚlibmemcachedÚ1Ú2r
   )Úurir
   r
   r   Ú<lambda>C   s    zLTestJail.test_django_cache_connection_is_closed_django_1_1.<locals>.<lambda>c                  _   r#   r   r$   r   r
   r
   r   r%   E   r&   z]TestJail.test_django_cache_connection_is_closed_django_1_1.<locals>.monkeypatched_cache_closer'   )r(   r   r   r"   r   r   r   r   r   r   r   r*   r
   r
   r   Ú1test_django_cache_connection_is_closed_django_1_1?   s   

z:TestJail.test_django_cache_connection_is_closed_django_1_1N)Ú__name__Ú
__module__Ú__qualname__r!   r-   r3   r
   r
   r
   r   r      s    r   N)Ú
__future__r   r   Údjango.corer   Úcelery.utilsr   Úcelery.taskr   Útask_decÚ celery.tests.worker.test_requestr   Údjcelery.appr   Údjcelery.tests.utilsr	   r   r   Úignore_resultÚTestCaser   r
   r
   r
   r   Ú<module>   s   

