+
    5mj  c                      ^ RI t^ RIHu Ht ^ RIt^ RIt^ RIt^ RI	t	^ RI
t
^ RIHt ^ RItR^/R R lltR R ltR R	 ltR
 R ltR R ltR R ltR R ltR R ltR R ltR R ltR R ltR R ltR R ltR R ltR# )     NPathversionc                H    V ^8  d   QhR\         R\        R\        R\         /# )   
codex_homestatusr   return)r   strint)formats   "`/home/user/overdeck-private-tests/workstation-ai-broad/modules/systray/tests/test_codex_state.py__annotate__r      s(      $   T     c          
      `   V P                  R R R7       V RV R2,          p\        P                  ! V4      pV;_uu_ 4        VP                  R4       VP                  RV\	        \
        P
                  ! 4       4      34       RRR4       VP                  4        V#   + '       g   i     L"; i)T)parentsexist_okstate_z.sqlitezCREATE TABLE backfill_state (id INTEGER PRIMARY KEY, status TEXT, last_watermark TEXT, last_success_at INTEGER, updated_at INTEGER)zDINSERT INTO backfill_state (id, status, updated_at) VALUES (1, ?, ?)N)mkdirsqlite3connectexecuter   timeclose)r   r   r   db_path
connections   &&$  r   	_state_dbr      s    TD1VG9G44G)J	P	
 	RS%&	
 
 N 
s   ABB-	c                0    V ^8  d   QhR\         R\        /# )r   r   r	   )r   r
   )r   s   "r   r   r      s      T c r   c                     \         P                  ! V 4      p VP                  R 4      P                  4       ^ ,          VP	                  4        #   TP	                  4        i ; i)z.SELECT status FROM backfill_state WHERE id = 1)r   r   r   fetchoner   )r   r   s   & r   _statusr       sL    )J!!"RS\\^_`a
s   %A A c                (    V ^8  d   QhR\         RR/# r   tmp_pathr	   Nr   )r   s   "r   r   r   $   s     7 7T 7d 7r   c                    \         P                  q! V 4      pR q2VJ qD'       Eg3   \        P                  ! RV3RW#34      RR\        P
                  ! 4       9   g!   \        P                  ! \         4      '       d   \        P                  ! \         4      MRR\        P                  ! V4      RR\        P
                  ! 4       9   g   \        P                  ! V 4      '       d   \        P                  ! V 4      MRR\        P                  ! V4      R\        P                  ! V4      /,          pRR	V/,          p\        \        P                  ! V4      4      hR
;p;p;rCR
# )Fpy0codex_statepy2py3r#   py5py8assert %(py10)spy10NiszR%(py5)s
{%(py5)s = %(py2)s
{%(py2)s = %(py0)s.needs_warmup
}(%(py3)s)
} is %(py8)s)
r&   needs_warmup
@pytest_ar_call_reprcompare@py_builtinslocals_should_repr_global_name	_safereprAssertionError_format_explanationr#   @py_assert1@py_assert4@py_assert7@py_assert6@py_format9@py_format11s   &      r   .test_a_home_without_a_state_db_needs_no_warmupr@   $   s    ##6#H-6666666-666666;6666;666#666666H6666H666-66666666666r   c                (    V ^8  d   QhR\         RR/# r"   r   )r   s   "r   r   r   (   s     6 64 6D 6r   c                    \        V R ^R7       \        V R^
R7       \        P                  q! V 4      pRq2VJ qD'       Eg3   \        P                  ! RV3RW#34      RR\
        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRR\        P                  ! V4      RR\
        P                  ! 4       9   g   \        P                  ! V 4      '       d   \        P                  ! V 4      MRR	\        P                  ! V4      R
\        P                  ! V4      /,          pRRV/,          p\        \        P                  ! V4      4      hR;p;p;rCR# )complete)r   runningTr%   r&   r'   r(   r#   r)   r*   r+   r,   Nr-   r/   r   r&   r0   r1   r2   r3   r4   r5   r6   r7   r8   r9   s   &      r   4test_needs_warmup_reads_the_highest_state_db_versionrF   (   s    h
A.h	2.##5#H-5555555-555555;5555;555#555555H5555H555-55555555555r   c                (    V ^8  d   QhR\         RR/# r"   r   )r   s   "r   r   r   /   s     7 7t 7 7r   c                    \        V R 4       \        P                  q! V 4      pRq2VJ qD'       Eg3   \        P                  ! RV3RW#34      RR\
        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRR\        P                  ! V4      RR\
        P                  ! 4       9   g   \        P                  ! V 4      '       d   \        P                  ! V 4      MRR\        P                  ! V4      R\        P                  ! V4      /,          pR	R
V/,          p\        \        P                  ! V4      4      hR;p;p;rCR# )rC   Fr%   r&   r'   r(   r#   r)   r*   r+   r,   Nr-   r/   rE   r9   s   &      r   (test_a_finished_backfill_needs_no_warmuprI   /   s    h
###6#H-6666666-666666;6666;666#666666H6666H666-66666666666r   c                (    V ^8  d   QhR\         RR/# r"   r   )r   s   "r   r   r   5        ) )$ )SW )r   c                 L   \        V R 4      p\        P                  q"! V 4      pRqCVJ qU'       Eg3   \        P                  ! RV3RW434      RR\
        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRR\        P                  ! V4      RR\
        P                  ! 4       9   g   \        P                  ! V 4      '       d   \        P                  ! V 4      MRR\        P                  ! V4      R\        P                  ! V4      /,          pR	R
V/,          p\        \        P                  ! V4      4      hR;p;p;rT\        V4      pRqV	8H  q3'       Eg   \        P                  ! RV3RW34      RR\
        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRRR\
        P                  ! 4       9   g   \        P                  ! V4      '       d   \        P                  ! V4      MRR\        P                  ! V4      R\        P                  ! V	4      /,          p
RRV
/,          p\        \        P                  ! V4      4      hR;p;r9R# )rD   Tr%   r&   r'   r(   r#   r)   r*   r+   r,   Npendingr    py1r   py6assert %(py8)sr-   z\%(py5)s
{%(py5)s = %(py2)s
{%(py2)s = %(py0)s.release_orphaned_claim
}(%(py3)s)
} is %(py8)s==z0%(py3)s
{%(py3)s = %(py0)s(%(py1)s)
} == %(py6)sr   r&   release_orphaned_claimr1   r2   r3   r4   r5   r6   r7   r8   r    r#   r   r:   r;   r<   r=   r>   r?   @py_assert2@py_assert5@py_format7s   &          r   ;test_release_orphaned_claim_reopens_a_claim_no_worker_holdsr[   5   sW   ),G--?-h7?4?4?????7??????;????;???-??????h????h???7???4????????7(y(y(((((((((((7((((7(((((((7((((7((((((y((((((((r   c                (    V ^8  d   QhR\         RR/# r"   r   )r   s   "r   r   r   <   s     ) )D )UY )r   c                    \        V R 4      p\        P                  ! V 4      ;_uu_ 4        \        P                  q"! V 4      pRqCVJ qU'       Eg3   \        P
                  ! RV3RW434      RR\        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRR\        P                  ! V4      RR\        P                  ! 4       9   g   \        P                  ! V 4      '       d   \        P                  ! V 4      MRR\        P                  ! V4      R\        P                  ! V4      /,          pR	R
V/,          p\        \        P                  ! V4      4      hR;p;p;rTRRR4       \        V4      pR qV	8H  q3'       Eg   \        P
                  ! RV3RW34      RR\        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRRR\        P                  ! 4       9   g   \        P                  ! V4      '       d   \        P                  ! V4      MRR\        P                  ! V4      R\        P                  ! V	4      /,          p
RRV
/,          p\        \        P                  ! V4      4      hR;p;r9R#   + '       g   i     ELM; i)rD   Fr%   r&   r'   r(   r#   r)   r*   r+   r,   Nr    rN   r   rO   rP   r-   rQ   rR   rT   )r   r&   warmup_claimrV   r1   r2   r3   r4   r5   r6   r7   r8   r    rW   s   &          r   =test_release_orphaned_claim_keeps_a_claim_a_live_worker_holdsr_   <   s{   ),G		!	!(	+	+11D1(;DuDuDDDDD;DDDDDD{DDDD{DDD1DDDDDD(DDDD(DDD;DDDuDDDDDDDD 
, 7(y(y(((((((((((7((((7(((((((7((((7((((((y(((((((( 
,	+	+s   A9K%A'KA<KK	c                (    V ^8  d   QhR\         RR/# r"   r   )r   s   "r   r   r   E   rK   r   c                 L   \        V R 4      p\        P                  q"! V 4      pRqCVJ qU'       Eg3   \        P                  ! RV3RW434      RR\
        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRR\        P                  ! V4      RR\
        P                  ! 4       9   g   \        P                  ! V 4      '       d   \        P                  ! V 4      MRR\        P                  ! V4      R\        P                  ! V4      /,          pR	R
V/,          p\        \        P                  ! V4      4      hR;p;p;rT\        V4      pR qV	8H  q3'       Eg   \        P                  ! RV3RW34      RR\
        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRRR\
        P                  ! 4       9   g   \        P                  ! V4      '       d   \        P                  ! V4      MRR\        P                  ! V4      R\        P                  ! V	4      /,          p
RRV
/,          p\        \        P                  ! V4      4      hR;p;r9R# )rM   Fr%   r&   r'   r(   r#   r)   r*   r+   r,   Nr    rN   r   rO   rP   r-   rQ   rR   rT   rU   rW   s   &          r   ;test_release_orphaned_claim_leaves_a_pending_backfill_alonerb   E   sW   ),G--@-h7@5@5@@@@@7@@@@@@;@@@@;@@@-@@@@@@h@@@@h@@@7@@@5@@@@@@@@7(y(y(((((((((((7((((7(((((((7((((7((((((y((((((((r   c                (    V ^8  d   QhR\         RR/# r"   r   )r   s   "r   r   r   L   s     ) )QU )Z^ )r   c                 h   \        V R 4      pRp\        P                  ! \        P                  RV\        V4      .4      p \        WP                  4       \        P                  qD! V 4      pRqeVJ qw'       Eg3   \        P                  ! RV3RWV34      RR\        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRR\        P                  ! V4      RR\        P                  ! 4       9   g   \        P                  ! V 4      '       d   \        P                  ! V 4      MRR	\        P                  ! V4      R
\        P                  ! V4      /,          pRRV/,          p	\!        \        P"                  ! V	4      4      hR;p;p;rvVP%                  4        VP'                  4        \)        V4      p
R qV8H  qU'       Eg   \        P                  ! RV3RW34      RR\        P                  ! 4       9   g!   \        P                  ! \(        4      '       d   \        P                  ! \(        4      MRRR\        P                  ! 4       9   g   \        P                  ! V4      '       d   \        P                  ! V4      MRR\        P                  ! V
4      R\        P                  ! V4      /,          pRR
V/,          p\!        \        P"                  ! V4      4      hR;p
;r[R#   TP%                  4        TP'                  4        i ; i)rD   zEimport sqlite3,sys,time; sqlite3.connect(sys.argv[1]); time.sleep(30)-cFr%   r&   r'   r(   r#   r)   r*   r+   r,   Nr    rN   r   rO   rP   r-   rQ   rR   rT   )r   
subprocessPopensys
executabler
   _wait_until_openpidr&   rV   r1   r2   r3   r4   r5   r6   r7   r8   killwaitr    )r#   r   holdholderr:   r;   r<   r=   r>   r?   rX   rY   rZ   s   &            r   Btest_release_orphaned_claim_keeps_a_claim_an_unmarked_worker_holdsrp   L   s   ),GRDs~~tT3w<HIF**-11D1(;DuDuDDDDD;DDDDDD{DDDD{DDD1DDDDDD(DDDD(DDD;DDDuDDDDDDDD7(y(y(((((((((((7((((7(((((((7((((7((((((y(((((((( 	s   BL A'L 7A<L "L1c                4    V ^8  d   QhR\         R\        RR/# )r   r   rk   r	   N)r   r   )r   s   "r   r   r   [   s!     
> 
>d 
> 
> 
>r   c                   a  \         P                  ! 4       R ,           p\        R4      \        V4      ,          R,          p\         P                  ! 4       V8  dy    \        ;QJ d-    V 3R lVP                  4        4       F  '       g   K   RM"	  RM! V 3R lVP                  4        4       4      '       d   R#  \         P                  ! R4       K  \        RV R	S  24      h  \         d     L6i ; i)
g      $@z/procfdc              3   d   <"   T F%  qP                  4       SP                  4       8H  x  K'  	  R # 5i)N)resolve).0entryr   s   & r   	<genexpr>#_wait_until_open.<locals>.<genexpr>`   s#     VEUE==?goo&77EUs   -0TFNg?zpid z never opened )	r   	monotonicr   r
   anyiterdirOSErrorsleepr7   )r   rk   deadlinefd_dirs   f&  r   rj   rj   [   s    ~~$&H']SX%,F
..
X
%	sVV^^EUVsssVV^^EUVVV W 	

4
4uN7)<
==  		s   
C# &"C# +C# #C10C1c                (    V ^8  d   QhR\         RR/# r"   r   )r   s   "r   r   r   h   s     = =T =d =r   c                    \         P                  q! V 4      pR q2VJ qD'       Eg3   \        P                  ! RV3RW#34      RR\        P
                  ! 4       9   g!   \        P                  ! \         4      '       d   \        P                  ! \         4      MRR\        P                  ! V4      RR\        P
                  ! 4       9   g   \        P                  ! V 4      '       d   \        P                  ! V 4      MRR\        P                  ! V4      R\        P                  ! V4      /,          pRR	V/,          p\        \        P                  ! V4      4      hR
;p;p;rC\         P                  ! V 4      ;_uu_ 4        \         P                  q! V 4      pRq2VJ qD'       Eg3   \        P                  ! RV3RW#34      RR\        P
                  ! 4       9   g!   \        P                  ! \         4      '       d   \        P                  ! \         4      MRR\        P                  ! V4      RR\        P
                  ! 4       9   g   \        P                  ! V 4      '       d   \        P                  ! V 4      MRR\        P                  ! V4      R\        P                  ! V4      /,          pRR	V/,          p\        \        P                  ! V4      4      hR
;p;p;rCR
R
R
4       \         P                  q! V 4      pR q2VJ qD'       Eg3   \        P                  ! RV3RW#34      RR\        P
                  ! 4       9   g!   \        P                  ! \         4      '       d   \        P                  ! \         4      MRR\        P                  ! V4      RR\        P
                  ! 4       9   g   \        P                  ! V 4      '       d   \        P                  ! V 4      MRR\        P                  ! V4      R\        P                  ! V4      /,          pRR	V/,          p\        \        P                  ! V4      4      hR
;p;p;rCR
#   + '       g   i     ELp; i)Fr%   r&   r'   r(   r#   r)   r*   r+   r,   NTr-   zX%(py5)s
{%(py5)s = %(py2)s
{%(py2)s = %(py0)s.warmup_in_progress
}(%(py3)s)
} is %(py8)s)r&   warmup_in_progressr1   r2   r3   r4   r5   r6   r7   r8   r^   r9   s   &      r   6test_warmup_claim_marks_progress_only_inside_the_blockr   h   s3   ))<)(3<u<u<<<<<3<<<<<<;<<<<;<<<)<<<<<<(<<<<(<<<3<<<u<<<<<<<<		!	!(	+	+--?-h7?4?4?????7??????{????{???-??????h????h???7???4???????? 
, ))<)(3<u<u<<<<<3<<<<<<;<<<<;<<<)<<<<<<(<<<<(<<<3<<<u<<<<<<<< 
,	+	+s   <A9Q6A'QA<QQ	c                (    V ^8  d   QhR\         RR/# r"   r   )r   s   "r   r   r   q   s     = =4 =TX =r   c                    \        4       pV R ,          P                  V R2RR7       \        P                  q"! V 4      pRqCVJ qU'       Eg3   \        P
                  ! RV3RW434      RR\        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRR\        P                  ! V4      RR	\        P                  ! 4       9   g   \        P                  ! V 4      '       d   \        P                  ! V 4      MR	R
\        P                  ! V4      R\        P                  ! V4      /,          pRRV/,          p\        \        P                  ! V4      4      hR;p;p;rTR# z.systray-backfill-worker
zutf-8)encodingFr%   r&   r'   r(   r#   r)   r*   r+   r,   Nr-   r   )	_dead_pid
write_textr&   r   r1   r2   r3   r4   r5   r6   r7   r8   )r#   dead_pidr:   r;   r<   r=   r>   r?   s   &       r   <test_warmup_in_progress_ignores_a_marker_from_a_dead_processr   q   s    {H**66(2QX6Y))<)(3<u<u<<<<<3<<<<<<;<<<<;<<<)<<<<<<(<<<<(<<<3<<<u<<<<<<<<r   c                (    V ^8  d   QhR\         RR/# r"   r   )r   s   "r   r   r   x   s     = ==	=r   c                    V R ,          pVP                  \        P                  ! 4        R2RR7       \        P                  ! 4       \        P
                  ,
          ^,
          p\        P                  ! WV34       \        P                  q3! V 4      pRqTVJ qf'       Eg3   \        P                  ! RV3RWE34      RR\        P                  ! 4       9   g!   \        P                  ! \        4      '       d   \        P                  ! \        4      MRR\        P                  ! V4      RR	\        P                  ! 4       9   g   \        P                  ! V 4      '       d   \        P                  ! V 4      MR	R
\        P                  ! V4      R\        P                  ! V4      /,          pRRV/,          p\        \        P                  ! V4      4      hR;p;p;reR# r   )r   osgetpidr   r&   WARMUP_TIMEOUT_Sutimer   r1   r2   r3   r4   r5   r6   r7   r8   )	r#   markerstaler:   r;   r<   r=   r>   r?   s	   &        r   Etest_warmup_in_progress_ignores_a_marker_older_than_the_warmup_budgetr   x   s	    22F
R(7;IIK+666:EHHVU^$))<)(3<u<u<<<<<3<<<<<<;<<<<;<<<)<<<<<<(<<<<(<<<3<<<u<<<<<<<<r   c                $    V ^8  d   QhR\         /# )r   r	   )r   )r   s   "r   r   r      s      3 r   c                      \         P                  ! \        P                  R R.4      p V P	                  4        V P
                  # )re    )rf   rg   rh   ri   rm   rk   )processs    r   r   r      s0    b9:GLLN;;r   )builtinsr3   _pytest.assertion.rewrite	assertionrewriter1   r   r   rf   rh   r   pathlibr   r&   r   r    r@   rF   rI   r[   r_   rb   rp   rj   r   r   r   r    r   r   <module>r      sk      	   
   q "767))))
>===r   