MCPcopy Create free account
hub / github.com/QuantumNous/new-api / TestFinishSystemTaskRetainsExecutor

Function TestFinishSystemTaskRetainsExecutor

model/system_task_test.go:286–308  ·  view source on GitHub ↗
(t *testing.T)

Source from the content-addressed store, hash-verified

284}
285
286func TestFinishSystemTaskRetainsExecutor(t *testing.T) {
287 truncateTables(t)
288
289 task, err := CreateSystemTask(SystemTaskTypeLogCleanup, nil, nil)
290 require.NoError(t, err)
291
292 runnerID := "node-1-abc123"
293 _, claimed, err := ClaimSystemTask(task.ID, SystemTaskTypeLogCleanup, runnerID, common.GetTimestamp()+60)
294 require.NoError(t, err)
295 require.True(t, claimed)
296
297 require.NoError(t, FinishSystemTask(task.TaskID, runnerID, SystemTaskStatusSucceeded, nil, ""))
298
299 reloaded, err := GetSystemTaskByTaskID(task.TaskID)
300 require.NoError(t, err)
301 require.NotNil(t, reloaded)
302 assert.Equal(t, SystemTaskStatusSucceeded, reloaded.Status)
303 assert.Equal(t, runnerID, reloaded.LockedBy, "executor-of-record must be retained for history")
304
305 var lockCount int64
306 require.NoError(t, DB.Model(&SystemTaskLock{}).Where("task_id = ?", task.TaskID).Count(&lockCount).Error)
307 assert.Equal(t, int64(0), lockCount)
308}
309
310func TestSystemTaskUpdatesRequireCurrentLock(t *testing.T) {
311 truncateTables(t)

Callers

nothing calls this directly

Calls 8

GetTimestampFunction · 0.92
truncateTablesFunction · 0.85
CreateSystemTaskFunction · 0.85
ClaimSystemTaskFunction · 0.85
FinishSystemTaskFunction · 0.85
GetSystemTaskByTaskIDFunction · 0.85
EqualMethod · 0.80
CountMethod · 0.80

Tested by

no test coverage detected