PyVulnerabilityLookup/tests/test_web.py

#!/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)