Fixed some of the operations payloads, Updated integration test

Author: jexp

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

@@ -190,10 +190,10 @@ def create_instance(self, tenant_id: str, name: str, memory: int = 1, region: st
 
         # Add optional parameters only if they're provided and applicable            
         if graph_analytics_plugin and type in ["professional-db", "enterprise-db", "business-critical"]:
-            payload["graph_analytics_plugin"] = graph_analytics_plugin
+            payload["graph_analytics_plugin"] = lower(str(graph_analytics_plugin))
 
         if vector_optimized and type in ["professional-db", "enterprise-db", "business-critical"]:
-            payload["vector_optimized"] = vector_optimized
+            payload["vector_optimized"] = lower(str(vector_optimized))
 
         if source_instance_id and type in ["professional-db", "enterprise-db", "business-critical"]:
             payload["source_instance_id"] = source_instance_id
@@ -212,11 +212,15 @@ def update_instance(self, instance_id: str, name: Optional[str] = None,
         if name is not None:
             payload["name"] = name
         if memory is not None:
-            payload["memory"] = memory
+            payload["memory"] = f"{memory}GB"
         if vector_optimized is not None:
-            payload["vector_optimized"] = vector_optimized
+            payload["vector_optimized"] = lower(str(vector_optimized))
 
+        print("Update instance payload:")
+        print(payload)
         response = requests.patch(url, headers=self._get_headers(), json=payload)
+        print("Update instance response: "+str(response.status_code))
+        print(response.json())
         return self._handle_response(response)
 
     def pause_instance(self, instance_id: str) -> Dict[str, Any]:

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

@@ -125,7 +125,7 @@ def test_integration_flow(aura_client, test_type):
     # Only run this if explicitly enabled and you understand the implications
     if test_type == "create_instance" and os.environ.get("ENABLE_INSTANCE_CREATION") == "true":
         # Create a test instance
-        test_instance_name = f"pytest-integration-{os.urandom(4).hex()}"
+        test_instance_name = f"Pro Test Instance {uuid.uuid4().hex[:8]}"
 
         try:
             # Create a small instance for testing
@@ -143,26 +143,40 @@ def test_integration_flow(aura_client, test_type):
             assert instance["name"] == test_instance_name
 
             # Update the instance name
-            updated_name = f"{test_instance_name}-updated"
+            print("Updating instance name")
+            updated_name = f"{test_instance_name}-U"
             updated = aura_client.update_instance(instance_id=instance_id, name=updated_name)
             assert updated["name"] == updated_name
 
+            print("Getting instance details")
             instance_details = aura_client.get_instance_details([instance_id])[0]
             assert instance_details["name"] == updated_name
+            
+            instance_details = wait_for_instance_status(aura_client, instance_id,"running")
+            assert instance_details["status"] == "running"
+
             # Pause the instance
+            print("Pausing instance")
             paused = aura_client.pause_instance(instance_id)
             assert paused["status"] in ["paused", "pausing"]
 
+            print("Waiting for instance to be paused")
             instance_details = wait_for_instance_status(aura_client, instance_id,"paused")
             assert instance_details["status"] == "paused"
 
-            # Resume the instance
+            print("Resuming instance")
             resumed = aura_client.resume_instance(instance_id)
-            assert resumed["status"] in ["running", "starting"]
+            assert resumed["status"] in ["resuming", "running"]
 
+            print("Waiting for instance to be running")
             instance_details = wait_for_instance_status(aura_client, instance_id,"running")
             assert instance_details["status"] == "running"
-            
+
+            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["memory"] == "2GB"
+
         except Exception as e:
             logger.error(f"Error during instance creation test: {str(e)}")
             raise