31 lines
779 B
Go
31 lines
779 B
Go
|
package sectorstorage
|
||
|
|
|
||
|
|
import (
|
||
|
|
"container/heap"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/filecoin-project/sector-storage/sealtasks"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestRequestQueue(t *testing.T) {
|
||
|
|
rq := &requestQueue{}
|
||
|
|
|
||
|
|
heap.Push(rq, &workerRequest{taskType: sealtasks.TTAddPiece})
|
||
|
|
heap.Push(rq, &workerRequest{taskType: sealtasks.TTPreCommit1})
|
||
|
|
heap.Push(rq, &workerRequest{taskType: sealtasks.TTPreCommit2})
|
||
|
|
heap.Push(rq, &workerRequest{taskType: sealtasks.TTPreCommit1})
|
||
|
|
heap.Push(rq, &workerRequest{taskType: sealtasks.TTAddPiece})
|
||
|
|
|
||
|
|
pt := heap.Pop(rq).(*workerRequest)
|
||
|
|
|
||
|
|
if pt.taskType != sealtasks.TTPreCommit2 {
|
||
|
|
t.Error("expected precommit2, got", pt.taskType)
|
||
|
|
}
|
||
|
|
|
||
|
|
pt = heap.Pop(rq).(*workerRequest)
|
||
|
|
|
||
|
|
if pt.taskType != sealtasks.TTPreCommit1 {
|
||
|
|
t.Error("expected precommit1, got", pt.taskType)
|
||
|
|
}
|
||
|
|
}
|