func TestPriorityQueue(t *testing.T) {
c := 100
pq := New(c)
for i := 0; i < c+1; i++ {
heap.Push(&pq, &Item{Value: i, Priority: int64(i)})
}
ensure.DeepEqual(t, pq.Len(), c+1)
ensure.DeepEqual(t, cap(pq), c*2)
for i := 0; i < c+1; i++ {
item := heap.Pop(&pq)
ensure.DeepEqual(t, item.(*Item).Value.(int), i)
}
ensure.DeepEqual(t, cap(pq), c/4)
}
func TestUnsortedInsert(t *testing.T) {
c := 100
pq := New(c)
ints := make([]int, 0, c)
for i := 0; i < c; i++ {
v := rand.Int()
ints = append(ints, v)
heap.Push(&pq, &Item{Value: i, Priority: int64(v)})
}
ensure.DeepEqual(t, pq.Len(), c)
ensure.DeepEqual(t, cap(pq), c)
sort.Sort(sort.IntSlice(ints))
for i := 0; i < c; i++ {
item, _ := pq.PeekAndShift(int64(ints[len(ints)-1]))
ensure.DeepEqual(t, item.Priority, int64(ints[i]))
}
}
func TestStatsString(t *testing.T) {
s := httpcontrol.Stats{
Request: &http.Request{
Method: "GET",
URL: &url.URL{Path: "/"},
},
Response: &http.Response{
Status: "200 OK",
},
}
ensure.DeepEqual(t, s.String(), "GET / got response with status 200 OK")
}
func TestRemove(t *testing.T) {
c := 100
pq := New(c)
for i := 0; i < c; i++ {
v := rand.Int()
heap.Push(&pq, &Item{Value: "test", Priority: int64(v)})
}
for i := 0; i < 10; i++ {
heap.Remove(&pq, rand.Intn((c-1)-i))
}
lastPriority := heap.Pop(&pq).(*Item).Priority
for i := 0; i < (c - 10 - 1); i++ {
item := heap.Pop(&pq)
ensure.DeepEqual(t, lastPriority < item.(*Item).Priority, true)
lastPriority = item.(*Item).Priority
}
}
func TestPriorityQueueWithZeroCapacity(t *testing.T) {
pq := New(0)
ensure.DeepEqual(t, cap(pq), 1)
}
