
    iF                         d dl Zd dlmc mZ d dlZej                  j                  d d       d dl	Z	d dl
m
Z
 d dlmZ d Zd Zy)    Nz!/home/jang/Projects/ai-researcher)datetime)Taskc                  ~   t        dt         j                  dd      } | j                  }d}||k(  }|st        j                  d|fd||f      dt        j                         v st        j                  |       rt        j                  |       ndt        j                  |      t        j                  |      dz  }d	d
|iz  }t        t        j                  |            dx}x}}| j                  }d}||k(  }|st        j                  d|fd||f      dt        j                         v st        j                  |       rt        j                  |       ndt        j                  |      t        j                  |      dz  }d	d
|iz  }t        t        j                  |            dx}x}}| j                  }d}||k(  }|st        j                  d|fd||f      dt        j                         v st        j                  |       rt        j                  |       ndt        j                  |      t        j                  |      dz  }d	d
|iz  }t        t        j                  |            dx}x}}| j                  }d}||k(  }|st        j                  d|fd||f      dt        j                         v st        j                  |       rt        j                  |       ndt        j                  |      t        j                  |      dz  }d	d
|iz  }t        t        j                  |            dx}x}}y)u   测试Task创建task_123u   查询技术部员工g?)idstatusquerycomplexity_score==)z*%(py2)s
{%(py2)s = %(py0)s.id
} == %(py5)staskpy0py2py5assert %(py7)spy7Npending)z.%(py2)s
{%(py2)s = %(py0)s.status
} == %(py5)s)z-%(py2)s
{%(py2)s = %(py0)s.query
} == %(py5)s)z8%(py2)s
{%(py2)s = %(py0)s.complexity_score
} == %(py5)s)r   STATUS_PENDINGr   
@pytest_ar_call_reprcompare@py_builtinslocals_should_repr_global_name	_safereprAssertionError_format_explanationr   r	   r
   )r   @py_assert1@py_assert4@py_assert3@py_format6@py_format8s         ?/home/jang/Projects/ai-researcher/tests/unit/test_task_model.pytest_task_creationr$      s   ""%	D 77 j 7j    7j      4   4   7   j       ;;#)#;)####;)######4###4###;###)#######::000:00000:000000040004000:00000000000  (D( D(((( D((((((4(((4((( (((D(((((((    c                  L   t         j                  } d}| |k(  }|st        j                  d|fd| |f      dt	        j
                         v st        j                  t               rt        j                  t               ndt        j                  |       t        j                  |      dz  }dd|iz  }t        t        j                  |            dx} x}}t         j                  } d	}| |k(  }|st        j                  d|fd
| |f      dt	        j
                         v st        j                  t               rt        j                  t               ndt        j                  |       t        j                  |      dz  }dd|iz  }t        t        j                  |            dx} x}}t         j                  } d}| |k(  }|st        j                  d|fd| |f      dt	        j
                         v st        j                  t               rt        j                  t               ndt        j                  |       t        j                  |      dz  }dd|iz  }t        t        j                  |            dx} x}}t         j                  } d}| |k(  }|st        j                  d|fd| |f      dt	        j
                         v st        j                  t               rt        j                  t               ndt        j                  |       t        j                  |      dz  }dd|iz  }t        t        j                  |            dx} x}}t         j                  } d}| |k(  }|st        j                  d|fd| |f      dt	        j
                         v st        j                  t               rt        j                  t               ndt        j                  |       t        j                  |      dz  }dd|iz  }t        t        j                  |            dx} x}}y)u   测试任务状态r   r   )z6%(py2)s
{%(py2)s = %(py0)s.STATUS_PENDING
} == %(py5)sr   r   r   r   Nrunning)z6%(py2)s
{%(py2)s = %(py0)s.STATUS_RUNNING
} == %(py5)s	completed)z8%(py2)s
{%(py2)s = %(py0)s.STATUS_COMPLETED
} == %(py5)sfailed)z5%(py2)s
{%(py2)s = %(py0)s.STATUS_FAILED
} == %(py5)s	cancelled)z8%(py2)s
{%(py2)s = %(py0)s.STATUS_CANCELLED
} == %(py5)s)r   r   r   r   r   r   r   r   r   r   STATUS_RUNNINGSTATUS_COMPLETEDSTATUS_FAILEDSTATUS_CANCELLED)r   r   r    r!   r"   s        r#   test_task_status_transitionsr/      s<   +)+)++++)++++++4+++4++++++)++++++++)+)++++)++++++4+++4++++++)+++++++  /K/ K//// K//////4///4/// ///K///////))))))))))))4)))4)))))))))))))  /K/ K//// K//////4///4/// ///K///////r%   )builtinsr   _pytest.assertion.rewrite	assertionrewriter   syspathinsertpytestr   src.models.taskr   r$   r/    r%   r#   <module>r:      s2      
 6 7    )0r%   