sneak
7981ebc33a
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 still write straight to stdout, as the `database delete` prompt does, together with what they ask to delete: the snapshot list before the purge prompt, and the blob count and size before the prune prompt. With --force those lines are status lines. The two purge test fixtures gained the UI writer production always sets. Model: opus-5-5
149 lines
4.3 KiB
Go
149 lines
4.3 KiB
Go
package vaultik_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"sneak.berlin/go/vaultik/internal/log"
|
|
"sneak.berlin/go/vaultik/internal/ui"
|
|
"sneak.berlin/go/vaultik/internal/vaultik"
|
|
)
|
|
|
|
// TestPurgeSnapshots_QuietSuppressesReport checks that `snapshot purge
|
|
// --keep-latest --force` under --quiet deletes the older snapshot and
|
|
// prints nothing.
|
|
func TestPurgeSnapshots_QuietSuppressesReport(t *testing.T) {
|
|
log.Initialize(log.Config{})
|
|
t.Parallel()
|
|
|
|
v := setupPurgeTest(t, snapHostname, []string{snapHomeT0, snapHomeT1})
|
|
|
|
var out bytes.Buffer
|
|
|
|
v.Stdout = &out
|
|
v.UI = ui.NewWithColor(&out, false)
|
|
v.UI.SetQuiet(true)
|
|
|
|
require.NoError(t, v.PurgeSnapshotsWithOptions(&vaultik.SnapshotPurgeOptions{
|
|
KeepLatest: true,
|
|
Force: true,
|
|
}))
|
|
|
|
assert.Equal(t, []string{snapHomeT1}, listRemainingSnapshots(t, v))
|
|
assert.Empty(t, out.String())
|
|
}
|
|
|
|
// TestPurgeSnapshots_QuietPromptListsSnapshots checks that `snapshot
|
|
// purge --keep-latest` without --force still lists the snapshots its
|
|
// prompt asks to delete under --quiet.
|
|
func TestPurgeSnapshots_QuietPromptListsSnapshots(t *testing.T) {
|
|
log.Initialize(log.Config{})
|
|
t.Parallel()
|
|
|
|
v := setupPurgeTest(t, snapHostname, []string{snapHomeT0, snapHomeT1})
|
|
|
|
var out bytes.Buffer
|
|
|
|
v.Stdout = &out
|
|
v.UI = ui.NewWithColor(&out, false)
|
|
v.UI.SetQuiet(true)
|
|
|
|
// Stdin is empty, so the prompt reads no answer and cancels.
|
|
require.NoError(t, v.PurgeSnapshotsWithOptions(&vaultik.SnapshotPurgeOptions{
|
|
KeepLatest: true,
|
|
}))
|
|
|
|
assert.Contains(t, out.String(), snapHomeT0)
|
|
assert.Contains(t, out.String(), "Delete 1 snapshot(s)? [y/N]")
|
|
assert.Len(t, listRemainingSnapshots(t, v), 2)
|
|
}
|
|
|
|
// TestPruneBlobs_QuietPromptShowsBlobTotal checks that `prune` without
|
|
// --force still states the count and total size of the blobs its prompt
|
|
// asks to delete under --quiet.
|
|
func TestPruneBlobs_QuietPromptShowsBlobTotal(t *testing.T) {
|
|
log.Initialize(log.Config{})
|
|
t.Parallel()
|
|
|
|
env := newListEnv(t)
|
|
env.v.UI.SetQuiet(true)
|
|
|
|
// The store holds no manifest, so nothing references this blob.
|
|
addBlob(t, env.store.testStorer, testBlobHashA)
|
|
|
|
// Stdin is empty, so the prompt reads no answer and cancels.
|
|
require.NoError(t, env.v.PruneBlobs(&vaultik.PruneOptions{}))
|
|
|
|
assert.Contains(t, env.stdout.String(),
|
|
"Found 1 unreferenced blob(s) totaling")
|
|
assert.Contains(t, env.stdout.String(),
|
|
"Delete 1 unreferenced blob(s)? [y/N]")
|
|
}
|
|
|
|
// TestRemoveSnapshot_QuietSuppressesReport checks that `snapshot remove`
|
|
// under --quiet prints nothing with --dry-run or with --force, and that
|
|
// the --json document still prints.
|
|
func TestRemoveSnapshot_QuietSuppressesReport(t *testing.T) {
|
|
log.Initialize(log.Config{})
|
|
t.Parallel()
|
|
|
|
newQuietRemover := func() (*vaultik.TestVaultik, *testStorer) {
|
|
store := newTestStorer()
|
|
addManifest(t, store, "snapshot-001", nil)
|
|
|
|
tv := vaultik.NewForTesting(store)
|
|
tv.Repositories = newLocalIndex(t, "snapshot-001")
|
|
tv.UI.SetQuiet(true)
|
|
|
|
return tv, store
|
|
}
|
|
|
|
tv, store := newQuietRemover()
|
|
|
|
_, err := tv.RemoveSnapshot("snapshot-001",
|
|
&vaultik.RemoveOptions{DryRun: true})
|
|
require.NoError(t, err)
|
|
assert.Empty(t, tv.Stdout.String(),
|
|
"--dry-run must print nothing under --quiet")
|
|
|
|
_, err = tv.RemoveSnapshot("snapshot-001",
|
|
&vaultik.RemoveOptions{Force: true})
|
|
require.NoError(t, err)
|
|
assert.False(t, store.hasKey(remoteKeyPath("snapshot-001")))
|
|
assert.Empty(t, tv.Stdout.String(),
|
|
"--force must print nothing under --quiet")
|
|
|
|
tv, _ = newQuietRemover()
|
|
|
|
_, err = tv.RemoveSnapshot("snapshot-001",
|
|
&vaultik.RemoveOptions{JSON: true})
|
|
require.NoError(t, err)
|
|
|
|
var result vaultik.RemoveResult
|
|
|
|
require.NoError(t, json.Unmarshal(tv.Stdout.Bytes(), &result),
|
|
"the --json document must still print under --quiet")
|
|
assert.True(t, result.RemoteRemoved)
|
|
}
|
|
|
|
// TestNukeRemote_QuietSuppressesReport checks that `remote nuke --force`
|
|
// under --quiet prints nothing while it removes a snapshot's metadata
|
|
// and its blobs.
|
|
func TestNukeRemote_QuietSuppressesReport(t *testing.T) {
|
|
log.Initialize(log.Config{})
|
|
t.Parallel()
|
|
|
|
ctx := context.Background()
|
|
storeDir := filepath.Join(t.TempDir(), "store")
|
|
v, _, out := backUpToFileDestination(ctx, t, storeDir)
|
|
v.UI.SetQuiet(true)
|
|
|
|
require.NoError(t, v.NukeRemote(true))
|
|
assert.Empty(t, out.String())
|
|
}
|