You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
537 B
Go
32 lines
537 B
Go
package collection
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewRing(t *testing.T) {
|
|
assert.Panics(t, func() {
|
|
NewRing(0)
|
|
})
|
|
}
|
|
|
|
func TestRingLess(t *testing.T) {
|
|
ring := NewRing(5)
|
|
for i := 0; i < 3; i++ {
|
|
ring.Add(i)
|
|
}
|
|
elements := ring.Take()
|
|
assert.ElementsMatch(t, []interface{}{0, 1, 2}, elements)
|
|
}
|
|
|
|
func TestRingMore(t *testing.T) {
|
|
ring := NewRing(5)
|
|
for i := 0; i < 11; i++ {
|
|
ring.Add(i)
|
|
}
|
|
elements := ring.Take()
|
|
assert.ElementsMatch(t, []interface{}{6, 7, 8, 9, 10}, elements)
|
|
}
|