mirror of
https://github.com/rclone/rclone.git
synced 2025-03-10 20:45:14 +08:00
401cf81034
This commit modernizes Go usage. This was done with:
go run golang.org/x/tools/gopls/internal/analysis/modernize/cmd/modernize@latest -fix -test ./...
Then files needed to be `go fmt`ed and a few comments needed to be
restored.
The modernizations include replacing
- if/else conditional assignment by a call to the built-in min or max functions added in go1.21
- sort.Slice(x, func(i, j int) bool) { return s[i] < s[j] } by a call to slices.Sort(s), added in go1.21
- interface{} by the 'any' type added in go1.18
- append([]T(nil), s...) by slices.Clone(s) or slices.Concat(s), added in go1.21
- loop around an m[k]=v map update by a call to one of the Collect, Copy, Clone, or Insert functions from the maps package, added in go1.21
- []byte(fmt.Sprintf...) by fmt.Appendf(nil, ...), added in go1.19
- append(s[:i], s[i+1]...) by slices.Delete(s, i, i+1), added in go1.21
- a 3-clause for i := 0; i < n; i++ {} loop by for i := range n {}, added in go1.22
40 lines
883 B
Go
40 lines
883 B
Go
package accounting
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/rclone/rclone/fs"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestLimitTPS(t *testing.T) {
|
|
timeTransactions := func(n int, minTime, maxTime time.Duration) {
|
|
start := time.Now()
|
|
for range n {
|
|
LimitTPS(context.Background())
|
|
}
|
|
dt := time.Since(start)
|
|
assert.True(t, dt >= minTime && dt <= maxTime, "Expecting time between %v and %v, got %v", minTime, maxTime, dt)
|
|
}
|
|
|
|
t.Run("Off", func(t *testing.T) {
|
|
assert.Nil(t, tpsBucket)
|
|
timeTransactions(100, 0*time.Millisecond, 100*time.Millisecond)
|
|
})
|
|
|
|
t.Run("On", func(t *testing.T) {
|
|
ctx, ci := fs.AddConfig(context.Background())
|
|
ci.TPSLimit = 100.0
|
|
ci.TPSLimitBurst = 0
|
|
StartLimitTPS(ctx)
|
|
assert.NotNil(t, tpsBucket)
|
|
defer func() {
|
|
tpsBucket = nil
|
|
}()
|
|
|
|
timeTransactions(100, 900*time.Millisecond, 5000*time.Millisecond)
|
|
})
|
|
}
|