sneak
cd573def55
check / check (push) Waiting to run
PruneBlobs, CleanupLocalSnapshots, RemoveSnapshot, RemoveAllSnapshots and snapshot purge wrote their status lines straight to stdout, so `snapshot create --cron --prune` printed on every successful run and --quiet did not silence `prune`, `snapshot purge`, `snapshot remove` or `remote nuke`. Those lines now go through internal/ui, which drops them under --cron and --quiet; warnings, errors and --json documents still print. The confirmation prompts and their "Cancelled" answer still write to stdout directly, as the `database delete` prompt does. Under --quiet without --force, the list of snapshots shown before the purge prompt is now silenced. The two purge test fixtures gained the UI writer production always sets. Model: opus-5-5
127 lines
3.4 KiB
Go
127 lines
3.4 KiB
Go
package cli //nolint:testpackage // shares runEntry and the hermetic config helpers
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// Flags used to build the argument vectors below.
|
|
const (
|
|
flagCron = "--cron"
|
|
flagForce = "--force"
|
|
)
|
|
|
|
// plantUnreferencedBlob writes a blob that no snapshot references to the
|
|
// destination store at storeDir, so that a prune deletes it. Returns its
|
|
// path.
|
|
func plantUnreferencedBlob(t *testing.T, storeDir string) string {
|
|
t.Helper()
|
|
|
|
hash := strings.Repeat("b", 64)
|
|
path := filepath.Join(storeDir, "blobs", hash[:2], hash[2:4], hash)
|
|
|
|
require.NoError(t, os.MkdirAll(filepath.Dir(path), 0o750))
|
|
require.NoError(t, os.WriteFile(path, []byte("unreferenced"), 0o600))
|
|
|
|
return path
|
|
}
|
|
|
|
// TestEntrySnapshotCreatePruneIsSilent checks that a successful
|
|
// `snapshot create --prune` under --cron or --quiet prints nothing on
|
|
// stdout or stderr, whether or not the prune finds a blob to delete.
|
|
//
|
|
//nolint:paralleltest // replaces os.Args, os.Stdout, os.Stderr and the xdg globals
|
|
func TestEntrySnapshotCreatePruneIsSilent(t *testing.T) {
|
|
for _, testCase := range []struct {
|
|
name string
|
|
args []string
|
|
plantBlob bool
|
|
}{
|
|
{
|
|
name: "cron, no blob to delete",
|
|
args: []string{cmdSnapshot, cmdCreate, flagCron, flagPrune},
|
|
},
|
|
{
|
|
name: "cron, blob to delete",
|
|
args: []string{cmdSnapshot, cmdCreate, flagCron, flagPrune},
|
|
plantBlob: true,
|
|
},
|
|
{
|
|
name: "cron with keep-newer-than",
|
|
args: []string{
|
|
cmdSnapshot, cmdCreate, flagCron, flagPrune,
|
|
flagKeepNewerThan, "4w",
|
|
},
|
|
plantBlob: true,
|
|
},
|
|
{
|
|
name: "quiet",
|
|
args: []string{flagQuiet, cmdSnapshot, cmdCreate, flagPrune},
|
|
plantBlob: true,
|
|
},
|
|
} {
|
|
t.Run(testCase.name, func(t *testing.T) {
|
|
configPath, storeDir := writeBackupConfig(t)
|
|
|
|
var blob string
|
|
if testCase.plantBlob {
|
|
blob = plantUnreferencedBlob(t, storeDir)
|
|
}
|
|
|
|
code, stdout, stderr := runEntry(t,
|
|
append([]string{flagConfig, configPath}, testCase.args...)...)
|
|
|
|
assert.Equal(t, 0, code)
|
|
assert.Empty(t, stdout)
|
|
assert.Empty(t, stderr)
|
|
|
|
if testCase.plantBlob {
|
|
assert.NoFileExists(t, blob, "the prune must delete the blob")
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
// TestEntryPruneQuietIsSilent checks that a successful `prune --force`
|
|
// under --quiet prints nothing on stdout or stderr, both when there is
|
|
// nothing to remove and when it removes a stale local snapshot record
|
|
// and an unreferenced blob.
|
|
//
|
|
//nolint:paralleltest // replaces os.Args, os.Stdout, os.Stderr and the xdg globals
|
|
func TestEntryPruneQuietIsSilent(t *testing.T) {
|
|
for _, testCase := range []struct {
|
|
name string
|
|
staleRecordAndBlob bool
|
|
}{
|
|
{name: "nothing to remove"},
|
|
{name: "stale record and blob to remove", staleRecordAndBlob: true},
|
|
} {
|
|
t.Run(testCase.name, func(t *testing.T) {
|
|
configPath := writeHermeticPruneConfig(t, testCase.staleRecordAndBlob)
|
|
|
|
var blob string
|
|
if testCase.staleRecordAndBlob {
|
|
// writeHermeticPruneConfig puts the store beside the config.
|
|
blob = plantUnreferencedBlob(t,
|
|
filepath.Join(filepath.Dir(configPath), "store"))
|
|
}
|
|
|
|
code, stdout, stderr := runEntry(t,
|
|
flagConfig, configPath, flagQuiet, cmdPrune, flagForce)
|
|
|
|
assert.Equal(t, 0, code)
|
|
assert.Empty(t, stdout)
|
|
assert.Empty(t, stderr)
|
|
|
|
if testCase.staleRecordAndBlob {
|
|
assert.NoFileExists(t, blob, "the prune must delete the blob")
|
|
}
|
|
})
|
|
}
|
|
}
|