Add optional entry_point parameter

Author: peterschrammel

src/mcp_diffblue_server.py

@@ -25,6 +25,10 @@ def __init__(self):
                         "working_directory": {
                             "type": "string",
                             "description": "Working directory for running dcover create (optional)"
+                        },
+                        "entry_point": {
+                            "type": "string",
+                            "description": "Fully qualified class name to test (optional)"
                         }
                     }
                 }
@@ -83,7 +87,10 @@ def handle_tools_call(self, request: Dict[str, Any]) -> Dict[str, Any]:
             }
 
         if tool_name == "dcover_create":
-            result = self.run_dcover_create(tool_args.get("working_directory"))
+            result = self.run_dcover_create(
+                tool_args.get("working_directory"),
+                tool_args.get("entry_point")
+            )
             return {
                 "jsonrpc": "2.0",
                 "id": request.get("id"),
@@ -97,14 +104,18 @@ def handle_tools_call(self, request: Dict[str, Any]) -> Dict[str, Any]:
                 }
             }
 
-    def run_dcover_create(self, working_dir: str = None) -> Dict[str, Any]:
+    def run_dcover_create(self, working_dir: str = None, entry_point: str = None) -> Dict[str, Any]:
         """
         Executes 'dcover create' in the specified working directory.
         """
         if not working_dir:
             working_dir = os.getcwd()
 
         command = ["dcover", "create"]
+
+        # Add entry point if provided
+        if entry_point:
+            command.append(entry_point)
 
         try:
             result = subprocess.run(
@@ -210,4 +221,4 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
+    main()