mirror of
https://github.com/temporalio/temporal.git
synced 2026-08-31 02:51:51 -07:00
5b49acfaf9
## What changed?
Integrate Go 1.26's new `go fix` into workflow.
NOTE that the changes caused our linter to fire; a [separate
commit](1b23f787ae)
addresses those.
## Why?
Ensure Go code is standardized/modernized.
## How did you test it?
- [ ] built
- [ ] run locally and tested manually
- [x] covered by existing tests
- [ ] added new unit test(s)
- [ ] added new functional test(s)
114 lines
2.4 KiB
Go
114 lines
2.4 KiB
Go
package collection
|
|
|
|
import (
|
|
"math/rand"
|
|
"sort"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type (
|
|
PriorityQueueSuite struct {
|
|
suite.Suite
|
|
pq Queue[*testPriorityQueueItem]
|
|
}
|
|
|
|
testPriorityQueueItem struct {
|
|
value int
|
|
}
|
|
)
|
|
|
|
func testPriorityQueueItemCompareLess(this *testPriorityQueueItem, that *testPriorityQueueItem) bool {
|
|
return this.value < that.value
|
|
}
|
|
|
|
func TestPriorityQueueSuite(t *testing.T) {
|
|
suite.Run(t, new(PriorityQueueSuite))
|
|
}
|
|
|
|
func (s *PriorityQueueSuite) SetupTest() {
|
|
s.pq = NewPriorityQueue(testPriorityQueueItemCompareLess)
|
|
}
|
|
|
|
func (s *PriorityQueueSuite) TestNewPriorityQueueWithItems() {
|
|
items := []*testPriorityQueueItem{
|
|
{value: 10},
|
|
{value: 3},
|
|
{value: 5},
|
|
{value: 4},
|
|
{value: 1},
|
|
{value: 16},
|
|
{value: -10},
|
|
}
|
|
s.pq = NewPriorityQueueWithItems(
|
|
testPriorityQueueItemCompareLess,
|
|
items,
|
|
)
|
|
|
|
expected := []int{-10, 1, 3, 4, 5, 10, 16}
|
|
result := []int{}
|
|
|
|
for !s.pq.IsEmpty() {
|
|
result = append(result, s.pq.Remove().value)
|
|
}
|
|
s.Equal(expected, result)
|
|
}
|
|
|
|
func (s *PriorityQueueSuite) TestInsertAndPop() {
|
|
s.pq.Add(&testPriorityQueueItem{10})
|
|
s.pq.Add(&testPriorityQueueItem{3})
|
|
s.pq.Add(&testPriorityQueueItem{5})
|
|
s.pq.Add(&testPriorityQueueItem{4})
|
|
s.pq.Add(&testPriorityQueueItem{1})
|
|
s.pq.Add(&testPriorityQueueItem{16})
|
|
s.pq.Add(&testPriorityQueueItem{-10})
|
|
|
|
expected := []int{-10, 1, 3, 4, 5, 10, 16}
|
|
result := []int{}
|
|
|
|
for !s.pq.IsEmpty() {
|
|
result = append(result, s.pq.Remove().value)
|
|
}
|
|
s.Equal(expected, result)
|
|
|
|
s.pq.Add(&testPriorityQueueItem{1000})
|
|
s.pq.Add(&testPriorityQueueItem{1233})
|
|
s.pq.Remove() // remove 1000
|
|
s.pq.Add(&testPriorityQueueItem{4})
|
|
s.pq.Add(&testPriorityQueueItem{18})
|
|
s.pq.Add(&testPriorityQueueItem{192})
|
|
s.pq.Add(&testPriorityQueueItem{255})
|
|
s.pq.Remove() // remove 4
|
|
s.pq.Remove() // remove 18
|
|
s.pq.Add(&testPriorityQueueItem{59})
|
|
s.pq.Add(&testPriorityQueueItem{727})
|
|
|
|
expected = []int{59, 192, 255, 727, 1233}
|
|
result = []int{}
|
|
|
|
for !s.pq.IsEmpty() {
|
|
result = append(result, s.pq.Remove().value)
|
|
}
|
|
s.Equal(expected, result)
|
|
}
|
|
|
|
func (s *PriorityQueueSuite) TestRandomNumber() {
|
|
for range 1000 {
|
|
|
|
expected := []int{}
|
|
result := []int{}
|
|
for range 1000 {
|
|
num := rand.Int()
|
|
s.pq.Add(&testPriorityQueueItem{num})
|
|
expected = append(expected, num)
|
|
}
|
|
sort.Ints(expected)
|
|
|
|
for !s.pq.IsEmpty() {
|
|
result = append(result, s.pq.Remove().value)
|
|
}
|
|
s.Equal(expected, result)
|
|
}
|
|
}
|