Adding auth header to rest calls

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()