V1 fix entrypoint schema

src/dioptra/restapi/v1/entrypoints/schema.py

@@ -151,8 +151,10 @@ class EntrypointMutableFieldsSchema(Schema):
         fields.Integer(),
         attribute="queue_ids",
         data_key="queues",
+        allow_none=True,
         metadata=dict(description="The queue for the entrypoint."),
         load_only=True,
+        load_default=list(),
     )
 
 

tests/unit/restapi/lib/actions.py

@@ -66,7 +66,7 @@ def register_entrypoint(
     task_graph: str,
     parameters: list[dict[str, Any]],
     plugin_ids: list[int],
-    queue_ids: list[int],
+    queue_ids: list[int] | None = None,
 ) -> TestResponse:
     """Register an entrypoint using the API.
 
@@ -87,9 +87,11 @@ def register_entrypoint(
         "taskGraph": task_graph,
         "parameters": parameters,
         "plugins": plugin_ids,
-        "queues": queue_ids,
     }
 
+    if queue_ids:
+        payload["queues"] = queue_ids
+
     if description:
         payload["description"] = description
 

tests/unit/restapi/v1/test_entrypoint.py

@@ -506,6 +506,21 @@ def test_create_entrypoint(
         queue_ids=queue_ids,
     )
 
+    # Testing that queue_ids feild can be excluded
+    entrypoint_response = actions.register_entrypoint(
+        client,
+        name="queues_not_included",
+        description=description,
+        group_id=group_id,
+        task_graph=task_graph,
+        parameters=parameters,
+        plugin_ids=plugin_ids,
+    )
+    entrypoint_expected = entrypoint_response.get_json()
+    assert_retrieving_entrypoint_by_id_works(
+        client, entrypoint_id=entrypoint_expected["id"], expected=entrypoint_expected
+    )
+
 
 def test_entrypoint_get_all(
     client: FlaskClient,