o
Mf7 @ s\ d Z ddlZddlZddlmZ ddlmZmZmZm Z G dd deZ
G dd deZ dS ) zThis file and its contents are licensed under the Apache License 2.0. Please see the included NOTICE for copyright information and LICENSE for a copy of the license.
N)uuid4)HttpUserTaskSetbetweentaskc @ s e Zd Zdd Zeddd Zeddd Zeddd Zed
d d Zedd
d Z eddd Z eddd Z
eddd Z eddd Z eddd Z
ed
dd Zeddd Zd S ) UserWorksWithProjectc C s t t }t|dddd}| jjd|dd| jj ddd1}|jd kr.||j n | d
| _ t
d | j d | jj W d d S W d d S 1 sUw Y d S )
NTz)title is_publishedskip_onboarding label_configz
/api/projectsapplication/jsonToken z content-type
Authorization)dataheaderscatch_response idzProject z has been created by user ) strr jsondumpsclientposttoken status_codefailure
project_idprintname)selfr payloadr r# i/var/www/html/analyze/labelStudio/lib/python3.10/site-packages/label_studio/tests/loadtests/locustfile.pyon_start s,
"z UserWorksWithProject.on_start c C | j d d S )Nz
/projects/r getr r# r# r$ project_list" z!UserWorksWithProject.project_listc C | j jd| j dd d S )Nz /projects/%iz/projects/r r r) r r* r# r# r$ project_dashboard& z&UserWorksWithProject.project_dashboardc C r- )Nz/projects/%i/dataz/projects//datar. r/ r* r# r# r$ project_data* r1 z!UserWorksWithProject.project_data c C r- )Nz/projects/%i/label-streamz/projects//label-streamr. r/ r* r# r# r$ label_stream. r1 z!UserWorksWithProject.label_streamc C r- Nz/projects/%i/expertsz/projects//expertsr. r/ r* r# r# r$ expert_page2 r1 z UserWorksWithProject.expert_pagec C r- r5 r/ r* r# r# r$ r6 6 r1 c C r' )Nz/business/statsr( r* r# r# r$ stats: r, zUserWorksWithProject.statsc C r- )Nz/projects/%i/plotsz/projects//plotsr. r/ r* r# r# r$
project_stats> r1 z"UserWorksWithProject.project_statsc C r' )Nz/business/expertsr( r* r# r# r$ expertsB r, z UserWorksWithProject.expertsc C sH t ddiddig}dd| jj d}| jjd| j ||dd d S )
Ntextzexample positive reviewzexample negative reviewr r
r z/api/projects/%i/tasks/bulkz /api/projects//tasks/bulkr r )r r r r r r )r r! r r# r# r$ import_tasksF s
z!UserWorksWithProject.import_tasksc C s | j jd| j dd| j j idd}| }t|rOt|}tdddd dtd
d ggid gi}d
d| j j d}| j j d|d ||dd d S d S )Nz/api/projects/%i/tasksr r
z/api/projects//tasksr; resultchoicesmy_classmy_textposneg)type from_nameto_namevaluer r z/api/tasks/%i/annotationsr z/api/tasks//annotations)
r r) r r r lenrandomchoicer r )r r"
tasks_listany_taskr! r r# r# r$ complete_task_via_apiQ s4
z*UserWorksWithProject.complete_task_via_api c C | d S N) interruptr* r# r# r$ stopo s zUserWorksWithProject.stopN)__name__
__module__ __qualname__r% r r+ r0 r2 r4 r6 r7 r8 r9 r< rL rQ r# r# r# r$ r
s4
r c @ s. e Zd ZeddZediZdd Zdd ZdS ) WebsiteUser
c C rN rO )signupr* r# r# r$ r% x s zWebsiteUser.on_startc C s | j d}|jd }tt }| dd| d}| j jd|d|id | j d }|d
| j _|| j _ t
d | d | j j d S )
N/ csrftokenz @loadtest.me12345678)emailpasswordr z /user/signupz X-CSRFToken)r z/api/current-user/tokendetailzClient z successfully signed up. Token: ) r r) cookiesr r upperr r r r r )r responser[ usernamer! r# r# r$ rY { s
zWebsiteUser.signupN) rR rS rT r wait_timer tasksr% rY r# r# r# r$ rU t s
rU ) __doc__r rH uuidr locustr r r r r rU r# r# r# r$ s j