Ë
    ·ðÜj3  ã                   ó€   — d dl Z 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)ÚBackendÚ
JobRequestÚPrivacyTierÚModeÚroutec                   ó   — e Zd Zd„ Zd„ Zd„ Zy)ÚSecurityTestsc                 ó  — t        ddt        j                  «      }t        dt	        dh«      t	        dh«      d¬«      }t        ||g«      }| j                  |j                  «       | j                  |j                  «       y )NÚs1Úanalysisúunknown-providerÚtextéd   ©Úquality_score)
r   r   ÚPUBLICr   Ú	frozensetr   ÚassertFalseÚallowedÚassertIsNoneÚbackend)ÚselfÚjobÚunknownÚresults       ú4/home/aria/aria-v3-dev/python/tests/test_security.pyÚtest_unknown_backend_rejectedz+SecurityTests.test_unknown_backend_rejected
   sn   € Ü˜˜z¬;×+=Ñ+=Ó>ˆäØÜ�v�hÓÜ�z�lÓ#Øô	
ˆô �s˜W˜IÓ&ˆà×Ñ˜Ÿ™Ô(Ø×Ñ˜&Ÿ.™.Õ)ó    c                 óL  — t        ddt        j                  «      }t        dt	        dh«      t	        dh«      d¬«      t        dt	        dh«      t	        dh«      d¬«      g}t        ||«      }| j                  |j                  «       | j                  |j                  d«       y )	NÚs2r   r   r   r   r   ÚcodexéP   )
r   r   r   r   r   r   Ú
assertTruer   ÚassertEqualr   )r   r   Úbackendsr   s       r   Ú"test_unknown_cannot_override_codexz0SecurityTests.test_unknown_cannot_override_codex   s—   € Ü˜˜z¬;×+=Ñ+=Ó>ˆô Ø"Ü˜6˜(Ó#Ü˜:˜,Ó'Ø!ô	ô ØÜ˜6˜(Ó#Ü˜:˜,Ó'Ø ô	ð
ˆô �s˜HÓ%ˆà�‰˜Ÿ™Ô'Ø×Ñ˜Ÿ™¨Õ1r   c                 óè   — t        ddt        j                  t        j                  ¬«      }t        dt        dh«      t        dh«      «      }| j                  t        ||g«      j                  «       y )NÚs3r   )Úmoder!   r   )
r   r   ÚCONFIDENTIALr   ÚCODEXr   r   r   r   r   )r   r   r!   s      r   Ú test_privacy_blocks_manual_codexz.SecurityTests.test_privacy_blocks_manual_codex0   sc   € ÜØ�*Ü×$Ñ$Ü—‘ô
ˆô ØÜ�v�hÓÜ�z�lÓ#ó
ˆð 	×Ñœ˜s U GÓ,×4Ñ4Õ5r   N)Ú__name__Ú
__module__Ú__qualname__r   r&   r,   © r   r   r	   r	      s   „ ò*ò2ó.6r   r	   Ú__main__)ÚunittestÚaria_router.corer   r   r   r   r   ÚTestCaser	   r-   Úmainr0   r   r   ú<module>r6      s?   ðÛ ÷õ ô
56�H×%Ñ%ô 56ðp ˆzÒØ€H‡M�M…Oð r   