]j% d dl Z d dlZd dlmZ d dlmZ d dlmZmZ dZ dZ
ej d Z ej d Z
ej j d Zej j d
Zej j d Zej j d Zej j d
Zej j d Zy) N)Task)status) make_projectregister_ml_backend_mockbasic_ml_backendzc # 0 K t | dd | y w)N https://ml_backend_for_test_apiz1.0.0)urlsetup_model_version)r )
ml_backends H/root/env/lib/python3.12/site-packages/label_studio/tests/ml/test_api.pyml_backend_for_test_apir s - #
s c * | j dd y )Nzsocket.gethostbynamez 321.21.21.21) return_value)patch)mockers r
mock_gethostbynamer s
LL'nLE c t t dt d | j }|j dk( sJ | j d|j ddd
}|j d k( sJ |j |j dk( sJ y ) NTtest_ml_backend_creation is_published label_configtitleconfiguser /api/ml/ml_backend_titler projectr r
data ) r dictPROJECT_CONFIGr
model_versionpostid status_coderefresh_from_dbbusiness_clientr r r" responses r
#test_ml_backend_set_for_prelabelingr0 s ',
!
!
G B && & ## zz'4
$ H 3 && & $6 66 6r c 6 t t dt d | j }t |_ |j
| j d|j ddd }|j d
k( sJ |j |j
t k( sJ y ) zrWe are not setting it when its already set for another name,
for example when predictions were uploaded beforeTr r r r r r r! r# r% N) r r& r' r ORIG_MODEL_NAMEr( saver) r* r+ r, r- s r
'test_ml_backend_not_set_for_prelabelingr4 9 s
',
!
!
G ,G LLN ## zz'4
$ H 3 && & O 33 3r c t t dt d | j }|j dk( sJ | j d|j ddd
}|j d k( sJ |j }|d }| j d| }|j d
k( sJ |j d dk( sJ | j d|j t j ddi d j d
k( sJ | j d| t j |j ddd d j d
k( sJ |j |j dk( sJ y )NTr r r r r #test_ml_backend_creation_ML_backendr r! r# r% r* stateCO/api/projects/r( application/jsonr$ content_type new_title)
r r& r' r r( r) r* r+ jsongetr dumpsr, r. r r r" r/ r
ml_backend_ids r
test_model_version_on_saverE Y s ',
!
!
G B && & ## zz:4
$ H 3 && &
AdGM ""Xm_#=>H 3 && & ==?7 #t ++ + WZZL )_.S TU+
+ }o & &zz (< ,
+
K // /r c t t dt d | j }|j dk( sJ | j d|j ddd
}|j d k( sJ |j }|d }| j d| }|j d
k( sJ |j d dk( sJ | j d|j t j ddi d j d
k( sJ |j |j dk( sJ | j d| j dk( sJ |j |j dk( sJ y )NTr r r r r r6 r r! r# r% r* r7 r8 r9 r: r( r; r< )r r& r' r r( r) r* r+ r? r@ r rA r, deleterB s r
test_model_version_on_deleterI s ',
!
!
G B && & ## zz:4
$ H 3 && &
AdGM ""Xm_#=>H 3 && & ==?7 #t ++ + WZZL )_.S TU+
+ $I II I ! !H]O"< = I IS PP P B && &r c
t t dt d | j }| j d|j
dddd
}|j d k( sJ |j }|d d
d dk( sJ | j d|j
dddddd
}|j dk( sJ |j }d|vsJ |d }| j d| }|j dk( sJ d|j vsJ | j d| t j |j
ddd d }|j dk( sJ d|j vsJ | j d| t j |j
ddd d d }d|j vsJ dd l m } |j j | }|j d k( sJ y ) NTr r r r r6 r
BASIC_AUTH)r" r r
auth_methodr# i validation_errorsnon_field_errorsr zJAuthentication username and password is required for Basic Authentication.r z)r" r r
rL basic_auth_userbasic_auth_passr% rP r* r7 new_title_1r! r; r< r> z)r" r r
rP ) MLBackend)r* )r r& r' r r) r* r+ r? r@ r rA ml.modelsrR objectsrP ) r. r r r" r/ rC rD rR r s r
test_security_write_only_payloadrU sH ',
!
!
G ## zz:4'
$
H 3 && &
A
1215 W X X
## zz:4' % )
$
H 3 && &
A A %% %dGM ""Xm_#=>H 3 && & HMMO 33 3 $$
=/"
ZZ"::&8
( %
H 3 && & HMMO 33 3 $$
=/"
ZZ"::$8#5
( % H HMMO 33 3#""&&-&8J % %); ;; ;r c t t dt d | j d }t j
j
|ddi |j |j j }| j d|j d }|j t j k( sJ |j }|d
d k( sJ |d d
k( sJ y )NTr r )r r use_ml_backendimagez http://example.com/image.jpg)r" r$ r z/predict/test?random=truer
z http://localhost:8999/predictr r7 )r r& r' r r rT creater, get_ml_backendsfirstr) r* r+ r HTTP_200_OKr? )r. r" r r/ rC s r
1test_ml_backend_predict_test_api_post_random_truer] s ',
!
!G LLw8V.WX ((*002J ##hz}}o=V$WXH 6#5#5 55 5
A U86 66 6 X;# r )r? pytestprojects.modelsr rest_frameworkr label_studio.tests.utilsr r r2 r' fixturer r mark django_dbr0 r4 rE rI rU r] r r
rf s
! K$X F F 7 76 4 4> 60 60r ,' ,'^ V< V