Fix linting

Ostorlab · Nov 10, 2023 · 9c1fa2e · 9c1fa2e
1 parent 8412e8c
commit 9c1fa2e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/tests/exploits_test.py b/tests/exploits_test.py
@@ -1,6 +1,5 @@
 """Unit tests for Agent Asteriod exploits"""
 import datetime
-import random
 
 import requests
 from pytest_mock import plugin
@@ -37,6 +36,7 @@ def side_effect(*args, **kwargs):  # type: ignore[no-untyped-def]
         == "A heap-based buffer overflow vulnerability in FortiOS"
     )
     assert (
-        "https://91.135.170.42:8443 is vulnerable to CVE-2023-27997, "
+        vulnerability.technical_detail
+        == "https://91.135.170.42:8443 is vulnerable to CVE-2023-27997, "
         "A heap-based buffer overflow vulnerability in FortiOS"
     )
diff --git a/tests/test-requirement.txt b/tests/test-requirement.txt
@@ -3,3 +3,4 @@ pytest
 black
 mypy
 typing-extensions
+pytest_mock
-Original file line number
+Diff line change
@@ Expand Up / @@ -3,3 +3,4 @@ pytest @@
     black
     mypy
     typing-extensions
+    pytest_mock