diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml index 2dda8809b..525f34067 100644 --- a/.pre-commit-config.yaml +++ b/.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 diff --git a/requirements.txt b/requirements.txt index 5da113cf9..9855f01b6 100644 --- a/requirements.txt +++ b/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 diff --git a/test/modules/test_wordpress_bruter.py b/test/modules/test_wordpress_bruter.py index d10b0bf2f..1368c0b2b 100644 --- a/test/modules/test_wordpress_bruter.py +++ b/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"},