76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
|
// Copyright (C) 2020 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package objectdeletion_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strconv"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"github.com/zeebo/errs"
|
||
|
|
"go.uber.org/zap/zaptest"
|
||
|
|
|
||
|
|
"storj.io/common/testcontext"
|
||
|
|
"storj.io/common/testrand"
|
||
|
|
"storj.io/storj/satellite/metainfo/objectdeletion"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestReport(t *testing.T) {
|
||
|
|
logger := zaptest.NewLogger(t)
|
||
|
|
|
||
|
|
var testCases = []struct {
|
||
|
|
description string
|
||
|
|
numRequests int
|
||
|
|
numDeletedPaths int
|
||
|
|
expectedFailure bool
|
||
|
|
}{
|
||
|
|
{"has-failure", 2, 1, true},
|
||
|
|
{"all-deleted", 2, 2, false},
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, tt := range testCases {
|
||
|
|
tt := tt
|
||
|
|
t.Run(tt.description, func(t *testing.T) {
|
||
|
|
ctx := testcontext.New(t)
|
||
|
|
defer ctx.Cleanup()
|
||
|
|
requests := createRequests(tt.numRequests)
|
||
|
|
deletedSegmentPaths, err := createDeletedSegmentPaths(requests, tt.numDeletedPaths)
|
||
|
|
require.NoError(t, err)
|
||
|
|
report := objectdeletion.GenerateReport(ctx, logger, requests, deletedSegmentPaths)
|
||
|
|
require.Equal(t, tt.expectedFailure, report.HasFailures())
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func createDeletedSegmentPaths(requests []*objectdeletion.ObjectIdentifier, numDeleted int) ([][]byte, error) {
|
||
|
|
if numDeleted > len(requests) {
|
||
|
|
return nil, errs.New("invalid argument")
|
||
|
|
}
|
||
|
|
deletedSegmentPaths := make([][]byte, 0, numDeleted)
|
||
|
|
for i := 0; i < numDeleted; i++ {
|
||
|
|
path, err := requests[i].SegmentPath(int64(testrand.Intn(10)))
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
deletedSegmentPaths = append(deletedSegmentPaths, path)
|
||
|
|
}
|
||
|
|
return deletedSegmentPaths, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func createRequests(numRequests int) []*objectdeletion.ObjectIdentifier {
|
||
|
|
requests := make([]*objectdeletion.ObjectIdentifier, 0, numRequests)
|
||
|
|
|
||
|
|
for i := 0; i < numRequests; i++ {
|
||
|
|
obj := objectdeletion.ObjectIdentifier{
|
||
|
|
ProjectID: testrand.UUID(),
|
||
|
|
Bucket: []byte("test"),
|
||
|
|
EncryptedPath: []byte(strconv.Itoa(i) + "test"),
|
||
|
|
}
|
||
|
|
requests = append(requests, &obj)
|
||
|
|
}
|
||
|
|
|
||
|
|
return requests
|
||
|
|
}
|