o
    ¨@TÑ  ã                   @   sŽ   d dl mZmZ 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mZ d dlmZ d dlmZ d d	lmZ G d
d„ dejƒZdS )é    )Úabsolute_importÚunicode_literals)ÚdatetimeÚ	timedelta)Ústates)Úgen_unique_id)Úcelery)ÚTaskMetaÚTaskSetMeta)Únow)Úunittest)Úunicodec                   @   s,   e Zd Zdd„ Zdd„ Zdd„ Zdd„ Zd	S )
Ú
TestModelsc                 C   ó   t ƒ }tjj|d�\}}|S )N©Útask_id)r   r	   ÚobjectsÚget_or_create)ÚselfÚidÚtaskmetaÚcreated© r   úL/var/www/html/env/lib/python3.10/site-packages/djcelery/tests/test_models.pyÚcreateTaskMeta   ó   zTestModels.createTaskMetac                 C   r   )N©Ú
taskset_id)r   r
   r   r   )r   r   Útasksetmetar   r   r   r   ÚcreateTaskSetMeta   r   zTestModels.createTaskSetMetac                 C   sX  |   ¡ }|   ¡ }|   ¡ }|  t|ƒ d¡¡ |  |j¡ |  |jt¡ |  t	j
 |j¡j|j¡ |  t	j
 |j¡jtj¡ t	j
j|jdtjd� t	j
j|jdtjd� |  t	j
 |j¡jtj¡ |  t	j
 |j¡jtj¡ t	j
j|jd�jtƒ tdd� d� t	j
 tjj¡}|  ||¡ |  ||¡ |  ||¡ t	j
 tjj¡ |  |t	j
 ¡ ¡ d S )Nz<Task:T)Ústatusr   é
   ©Údays©Ú	date_done)r   Ú
assertTruer   Ú
startswithr   ÚassertIsInstancer%   r   ÚassertEqualr	   r   Úget_taskÚassertNotEqualr    r   ÚSUCCESSÚstore_resultÚfilterÚupdater   r   Úget_all_expiredr   ÚconfÚCELERY_TASK_RESULT_EXPIRESÚassertInÚassertNotInÚdelete_expiredÚall)r   Úm1Úm2Úm3Úexpiredr   r   r   Útest_taskmeta   sH   þÿÿÿÿþÿÿzTestModels.test_taskmetac                 C   s"  |   ¡ }|   ¡ }|   ¡ }|  t|ƒ d¡¡ |  |j¡ |  |jt¡ |  t	j
 |j¡j|j¡ t	j
j|jd�jtƒ tdd� d� t	j
 tjj¡}|  ||¡ |  ||¡ |  ||¡ t	j
 tjj¡ |  |t	j
 ¡ ¡ |   ¡ }|  t	j
 |j¡j|j¡ t	j
 |j¡ |  t	j
 |j¡¡ d S )Nz	<TaskSet:r   r!   r"   r$   )r   r&   r   r'   r   r(   r%   r   r)   r
   r   Úrestore_tasksetr.   r/   r   r   r0   r   r1   r2   r3   r4   r5   r6   Údelete_tasksetÚassertIsNone)r   r7   r8   r9   r:   Úm4r   r   r   Útest_tasksetmetaA   s@   þÿþÿÿþzTestModels.test_tasksetmetaN)Ú__name__Ú
__module__Ú__qualname__r   r   r;   r@   r   r   r   r   r      s
    &r   N)Ú
__future__r   r   r   r   r   r   Úcelery.utilsr   ÚdjceleryÚdjcelery.modelsr	   r
   Údjcelery.utilsr   Údjcelery.tests.utilsr   Údjcelery.compatr   ÚTestCaser   r   r   r   r   Ú<module>   s    