1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
package priority
import (
"container/heap"
"testing"
"github.com/stretchr/testify/require"
)
func TestQueue(t *testing.T) {
pq := MakeQueue()
require.Equal(t, pq.Len(), 0)
heap.Push(&pq, &Item{Value: "2", Priority: 2})
heap.Push(&pq, &Item{Value: "1", Priority: 1})
heap.Push(&pq, &Item{Value: "3", Priority: 3})
require.Equal(t, pq.Len(), 3)
item := heap.Pop(&pq).(*Item)
require.Equal(t, item.Value, "1")
item = heap.Pop(&pq).(*Item)
require.Equal(t, item.Value, "2")
item = heap.Pop(&pq).(*Item)
require.Equal(t, item.Value, "3")
require.Equal(t, pq.Len(), 0)
}
|
