Use data models from official sdk

Author: NickCao

packages/jumpstarter-driver-corellium/fixtures/http/get-models-200.json

@@ -5,10 +5,6 @@
         "flavor": "rpi4b",
         "description": "Raspberry Pi 4",
         "model": "rpi4b",
-        "peripherals": false,
-        "quotas": {
-            "cores": 4,
-            "cpus": 4
-        }
+        "peripherals": false
     }
 ]

packages/jumpstarter-driver-corellium/jumpstarter_driver_corellium/corellium/api_test.py

@@ -133,7 +133,12 @@ def test_create_instance_ok(requests_mock):
 
     project = Project("d59db33d-27bd-4b22-878d-49e4758a648e", "Default Project")
     device = Device(
-        name="rd1ae", type="automotive", flavor="kronos", description="", model="kronos", peripherals=False, quotas={}
+        name="rd1ae",
+        type="automotive",
+        flavor="kronos",
+        description="",
+        model="kronos",
+        peripherals=False,
     )
     instance = api.create_instance("my-instance", project, device, "1.1.1", "Critical Application Monitor (Baremetal)")
 
@@ -162,7 +167,6 @@ def test_create_instance_error(requests_mock, status_code, data, msg):
             description="",
             model="kronos",
             peripherals=False,
-            quotas={},
         )
         api.create_instance("my-instance", project, device, "1.1.1", "Critical Application Monitor (Baremetal)")
 

packages/jumpstarter-driver-corellium/jumpstarter_driver_corellium/corellium/types.py

@@ -2,8 +2,14 @@
 Corellium API types.
 """
 
-from dataclasses import dataclass, field
-from typing import Dict, Optional
+from dataclasses import dataclass
+from typing import Dict
+
+from corellium_api import Instance, Model, Project
+
+__all__ = ["Project", "Device", "Instance"]
+
+Device = Model
 
 
 @dataclass
@@ -20,40 +26,3 @@ def as_header(self) -> Dict[str, str]:
         Return a dict to be used as HTTP header for authenticated requests.
         """
         return {"Authorization": f"Bearer {self.token}"}
-
-
-@dataclass
-class Project:
-    """
-    Dataclass that represents a Corellium project.
-    """
-
-    id: str
-    name: str
-
-
-@dataclass
-class Device:
-    """
-    Dataclass to represent a Corellium Device.
-
-    A device object is used to create virtual instances.
-    """
-
-    name: str
-    type: str
-    flavor: str
-    description: str
-    model: str
-    peripherals: bool
-    quotas: dict
-
-
-@dataclass
-class Instance:
-    """
-    Virtual instance dataclass.
-    """
-
-    id: str
-    state: Optional[str] = field(default=None)

packages/jumpstarter-driver-corellium/jumpstarter_driver_corellium/driver_test.py

@@ -71,7 +71,12 @@ def test_driver_power_on_ok(monkeypatch):
 
     project = Project("1", "Default Project")
     device = Device(
-        name="rd1ae", type="automotive", flavor="kronos", description="", model="kronos", peripherals=False, quotas={}
+        name="rd1ae",
+        type="automotive",
+        flavor="kronos",
+        description="",
+        model="kronos",
+        peripherals=False,
     )
     instance = Instance(id="7f4f241c-821f-4219-905f-c3b50b0db5dd", state="on")
     root = Corellium(project_id="1", device_name="jmp", device_flavor="kronos", device_os="1.0")