Ë
    ·ðÜj©  ã                   ó  — d dl Z d dlZd dlmZ e j                  j                  d  e ee«      j                  «       j                  d   «      «       d dl
mZmZmZmZmZ  G d„ dej                   «      Zedk(  r ej&                  «        yy)é    N)ÚPathé   )ÚBackendÚ
JobRequestÚModeÚPrivacyTierÚroutec                   ó6   — e Zd Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Z	y)	ÚRoutingTestsc           	      óš   — t        dt        ddh«      t        ddh«      «      t        dt        h d£«      t        h d£«      «      g| _        y )	NÚcodexÚcodeÚtextÚcodingÚwritingÚclaude>   r   r   Úvision>   r   r   r   )r   Ú	frozensetÚbackends)Úselfs    ú0/home/aria/aria-v3-dev/python/tests/test_core.pyÚsetUpzRoutingTests.setUp   sO   € äØÜ˜6 6Ð*Ó+Ü˜8 YÐ/Ó0óô
 ØÜÒ4Ó5ÜÒ9Ó:óð
ˆ�ó    c                 óš   — t        ddt        j                  «      }| j                  t	        || j
                  «      j                  d«       y )NÚ1r   r   )r   r   ÚPUBLICÚassertEqualr	   r   Úbackend©r   Újobs     r   Útest_codex_wins_tiez RoutingTests.test_codex_wins_tie   s7   € Ü˜˜h¬×(:Ñ(:Ó;ˆØ×Ñœ˜s D§M¡MÓ2×:Ñ:¸GÕDr   c                 ó°   — t        ddt        j                  t        dh«      «      }| j	                  t        || j                  «      j                  d«       y )NÚ2r   r   )r   r   r   r   r   r	   r   r   r   s     r   Útest_claude_for_visionz#RoutingTests.test_claude_for_vision    sF   € ÜØ�œ;×-Ñ-Ü�x�jÓ!ó
ˆð 	×Ñœ˜s D§M¡MÓ2×:Ñ:¸HÕEr   c                 ó˜   — t        ddt        j                  «      }| j                  t	        || j
                  «      j                  «       y )NÚ3r   )r   r   ÚCONFIDENTIALÚassertFalser	   r   Úallowedr   s     r   Útest_confidential_blockedz&RoutingTests.test_confidential_blocked'   s5   € Ü˜˜h¬×(@Ñ(@ÓAˆØ×Ñœ˜s D§M¡MÓ2×:Ñ:Õ;r   c                 ó˜   — t        ddt        j                  «      }| j                  t	        || j
                  «      j                  «       y )NÚ4r   )r   r   ÚUNKNOWNr(   r	   r   r)   r   s     r   Útest_unknown_blockedz!RoutingTests.test_unknown_blocked+   s5   € Ü˜˜h¬×(;Ñ(;Ó<ˆØ×Ñœ˜s D§M¡MÓ2×:Ñ:Õ;r   c                 óº   — t        ddt        j                  t        j                  ¬«      }| j                  t        || j                  «      j                  d«       y )NÚ5r   )Úmoder   )	r   r   r   r   ÚCLAUDEr   r	   r   r   r   s     r   Útest_manual_claudezRoutingTests.test_manual_claude/   sC   € ÜØ�œ;×-Ñ-Ü—‘ô
ˆð 	×Ñœ˜s D§M¡MÓ2×:Ñ:¸HÕEr   c                 óÌ   — t        dt        dh«      t        dh«      d¬«      g}t        ddt        j                  «      }| j                  t        ||«      j                  «       y )Nr   r   r   F)Úbudget_allowedÚ6)r   r   r   r   r   r(   r	   r)   )r   Úblockedr    s      r   Útest_budget_blockzRoutingTests.test_budget_block6   s\   € äØÜ˜6˜(Ó#Ü˜8˜*Ó%Ø$ô	ð
ˆô ˜˜h¬×(:Ñ(:Ó;ˆØ×Ñœ˜s GÓ,×4Ñ4Õ5r   N)
Ú__name__Ú
__module__Ú__qualname__r   r!   r$   r*   r.   r3   r8   © r   r   r   r      s(   „ ò
òEòFò<ò<òFó
6r   r   Ú__main__)ÚsysÚunittestÚpathlibr   ÚpathÚinsertÚstrÚ__file__ÚresolveÚparentsÚaria_router.corer   r   r   r   r	   ÚTestCaser   r9   Úmainr<   r   r   ú<module>rJ      st   ðÛ 
Û Ý à ‡�‡��‘3‘t˜H“~×-Ñ-Ó/×7Ñ7¸Ñ:Ó;Ô <÷õ ô
46�8×$Ñ$ô 46ðn ˆzÒØ€H‡M�M…Oð r   