b12bb50a4a
Create new operation type to update retention parameters for immutable backups --- #### Does this PR need a docs update or release note? - [ ] ✅ Yes, it's included - [ ] 🕐 Yes, but in a later PR - [x] ⛔ No #### Type of change - [x] 🌻 Feature - [ ] 🐛 Bugfix - [ ] 🗺️ Documentation - [ ] 🤖 Supportability/Tests - [ ] 💻 CI/Deployment - [ ] 🧹 Tech Debt/Cleanup #### Test Plan - [ ] 💪 Manual - [x] ⚡ Unit test - [ ] 💚 E2E
75 lines
2.0 KiB
Go
75 lines
2.0 KiB
Go
package operations
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/alcionai/clues"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/alcionai/corso/src/internal/common/ptr"
|
|
evmock "github.com/alcionai/corso/src/internal/events/mock"
|
|
"github.com/alcionai/corso/src/internal/kopia"
|
|
"github.com/alcionai/corso/src/internal/tester"
|
|
"github.com/alcionai/corso/src/pkg/control"
|
|
"github.com/alcionai/corso/src/pkg/control/repository"
|
|
storeTD "github.com/alcionai/corso/src/pkg/storage/testdata"
|
|
)
|
|
|
|
type RetentionConfigOpIntegrationSuite struct {
|
|
tester.Suite
|
|
}
|
|
|
|
func TestRetentionConfigOpIntegrationSuite(t *testing.T) {
|
|
suite.Run(t, &RetentionConfigOpIntegrationSuite{
|
|
Suite: tester.NewIntegrationSuite(
|
|
t,
|
|
[][]string{storeTD.AWSStorageCredEnvs}),
|
|
})
|
|
}
|
|
|
|
func (suite *RetentionConfigOpIntegrationSuite) TestRepoRetentionConfig() {
|
|
var (
|
|
t = suite.T()
|
|
// need to initialize the repository before we can test connecting to it.
|
|
st = storeTD.NewPrefixedS3Storage(t)
|
|
k = kopia.NewConn(st)
|
|
)
|
|
|
|
ctx, flush := tester.NewContext(t)
|
|
defer flush()
|
|
|
|
err := k.Initialize(ctx, repository.Options{}, repository.Retention{})
|
|
require.NoError(t, err, clues.ToCore(err))
|
|
|
|
kw, err := kopia.NewWrapper(k)
|
|
// kopiaRef comes with a count of 1 and Wrapper bumps it again so safe
|
|
// to close here.
|
|
k.Close(ctx)
|
|
|
|
require.NoError(t, err, clues.ToCore(err))
|
|
|
|
defer kw.Close(ctx)
|
|
|
|
// Only set extend locks parameter as other retention options require a bucket
|
|
// with object locking enabled. There's more complete tests in the kopia
|
|
// package.
|
|
rco, err := NewRetentionConfigOperation(
|
|
ctx,
|
|
control.DefaultOptions(),
|
|
kw,
|
|
repository.Retention{
|
|
Extend: ptr.To(true),
|
|
},
|
|
evmock.NewBus())
|
|
require.NoError(t, err, clues.ToCore(err))
|
|
|
|
err = rco.Run(ctx)
|
|
assert.NoError(t, err, clues.ToCore(err))
|
|
assert.Equal(t, Completed, rco.Status)
|
|
assert.NotZero(t, rco.Results.StartedAt)
|
|
assert.NotZero(t, rco.Results.CompletedAt)
|
|
assert.NotEqual(t, rco.Results.StartedAt, rco.Results.CompletedAt)
|
|
}
|