88import pytest
99
1010from taskgraph .transforms .run import make_task_description
11- from taskgraph .transforms .task import payload_builders
11+ from taskgraph .transforms .task import payload_builders , set_defaults
1212from taskgraph .util .caches import CACHES
1313from taskgraph .util .schema import Schema , validate_schema
1414from taskgraph .util .templates import merge
@@ -70,6 +70,7 @@ def assert_docker_worker(task):
7070 "-cx" ,
7171 "echo hello world" ,
7272 ],
73+ "docker-image" : {"in-tree" : "image" },
7374 "env" : {
7475 "CI_BASE_REPOSITORY" : "http://hg.example.com" ,
7576 "CI_HEAD_REF" : "default" ,
@@ -87,6 +88,13 @@ def assert_docker_worker(task):
8788 },
8889 "worker-type" : "t-linux" ,
8990 }
91+ taskdesc = next (set_defaults ({}, [task ]))
92+ taskdesc ["worker" ]["max-run-time" ] = 0
93+ validate_schema (
94+ payload_builders [taskdesc ["worker" ]["implementation" ]].schema ,
95+ taskdesc ["worker" ],
96+ "validation error" ,
97+ )
9098
9199
92100def assert_generic_worker (task ):
@@ -123,7 +131,7 @@ def assert_generic_worker(task):
123131 "content" : {
124132 "artifact" : "public/run-task" ,
125133 "sha256" : "581ca6876fac84fa2dd8e8c2c18677d790890e9675229fd34c912c937ae19fae" ,
126- "taskId " : {"task-reference" : "<decision>" },
134+ "task-id " : {"task-reference" : "<decision>" },
127135 },
128136 "file" : "./run-task" ,
129137 },
@@ -132,6 +140,13 @@ def assert_generic_worker(task):
132140 },
133141 "worker-type" : "b-win2012" ,
134142 }
143+ taskdesc = next (set_defaults ({}, [task ]))
144+ taskdesc ["worker" ]["max-run-time" ] = 0
145+ validate_schema (
146+ payload_builders [taskdesc ["worker" ]["implementation" ]].schema ,
147+ taskdesc ["worker" ],
148+ "validation error" ,
149+ )
135150
136151
137152def assert_exec_with (task ):
@@ -177,7 +192,7 @@ def assert_run_task_command_generic_worker(task):
177192 "task" ,
178193 (
179194 pytest .param (
180- {"worker" : {"os" : "linux" }},
195+ {"worker" : {"os" : "linux" , "docker-image" : { "in-tree" : "image" } }},
181196 id = "docker_worker" ,
182197 ),
183198 pytest .param (
0 commit comments