fix new requirement of providing both memory and storage when updating memory

Author: jexp

servers/mcp-neo4j-cloud-aura-api/src/mcp_neo4j_aura_manager/server.py

@@ -206,7 +206,8 @@ def create_instance(self, tenant_id: str, name: str, memory: int = 1, region: st
 
     def update_instance(self, instance_id: str, name: Optional[str] = None, 
                         memory: Optional[int] = None, 
-                        vector_optimized: Optional[bool] = None) -> Dict[str, Any]:
+                        vector_optimized: Optional[bool] = None, 
+                        storage: Optional[int] = None) -> Dict[str, Any]:
         """Update an existing instance."""
         url = f"{self.BASE_URL}/instances/{instance_id}"
 
@@ -215,6 +216,9 @@ def update_instance(self, instance_id: str, name: Optional[str] = None,
             payload["name"] = name
         if memory is not None:
             payload["memory"] = f"{memory}GB"
+            payload["storage"] = f"{2*memory}GB"
+        if storage is not None:
+            payload["storage"] = f"{storage}GB"
         if vector_optimized is not None:
             payload["vector_optimized"] = str(vector_optimized).lower()
 

servers/mcp-neo4j-cloud-aura-api/tests/test_aura_integration.py

@@ -175,6 +175,7 @@ def test_integration_flow(aura_client, test_type):
             print("Updating instance memory")
             updated = aura_client.update_instance(instance_id=instance_id, memory=2)
             instance_details = wait_for_instance_status(aura_client, instance_id,"running")
+            assert instance_details["status"] == "running"
             assert instance_details["memory"] == "2GB"
 
         except Exception as e: