blender/flamenco
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix two unit tests for the previous commit

Browse Source 
I pushed too soon :'(
This commit is contained in:
Sybren A. Stüvel 2022-06-16 10:42:04 +02:00
parent be0b10400f
commit 40f711bf69
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
internal/manager/api_impl/workers_test.go
View File

@ -88,11 +88,13 @@ func TestTaskScheduleNonActiveStatus(t *testing.T) {
worker := testWorker() worker := testWorker()
worker.Status = api.WorkerStatusAsleep worker.Status = api.WorkerStatusAsleep
// Explicitly NO expected calls to the persistence layer. Since the worker is
// not in a state that allows task execution, there should be no DB queries.
echoCtx := mf.prepareMockedRequest(nil) echoCtx := mf.prepareMockedRequest(nil)
requestWorkerStore(echoCtx, &worker) requestWorkerStore(echoCtx, &worker)
// The worker should be marked as 'seen', even when it's in a state that
// doesn't allow task execution.
mf.persistence.EXPECT().WorkerSeen(echoCtx.Request().Context(), &worker)
err := mf.flamenco.ScheduleTask(echoCtx) err := mf.flamenco.ScheduleTask(echoCtx)
assert.NoError(t, err) assert.NoError(t, err)
@ -108,11 +110,13 @@ func TestTaskScheduleOtherStatusRequested(t *testing.T) {
worker := testWorker() worker := testWorker()
worker.StatusChangeRequest(api.WorkerStatusAsleep, false) worker.StatusChangeRequest(api.WorkerStatusAsleep, false)
// Explicitly NO expected calls to the persistence layer. Since the worker is
// not in a state that allows task execution, there should be no DB queries.
echoCtx := mf.prepareMockedRequest(nil) echoCtx := mf.prepareMockedRequest(nil)
requestWorkerStore(echoCtx, &worker) requestWorkerStore(echoCtx, &worker)
// The worker should be marked as 'seen', even when it's in a state that
// doesn't allow task execution.
mf.persistence.EXPECT().WorkerSeen(echoCtx.Request().Context(), &worker)
err := mf.flamenco.ScheduleTask(echoCtx) err := mf.flamenco.ScheduleTask(echoCtx)
assert.NoError(t, err) assert.NoError(t, err)