From 684cce198fdff0e876487102bf11ecbea0035ce6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ra=C3=BAl=20Kripalani?= Date: Mon, 21 Jun 2021 20:49:24 +0100 Subject: [PATCH] add a unit test. --- extern/sector-storage/sched_test.go | 17 +++++++---------- 1 file changed, 7 insertions(+), 10 deletions(-) diff --git a/extern/sector-storage/sched_test.go b/extern/sector-storage/sched_test.go index 38b3efda9..fbc4d83ee 100644 --- a/extern/sector-storage/sched_test.go +++ b/extern/sector-storage/sched_test.go @@ -338,18 +338,15 @@ func TestSched(t *testing.T) { } } - t.Run("constrained-resources-not-scheduled", testFunc([]workerSpec{ - {name: "fred", resources: constrainedWorkerResources, taskTypes: map[sealtasks.TaskType]struct{}{sealtasks.TTPreCommit1: {}}}, + // checks behaviour with workers with constrained resources + // the first one is not ignoring resource constraints, so we assign to the second worker, who is + t.Run("constrained-resources", testFunc([]workerSpec{ + {name: "fred1", resources: constrainedWorkerResources, taskTypes: map[sealtasks.TaskType]struct{}{sealtasks.TTPreCommit1: {}}}, + {name: "fred2", resources: constrainedWorkerResources, ignoreResources: true, taskTypes: map[sealtasks.TaskType]struct{}{sealtasks.TTPreCommit1: {}}}, }, []task{ - sched("pc1-1", "fred", 8, sealtasks.TTPreCommit1), - taskNotScheduled("pc1-1"), - })) - - t.Run("constrained-resources-ignored-scheduled", testFunc([]workerSpec{ - {name: "fred", resources: constrainedWorkerResources, ignoreResources: true, taskTypes: map[sealtasks.TaskType]struct{}{sealtasks.TTPreCommit1: {}}}, - }, []task{ - sched("pc1-1", "fred", 8, sealtasks.TTPreCommit1), + sched("pc1-1", "fred2", 8, sealtasks.TTPreCommit1), taskStarted("pc1-1"), + taskDone("pc1-1"), })) t.Run("one-pc1", testFunc([]workerSpec{