From 2d05e1c773492f0d7778973be9e43a6deb8c8be4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sybren=20A=2E=20St=C3=BCvel?= Date: Mon, 20 Jun 2022 16:05:36 +0200 Subject: [PATCH] Fix unit test for recent scheduler change Fix unit test for rF1586c37b. --- internal/manager/api_impl/workers_test.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/internal/manager/api_impl/workers_test.go b/internal/manager/api_impl/workers_test.go index 680c3500..5294756b 100644 --- a/internal/manager/api_impl/workers_test.go +++ b/internal/manager/api_impl/workers_test.go @@ -42,6 +42,8 @@ func TestTaskScheduleHappy(t *testing.T) { mf.logStorage.EXPECT().WriteTimestamped(gomock.Any(), job.UUID, task.UUID, "Task assigned to worker дрон (e7632d62-c3b8-4af0-9e78-01752928952c)") + mf.stateMachine.EXPECT().TaskStatusChange(gomock.Any(), &task, api.TaskStatusActive) + err := mf.flamenco.ScheduleTask(echo) assert.NoError(t, err)