mirror of
https://github.com/rclone/rclone.git
synced 2025-04-21 19:38:53 +08:00

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
69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
//go:build !plan9 && !js
|
|
|
|
package kv
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestKvConcurrency(t *testing.T) {
|
|
require.Equal(t, 0, len(dbMap), "no databases can be started initially")
|
|
|
|
const threadNum = 5
|
|
var wg sync.WaitGroup
|
|
ctx := context.Background()
|
|
results := make([]*DB, threadNum)
|
|
wg.Add(threadNum)
|
|
for i := range threadNum {
|
|
go func(i int) {
|
|
db, err := Start(ctx, "test", nil)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, db)
|
|
results[i] = db
|
|
wg.Done()
|
|
}(i)
|
|
}
|
|
wg.Wait()
|
|
|
|
// must have a single multi-referenced db
|
|
db := results[0]
|
|
assert.Equal(t, 1, len(dbMap))
|
|
assert.Equal(t, threadNum, db.refs)
|
|
for i := range threadNum {
|
|
assert.Equal(t, db, results[i])
|
|
}
|
|
|
|
for i := range threadNum {
|
|
assert.Equal(t, 1, len(dbMap))
|
|
err := db.Stop(false)
|
|
assert.NoError(t, err, "unexpected error %v at retry %d", err, i)
|
|
}
|
|
|
|
assert.Equal(t, 0, len(dbMap), "must be closed in the end")
|
|
err := db.Stop(false)
|
|
assert.ErrorIs(t, err, ErrInactive, "missing expected stop indication")
|
|
}
|
|
|
|
func TestKvExit(t *testing.T) {
|
|
require.Equal(t, 0, len(dbMap), "no databases can be started initially")
|
|
const dbNum = 5
|
|
ctx := context.Background()
|
|
for i := range dbNum {
|
|
facility := fmt.Sprintf("test-%d", i)
|
|
for j := 0; j <= i; j++ {
|
|
db, err := Start(ctx, facility, nil)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, db)
|
|
}
|
|
}
|
|
assert.Equal(t, dbNum, len(dbMap))
|
|
Exit()
|
|
assert.Equal(t, 0, len(dbMap))
|
|
}
|