Class OpenVulnQueryClientTestCvrf

tests/test_query_client_cvrf.py:154–224

Unit Test for all function in OpenVulnQueryClient

@pytest.mark.skip(reason='Out of sync and requires / triggers token usage as'
                         ' well as api access')
class OpenVulnQueryClientTestCvrf(unittest.TestCase):
    """Unit Test for all function in OpenVulnQueryClient"""

    def setUp(self):
        self.open_vuln_client = query_client.OpenVulnQueryClient(
            config.CLIENT_ID,
            config.CLIENT_SECRET)
        self.adv_format = "cvrf"

    @mock.patch("query_client.OpenVulnQueryClient.get_request",
                side_effect=mocked_get_requests)
    def test_get_by_cve(self, mock_get_requests):
        """Checks if get_by_cve function calls request args with correct arguments"""
        exp_args = "cvrf/cve/%s" % test_cve
        response = self.open_vuln_client.get_by_cve(self.adv_format, test_cve)
        mock_get_requests.assert_called_with(exp_args)

    @mock.patch("query_client.OpenVulnQueryClient.get_request",
                side_effect=mocked_get_requests)
    def test_get_by_advisory(self, mock_get_requests):
        """Checks if get_by_advisory function calls request args with correct arguments"""
        exp_args = "cvrf/advisory/%s" % test_advisory_id
        response = self.open_vuln_client.get_by_advisory(self.adv_format,
                                                         test_advisory_id)
        mock_get_requests.assert_called_with(exp_args)

    @mock.patch("query_client.OpenVulnQueryClient.get_request",
                side_effect=mocked_get_requests)
    def test_get_by_year(self, mock_get_requests):
        """Checks if get_by_year function calls request args with correct arguments"""
        exp_args = "cvrf/year/%s" % test_year
        response = self.open_vuln_client.get_by_year(self.adv_format,
                                                     year=test_year)
        mock_get_requests.assert_called_with(exp_args)

    @mock.patch("query_client.OpenVulnQueryClient.get_request",
                side_effect=mocked_get_requests)
    def test_get_by_severity(self, mock_get_requests):
        """Checks if get_by_severity function calls request args with correct arguments"""
        exp_args = "cvrf/severity/%s" % test_severity
        response = self.open_vuln_client.get_by_severity(self.adv_format,
                                                         severity=test_severity)
        mock_get_requests.assert_called_with(exp_args)

    @mock.patch("query_client.OpenVulnQueryClient.get_request",
                side_effect=mocked_get_requests)
    def test_get_by_all(self, mock_get_requests):
        """Checks if get_by_all function calls request args with correct arguments"""
        exp_args = "cvrf/all"
        response = self.open_vuln_client.get_by_all(self.adv_format, "all")
        mock_get_requests.assert_called_with(exp_args)

    @mock.patch("requests.get", side_effect=mocked_requests_lib_get)
    def test_get_requests(self, mock_get):
        """Checks if _get_requests function returns correct output"""
        test_path = "cvrf/cve/%s" % test_cve
        response = self.open_vuln_client.get_request(test_path)
        self.assertDictEqual(json.loads(response_generic), response)