Merge pull request #111 from zkarpinski/feat/full-mocking-support-7041241834180989467

feat: Add full mocking support and missing endpoints

Author: zkarpinski

codeinsight_sdk/__init__.py

@@ -16,3 +16,9 @@
 """
 
 from .client import CodeInsightClient
+
+# Optional mocking support
+try:
+    from .testing import CodeInsightMockAdapter, mock_client, codeinsight_mock
+except ImportError:
+    pass

codeinsight_sdk/client.py

@@ -92,31 +92,56 @@ def experimental(self) -> ExperimentalHandler:
             raise CodeInsightError("Experimental API is not enabled for this instance")
         return ExperimentalHandler(self)
 
-    # Coming soon...?
+    @property
+    def vulnerabilities(self):
+        from .handlers import VulnerabilityHandler
 
-    def vulnerabilites(self):
-        raise NotImplementedError
+        return VulnerabilityHandler(self)
 
+    @property
     def users(self):
-        raise NotImplementedError
+        from .handlers import UserHandler
+
+        return UserHandler(self)
 
+    @property
     def licenses(self):
-        raise NotImplementedError
+        from .handlers import LicenseHandler
+
+        return LicenseHandler(self)
 
+    @property
     def tasks(self):
-        raise NotImplementedError
+        from .handlers import TaskHandler
+
+        return TaskHandler(self)
 
+    @property
     def rules(self):
-        raise NotImplementedError
+        from .handlers import RuleHandler
+
+        return RuleHandler(self)
 
+    @property
     def files(self):
-        raise NotImplementedError
+        from .handlers import FileHandler
+
+        return FileHandler(self)
 
+    @property
     def folders(self):
-        raise NotImplementedError
+        from .handlers import FolderHandler
+
+        return FolderHandler(self)
 
+    @property
     def jobs(self):
-        raise NotImplementedError
+        from .handlers import JobHandler
+
+        return JobHandler(self)
 
+    @property
     def components(self):
-        raise NotImplementedError
+        from .handlers import ComponentHandler
+
+        return ComponentHandler(self)

codeinsight_sdk/handlers/__init__.py

@@ -1,3 +1,12 @@
 from .inventory import InventoryHandler
 from .project import ProjectHandler
 from .report import ReportHandler
+from .vulnerability import VulnerabilityHandler
+from .user import UserHandler
+from .license import LicenseHandler
+from .task import TaskHandler
+from .rule import RuleHandler
+from .file import FileHandler
+from .folder import FolderHandler
+from .job import JobHandler
+from .component import ComponentHandler

codeinsight_sdk/handlers/component.py

@@ -0,0 +1,21 @@
+from typing import List
+from ..handler import Handler
+from ..models import Component
+
+
+class ComponentHandler(Handler):
+    def all(self) -> List[Component]:
+        """
+        Retrieves all components.
+        """
+        path = "components"
+        resp = self.client.request("GET", url_part=path)
+        return [Component.from_dict(component) for component in resp.json()["data"]]
+
+    def get(self, id: int) -> Component:
+        """
+        Retrieves a component by its ID.
+        """
+        path = f"components/{id}"
+        resp = self.client.request("GET", url_part=path)
+        return Component.from_dict(resp.json()["data"])

codeinsight_sdk/handlers/file.py

@@ -0,0 +1,13 @@
+from typing import List
+from ..handler import Handler
+from ..models import File
+
+
+class FileHandler(Handler):
+    def get(self, id: int) -> File:
+        """
+        Retrieves a file by its ID.
+        """
+        path = f"files/{id}"
+        resp = self.client.request("GET", url_part=path)
+        return File.from_dict(resp.json()["data"])

codeinsight_sdk/handlers/folder.py

@@ -0,0 +1,13 @@
+from typing import List
+from ..handler import Handler
+from ..models import Folder
+
+
+class FolderHandler(Handler):
+    def get(self, id: int) -> Folder:
+        """
+        Retrieves a folder by its ID.
+        """
+        path = f"folders/{id}"
+        resp = self.client.request("GET", url_part=path)
+        return Folder.from_dict(resp.json()["data"])

codeinsight_sdk/handlers/job.py

@@ -0,0 +1,21 @@
+from typing import List
+from ..handler import Handler
+from ..models import Job
+
+
+class JobHandler(Handler):
+    def all(self) -> List[Job]:
+        """
+        Retrieves all jobs.
+        """
+        path = "jobs"
+        resp = self.client.request("GET", url_part=path)
+        return [Job.from_dict(job) for job in resp.json()["data"]]
+
+    def get(self, id: int) -> Job:
+        """
+        Retrieves a job by its ID.
+        """
+        path = f"jobs/{id}"
+        resp = self.client.request("GET", url_part=path)
+        return Job.from_dict(resp.json()["data"])

codeinsight_sdk/handlers/license.py

@@ -0,0 +1,21 @@
+from typing import List
+from ..handler import Handler
+from ..models import License
+
+
+class LicenseHandler(Handler):
+    def all(self) -> List[License]:
+        """
+        Retrieves all licenses.
+        """
+        path = "licenses"
+        resp = self.client.request("GET", url_part=path)
+        return [License.from_dict(license) for license in resp.json()["data"]]
+
+    def get(self, id: int) -> License:
+        """
+        Retrieves a license by its ID.
+        """
+        path = f"licenses/{id}"
+        resp = self.client.request("GET", url_part=path)
+        return License.from_dict(resp.json()["data"])

codeinsight_sdk/handlers/rule.py

@@ -0,0 +1,21 @@
+from typing import List
+from ..handler import Handler
+from ..models import Rule
+
+
+class RuleHandler(Handler):
+    def all(self) -> List[Rule]:
+        """
+        Retrieves all rules.
+        """
+        path = "rules"
+        resp = self.client.request("GET", url_part=path)
+        return [Rule.from_dict(rule) for rule in resp.json()["data"]]
+
+    def get(self, id: int) -> Rule:
+        """
+        Retrieves a rule by its ID.
+        """
+        path = f"rules/{id}"
+        resp = self.client.request("GET", url_part=path)
+        return Rule.from_dict(resp.json()["data"])

codeinsight_sdk/handlers/task.py

@@ -0,0 +1,21 @@
+from typing import List
+from ..handler import Handler
+from ..models import Task
+
+
+class TaskHandler(Handler):
+    def all(self) -> List[Task]:
+        """
+        Retrieves all tasks.
+        """
+        path = "tasks"
+        resp = self.client.request("GET", url_part=path)
+        return [Task.from_dict(task) for task in resp.json()["data"]]
+
+    def get(self, id: int) -> Task:
+        """
+        Retrieves a task by its ID.
+        """
+        path = f"tasks/{id}"
+        resp = self.client.request("GET", url_part=path)
+        return Task.from_dict(resp.json()["data"])