
    Dlj/                         d dl mZm Z  d dlZd dlmZmZmZ ddZej                  j                  dg d      de
ddfd       Zdd	Zdd
ZddZy)    )UTCdatetimeN)JobDescriptorRunDescriptorvalidate_job_keyreturnc                  $    t        d      dk(  sJ y )Nzyoutube.transcript.batch)r        tests/unit/test_models.pytest_valid_job_keyr      s    67;UUUUr   key)zYoutube.Transcript.Batchza..bza.bza.b.c-1za.b.c_1c                 v    t        j                  t              5  t        |        d d d        y # 1 sw Y   y xY w)N)pytestraises
ValueErrorr   )r   s    r   test_invalid_job_keyr      s+    	z	"   s   /8c                      t        j                  t              5  t        t	        ddd             d d d        y # 1 sw Y   y xY w)N     )
started_at)r   r   r   r   r   r
   r   r   test_naive_datetime_rejectedr      s4    	z	" 7$1!567 7 7s	   ;Ac            
          t        j                  t              5  t        t	        dddt
              t	        dddt
                     d d d        y # 1 sw Y   y xY w)Nr   r      )tzinfo)r   finished_at)r   r   r   r   r   r   r
   r   r   test_run_time_orderr      sK    	z	" q$1S!AxX\^_abknOopq q qs   0AAc                      t        dddd      } ddlm}  || ddd	d
      } || dddd
      }|j                  |j                  k(  sJ |j                  dk(  sJ y )Nzbackup.vps.fullBackuptestbatchr   )InstanceDescriptorhermesprocessz/opt/backupprimary)instance_namez
/differentz&backup.vps.full|hermes|process|primary)r   worker_watcher.modelsr"   instance_key)jobr"   firstseconds       r   test_instance_key_is_stabler,      sk    
)8VW
EC8sHiV_`EXy,V_`F!4!4444!IIIIr   )r   N)r   r   r   r'   r   r   r   r   markparametrizestrr   r   r   r,   r
   r   r   <module>r0      s]    "  P PV  abc d  c
7
q
Jr   