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.
26 lines
457 B
Go
26 lines
457 B
Go
4 years ago
|
package collection
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
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)
|
||
|
|
}
|