File tree 2 files changed +19
-2
lines changed
2 files changed +19
-2
lines changed Original file line number Diff line number Diff line change @@ -66,7 +66,7 @@ def register_entrypoint(
66
66
task_graph : str ,
67
67
parameters : list [dict [str , Any ]],
68
68
plugin_ids : list [int ],
69
- queue_ids : list [int ],
69
+ queue_ids : list [int ] | None = None ,
70
70
) -> TestResponse :
71
71
"""Register an entrypoint using the API.
72
72
@@ -87,9 +87,11 @@ def register_entrypoint(
87
87
"taskGraph" : task_graph ,
88
88
"parameters" : parameters ,
89
89
"plugins" : plugin_ids ,
90
- "queues" : queue_ids ,
91
90
}
92
91
92
+ if queue_ids :
93
+ payload ["queues" ] = queue_ids
94
+
93
95
if description :
94
96
payload ["description" ] = description
95
97
Original file line number Diff line number Diff line change @@ -549,6 +549,21 @@ def test_create_entrypoint(
549
549
queue_ids = queue_ids ,
550
550
)
551
551
552
+ # Testing that queue_ids feild can be excluded
553
+ entrypoint_response = actions .register_entrypoint (
554
+ client ,
555
+ name = "queues_not_included" ,
556
+ description = description ,
557
+ group_id = group_id ,
558
+ task_graph = task_graph ,
559
+ parameters = parameters ,
560
+ plugin_ids = plugin_ids ,
561
+ )
562
+ entrypoint_expected = entrypoint_response .get_json ()
563
+ assert_retrieving_entrypoint_by_id_works (
564
+ client , entrypoint_id = entrypoint_expected ["id" ], expected = entrypoint_expected
565
+ )
566
+
552
567
553
568
def test_entrypoint_get_all (
554
569
dioptra_client : DioptraClient [DioptraResponseProtocol ],
You can’t perform that action at this time.
0 commit comments