diff --git a/codeinsight_sdk/handler.py b/codeinsight_sdk/handler.py index 8a0b5b3..302ebff 100644 --- a/codeinsight_sdk/handler.py +++ b/codeinsight_sdk/handler.py @@ -1,8 +1,6 @@ import abc from typing import List -from .models import Project, ProjectInventory, ProjectInventoryItem, Report -from .exceptions import CodeInsightError class Handler(abc.ABC): def __init__(self, client): @@ -12,4 +10,3 @@ def __init__(self, client): @abc.abstractmethod def get(self): pass - diff --git a/tests/test_experimental.py b/tests/test_experimental.py index fe0e0ea..551beff 100644 --- a/tests/test_experimental.py +++ b/tests/test_experimental.py @@ -5,7 +5,6 @@ from codeinsight_sdk import CodeInsightClient -from codeinsight_sdk.exceptions import CodeInsightError logger = logging.getLogger(__name__) @@ -13,14 +12,15 @@ TEST_URL = "https://api.revenera.com" TEST_API_TOKEN = "your_api_token" + class TestExperimental: @pytest.fixture def client(self): return CodeInsightClient(TEST_URL, TEST_API_TOKEN, experimental=True) - + def test_experimental_enabled(self, client): assert client.experimental_enabled == True - + def test_get_project_vulnerabilities(self, client): project_id = 1 total_pages = 4 @@ -86,11 +86,21 @@ def test_get_project_vulnerabilities(self, client): "vulnerabilityCvssV3Severity":"CRITICAL"} ] } """ with requests_mock.Mocker() as m: - m.get(f"{TEST_URL}/codeinsight/api/projects/{project_id}/inventorySummary", - text=fake_response_json, headers=response_header) - m.get(f"{TEST_URL}/codeinsight/api/inventories/12346/vulnerabilities", - text=mock_resp_vuln, headers=response_header) - vulnerable_items = client.experimental.get_project_vulnerabilities(project_id) + m.get( + f"{TEST_URL}/codeinsight/api/projects/{project_id}/inventorySummary", + text=fake_response_json, + headers=response_header, + ) + m.get( + f"{TEST_URL}/codeinsight/api/inventories/12346/vulnerabilities", + text=mock_resp_vuln, + headers=response_header, + ) + vulnerable_items = client.experimental.get_project_vulnerabilities( + project_id + ) assert len(vulnerable_items) > 0 assert vulnerable_items[0].vulnerabilities is not None - assert vulnerable_items[0].vulnerabilities[1].vulnerabilityName == "CVE-987-65432" + assert ( + vulnerable_items[0].vulnerabilities[1].vulnerabilityName == "CVE-987-65432" + ) diff --git a/tests/test_handlers.py b/tests/test_handlers.py index 6bc3bdd..b37081a 100644 --- a/tests/test_handlers.py +++ b/tests/test_handlers.py @@ -1,11 +1,9 @@ import pytest from codeinsight_sdk import CodeInsightClient -from codeinsight_sdk.handlers import ProjectHandler, ReportHandler -from codeinsight_sdk.models import Project, Report class TestHandlers(object): @pytest.fixture def client(self): - return CodeInsightClient("","") + return CodeInsightClient("", "")