diff --git a/tests/test_web.py b/tests/test_web.py index 73d8098..c41a3eb 100644 --- a/tests/test_web.py +++ b/tests/test_web.py @@ -2,6 +2,7 @@ import unittest import time +import os from pyvulnerabilitylookup import PyVulnerabilityLookup @@ -9,7 +10,8 @@ from pyvulnerabilitylookup import PyVulnerabilityLookup class TestPublic(unittest.TestCase): def setUp(self) -> None: - self.client = PyVulnerabilityLookup(root_url="https://vulnerability.circl.lu") + token = os.getenv("API_KEY", "") + self.client = PyVulnerabilityLookup(root_url="https://vulnerability.circl.lu", token=token) # Test default