<jZqdZddlmZddlZddlZddlZddlmZmZddlm Z e e jjdZ ee ejvr"ejj!dee ddlmZmZmZdZd Zd Zd Zd!d Z d" d#d Zeeeef d$dZd%dZd&dZd&dZd&dZd&dZ d&dZ!d&dZ"d&dZ#d&dZ$d&dZ%d&dZ&d&dZ'd&dZ(d&dZ)d&dZ*d&dZ+d&dZ,d&d Z-y)'u[회귀 테스트 17건 — anu_v2.worktree_cleanup (task-2550). pytest 사용. 외부 부수효과(subprocess / file write / clock)는 모두 fake callable로 주입. ⚠️ 테스트 코드 내 "ghp_faketoken123abc" 등 raw token placeholder는 실제 토큰이 아닌 테스트 fake 값임 — leak detector 오탐 방지를 위해 명시. ) annotationsN)datetimetimezone)Path)DEFAULT_CHAT_IDWorktreeCandidateWorktreeCleanup(aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa task-2550ztask/task-2550-dev5-/home/jay/workspace/.worktrees/task-2550-dev5c BtddddddtjS)u4고정 시각 반환 — 테스트 결정론 보장.i r)tzinfo)rrutc*anu_v2/tests/test_worktree_cleanup_2550.py _fake_clockr$s D!RQ(,, ??rc4tjg|||S)N)args returncodestdoutstderr) subprocessCompletedProcess)rrrs r_make_completed_processr)s#  & &   rc t||||S)Npathbranchtask_idhead_sha)r r!s r_make_candidater&6s $vwQY ZZrc~~tdS)ue기본 fake subprocess runner — rc=0, no output. unused parameter는 명시적으로 del로 처리.rr)_args_kwargss r _noop_runnerr+?s w "1 %%rcJ|dz dz }|jdd|tdz jttt |}|j t}|jdk(sJ|jdk(sJ|jdusJ|jd k(sJy ) u*.done.acked 마커 존재 → passed=True.memoryeventsTparentsexist_ok .done.ackedsubprocess_runnerclockworkspace_root done_ackedokN) mkdir FAKE_TASK_IDtouchr r+rcheck_safety_1_done_acked conditionnamepasseddetailtmp_path events_dircleanupresults rtest_safety_1_done_acked_passrGHsH$x/JTD1\N+..557&G  . .| r?r@rA)rCrErFs rtest_safety_1_done_acked_failrJ_sq&G  . .| .fake_runnerw. 4 >&q"b11rr3r pr_mergedTN jsondumps FAKE_BRANCHr rcheck_safety_2_pr_mergedr;r>r?r@rArCrZrErFrYs @rtest_safety_2_pr_merged_passrcqs**+FK2 %G  - -l ;F   q  ;;+ %% % ==D  v}} $$ $rctjddtdgfd}t|t|}|j t }|jdk(sJ|jdk(sJ|jdusJd|jvsJy ) u8gh API mock으로 PR state=OPEN 반환 → passed=False.fOPENrNcHd|vrd|vr tddStdddSrSr(rWs rrZz6test_safety_2_pr_merged_fail_open..fake_runnerr[rr3rr\FNr]rbs @r!test_safety_2_pr_merged_fail_openrhs** DK2 %G  - -l ;F   q  ;;+ %% % ==E !! ! V]] "" "rcJ|dz dz }|jdd|tdz jttt |}|j t}|jdk(sJ|jdk(sJ|jdusJ|jd k(sJy ) u*.merge-done 마커 존재 → passed=True.r-r.Tr/ .merge-doner3 merge_doner9N) r:r;r<r r+rcheck_safety_3_merge_doner>r?r@rArBs rtest_safety_3_merge_done_passrnsH$x/JTD1\N+..557&G  . .| .fake_runner/ 4 Ot$;*1b"5 5&q"b11rr3branch_in_mainTancestorNr rcheck_safety_4_branch_in_mainr`r>r?r@rArCrZrErFs r!test_safety_4_branch_in_main_passr{sy2 %G  2 2; ?F   q  ;;* ** * ==D   && &rcd}t|t|}|jt}|jdk(sJ|j dk(sJ|j dusJd|jvsd|jvsJyy) u3git merge-base --is-ancestor rc=1 → passed=False.cFd|vrd|vr tdddStdddS)Nrqrrr7rVrr(rss rrZz6test_safety_4_branch_in_main_fail..fake_runnerrtrr3rurvFz NOT ancestorzrc=1Nrxrzs r!test_safety_4_branch_in_main_failr~s2 %G  2 2; ?F   q  ;;* ** * ==E !! ! V]] *f .EE E.E *rcd}t|t|}|jt}|jdk(sJ|j dk(sJ|j dusJd|jvsJy)u+pgrep rc=1 (매치 없음) → passed=True.c>d|vr tdddStdddS)Npgrepr7rVrr(rss rrZz2test_safety_5_not_in_use_pass..fake_runners' d?*1b"5 5&q"b11rr3r not_in_useTz no processNr rcheck_safety_5_not_in_useFAKE_WORKTREE_PATHr>r?r@rArzs rtest_safety_5_not_in_use_passrsy2 %G  . ./A BF   q  ;;, && & ==D  6== (( (rcd}t|t|}|jt}|jdk(sJ|j dk(sJ|j dusJd|jvsJy)u,pgrep rc=0 (매치 있음) → passed=False.c>d|vr tdddStdddS)Nrrz12345 rVr(rss rrZz2test_safety_5_not_in_use_fail..fake_runners' d?*1i< <&q"b11rr3rrFprocessNrrzs rtest_safety_5_not_in_use_failrsy2 %G  . ./A BF   q  ;;, && & ==E !! !  %% %rc\ttt|}|jd}|jdk(sJ|j dk(sJ|j dusJ|jd}|jdk(sJ|j dk(sJ|j dusJd|jvsJy)u9apply=True → passed=True; apply=False → passed=False.r3Tapply_explicitFzdry-runN)r r+rcheck_safety_6_apply_explicitr>r?r@rA)rCrE result_true result_falses rtest_safety_6_apply_explicitrs&G 77=K  A %% %   / // /    %% %88?L  ! !Q && &    0 00 0   % '' '  ++ ++ +rcLd}t|t|}tt|dz }|j |d}|j dusJ|j dusJ|jdusJ|jJd |jjvsJ|d z d z }t|jd }t|d k\sJdtj|djd}|dt k(sJdt d|dt"k(sJy)utdirty=True인 worktree → CleanupResult.skipped=True, skip_reason 포함 "dirty", applied=False, log 파일 생성.cFd|vrd|vr tdddStdddS)Nstatus --porcelainrzM modified_file.py rVr(rss rrZz0test_dirty_worktree_skipped..fake_runner7s0 t   5*1.ErJ J&q"b11rr3 some-worktreer"FapplyTNdirtyr-r.zworktree-cleanup-skipped-*.jsonr7u0dirty skip 시 log 파일이 생성되어야 함rzutf-8)encodingchat_idu chat_id가 u 이어야 함r$)r rr&strcleanup_worktreerskippedapplied skip_reasonlowerrUgloblenr^loads read_textrr;)rCrZrE candidaterFrD log_fileslog_datas rtest_dirty_worktree_skippedr4sA2%G  SO)C%DEI  % %iu % =F <<4   >>T !! ! >>U "" "    )) ) f((..0 00 0H$x/JZ__%FGHI y>Q R RR zz)A,00'0BCH I / 1_[@QQ^3__ 1 I , .. .rc8ttt|}tt |}|j |d}|j dusJ|jdusJ|jdusJ|jJd|jjvsJy)ueworkspace_root와 동일 path → is_main=True, applied=False, skip_reason에 'main worktree' 포함.r3rTrFNz main worktree) r r+rr&rris_mainrrrr)rCrErrFs r test_main_worktree_never_deletedr[s&G S]3I  % %it % >T !! ! >>U "" " >>T !! !    )) ) f00668 88 8rczdz dz }|jdd|tdz j|tdz jtjddt d gfd }t |t }|jd }td|D}|dk(s Jd|dy)u@apply=False, 모든 safety PASS여도 applied_count=0 어설션.r-r.Tr/r2rjgrMrNc$d|vrd|vr tdddSd|vrd|vr tddSd|vr tdddSd|vr td ddSd |vr3d|vr/td z }d |d tdtd}td|dStdddS)NrrrrVrTrUrqrr7worktreer worktree  HEAD z branch refs/heads/ )rrFAKE_SHAr`)rrXwt_path porcelainrYrCs rrZz4test_dry_run_zero_actual_delete..fake_runner~s t   5*1b"5 5 4 > 4 *1b"5 5 d?*1b"5 5  &D.(_45GG9% z"%%0M5  +1i< <&q"b11rr3Frc3:K|]}|jsdyw)r7N)r).0rs r z2test_dry_run_zero_actual_delete..s8aaii8sru3dry-run에서 applied_count가 0이어야 함 (got )N) r:r;r<r^r_r`r rcleanup_all_dry_runsum)rCrDrZrEresults applied_countrYs` @rtest_dry_run_zero_actual_deleterrsH$x/JTD1\N+..557\N+..557**+FK2(%G )))6G8788M A e!TUbTccdee rc|dz dz }|jdd|tdz j|tdz jtjddt d gd d ifd }t |t| }tt|dz }|j|d}|jdusJ|jdusJ|jd usJd dusJdy)uaapply=True, 6대 safety 모두 PASS, mock subprocess 'git worktree remove' rc=0 → applied=True.r-r.Tr/r2rjhrMrNcalledFcd|vrd|vr tdddSd|vrd|vr tddSd|vr tdddSd|vr td ddSd |vrd |vrd d <tdddStdddS)NrrrrVrTrUrqrr7rremoveTrr()rrXrYremoveds rrZz:test_apply_mode_with_all_safe_deletes..fake_runners t   5*1b"5 5 4 > 4 *1b"5 5 d?*1b"5 5  (d"2 $GH *1b"5 5&q"b11rr3rrru*git worktree remove가 호출되어야 함N)r:r;r<r^r_r`r rr&rrall_saferr)rCrDrZrErrFrYrs @@r%test_apply_mode_with_all_safe_deletesrsH$x/JTD1\N+..557\N+..557**+FKG 2%G SO)C%DEI  % %it % >T !! ! >>U "" " 8  $R&RR $rcV|dz dz }|jdd|tdz j|tdz jtjddt d gfd }t |t| }tt|d z  }|j|d}|jdusJ|jdusJ|jdusJ|jJ|jDcgc]}|j r|j"}}d|vsJycc}w)uYsafety_2 FAIL (PR OPEN) → cleanup_worktree all_safe=False, skipped=True, applied=False.r-r.Tr/r2rjirfrNcd|vrd|vr tdddSd|vrd|vr tddSd|vr tdddSd|vr td ddStdddS) NrrrrVrTrUrqrr7r(rWs rrZz-test_unmerged_pr_skipped..fake_runnersy t   5*1b"5 5 4 > 4 *1b"5 5 d?*1b"5 5&q"b11rr3rrrFNr\)r:r;r<r^r_r`r rr&rrrrrrsafety_resultsr@r?) rCrDrZrErrFrfailed_safety_namesrYs @rtest_unmerged_pr_skippedrs8H$x/JTD1\N+..557\N+..557** DK 2%G  SO)C%DEI  % %it % >T !! ! >>U "" "    )) )+1+@+@Qa166QQ - -- -Rs =D&D&cd}d||dz dz }|jdd|tdz j|tdz jtjd d t d gfd }t |t| }tt|dz }|j|d}|jJ||jvsJd|jd|jvsJd|jy)usubprocess error 시 raw token 'ghp_faketoken123abc'가 stderr에 있어도 skip_reason에는 ***MASKED***로 출력. ⚠️ 'ghp_faketoken123abc'는 테스트 fake 값 (실제 토큰 아님). ghp_faketoken123abcz(error: authentication failed with token r-r.Tr/r2rjjrMrNcd|vrd|vr tdddSd|vrd|vr tddSd|vr tdddSd|vr td ddSd |vrd |vr td dStdddS) NrrrrVrTrUrqrr7rrr()rrXfake_stderr_with_tokenrYs rrZz9test_token_not_leaked_in_skip_reason..fake_runner s t   5*1b"5 5 4 > 4 *1b"5 5 d?*1b"5 5  (d"2*1b2HI I&q"b11rr3rrrNu'skip_reason에 raw token이 노출됨: z ***MASKED***u.skip_reason에 ***MASKED*** 처리가 없음: ) r:r;r<r^r_r`r rr&rrr) rCfake_raw_tokenrDrZrErrFrrYs @@r$test_token_not_leaked_in_skip_reasonrs: +NGGWXH$x/JTD1\N+..557\N+..557**+FK 2%G  SO)C%DEI  % %it % .fake_runnerEs7  &D.]d5J*1.>C C&q"b11rr3rku*3개 worktree가 파싱되어야 함 (got rr z task-2545N)rr renumerate_worktreesrr"r$r%) rC wt_path_1 wt_path_2 wt_path_3rZrE candidatescpathstask_mapsha_maprs @r-test_enumerate_parses_worktree_list_porcelainr1s%I?I?I I;z;z;z2 32 %G ,,.J z?a `#McR\oM^^_!`` ( (QVV (E (         ,66a !6H6 I + -- - I + -- -,66aqvvqzz!6G6 9  )) ) 9  )) ) 9  )) ) ) 7 7s.DDD)returnr)rrVrV)rintrrrrrsubprocess.CompletedProcess) r"rr#rr$z str | Noner%rrr )r)objectr*rrr)rCrrNone).__doc__ __future__rr^rsysrrpathlibr__file__resolver0WORKSPACE_ROOTrr"insertanu_v2.worktree_cleanuprr r rr;r`rrrr&r+rGrJrcrhrnr{r~rrrrrrrrrrrrrrsi#  'h'')11!4~chh&HHOOAs>*+  # D@       ! #& [ [ [[ [  [&!. &$%6#6!,'0F0)0&0,.!/N9.*f`(S\%.V0l2*r