@@ -2186,6 +2186,7 @@ func TestGetProvisionerJobsByIDsWithQueuePosition(t *testing.T) {
21862186// but this should not affect the queue position or queue size of other jobs.
21872187skipJobIDs map [int ]struct {}
21882188}{
2189+ // Baseline test case
21892190{
21902191name :"test-case-1" ,
21912192jobTags : []database.StringMap {
@@ -2200,7 +2201,7 @@ func TestGetProvisionerJobsByIDsWithQueuePosition(t *testing.T) {
22002201queueSizes : []int64 {2 ,2 ,0 },
22012202queuePositions : []int64 {1 ,1 ,0 },
22022203},
2203- //Similar to the previous case, but includes an additional provisioner.
2204+ //Includes an additional provisioner
22042205{
22052206name :"test-case-2" ,
22062207jobTags : []database.StringMap {
@@ -2216,7 +2217,7 @@ func TestGetProvisionerJobsByIDsWithQueuePosition(t *testing.T) {
22162217queueSizes : []int64 {3 ,3 ,3 },
22172218queuePositions : []int64 {1 ,1 ,3 },
22182219},
2219- //Similar to the previous case, but skips job at index 0
2220+ //Skips job at index 0
22202221{
22212222name :"test-case-3" ,
22222223jobTags : []database.StringMap {
@@ -2293,6 +2294,46 @@ func TestGetProvisionerJobsByIDsWithQueuePosition(t *testing.T) {
229322942 : {},
22942295},
22952296},
2297+ // Includes two additional jobs that any provisioner can execute.
2298+ {
2299+ name :"test-case-7" ,
2300+ jobTags : []database.StringMap {
2301+ {},
2302+ {},
2303+ {"a" :"1" ,"b" :"2" },
2304+ {"a" :"1" },
2305+ {"a" :"1" ,"c" :"3" },
2306+ },
2307+ daemonTags : []database.StringMap {
2308+ {"a" :"1" ,"b" :"2" },
2309+ {"a" :"1" },
2310+ {"a" :"1" ,"b" :"2" ,"c" :"3" },
2311+ },
2312+ queueSizes : []int64 {5 ,5 ,5 ,5 ,5 },
2313+ queuePositions : []int64 {1 ,2 ,3 ,3 ,5 },
2314+ },
2315+ // Includes two additional jobs that any provisioner can execute, but they are intentionally skipped.
2316+ {
2317+ name :"test-case-8" ,
2318+ jobTags : []database.StringMap {
2319+ {},
2320+ {},
2321+ {"a" :"1" ,"b" :"2" },
2322+ {"a" :"1" },
2323+ {"a" :"1" ,"c" :"3" },
2324+ },
2325+ daemonTags : []database.StringMap {
2326+ {"a" :"1" ,"b" :"2" },
2327+ {"a" :"1" },
2328+ {"a" :"1" ,"b" :"2" ,"c" :"3" },
2329+ },
2330+ queueSizes : []int64 {5 ,5 ,5 },
2331+ queuePositions : []int64 {3 ,3 ,5 },
2332+ skipJobIDs :map [int ]struct {}{
2333+ 0 : {},
2334+ 1 : {},
2335+ },
2336+ },
22962337}
22972338
22982339for _ ,tc := range testCases {