load tests for api's

Author: akhil27051999

tests/load_test.py

@@ -1,9 +1,17 @@
-# load_test.py
 from locust import HttpUser, task, between
 import random
 
 class StudentApiUser(HttpUser):
     wait_time = between(1, 2)
+    last_student_ids = []
+
+    @task(1)
+    def home(self):
+        self.client.get("/")
+
+    @task(1)
+    def health(self):
+        self.client.get("/health")
 
     @task(2)
     def get_students(self):
@@ -18,5 +26,27 @@ def create_student(self):
             "gpa": round(random.uniform(6.0, 10.0), 2),
             "email": f"testuser{student_id}@example.com"
         }
-        self.client.post("/students", json=payload) 
+        response = self.client.post("/students", json=payload)
+        if response.status_code == 201:
+            # store created ID for subsequent GET/PUT/DELETE
+            self.last_student_ids.append(response.json()["id"])
+
+    @task(1)
+    def get_student_by_id(self):
+        if self.last_student_ids:
+            student_id = random.choice(self.last_student_ids)
+            self.client.get(f"/students/{student_id}")
+
+    @task(1)
+    def update_student(self):
+        if self.last_student_ids:
+            student_id = random.choice(self.last_student_ids)
+            payload = {"gpa": round(random.uniform(6.0, 10.0), 2)}
+            self.client.put(f"/students/{student_id}", json=payload)
+
+    @task(1)
+    def delete_student(self):
+        if self.last_student_ids:
+            student_id = self.last_student_ids.pop(0)  # remove from list after deletion
+            self.client.delete(f"/students/{student_id}")