mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-02 11:59:00 +08:00
a88c896733
https://github.com/milvus-io/milvus/issues/31007 --------- Signed-off-by: longjiquan <jiquan.long@zilliz.com>
24 lines
419 B
Go
24 lines
419 B
Go
package connection
|
|
|
|
import (
|
|
"container/heap"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func Test_priorityQueue(t *testing.T) {
|
|
q := newPriorityQueue()
|
|
repeat := 10
|
|
for i := 0; i < repeat; i++ {
|
|
heap.Push(&q, newQueryItem(int64(i), time.Now()))
|
|
}
|
|
counter := repeat - 1
|
|
for q.Len() > 0 {
|
|
item := heap.Pop(&q).(*queueItem)
|
|
assert.Equal(t, int64(counter), item.identifier)
|
|
counter--
|
|
}
|
|
}
|