Merge pull request #198 from AikidoSec/simplify-tests

Fix flickering test and simplify generate data for limit tests

Author: tudor-timcu

tests/server/test_domains_limits/test.py

@@ -5,12 +5,11 @@
 
 
 def generate_domains():
-    generated_domains = set()
-    while len(generated_domains) < 2100:
-        domain_len = random.randint(20, 30)  # Random route length between 1-20 segments
-        domain = ''.join(random.choices('abcdefghijklmnopqrstuvwxyz0123456789-_', k=domain_len))
-        generated_domains.add(domain)
-    return list(generated_domains)
+    generated_domains = []
+    for i in range(2100):
+        domain = f"domain{i}"
+        generated_domains.append(domain)
+    return generated_domains
 
 
 def run_test():

tests/server/test_routes_limits/test.py

@@ -10,17 +10,11 @@
 '''
 
 def generate_routes():
-    generated_routes = set()
-    while len(generated_routes) < 6000:
-        route_len = random.randint(1, 20)  # Random route length between 1-20 segments
-        route_parts = []
-        for j in range(route_len):
-            part_len = random.randint(3, 15)  # Random segment length between 3-15 chars
-            part = ''.join(random.choices('abcdefghijklmnopqrstuvwxyz0123456789-_', k=part_len))
-            route_parts.append(part)
-        route = '/' + '/'.join(route_parts)
-        generated_routes.add(route)
-    return list(generated_routes)
+    generated_routes = []
+    for i in range(6000):
+        route = f"/route{i}"
+        generated_routes.append(route)
+    return generated_routes
 
 def run_test():
     routes = generate_routes()

tests/server/test_user_limits/test.py

@@ -5,12 +5,11 @@
 
 
 def generate_users():
-    generated_users = set()
-    while len(generated_users) < 3500:
-        user_len = random.randint(1, 20)  # Random route length between 1-20 segments
-        user = ''.join(random.choices('abcdefghijklmnopqrstuvwxyz0123456789-_', k=user_len))
-        generated_users.add(user)
-    return list(generated_users)
+    generated_users = []
+    for i in range(3500):
+        user = f"user{i}"
+        generated_users.append(user)
+    return generated_users
 
 
 def run_test():