mirror of
https://github.com/nadoo/glider.git
synced 2025-02-24 01:45:39 +08:00
31 lines
519 B
Go
31 lines
519 B
Go
package smux
|
|
|
|
import (
|
|
"container/heap"
|
|
"testing"
|
|
)
|
|
|
|
func TestShaper(t *testing.T) {
|
|
w1 := writeRequest{prio: 10}
|
|
w2 := writeRequest{prio: 10}
|
|
w3 := writeRequest{prio: 20}
|
|
w4 := writeRequest{prio: 100}
|
|
|
|
var reqs shaperHeap
|
|
heap.Push(&reqs, w4)
|
|
heap.Push(&reqs, w3)
|
|
heap.Push(&reqs, w2)
|
|
heap.Push(&reqs, w1)
|
|
|
|
var lastPrio uint64
|
|
for len(reqs) > 0 {
|
|
w := heap.Pop(&reqs).(writeRequest)
|
|
if w.prio < lastPrio {
|
|
t.Fatal("incorrect shaper priority")
|
|
}
|
|
|
|
t.Log("prio:", w.prio)
|
|
lastPrio = w.prio
|
|
}
|
|
}
|