27 lines
511 B
Go
27 lines
511 B
Go
package writecache
|
|
|
|
import (
|
|
"sync/atomic"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"golang.org/x/sync/errgroup"
|
|
)
|
|
|
|
func TestLimiter(t *testing.T) {
|
|
var maxSize uint64 = 10
|
|
var single uint64 = 3
|
|
l := newFlushLimiter(uint64(maxSize))
|
|
var currSize atomic.Int64
|
|
var eg errgroup.Group
|
|
for range 10_000 {
|
|
eg.Go(func() error {
|
|
defer l.release(single)
|
|
defer currSize.Add(-1)
|
|
l.acquire(single)
|
|
require.True(t, currSize.Add(1) <= 3)
|
|
return nil
|
|
})
|
|
}
|
|
require.NoError(t, eg.Wait())
|
|
}
|