PyVulnerabilityLookup/tests/test_web.py
2024-01-17 12:41:24 +01:00

19 lines
539 B
Python

#!/usr/bin/env python3
import unittest
from pyvulnerabilitylookup import PyVulnerabilityLookup
class TestBasic(unittest.TestCase):
def setUp(self) -> None:
self.client = PyVulnerabilityLookup(root_url="http://127.0.0.1:10001")
def test_up(self) -> None:
self.assertTrue(self.client.is_up)
self.assertTrue(self.client.redis_up())
def test_get_vulnerability(self) -> None:
vuln = self.client.get_vulnerability('CVE-2023-23059')
self.assertEqual(vuln['cve']['id'], 'CVE-2023-23059')