Retrying a flaky test (#847)

.pre-commit-config.yaml

@@ -41,6 +41,7 @@ repos:
     - types-PyYAML==6.0.12.11
     - types-redis==4.6.0.5
     - types-requests==2.31.0.2
+    - types-retry==0.9.9.4
     - types-termcolor==1.1.6.2
     - types-tqdm==4.66.0.2
     - uvicorn==0.23.2

requirements.txt

@@ -36,6 +36,7 @@ PyYAML==6.0.1
 redis==5.0.3
 requests==2.31.0
 requests-mock==1.11.0
+retry==0.9.2
 SQLAlchemy==2.0.28
 semver==3.0.2
 shodan==1.31.0

test/modules/test_wordpress_bruter.py

@@ -2,6 +2,7 @@
 
 from freezegun import freeze_time
 from karton.core import Task
+from retry import retry
 
 from artemis.binds import TaskStatus, TaskType, WebApplication
 from artemis.modules.wordpress_bruter import PASSWORDS, WordPressBruter
@@ -41,7 +42,10 @@ def test_getting_passwords_ip(self) -> None:
         )
         self.assertEqual(passwords, PASSWORDS)
 
+    @retry(tries=3)
     def test_simple(self) -> None:
+        self.setUp()  # @retry() will not rerun setUp
+
         task = Task(
             headers={"type": TaskType.WEBAPP, "webapp": WebApplication.WORDPRESS},
             payload={"url": "http://test-wordpress-easy-password"},