From a4a21f2a561cb1b519d548abd4ca96f7beb2adb3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jan=20Pokorn=C3=BD?= <xpokorn8@fi.muni.cz>
Date: Sun, 30 Apr 2023 22:19:36 +0200
Subject: [PATCH] Adding auth header to rest calls
---
application/locustfile.py | 55 +++++++++++++++++++++++++++------------
1 file changed, 38 insertions(+), 17 deletions(-)
diff --git a/application/locustfile.py b/application/locustfile.py
index afe71048..bb4fde39 100644
--- a/application/locustfile.py
+++ b/application/locustfile.py
@@ -2,6 +2,7 @@ import random
from locust import HttpUser, task, between
from locust.exception import StopUser
+header={'Authorization': 'Bearer YOUR_ACCESS_TOKEN'}
class StudentActor(HttpUser):
@@ -14,25 +15,41 @@ class StudentActor(HttpUser):
@task(3)
def get_courses_english(self):
- self.client.get('/courses/findAllByLang?lang=ENGLISH')
+ self.client.get(
+ url='/courses/findAllByLang?lang=ENGLISH',
+ auth=None,
+ headers=header,
+ name="Find Courses"
+ )
@task(3)
def get_courses_spanish(self):
- self.client.get('/courses/findAllByLang?lang=SPANISH')
+ self.client.get(
+ url='/courses/findAllByLang?lang=SPANISH',
+ auth=None,
+ headers=header,
+ name="Find Courses"
+ )
@task
def enrol(self):
- self.client.get(f'http://localhost:5000/courses/enrol/{1}?studentId={1}')
+ self.client.patch(
+ url=f'/courses/enrol/{1}?studentId={1}',
+ auth=None,
+ headers=header,
+ name="Enrol Courses"
+ )
@task
def get_lecture_by_id(self):
- self.client.get(f'/lectures/findByCourse?courseId={1}')
-
- def on_start(self):
- # TODO auth
- return super().on_start()
+ self.client.get(
+ url=f'/lectures/findByCourse?courseId={1}',
+ auth=None,
+ headers=header,
+ name="Find Lectures"
+ )
class LecturerActor(HttpUser):
weight = 0
@@ -48,9 +65,13 @@ class LecturerActor(HttpUser):
def new_course(self):
if self.courses_created < 3:
self.client.headers['Content-Type'] = "application/json"
- response = self.client.post("/courses", json=
- {
- "name": "string",
+ response = self.client.post(
+ url="/courses",
+ auth=None,
+ headers=header,
+ name="Create Course",
+ json={
+ "name": "string",
"capacity": 10,
"language": random.choice(["SPANISH", "ENGLISH"]),
"proficiency": random.choice(["A1", "B1", "C1"])
@@ -66,8 +87,12 @@ class LecturerActor(HttpUser):
def new_lecture(self):
if self.courses_created > 0 and self.lecture_created < 10:
self.client.headers['Content-Type'] = "application/json"
- response = self.client.post("/lectures", json=
- {
+ response = self.client.post(
+ url="/lectures",
+ auth=None,
+ headers=header,
+ name="Create Lecture",
+ json={
"lectureFrom": "2024-04-26T18:06:30.658Z",
"lectureTo": "2024-04-27T18:06:30.658Z",
"topic": "string",
@@ -82,7 +107,3 @@ class LecturerActor(HttpUser):
if self.lecture_created >= 9:
raise StopUser()
-
- def on_start(self):
- # TODO auth
- return super().on_start()
--
GitLab