github/woodpecker
1
0
Fork 0
mirror of https://github.com/woodpecker-ci/woodpecker.git synced 2025-10-22 08:49:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Restart tasks on dead agents sooner (#4114)

Browse Source
This commit is contained in:
6543
2024-09-20 09:40:48 +02:00
committed by GitHub
parent cfe6df5add
commit eebaa10d10
4 changed files with 57 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
server/queue/fifo_test.go
View File

@@ -52,17 +52,23 @@ func TestFifo(t *testing.T) {
func TestFifoExpire(t *testing.T) {
want := &model.Task{ID: "1"}
ctx, cancel := context.WithCancelCause(context.Background())
q, _ := New(context.Background()).(*fifo)
q, _ := New(ctx).(*fifo)
q.extension = 0
assert.NoError(t, q.Push(noContext, want))
info := q.Info(noContext)
assert.NoError(t, q.Push(ctx, want))
info := q.Info(ctx)
assert.Len(t, info.Pending, 1, "expect task in pending queue")
got, err := q.Poll(noContext, 1, func(*model.Task) bool { return true })
got, err := q.Poll(ctx, 1, func(*model.Task) bool { return true })
assert.NoError(t, err)
assert.Equal(t, want, got)
// cancel the context to let the process func end
go func() {
time.Sleep(time.Millisecond)
cancel(nil)
}()
q.process()
assert.Len(t, info.Pending, 1, "expect task re-added to pending queue")
}