Files
kopia/cli/command_snapshot_pin_test.go
Julio Lopez bb20d9e11a chore(ci): enable wsl_v5:{assign,expr} linter settings (#4982)
Enable wsl_v5 settings:
- assign
- expr
2025-11-12 23:12:06 -08:00

99 lines
3.2 KiB
Go

package cli_test
import (
"os"
"path/filepath"
"testing"
"github.com/stretchr/testify/require"
"github.com/kopia/kopia/cli"
"github.com/kopia/kopia/internal/testutil"
"github.com/kopia/kopia/snapshot"
"github.com/kopia/kopia/tests/testenv"
)
func TestSnapshotPin(t *testing.T) {
t.Parallel()
runner := testenv.NewInProcRunner(t)
e := testenv.NewCLITest(t, testenv.RepoFormatNotImportant, runner)
defer e.RunAndExpectSuccess(t, "repo", "disconnect")
e.RunAndExpectSuccess(t, "repo", "create", "filesystem", "--path", e.RepoDir)
srcdir := testutil.TempDirectory(t)
require.NoError(t, os.WriteFile(filepath.Join(srcdir, "some-file2"), []byte{1, 2, 3}, 0o755))
var man snapshot.Manifest
e.RunAndExpectSuccess(t, "policy", "set", srcdir, "--keep-latest=3", "--keep-hourly=0", "--keep-daily=0", "--keep-monthly=0", "--keep-weekly=0", "--keep-annual=0")
testutil.MustParseJSONLines(t, e.RunAndExpectSuccess(t, "snapshot", "create", srcdir, "--pin=a", "--pin=b", "--json"), &man)
require.Equal(t, []string{"a", "b"}, man.Pins)
e.RunAndExpectSuccess(t, "snapshot", "list")
// create more unpinned snapshots
e.RunAndExpectSuccess(t, "snapshot", "create", srcdir)
e.RunAndExpectSuccess(t, "snapshot", "create", srcdir)
e.RunAndExpectSuccess(t, "snapshot", "create", srcdir)
e.RunAndExpectSuccess(t, "snapshot", "create", srcdir)
e.RunAndExpectSuccess(t, "snapshot", "create", srcdir)
snapshots := mustListSnapshots(t, e)
// make sure the pinned one is on top.
require.Len(t, snapshots, 4)
require.Equal(t, []string{"a", "b"}, snapshots[0].Pins)
require.Empty(t, snapshots[1].Pins)
require.Empty(t, snapshots[2].Pins)
require.Empty(t, snapshots[3].Pins)
// neither --add nor --remove were provided
e.RunAndExpectFailure(t, "snapshot", "pin", string(snapshots[3].ID))
e.RunAndExpectSuccess(t, "snapshot", "pin", string(snapshots[0].ID), "--add=c", "--remove=b")
e.RunAndExpectSuccess(t, "snapshot", "pin", string(snapshots[3].ID), "--add=d")
snapshots2 := mustListSnapshots(t, e)
require.Len(t, snapshots2, 4)
require.Equal(t, []string{"a", "c"}, snapshots2[0].Pins)
require.Empty(t, snapshots2[1].Pins)
require.Empty(t, snapshots2[2].Pins)
require.Equal(t, []string{"d"}, snapshots2[3].Pins)
// create more unpinned snapshots
e.RunAndExpectSuccess(t, "snapshot", "create", srcdir)
e.RunAndExpectSuccess(t, "snapshot", "create", srcdir)
e.RunAndExpectSuccess(t, "snapshot", "create", srcdir)
e.RunAndExpectSuccess(t, "snapshot", "create", srcdir)
e.RunAndExpectSuccess(t, "snapshot", "create", srcdir)
snapshots3 := mustListSnapshots(t, e)
require.Len(t, snapshots3, 5)
require.Equal(t, []string{"a", "c"}, snapshots3[0].Pins)
require.Equal(t, []string{"d"}, snapshots3[1].Pins)
require.Empty(t, snapshots3[2].Pins)
require.Empty(t, snapshots3[3].Pins)
require.Empty(t, snapshots3[4].Pins)
}
func mustListSnapshots(t *testing.T, e *testenv.CLITest) []*snapshot.Manifest {
t.Helper()
var cliSnapshots []cli.SnapshotManifest
testutil.MustParseJSONLines(t, e.RunAndExpectSuccess(t, "snapshot", "list", "--json"), &cliSnapshots)
snapshots := make([]*snapshot.Manifest, 0, len(cliSnapshots))
for _, s := range cliSnapshots {
snapshots = append(snapshots, s.Manifest)
}
return snapshot.SortByTime(snapshots, false)
}