o
Mf% @ s d dl Z d dlZd dlmZ d dlmZ d dlmZmZ dZ dZ
ej dd Z ej d d
Z
ejjd d Zejjd
d Zejjdd Zejjdd Zejjdd Zejjdd ZdS ) N)Task)status) make_projectregister_ml_backend_mockbasic_ml_backendzc c s t | ddd | V d S )N https://ml_backend_for_test_apiz1.0.0)urlsetup_model_version)r )
ml_backend r `/var/www/html/analyze/labelStudio/lib/python3.10/site-packages/label_studio/tests/ml/test_api.pyml_backend_for_test_api s
r
c C s | j ddd d S )Nzsocket.gethostbynamez 321.21.21.21) return_value)patch)mockerr r r mock_gethostbyname s r c C sf t tdtdd| jd}|jdksJ | jd|jddd d
}|jd ks&J | |jdks1J d S ) 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 responser r r #test_ml_backend_set_for_prelabeling s& r, c C sf t tdtdd| jd}t|_| | jd|jdddd }|j d
ks&J |
|jtks1J d S ) 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) r r r 'test_ml_backend_not_set_for_prelabeling9 s( r/ c C s t tdtdd| jd}|jdksJ | jd|jddd d
}|jd ks&J | }|d }| d| }|jd
ks=J | d dksGJ | j
d|j t ddiddjd
ks]J | j
d| t |jddd ddjd
ksuJ | |jdksJ d S )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_idr r r test_model_version_on_saveY sZ
r? c C s t tdtdd| jd}|jdksJ | jd|jddd d
}|jd ks&J | }|d }| d| }|jd
ks=J | d dksGJ | j
d|j t ddiddjd
ks]J | |jdkshJ |
d| jdksuJ | |jdksJ d S )NTr r r r r r0 r r r r! r& r1 r2 r3 r4 r$ r5 r6 )r r" r# r r$ r% r& r' r9 r: r r; r( deleter< r r r test_model_version_on_delete sF
rB c C s~ t tdtdd| jd}| jd|jdddd d
}|jd ks J | }|d d
d dks0J | jd|jddddddd
}|jdksFJ | }d|vsPJ |d }| d| }|jdkscJ d| vskJ | j d| t
|jddddd}|jdksJ d| vsJ | j d| t
|jddd d dd}d| vsJ dd l m } |j
j|d }|jd ksJ d S ) NTr r r r r0 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 rD basic_auth_userbasic_auth_passr! rH r& r1 new_title_1r r5 r6 r8 z)r r r rH ) MLBackend)r& )r r" r# r r% r& r' r9 r: r r; ml.modelsrJ objectsrH ) r* r
r r r+ r= r> rJ r
r r r test_security_write_only_payload s
rM c C s t tdtdd| jdd}tjj|ddid | | }|
d|j d }|j t
jks3J | }|d
d ks?J |d d
ksGJ d S )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 r1 )r r" r# r r rL creater( get_ml_backendsfirstr% r& r' r HTTP_200_OKr9 )r* r r
r+ r= r r r 1test_ml_backend_predict_test_api_post_random_true s" rT )r9 pytestprojects.modelsr rest_frameworkr label_studio.tests.utilsr r r- r# fixturer
r mark django_dbr, r/ r? rB rM rT r r r r s.
9
/
Y