Function TestUpdateRetention

repo/format/format_manager_test.go:223–275

(t *testing.T)

}

func TestUpdateRetention(t *testing.T) {
	ctx := testlogging.Context(t)

	mode := blob.Governance
	period := time.Hour * 48

	ta := faketime.NewClockTimeWithOffset(0)
	nowFunc := ta.NowFunc()
	earliestExpiry := nowFunc().Add(period)

	st := blobtesting.NewVersionedMapStorage(nowFunc)
	blobCache := format.NewMemoryBlobCache(nowFunc)

	// success
	require.NoError(t, format.Initialize(ctx, st, &format.KopiaRepositoryJSON{}, rc, format.BlobStorageConfiguration{}, "some-password"))

	mgr, err := format.NewManagerWithCache(ctx, st, cacheDuration, "some-password", nowFunc, blobCache)
	require.NoError(t, err, "getting format manager")

	mp := mustGetMutableParameters(t, mgr)
	rf := mustGetRequiredFeatures(t, mgr)

	err = mgr.SetParameters(
		ctx,
		mp,
		format.BlobStorageConfiguration{
			RetentionMode:   mode,
			RetentionPeriod: period,
		},
		rf,
	)
	require.NoError(t, err, "setting repo parameters")

	// New retention parameters should be available from the format manager.
	blobCfg := mustGetBlobStorageConfiguration(t, mgr)
	assert.Equal(t, mode, blobCfg.RetentionMode)
	assert.Equal(t, period, blobCfg.RetentionPeriod)

	// Get the retention configuration that was added to the blob. Allow up to a
	// minute difference between the expected and returned values since that
	// should be large enough to avoid test flakes.
	gotMode, expiry, err := st.GetRetention(ctx, format.KopiaRepositoryBlobID)
	require.NoError(t, err, "getting repo blob retention info")

	assert.Equal(t, mode, gotMode)
	assert.WithinDuration(t, earliestExpiry, expiry, time.Minute)

	gotMode, expiry, err = st.GetRetention(ctx, format.KopiaBlobCfgBlobID)
	require.NoError(t, err, "getting storage blob config retention info")

	assert.Equal(t, mode, gotMode)
	assert.WithinDuration(t, earliestExpiry, expiry, time.Minute)
}

func TestUpdateRetentionNegativeValue(t *testing.T) {
	ctx := testlogging.Context(t)