Update performance_testing.py

performance_testing.py

@@ -8,13 +8,20 @@ def perform_performance_test(url):
     Returns:
         str: Performance test code snippet.
     """
-    # Extract base domain from URL (you may need a more robust method)
-    base_domain = url.split('/')[2]
-
+    # Ensure URL is properly formatted
+    if not url.startswith("http://") and not url.startswith("https://"):
+        raise ValueError("Invalid URL format. URL must start with 'http://' or 'https://'.")
+    
+    # Extract base domain from URL
+    try:
+        base_domain = url.split('/')[2]
+    except IndexError:
+        raise ValueError("URL does not contain enough segments to extract base domain.")
+    
     snippet = f"""
 import unittest
 from selenium import webdriver
-from locust import HttpLocust, TaskSet, task
+from locust import HttpUser, TaskSet, task
 
 class PerformanceTests(TaskSet):
 
@@ -33,9 +40,8 @@ class PerformanceTests(TaskSet):
         cart_count = self.client.find_element_by_class_name('cart-count-class').text  # Update with actual class
         self.assertEqual(cart_count, '1')
 
-class LocustTest(HttpLocust):
-    task_set = PerformanceTests
-    min_wait = 5000
-    max_wait = 15000
+class LocustTest(HttpUser):
+    tasks = [PerformanceTests]
+    wait_time = between(5, 15)
     """
-    return snippet
+    return snippet