storj/pkg/datarepair/checker/online_test.go
2019-07-15 22:58:39 +03:00

53 lines
1.1 KiB
Go

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.
package checker
import (
"context"
"testing"
"time"
"go.uber.org/zap"
"storj.io/storj/internal/testcontext"
"storj.io/storj/internal/testrand"
"storj.io/storj/pkg/overlay"
"storj.io/storj/pkg/pb"
"storj.io/storj/pkg/storj"
)
func TestReliabilityCache_Concurrent(t *testing.T) {
ctx := testcontext.New(t)
defer ctx.Cleanup()
ocache := overlay.NewCache(zap.NewNop(), fakeOverlayDB{}, overlay.NodeSelectionConfig{})
rcache := NewReliabilityCache(ocache, time.Millisecond)
for i := 0; i < 10; i++ {
ctx.Go(func() error {
for i := 0; i < 10000; i++ {
pieces := []*pb.RemotePiece{{NodeId: testrand.NodeID()}}
_, err := rcache.MissingPieces(ctx, time.Now(), pieces)
if err != nil {
return err
}
}
return nil
})
}
ctx.Wait()
}
type fakeOverlayDB struct{ overlay.DB }
func (fakeOverlayDB) Reliable(context.Context, *overlay.NodeCriteria) (storj.NodeIDList, error) {
return storj.NodeIDList{
testrand.NodeID(),
testrand.NodeID(),
testrand.NodeID(),
testrand.NodeID(),
}, nil
}