queue_test.go

func TestPriorityQueue(t *testing.T) {
	pq := NewPriorityQueue()
	val := &struct{}{}
	var wg sync.WaitGroup
	const nthreads = 32
	for range nthreads {
		wg.Add(1)
		go func() {
			for range 100_000 {
				pq.Put(1, 1, val)
			}
			wg.Done()
			fmt.Println("done")
		}()
	}
	wg.Add(1)
	go func() {
		for range 100_000 * nthreads {
			pq.Get()
		}
		wg.Done()
	}()
	wg.Wait()
}