New create_instance

Author: NickCao

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

@@ -82,26 +82,22 @@ async def get_device(self, model: str) -> Optional[Device]:
 
         return None
 
-    def create_instance(self, name: str, project: Project, device: Device, os_version: str, os_build: str) -> Instance:
+    async def create_instance(
+        self, name: str, project: Project, device: Device, os_version: str, os_build: str
+    ) -> Instance:
         """
         Create a new virtual instance from a device spec.
         """
-        data = {
-            "name": name,
-            "project": project.id,
-            "flavor": device.flavor,
-            "os": os_version,
-            "osbuild": os_build,
-        }
 
-        try:
-            res = self.req.post(f"{self.baseurl}/v1/instances", json=data)
-            data = res.json()
-            res.raise_for_status()
-        except requests.exceptions.RequestException as e:
-            raise CorelliumApiException(data.get("error", str(e))) from e
-
-        return Instance(**data)
+        return await self.api.v1_create_instance(
+            corellium_api.InstanceCreateOptions(
+                name=name,
+                project=project.id,
+                flavor=device.flavor,
+                os=os_version,
+                osbuild=os_build,
+            )
+        )
 
     async def get_instance(self, instance_ref: str) -> Optional[Instance]:
         """

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

@@ -174,7 +174,7 @@ async def on(self) -> None:
                 opts["os_version"] = self.parent.device_os
             if self.parent.device_build:
                 opts["os_build"] = self.parent.device_build
-            instance = self.parent.api.create_instance(self.parent.device_name, project, device, **opts)
+            instance = await self.parent.api.create_instance(self.parent.device_name, project, device, **opts)
         self.logger.info(f"Instance: {self.parent.device_name} (ID: {instance.id})")
 
         await self.wait_instance(instance, Instance(id=instance.id, state="on"))