add attempt

skoudoro · Jan 25, 2022 · a088a2c · a088a2c
1 parent b0f8f0d
commit a088a2c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/mailerlite/tests/test_group.py b/mailerlite/tests/test_group.py
@@ -2,6 +2,7 @@
 import random
 import string
 import time
+import itertools
 
 import pytest
 
@@ -103,7 +104,8 @@ def test_groups_subscriber(header):
 
     assert sub1.email == tmp_sub.email
 
-    while True:
+    attempt = itertools.count()
+    while True and next(attempt) < 15:
         try:
             num = random.randint(1000, 100000)
             mail = generate_random_email(length=15, seed=num)
@@ -157,7 +159,8 @@ def test_groups_single_subscriber(header):
 
     assert sub1.email == tmp_sub.email
 
-    while True:
+    attempt = itertools.count()
+    while True and next(attempt) < 20:
         try:
             num = random.randint(1000, 100000)
             mail = generate_random_email(length=15, seed=num)

diff --git a/mailerlite/tests/test_subscriber.py b/mailerlite/tests/test_subscriber.py
@@ -2,6 +2,7 @@
 import random
 import string
 import time
+import itertools
 
 import pytest
 
@@ -73,7 +74,8 @@ def test_subscribers_error(header):
 def test_subscribers_crud(header):
     subscriber = Subscribers(header)
 
-    while True:
+    attempt = itertools.count()
+    while True and next(attempt) < 20:
         try:
             num = random.randint(1000, 100000)
             mail = generate_random_email(length=15, seed=num)