31 lines
530 B
Go
31 lines
530 B
Go
|
package collection
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync/atomic"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestEachSync(t *testing.T) {
|
||
|
|
collection := []int{1, 2, 3, 4, 5, 6}
|
||
|
|
want := 21
|
||
|
|
sum := 0
|
||
|
|
EachSync(collection, func(v int, _ int) {
|
||
|
|
sum += v
|
||
|
|
})
|
||
|
|
if sum != want {
|
||
|
|
t.Errorf("Expected %d, got %d", want, sum)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestEach(t *testing.T) {
|
||
|
|
collection := []int{1, 2, 3, 4, 5, 6}
|
||
|
|
want := int64(21)
|
||
|
|
sum := int64(0)
|
||
|
|
Each(collection, func(v int, _ int) {
|
||
|
|
atomic.AddInt64(&sum, int64(v))
|
||
|
|
})
|
||
|
|
if sum != want {
|
||
|
|
t.Errorf("Expected %d, got %d", want, sum)
|
||
|
|
}
|
||
|
|
}
|