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 @@ -506,6 +506,21 @@ def test_create_entrypoint(
506
506
queue_ids = queue_ids ,
507
507
)
508
508
509
+ # Testing that queue_ids feild can be excluded
510
+ entrypoint_response = actions .register_entrypoint (
511
+ client ,
512
+ name = "queues_not_included" ,
513
+ description = description ,
514
+ group_id = group_id ,
515
+ task_graph = task_graph ,
516
+ parameters = parameters ,
517
+ plugin_ids = plugin_ids ,
518
+ )
519
+ entrypoint_expected = entrypoint_response .get_json ()
520
+ assert_retrieving_entrypoint_by_id_works (
521
+ client , entrypoint_id = entrypoint_expected ["id" ], expected = entrypoint_expected
522
+ )
523
+
509
524
510
525
def test_entrypoint_get_all (
511
526
client : FlaskClient ,
You can’t perform that action at this time.
0 commit comments