mirror of
https://github.com/cve-search/PyVulnerabilityLookup.git
synced 2024-11-25 16:27:23 +00:00
24 lines
679 B
Python
24 lines
679 B
Python
#!/usr/bin/env python3
|
|
|
|
import unittest
|
|
import time
|
|
|
|
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:
|
|
while True:
|
|
if vuln := self.client.get_vulnerability('PYSEC-2024-4'):
|
|
self.assertEqual(vuln['id'], 'PYSEC-2024-4')
|
|
break
|
|
print('waiting for pysec to be imported')
|
|
time.sleep(1)
|