75903a2da3
Add a new `clock` service to the Flamenco struct, which allows us to mock the passing of time, and thus test for timestamps in a stable fashion.
82 lines
2.4 KiB
Go
82 lines
2.4 KiB
Go
// SPDX-License-Identifier: GPL-3.0-or-later
|
|
package api_impl
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
"time"
|
|
|
|
"git.blender.org/flamenco/internal/manager/persistence"
|
|
"git.blender.org/flamenco/pkg/api"
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/stretchr/testify/assert"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func TestFetchTask(t *testing.T) {
|
|
mockCtrl := gomock.NewController(t)
|
|
defer mockCtrl.Finish()
|
|
|
|
mf := newMockedFlamenco(mockCtrl)
|
|
|
|
taskUUID := "19b62e32-564f-43a3-84fb-06e80ad36f16"
|
|
workerUUID := "b5725bb3-d540-4070-a2b6-7b4b26925f94"
|
|
jobUUID := "8b179118-0189-478a-b463-73798409898c"
|
|
|
|
dbTask := persistence.Task{
|
|
Model: gorm.Model{
|
|
ID: 327,
|
|
CreatedAt: mf.clock.Now().Add(-30 * time.Second),
|
|
UpdatedAt: mf.clock.Now(),
|
|
},
|
|
UUID: taskUUID,
|
|
Name: "симпатичная задача",
|
|
Type: "misc",
|
|
JobID: 0,
|
|
Job: &persistence.Job{UUID: jobUUID},
|
|
Priority: 47,
|
|
Status: api.TaskStatusQueued,
|
|
WorkerID: new(uint),
|
|
Worker: &persistence.Worker{UUID: workerUUID, Name: "Radnik", Address: "Slapić"},
|
|
Dependencies: []*persistence.Task{},
|
|
Activity: "used in unit test",
|
|
|
|
Commands: []persistence.Command{
|
|
{Name: "move-directory",
|
|
Parameters: map[string]interface{}{
|
|
"dest": "/render/_flamenco/tests/renders/2022-04-29 Weekly/2022-04-29_140531",
|
|
"src": "/render/_flamenco/tests/renders/2022-04-29 Weekly/2022-04-29_140531__intermediate-2022-04-29_140531",
|
|
}},
|
|
},
|
|
}
|
|
|
|
expectAPITask := api.Task{
|
|
Activity: "used in unit test",
|
|
Created: dbTask.CreatedAt,
|
|
Id: taskUUID,
|
|
JobId: jobUUID,
|
|
Name: "симпатичная задача",
|
|
Priority: 47,
|
|
Status: api.TaskStatusQueued,
|
|
TaskType: "misc",
|
|
Updated: dbTask.UpdatedAt,
|
|
Worker: &api.TaskWorker{Id: workerUUID, Name: "Radnik", Address: "Slapić"},
|
|
|
|
Commands: []api.Command{
|
|
{Name: "move-directory",
|
|
Parameters: map[string]interface{}{
|
|
"dest": "/render/_flamenco/tests/renders/2022-04-29 Weekly/2022-04-29_140531",
|
|
"src": "/render/_flamenco/tests/renders/2022-04-29 Weekly/2022-04-29_140531__intermediate-2022-04-29_140531",
|
|
}},
|
|
},
|
|
}
|
|
|
|
mf.persistence.EXPECT().FetchTask(gomock.Any(), taskUUID).Return(&dbTask, nil)
|
|
|
|
echoCtx := mf.prepareMockedRequest(nil)
|
|
err := mf.flamenco.FetchTask(echoCtx, taskUUID)
|
|
assert.NoError(t, err)
|
|
|
|
assertResponseJSON(t, echoCtx, http.StatusOK, expectAPITask)
|
|
}
|