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,10 @@ 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 (payload_builders [taskdesc ["worker" ]["implementation" ]].schema ,
94+ taskdesc ["worker" ], "validation error" )
9095
9196
9297def assert_generic_worker (task ):
@@ -123,7 +128,7 @@ def assert_generic_worker(task):
123128 "content" : {
124129 "artifact" : "public/run-task" ,
125130 "sha256" : "581ca6876fac84fa2dd8e8c2c18677d790890e9675229fd34c912c937ae19fae" ,
126- "taskId " : {"task-reference" : "<decision>" },
131+ "task-id " : {"task-reference" : "<decision>" },
127132 },
128133 "file" : "./run-task" ,
129134 },
@@ -132,6 +137,10 @@ def assert_generic_worker(task):
132137 },
133138 "worker-type" : "b-win2012" ,
134139 }
140+ taskdesc = next (set_defaults ({}, [task ]))
141+ taskdesc ["worker" ]["max-run-time" ] = 0
142+ validate_schema (payload_builders [taskdesc ["worker" ]["implementation" ]].schema ,
143+ taskdesc ["worker" ], "validation error" )
135144
136145
137146def assert_exec_with (task ):
@@ -177,7 +186,7 @@ def assert_run_task_command_generic_worker(task):
177186 "task" ,
178187 (
179188 pytest .param (
180- {"worker" : {"os" : "linux" }},
189+ {"worker" : {"os" : "linux" , "docker-image" : { "in-tree" : "image" } }},
181190 id = "docker_worker" ,
182191 ),
183192 pytest .param (
0 commit comments