Fix: Solved not passing tests and returned value as integer instead of decimal.

Author: nesitor

src/aleph/vm/orchestrator/utils.py

@@ -109,7 +109,7 @@ def get_compatible_gpus() -> list[Any]:
     return LAST_AGGREGATE_SETTINGS["compatible_gpus"]
 
 
-def get_execution_disk_size(message: InstanceContent | ProgramContent) -> Decimal:
+def get_execution_disk_size(message: InstanceContent | ProgramContent) -> int:
     disk_size_mib = 0
 
     # For Programs the disk size depends on the runtime
@@ -123,4 +123,4 @@ def get_execution_disk_size(message: InstanceContent | ProgramContent) -> Decima
         if getattr(volume, "size_mib", None):
             disk_size_mib += volume.size_mib
 
-    return Decimal(disk_size_mib)
+    return disk_size_mib

tests/supervisor/test_views.py

@@ -389,6 +389,11 @@ async def test_v2_executions_list_one_vm(aiohttp_client, mock_app_with_pool, moc
     assert await response.json() == {
         "decadecadecadecadecadecadecadecadecadecadecadecadecadecadecadeca": {
             "networking": {},
+            "resources": {
+                "vcpus": 1,
+                "memory": 256,
+                "disk_mib": 1000,
+            },
             "status": {
                 "defined_at": str(execution.times.defined_at),
                 "preparing_at": None,
@@ -464,6 +469,11 @@ async def test_v2_executions_list_vm_network(aiohttp_client, mocker, mock_app_wi
                 "ipv6_ip": "fc00:1:2:3:3:deca:deca:dec1",
                 "mapped_ports": {},
             },
+            "resources": {
+                "vcpus": 1,
+                "memory": 256,
+                "disk_mib": 1000,
+            },
             "status": {
                 "defined_at": str(execution.times.defined_at),
                 "preparing_at": None,